diff --git a/main.py b/main.py index c88fc76..6d82db2 100644 --- a/main.py +++ b/main.py @@ -1,6 +1,6 @@ -from PyInquirer import prompt import importlib import os +import fire def load_modules(): @@ -16,19 +16,5 @@ def load_modules(): return loaded_modules -def open_module_selection(module_names): - print("Available Modules:") - module_selection = [ - { - 'type': 'checkbox', - 'message': 'Select modules', - 'name': 'modules', - 'choices': list(map(lambda x: {"name": x}, list(module_names))) - } - ] - return prompt(module_selection)['modules'] - - modules = load_modules() - -[modules[module].run() for module in open_module_selection(modules.keys())] +fire.Fire(modules) diff --git a/requirements.txt b/requirements.txt index efa466d..c72786b 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1 +1,2 @@ +fire PyInquirer==1.0.3