diff options
Diffstat (limited to '')
| -rw-r--r-- | py/roles.py | 26 |
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 |