58 lines
1.6 KiB
Python
58 lines
1.6 KiB
Python
# Wizard Kit: Check or repair component store health via DISM
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Init
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
from functions.repairs import *
|
|
init_global_vars()
|
|
os.system('title {}: DISM helper Tool'.format(KIT_NAME_FULL))
|
|
set_log_file('DISM Helper.log')
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
stay_awake()
|
|
clear_screen()
|
|
other_results = {
|
|
'Error': {
|
|
'CalledProcessError': 'Unknown Error',
|
|
},
|
|
'Warning': {
|
|
'GenericRepair': 'Repaired',
|
|
'UnsupportedOSError': 'Unsupported OS',
|
|
}}
|
|
disabled = bool(global_vars['OS']['Version'] not in ('8', '8.1', '10'))
|
|
options = [
|
|
{'Name': 'Check Health', 'Repair': False, 'Disabled': disabled},
|
|
{'Name': 'Restore Health', 'Repair': True, 'Disabled': disabled}]
|
|
actions = [{'Name': 'Quit', 'Letter': 'Q'}]
|
|
selection = menu_select(
|
|
'{}: DISM Menu\n'.format(KIT_NAME_FULL),
|
|
main_entries=options,
|
|
action_entries=actions)
|
|
print_info('{}: DISM Menu\n'.format(KIT_NAME_FULL))
|
|
if selection == 'Q':
|
|
abort()
|
|
elif selection.isnumeric():
|
|
repair = options[int(selection)-1]['Repair']
|
|
if repair:
|
|
message='DISM RestoreHealth...'
|
|
else:
|
|
message='DISM ScanHealth...'
|
|
try_and_print(message=message, function=run_dism,
|
|
cs='No corruption', ns='Corruption detected',
|
|
other_results=other_results, repair=repair)
|
|
else:
|
|
abort()
|
|
|
|
# Done
|
|
print_success('Done.')
|
|
pause("Press Enter to exit...")
|
|
exit_script()
|
|
except SystemExit as sys_exit:
|
|
exit_script(sys_exit.code)
|
|
except:
|
|
major_exception()
|
|
|
|
# vim: sts=2 sw=2 ts=2
|