public/linux-tools
1
0
Fork 0
Code Issues 12 Pull Requests Packages Projects Releases Wiki Activity

Rename variables to not cover outer ones, remove transform method

Browse Source 
simplify module run process
This commit is contained in:
Marcel Schwarz 2020-01-05 17:56:53 +01:00
parent 058bdcd828
commit c7e2afbbc7
1 changed files with 7 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
main.py
View File

@ -3,46 +3,32 @@ import importlib
import os 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(): def load_modules():
print("Loading modules") print("Loading modules")
modules = {} loaded_modules = {}
for folder in os.listdir("./modules"): for folder in os.listdir("./modules"):
try: try:
curr_module = importlib.import_module('.' + folder, package="modules") curr_module = importlib.import_module('.' + folder, package="modules")
modules[folder] = curr_module.get_module() loaded_modules[folder] = curr_module.get_module()
except AttributeError: except AttributeError:
pass pass
print("Modules loaded successfully.\n") print("Modules loaded successfully.\n")
return modules return loaded_modules
def open_module_selection(modules): def open_module_selection(module_names):
print("Available Modules:") print("Available Modules:")
module_selection = [ module_selection = [
{ {
'type': 'checkbox', 'type': 'checkbox',
'message': 'Select modules', 'message': 'Select modules',
'name': 'modules', 'name': 'modules',
'choices': transform_list_into_choices( 'choices': list(map(lambda x: {"name": x}, list(module_names)))
list(modules.keys())
)
} }
] ]
return prompt(module_selection) return prompt(module_selection)['modules']
modules = load_modules() modules = load_modules()
selected_modules = open_module_selection(modules) [modules[module].run() for module in open_module_selection(modules.keys())]
for answer in selected_modules['modules']:
modules[answer].run()