110 lines
3.6 KiB
Python
110 lines
3.6 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=None, **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', 'auto_enable_regback'),
|
|
MenuEntry('Enable System Restore', 'auto_system_restore_enable'),
|
|
MenuEntry('Set System Restore Size', 'auto_system_restore_set_size'),
|
|
MenuEntry('Create System Restore', 'auto_system_restore_create'),
|
|
MenuEntry('Backup Browsers', placeholder_function),
|
|
MenuEntry('Backup Power Plans', 'auto_backup_power_plans'),
|
|
MenuEntry('Backup Registry', 'auto_backup_registry'),
|
|
),
|
|
'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', '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'),
|
|
MenuEntry('Run RKill at startup'),
|
|
MenuEntry('Use Autologon'),
|
|
),
|
|
'Actions': (
|
|
MenuEntry('Options'),
|
|
MenuEntry('Start', Separator=True),
|
|
MenuEntry('Quit'),
|
|
),
|
|
}
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
wk.repairs.win.run_auto_repairs(BASE_MENUS)
|
|
except SystemExit:
|
|
raise
|
|
except: #pylint: disable=bare-except
|
|
wk.std.major_exception()
|