2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Get repair functions by name instead of full path

Browse source
This commit is contained in:
2Shirt 2021-04-24 20:23:08 -06:00
parent 1dbad4bafe
commit a7db972ba5
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
2 changed files with 19 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
scripts/auto_repairs.py
View file

@ -15,7 +15,7 @@ import wk # pylint: disable=wrong-import-position
class MenuEntry(): class MenuEntry():
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
"""Simple class to allow cleaner code below.""" """Simple class to allow cleaner code below."""
def __init__(self, name, function, **kwargs): def __init__(self, name, function=None, **kwargs):
self.name = name self.name = name
# Color reboot entries # Color reboot entries
@ -64,7 +64,7 @@ BASE_MENUS = {
MenuEntry('Reset Windows Updates', placeholder_function), MenuEntry('Reset Windows Updates', placeholder_function),
MenuEntry('Reboot', placeholder_reboot), MenuEntry('Reboot', placeholder_reboot),
MenuEntry('CHKDSK', placeholder_function), MenuEntry('CHKDSK', placeholder_function),
MenuEntry('DISM RestoreHealth', wk.repairs.win.auto_dism), MenuEntry('DISM RestoreHealth', 'auto_dism'),
MenuEntry('SFC Scan', placeholder_function), MenuEntry('SFC Scan', placeholder_function),
MenuEntry('Fix File Associations', placeholder_function), MenuEntry('Fix File Associations', placeholder_function),
MenuEntry('Clear Proxy Settings', placeholder_function), MenuEntry('Clear Proxy Settings', placeholder_function),
@ -88,14 +88,14 @@ BASE_MENUS = {
), ),
}, },
'Options': ( 'Options': (
MenuEntry('Kill Explorer', placeholder_function), MenuEntry('Kill Explorer'),
MenuEntry('Run RKill at startup', placeholder_function), MenuEntry('Run RKill at startup'),
MenuEntry('Use Autologon', placeholder_function), MenuEntry('Use Autologon'),
), ),
'Actions': ( 'Actions': (
MenuEntry('Options', placeholder_function), MenuEntry('Options'),
MenuEntry('Start', placeholder_function, Separator=True), MenuEntry('Start', Separator=True),
MenuEntry('Quit', placeholder_function), MenuEntry('Quit'),
), ),
} }

11
scripts/wk/repairs/win.py
View file

@ -91,6 +91,17 @@ def build_menus(base_menus, title):
# Initialize main menu display names # Initialize main menu display names
menus['Main'].update() menus['Main'].update()
# Fix Function references
for group, menu in menus.items():
if group not in base_menus['Groups']:
continue
for name in menu.options:
_function = menu.options[name]['Function']
if isinstance(_function, str):
menu.options[name]['Function'] = getattr(
sys.modules[__name__], _function,
)
# Done # Done
return menus return menus