From ffe3ae76bbaf6baa084fe2e8e115684770c23a1d Mon Sep 17 00:00:00 2001 From: Ceda EI Date: Sun, 7 Apr 2024 20:10:35 +0530 Subject: [PATCH] Add status around module initialization --- inni/cli.py | 20 ++++++++++++-------- 1 file changed, 12 insertions(+), 8 deletions(-) diff --git a/inni/cli.py b/inni/cli.py index b5ef35d..4df0547 100644 --- a/inni/cli.py +++ b/inni/cli.py @@ -29,14 +29,18 @@ def _login_out(ctx: click.Context, login: bool): prompts = config["inni"]["prompts"] modules = {} - for module_name in module_names: - try: - module = load_module(module_name) - module_config = config["modules"].get(module_name, {}) - modules[module_name] = module(module_config) - except ModuleNotFoundError as err: - error_console.print(err.msg) - sys.exit(0) + with console.status("") as status: + for module_name in module_names: + status.update(f"[green]Initializing {module_name}") + try: + module = load_module(module_name) + module_config = config["modules"].get(module_name, {}) + modules[module_name] = module(module_config, console, error_console) + except ModuleNotFoundError as err: + error_console.print(err.msg) + sys.exit(0) + console.print("[green]✅ Initialized Modules") + console.print("") variables_to_prompt = [] for module in modules.values():