Add command line interface with PyInquirer
Add update requirements.txt
This commit is contained in:
parent
c47b255924
commit
79449c8ad3
42
main.py
42
main.py
@ -1,7 +1,19 @@
|
||||
from CommandRunner import CommandRunnerDebug as Runner
|
||||
from PyInquirer import prompt
|
||||
import importlib
|
||||
import os
|
||||
|
||||
|
||||
def transform_list_into_choices(choices_list):
|
||||
mapped_list = []
|
||||
for choice in choices_list:
|
||||
mapped_list.append({
|
||||
"name": choice
|
||||
})
|
||||
return mapped_list
|
||||
|
||||
|
||||
def load_modules():
|
||||
print("Loading modules")
|
||||
modules = {}
|
||||
for folder in os.listdir("./modules"):
|
||||
@ -11,21 +23,27 @@ for folder in os.listdir("./modules"):
|
||||
except AttributeError:
|
||||
pass
|
||||
print("Modules loaded successfully.\n")
|
||||
return modules
|
||||
|
||||
|
||||
def open_module_selection(modules):
|
||||
print("Available Modules:")
|
||||
for mod in modules.keys():
|
||||
print(mod)
|
||||
module_selection = [
|
||||
{
|
||||
'type': 'checkbox',
|
||||
'message': 'Select modules',
|
||||
'name': 'modules',
|
||||
'choices': transform_list_into_choices(
|
||||
list(modules.keys())
|
||||
)
|
||||
}
|
||||
]
|
||||
return prompt(module_selection)
|
||||
|
||||
print()
|
||||
|
||||
while True:
|
||||
chosen_module = input("Please specify your module: ")
|
||||
if chosen_module in modules.keys():
|
||||
break
|
||||
else:
|
||||
print("Module was not found.")
|
||||
modules = load_modules()
|
||||
|
||||
print("\nChosen Module to run: " + chosen_module)
|
||||
|
||||
Runner().run(modules[chosen_module])
|
||||
selected_modules = open_module_selection(modules)
|
||||
|
||||
for answer in selected_modules['modules']:
|
||||
Runner().run(modules[answer])
|
||||
|
@ -1 +1 @@
|
||||
|
||||
PyInquirer==1.0.3
|
||||
|
Loading…
Reference in New Issue
Block a user