b37a492db5
## MAJOR refactoring in progress ##
# BROKEN SCRIPTS #
(These scripts need updated to use the new global_vars & try_and_print)
* user_data_transfer.py
* install_sw_bundle.py
* All Python scripts
* PLAN: Replace vars_wk with global_vars
* functions.py
* PLAN: Will hold (nearly?) all functions going forward
* Should simplify the remaining scripts
* (e.g. reset_browsers and sw_diagnostics both had a backup_browsers())
* PLAN: Move all program definitions to functions.py
* (e.g. vars_wk['SevenZip'] --> global_vars['Programs']['SevenZip'])
* Added major_exception()
* Should be called if anything goes wrong to alert user and provide info
* Added non_clobber_rename()
* Appends a number (if necessary) to avoid overwriting data
* Added popen_program()
* Removes subprocess req for sw_checklist
* Added try_and_print()
* Run a function and show CS/NS/etc
* Can add additional "result" strings via Exception classes
* Passes unrecognized args/kwargs to the function
* Used by scripts to standardize message formatting and error handling
* exit_script() now handles opening the log (if set) and killing caffeine.exe
* Refactored init_vars_wk/os code to work with global_vars
* Uses the try_and_print function for better formatting and error handling
* BREAKING: 'OS' vars now located in global_vars['OS'] (including 'Arch')
* OS labeled as 'outdated', 'very outdated', and 'unrecognized' as needed
* BREAKING: renamed PROGRAMS to TOOLS
* Expanded TOOLS
* Removed log_file as a required argument from all functions
* Removed vars_wk/global_vars as a required argument from all functions
* Removed unused WinPE functions
* Sorted functions
* user_checklist() (formerly reset_browsers())
* Most functions moved to functions.py (see above)
* Browsers
* Work has been split into backup, reset, and config sections
* Script now asks all questions before starting backup/reset/config
* Script now forces the homepage to DEFAULT_HOMEPAGE (set in functions.py)
* (i.e. it no longer asks first, it just does it)
* Current homepages are listed where possible before backup/reset/config
* Launch/Launchers
* Added QuickBooks support
* Removed launchers
* CPU-Z
* HWMonitor
* PerfMonitor2
* SanDisk Express Cache
* Shortcut Cleaner
* SuperAntiSpyware (SAS)
* Replaced Notepad2-Mod with Notepad++
101 lines
3.7 KiB
Python
101 lines
3.7 KiB
Python
# Wizard Kit: Activate Windows using various methods
|
|
|
|
import csv
|
|
import os
|
|
import re
|
|
import subprocess
|
|
|
|
# Init
|
|
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
|
os.system('title Wizard Kit: Windows Activation Tool')
|
|
from functions import *
|
|
init_global_vars()
|
|
set_global_vars()
|
|
|
|
def abort():
|
|
print_warning('Aborted.')
|
|
exit_script()
|
|
|
|
def activate_with_bios():
|
|
"""Attempt to activate Windows with a key stored in the BIOS."""
|
|
extract_item('ProduKey', silent=True)
|
|
_args = [
|
|
'/nosavereg',
|
|
'/scomma', '{TmpDir}\\keys.csv'.format(**global_vars),
|
|
'/WindowsKeys', '1',
|
|
'/OfficeKeys', '0',
|
|
'/IEKeys', '0',
|
|
'/SQLKeys', '0',
|
|
'/ExchangeKeys', '0']
|
|
try:
|
|
run_program(global_vars['Tools']['ProduKey'], _args, pipe=False)
|
|
except subprocess.CalledProcessError:
|
|
print_error('ERROR: Failed to extract BIOS key')
|
|
abort()
|
|
else:
|
|
with open ('{TmpDir}\\keys.csv'.format(**global_vars), newline='') as key_file:
|
|
key_reader = csv.reader(key_file)
|
|
_key_found = False
|
|
for key in key_reader:
|
|
if 'BIOS' in key[0] and re.match(r'^\w{5}-\w{5}-\w{5}-\w{5}-\w{5}$', key[2]):
|
|
_key_found = True
|
|
print_standard('BIOS key found, installing...')
|
|
run_program('cscript {SYSTEMROOT}\\System32\\slmgr.vbs /ipk {pkey} //nologo'.format(**global_vars['Env'], pkey=key[2]), check=False, shell=True)
|
|
sleep(15)
|
|
print_standard('Attempting activation...')
|
|
run_program('cscript {SYSTEMROOT}\\System32\\slmgr.vbs /ato //nologo'.format(**global_vars['Env']), check=False, shell=True)
|
|
sleep(15)
|
|
# Open system properties for user verification
|
|
popen_program(['control', 'system'])
|
|
break
|
|
if not _key_found:
|
|
print_error('ERROR: BIOS not key found.')
|
|
abort()
|
|
|
|
def activate_with_hive():
|
|
"""Scan any transferred software hives for Windows keys and attempt activation."""
|
|
extract_item('ProduKey', silent=True)
|
|
|
|
def is_activated():
|
|
"""Updates activation status, checks if activated, and returns a bool."""
|
|
_out = run_program('cscript /nologo {SYSTEMROOT}\\System32\\slmgr.vbs /xpr'.format(**global_vars['Env']))
|
|
_out = _out.stdout.decode().splitlines()
|
|
_out = [l for l in _out if re.match(r'^\s', l)]
|
|
if len(_out) > 0:
|
|
global_vars['OS']['Activation'] = re.sub(r'^\s+', '', _out[0])
|
|
else:
|
|
global_vars['OS']['Activation'] = 'Activation status unknown'
|
|
return 'The machine is permanently activated.' in global_vars['OS']['Activation']
|
|
|
|
if __name__ == '__main__':
|
|
stay_awake()
|
|
# Bail early if already activated
|
|
if is_activated():
|
|
print_info('This system is already activated')
|
|
exit_script()
|
|
|
|
# Determine activation method
|
|
activation_methods = [
|
|
{'Name': 'Activate with BIOS key', 'Function': activate_with_bios},
|
|
{'Name': 'Activate with transferred SW hive', 'Function': activate_with_hive, 'Disabled': True},
|
|
]
|
|
if not re.match(r'^(8|10)$', global_vars['OS']['Version']):
|
|
activation_methods[0]['Disabled'] = True
|
|
actions = [
|
|
{'Name': 'Quit', 'Letter': 'Q'},
|
|
]
|
|
|
|
# Main loop
|
|
while True:
|
|
selection = menu_select('Wizard Kit: Windows Activation Menu', activation_methods, actions)
|
|
|
|
if (selection.isnumeric()):
|
|
activation_methods[int(selection)-1]['Function']()
|
|
break
|
|
elif selection == 'Q':
|
|
exit_script()
|
|
|
|
# Done
|
|
print_success('Done.')
|
|
pause("Press Enter to exit...")
|
|
exit_script()
|