2Shirt
1dbad4bafe
result is for the return variable from TryAndPrint calls. message is the text to display or save to the registry.
109 lines
3.4 KiB
Python
109 lines
3.4 KiB
Python
"""Wizard Kit: Auto-Repair Tool"""
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import os
|
|
import sys
|
|
import random # TODO: Deleteme
|
|
import time
|
|
|
|
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
|
sys.path.append(os.getcwd())
|
|
import wk # pylint: disable=wrong-import-position
|
|
|
|
|
|
# Classes
|
|
class MenuEntry():
|
|
# pylint: disable=too-few-public-methods
|
|
"""Simple class to allow cleaner code below."""
|
|
def __init__(self, name, function, **kwargs):
|
|
self.name = name
|
|
|
|
# Color reboot entries
|
|
if name == 'Reboot':
|
|
self.name = wk.std.color_string(
|
|
['Reboot', ' ', '(Forced)'],
|
|
['YELLOW', None, 'ORANGE'],
|
|
sep='',
|
|
)
|
|
|
|
# Set details
|
|
self.details = {
|
|
'Function': function,
|
|
'Selected': True,
|
|
**kwargs,
|
|
}
|
|
|
|
|
|
# TODO: Deleteme
|
|
TRY_AND_PRINT = wk.std.TryAndPrint()
|
|
TRY_AND_PRINT.width = 50
|
|
def placeholder_function(group, name):
|
|
result = TRY_AND_PRINT.run(f'{name}...', time.sleep, random.randint(1, 3))
|
|
wk.repairs.win.save_settings(group, name, result=result)
|
|
|
|
def placeholder_reboot(group, name):
|
|
print('"Rebooting" shortly...')
|
|
time.sleep(random.randint(1, 3))
|
|
wk.repairs.win.save_settings(group, name, done=True, message='DONE')
|
|
raise SystemExit
|
|
|
|
|
|
# STATIC VARIABLES
|
|
BASE_MENUS = {
|
|
'Groups': {
|
|
'Backup Settings': (
|
|
MenuEntry('Enable RegBack', placeholder_function),
|
|
MenuEntry('Enable System Restore', placeholder_function),
|
|
MenuEntry('Create System Restore', placeholder_function),
|
|
MenuEntry('Backup Browsers', placeholder_function),
|
|
MenuEntry('Backup Power Plans', placeholder_function),
|
|
MenuEntry('Backup Registry', placeholder_function),
|
|
),
|
|
'Windows Repairs': (
|
|
MenuEntry('Disable Windows Updates', placeholder_function),
|
|
MenuEntry('Reset Windows Updates', placeholder_function),
|
|
MenuEntry('Reboot', placeholder_reboot),
|
|
MenuEntry('CHKDSK', placeholder_function),
|
|
MenuEntry('DISM RestoreHealth', wk.repairs.win.auto_dism),
|
|
MenuEntry('SFC Scan', placeholder_function),
|
|
MenuEntry('Fix File Associations', placeholder_function),
|
|
MenuEntry('Clear Proxy Settings', placeholder_function),
|
|
MenuEntry('Disable Pending Renames', placeholder_function),
|
|
MenuEntry('Registry Repairs', placeholder_function),
|
|
MenuEntry('Repair Safe Mode', placeholder_function),
|
|
MenuEntry('Reset UAC', placeholder_function),
|
|
MenuEntry('Reset Windows Policies', placeholder_function),
|
|
),
|
|
'Malware Cleanup': (
|
|
MenuEntry('BleachBit', placeholder_function),
|
|
MenuEntry('HitmanPro', placeholder_function),
|
|
MenuEntry('KVRT', placeholder_function),
|
|
MenuEntry('Windows Defender', placeholder_function),
|
|
MenuEntry('Reboot', placeholder_reboot),
|
|
),
|
|
'Manual Steps': (
|
|
MenuEntry('AdwCleaner', placeholder_function),
|
|
MenuEntry('IO Bit Uninstaller', placeholder_function),
|
|
MenuEntry('Enable Windows Updates', placeholder_function),
|
|
),
|
|
},
|
|
'Options': (
|
|
MenuEntry('Kill Explorer', placeholder_function),
|
|
MenuEntry('Run RKill at startup', placeholder_function),
|
|
MenuEntry('Use Autologon', placeholder_function),
|
|
),
|
|
'Actions': (
|
|
MenuEntry('Options', placeholder_function),
|
|
MenuEntry('Start', placeholder_function, Separator=True),
|
|
MenuEntry('Quit', placeholder_function),
|
|
),
|
|
}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
wk.repairs.win.run_auto_repairs(BASE_MENUS)
|
|
except SystemExit:
|
|
raise
|
|
except: #pylint: disable=bare-except
|
|
wk.std.major_exception()
|