refactoring: import python when needed, run as functions

1 files changed, 10 insertions, 16 deletions

diff --git a/py/roles.py b/py/roles.py
index b7d19e1..16aa4e9 100644
--- a/py/roles.py
+++ b/py/roles.py
@@ -1,23 +1,17 @@
 import vim
 
-# import utils
-plugin_root = vim.eval("s:plugin_root")
-vim.command(f"py3file {plugin_root}/py/utils.py")
+roles_py_imported = True
 
-roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file"))
-if not os.path.exists(roles_config_path):
-    raise Exception(f"Role config file does not exist: {roles_config_path}")
+def load_ai_role_names():
+    roles_config_path = os.path.expanduser(vim.eval("g:vim_ai_roles_config_file"))
+    if not os.path.exists(roles_config_path):
+        raise Exception(f"Role config file does not exist: {roles_config_path}")
 
-roles = configparser.ConfigParser()
-roles.read(roles_config_path)
+    roles = configparser.ConfigParser()
+    roles.read(roles_config_path)
 
-enhance_roles_with_custom_function(roles)
+    enhance_roles_with_custom_function(roles)
 
-role_names = [name for name in roles.sections() if not '.' in name]
+    role_names = [name for name in roles.sections() if not '.' in name]
 
-role_list = [f'"{name}"' for name in role_names]
-role_list = ", ".join(role_list)
-
-role_list = f"[{role_list}]"
-
-vim.command(f'let l:role_list = {role_list}')
+    return role_names