278 lines
13 KiB
Python
278 lines
13 KiB
Python
'''Wizard Kit: System Setup'''
|
|
# pylint: disable=wildcard-import
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import os
|
|
import sys
|
|
|
|
# Init
|
|
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
|
from collections import OrderedDict
|
|
from functions.activation import *
|
|
from functions.browsers import *
|
|
from functions.cleanup import *
|
|
from functions.info import *
|
|
from functions.product_keys import *
|
|
from functions.setup import *
|
|
from functions.sw_diags import *
|
|
init_global_vars()
|
|
os.system('title {}: System Setup'.format(KIT_NAME_FULL))
|
|
set_log_file('System Setup.log')
|
|
|
|
|
|
# STATIC VARIABLES
|
|
# pylint: disable=bad-whitespace,line-too-long
|
|
OTHER_RESULTS = {
|
|
'Error': {
|
|
'BIOSKeyNotFoundError': 'BIOS key not found',
|
|
'CalledProcessError': 'Unknown Error',
|
|
'FileNotFoundError': 'File not found',
|
|
'GenericError': 'Unknown Error',
|
|
'Not4KAlignedError': 'False',
|
|
'SecureBootDisabledError': 'Disabled',
|
|
'WindowsUnsupportedError': 'Unsupported',
|
|
},
|
|
'Warning': {
|
|
'GenericRepair': 'Repaired',
|
|
'NoProfilesError': 'No profiles found',
|
|
'NotInstalledError': 'Not installed',
|
|
'OSInstalledLegacyError': 'OS installed Legacy',
|
|
'SecureBootNotAvailError': 'Not available',
|
|
'SecureBootUnknownError': 'Unknown',
|
|
'UnsupportedOSError': 'Unsupported OS',
|
|
'WindowsOutdatedError': 'Outdated',
|
|
},
|
|
}
|
|
SETUP_ACTIONS = OrderedDict({
|
|
# Install software
|
|
'Installing Programs': {'Info': True},
|
|
'VCR': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_vcredists, 'Just run': True,},
|
|
'ESET NOD32 AV': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_eset_nod32_av, 'If Answer': 'ESET', 'KWArgs': {'scan_pups': False},},
|
|
'LibreOffice': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_libreoffice,
|
|
'If Answer': 'LibreOffice', 'KWArgs': {'quickstart': False, 'register_mso_types': True, 'use_mso_formats': True, 'vcredist': False},
|
|
},
|
|
'Ninite bundle': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_ninite_bundle, 'KWArgs': {'cs': 'Started'},},
|
|
|
|
# Browsers
|
|
'Scanning for browsers': {'Info': True},
|
|
'Scan': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': scan_for_browsers, 'Just run': True, 'KWArgs': {'skip_ie': True},},
|
|
'Backing up browsers': {'Info': True},
|
|
'Backup browsers': {'New': False, 'Fab': True, 'Cur': True, 'HW': False, 'Function': backup_browsers, 'Just run': True,},
|
|
|
|
# Install extensions
|
|
'Installing Extensions': {'Info': True},
|
|
'Classic Shell skin': {'New': True, 'Fab': True, 'Cur': False, 'HW': False, 'Function': install_classicstart_skin, 'Win10 only': True,},
|
|
'Chrome extensions': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_chrome_extensions,},
|
|
'Firefox extensions': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_firefox_extensions,},
|
|
|
|
# Configure software'
|
|
'Configuring Programs': {'Info': True},
|
|
'Browser add-ons': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': install_adblock,
|
|
'Pause': 'Please enable uBlock Origin for all browsers',
|
|
},
|
|
'Classic Start': {'New': True, 'Fab': True, 'Cur': False, 'HW': False, 'Function': config_classicstart, 'Win10 only': True,},
|
|
'Explorer (user)': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': config_explorer_user, 'Win10 only': True,},
|
|
'Explorer (system)': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': config_explorer_system, 'Win10 only': True,},
|
|
#'Disable Fast Startup': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': TODO, 'Win10 only': True,},
|
|
#'Enable Hibernation': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': TODO, 'Win10 only': True,},
|
|
'Disable telemetry': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': disable_windows_telemetry, 'Win10 only': True,},
|
|
'Enable RegBack': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': enable_regback, 'Win10 only': True,},
|
|
'Windows 10 Updates': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': config_windows_updates, 'Win10 only': True,},
|
|
'Enable BSoD mini dumps': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': enable_mini_dumps,},
|
|
'Enable System Restore': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': enable_system_restore,},
|
|
'Update Clock': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': update_clock,},
|
|
'Restart Explorer': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': restart_explorer,},
|
|
|
|
# Cleanup
|
|
'Cleaning up': {'Info': True},
|
|
'AdwCleaner': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': cleanup_adwcleaner,},
|
|
'Desktop': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': cleanup_desktop,},
|
|
'Emsisoft s2cmd': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': cleanup_emsisoft,},
|
|
'Registry Backup(s)': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': cleanup_regbackups,},
|
|
'KIT_NAME_FULL': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': delete_empty_folders,},
|
|
|
|
# System Info
|
|
'Exporting system info': {'Info': True},
|
|
'AIDA64 Report': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': run_aida64,},
|
|
'File listing': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': backup_file_list,},
|
|
'Power plans': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': backup_power_plans,},
|
|
'Product Keys': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': run_produkey,},
|
|
'Registry': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': backup_registry,},
|
|
|
|
# Show Summary
|
|
'Summary': {'Info': True},
|
|
'Operating System': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': show_os_name, 'KWArgs': {'ns': 'Unknown', 'silent_function': False},},
|
|
'Activation': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': show_os_activation, 'KWArgs': {'ns': 'Unknown', 'silent_function': False},},
|
|
'BIOS Activation': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': activate_with_bios, 'If not activated': True,},
|
|
'Secure Boot': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': check_secure_boot_status, 'KWArgs': {'show_alert': True},},
|
|
'Installed RAM': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': show_installed_ram, 'KWArgs': {'ns': 'Unknown', 'silent_function': False},},
|
|
'Show free space': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': show_free_space, 'Just run': True,},
|
|
'Installed AV': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': get_installed_antivirus, 'KWArgs': {'ns': 'Unknown', 'print_return': True},},
|
|
'Installed Office': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': get_installed_office, 'KWArgs': {'ns': 'Unknown', 'print_return': True},},
|
|
'Partitions 4K aligned': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': check_4k_alignment, 'KWArgs': {'cs': 'True', 'ns': 'False'},},
|
|
|
|
# Open things
|
|
'Opening Programs': {'Info': True},
|
|
'Device Manager': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': open_device_manager, 'KWArgs': {'cs': 'Started'},},
|
|
'HWiNFO sensors': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': run_hwinfo_sensors, 'KWArgs': {'cs': 'Started'},},
|
|
'Snappy': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': open_snappy_driver_origin, 'KWArgs': {'cs': 'Started'},},
|
|
'Speed test': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': open_speedtest, 'KWArgs': {'cs': 'Started'},},
|
|
'Windows Updates': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': open_windows_updates, 'KWArgs': {'cs': 'Started'},},
|
|
'Windows Activation': {'New': True, 'Fab': True, 'Cur': True, 'HW': False, 'Function': open_windows_activation, 'If not activated': True, 'KWArgs': {'cs': 'Started'},},
|
|
'Sleep': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': sleep, 'Just run': True, 'KWArgs': {'seconds': 3},},
|
|
'XMPlay': {'New': True, 'Fab': True, 'Cur': True, 'HW': True, 'Function': run_xmplay, 'KWArgs': {'cs': 'Started'},},
|
|
})
|
|
SETUP_ACTION_KEYS = (
|
|
'Function',
|
|
'If answer',
|
|
'If not activated',
|
|
'Just run',
|
|
'KWArgs',
|
|
'Pause',
|
|
'Win10 only',
|
|
)
|
|
SETUP_QUESTIONS = {
|
|
# AV
|
|
'ESET': {'New': None, 'Fab': None, 'Cur': None, 'HW': False},
|
|
'MSE': {'New': None, 'Fab': None, 'Cur': None, 'HW': False},
|
|
|
|
# LibreOffice
|
|
'LibreOffice':{'New': None, 'Fab': None, 'Cur': None, 'HW': False},
|
|
|
|
# Ninite
|
|
'Base': {'New': True, 'Fab': True, 'Cur': True, 'HW': False},
|
|
'Current': {'New': False, 'Fab': False, 'Cur': True, 'HW': False},
|
|
'Missing': {'New': False, 'Fab': True, 'Cur': False, 'HW': False},
|
|
'Standard': {'New': True, 'Fab': True, 'Cur': False, 'HW': False},
|
|
}
|
|
# pylint: enable=bad-whitespace,line-too-long
|
|
|
|
|
|
# Functions
|
|
def get_actions(setup_mode):
|
|
"""Get actions to perform based on setup_mode, returns OrderedDict."""
|
|
actions = OrderedDict({})
|
|
for _key, _val in SETUP_ACTIONS.items():
|
|
_action = {'Enabled': _val.get(setup_mode, False)}
|
|
for _sub_key in SETUP_ACTION_KEYS:
|
|
_action[_sub_key] = _val.get(_sub_key, None)
|
|
actions[_key] = _action
|
|
|
|
return actions
|
|
|
|
|
|
def get_answers(setup_mode):
|
|
"""Get setup answers based on setup_mode and user input, returns dict."""
|
|
answers = {k: v.get(setup_mode, False) for k, v in SETUP_QUESTIONS.items()}
|
|
|
|
# Answer setup questions as needed
|
|
if answers['ESET'] is None or answers['MSE'] is None:
|
|
answers.update(get_av_selection())
|
|
|
|
if answers['LibreOffice'] is None:
|
|
answers['LibreOffice'] = ask('Install LibreOffice?')
|
|
|
|
return answers
|
|
|
|
|
|
def get_av_selection():
|
|
"""Get AV selection."""
|
|
av_answers = {
|
|
'ESET': False,
|
|
'MSE': False,
|
|
}
|
|
av_options = [
|
|
{'Name': 'ESET NOD32 AV'},
|
|
{
|
|
'Name': 'Microsoft Security Essentials',
|
|
'Disabled': global_vars['OS']['Version'] not in ['7'],
|
|
},
|
|
]
|
|
actions = [
|
|
{'Name': 'None', 'Letter': 'N'},
|
|
{'Name': 'Quit', 'Letter': 'Q'},
|
|
]
|
|
|
|
# Show menu
|
|
selection = menu_select(
|
|
'Please select an option to install',
|
|
main_entries=av_options,
|
|
action_entries=actions)
|
|
if selection.isnumeric():
|
|
index = int(selection) - 1
|
|
if 'ESET' in av_options[index]['Name']:
|
|
av_answers['ESET'] = True
|
|
av_answers['MSE'] = False
|
|
elif 'Microsoft' in av_options[index]['Name']:
|
|
av_answers['ESET'] = False
|
|
av_answers['MSE'] = True
|
|
elif selection == 'Q':
|
|
abort()
|
|
|
|
return av_answers
|
|
|
|
|
|
def get_mode():
|
|
"""Get mode via menu_select, returns str."""
|
|
setup_mode = None
|
|
mode_options = [
|
|
{'Name': 'New', 'Display Name': 'New / Clean install (no data)'},
|
|
{'Name': 'Fab', 'Display Name': 'Clean install with data migration'},
|
|
{'Name': 'Cur', 'Display Name': 'Original OS (post-d7II or overinstall)'},
|
|
{'Name': 'HW', 'Display Name': 'Hardware service (i.e. no software work)'},
|
|
]
|
|
actions = [
|
|
{'Name': 'Quit', 'Letter': 'Q'},
|
|
]
|
|
|
|
# Get selection
|
|
selection = menu_select(
|
|
'Please select a setup mode',
|
|
main_entries=mode_options,
|
|
action_entries=actions)
|
|
if selection.isnumeric():
|
|
index = int(selection) - 1
|
|
setup_mode = mode_options[index]['Name']
|
|
elif selection == 'Q':
|
|
abort()
|
|
|
|
return setup_mode
|
|
|
|
|
|
def main():
|
|
"""Main function."""
|
|
stay_awake()
|
|
clear_screen()
|
|
|
|
# Check installed OS
|
|
result = try_and_print(
|
|
message='OS support status...',
|
|
function=check_os_support_status,
|
|
cs='GOOD',
|
|
)
|
|
if not result['CS'] and 'Unsupported' in result['Error']:
|
|
print_warning('OS version not supported by this script')
|
|
if not ask('Continue anyway? (NOT RECOMMENDED)'):
|
|
abort()
|
|
|
|
# Get setup mode
|
|
setup_mode = get_mode()
|
|
|
|
# Get actions to perform
|
|
actions = get_actions(setup_mode)
|
|
|
|
# Get answers to setup questions
|
|
answers = get_answers(setup_mode)
|
|
|
|
# Perform actions
|
|
# TODO
|
|
|
|
|
|
if __name__ == '__main__':
|
|
try:
|
|
main()
|
|
exit_script()
|
|
except SystemExit as sys_exit:
|
|
exit_script(sys_exit.code)
|
|
except: # pylint: disable=bare-except
|
|
major_exception()
|