144 lines
4.7 KiB
Python
144 lines
4.7 KiB
Python
'''Wizard Kit: Functions - Cleanup'''
|
|
# pylint: disable=no-name-in-module,wildcard-import
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
from functions.setup import *
|
|
from settings.cleanup import *
|
|
|
|
def cleanup_adwcleaner():
|
|
"""Move AdwCleaner folders into the ClientDir."""
|
|
source_path = r'{SYSTEMDRIVE}\AdwCleaner'.format(**global_vars['Env'])
|
|
source_quarantine = r'{}\Quarantine'.format(source_path)
|
|
|
|
# Quarantine
|
|
if os.path.exists(source_quarantine):
|
|
os.makedirs(global_vars['QuarantineDir'], exist_ok=True)
|
|
dest_name = r'{QuarantineDir}\AdwCleaner_{Date-Time}'.format(
|
|
**global_vars)
|
|
dest_name = non_clobber_rename(dest_name)
|
|
shutil.move(source_quarantine, dest_name)
|
|
|
|
# Delete source folder if empty
|
|
delete_empty_folders(source_path)
|
|
|
|
# Main folder
|
|
if os.path.exists(source_path):
|
|
os.makedirs(global_vars['LogDir'], exist_ok=True)
|
|
dest_name = r'{LogDir}\Tools\AdwCleaner'.format(
|
|
**global_vars)
|
|
dest_name = non_clobber_rename(dest_name)
|
|
shutil.move(source_path, dest_name)
|
|
|
|
|
|
def cleanup_cbs(dest_folder):
|
|
"""Safely cleanup a known CBS archive bug under Windows 7.
|
|
|
|
If a CbsPersist file is larger than 2 Gb then the auto archive feature
|
|
continually fails and will fill up the system drive with temp files.
|
|
|
|
This function moves the temp files and CbsPersist file to a temp folder,
|
|
compresses the CbsPersist files with 7-Zip, and then opens the temp folder
|
|
for the user to manually save the backup files and delete the temp files.
|
|
"""
|
|
backup_folder = r'{dest_folder}\CbsFix'.format(dest_folder=dest_folder)
|
|
temp_folder = r'{backup_folder}\Temp'.format(backup_folder=backup_folder)
|
|
os.makedirs(backup_folder, exist_ok=True)
|
|
os.makedirs(temp_folder, exist_ok=True)
|
|
|
|
# Move files into temp folder
|
|
cbs_path = r'{SYSTEMROOT}\Logs\CBS'.format(**global_vars['Env'])
|
|
for entry in os.scandir(cbs_path):
|
|
# CbsPersist files
|
|
if entry.name.lower().startswith('cbspersist'):
|
|
dest_name = r'{}\{}'.format(temp_folder, entry.name)
|
|
dest_name = non_clobber_rename(dest_name)
|
|
shutil.move(entry.path, dest_name)
|
|
temp_path = r'{SYSTEMROOT}\Temp'.format(**global_vars['Env'])
|
|
for entry in os.scandir(temp_path):
|
|
# cab_ files
|
|
if entry.name.lower().startswith('cab_'):
|
|
dest_name = r'{}\{}'.format(temp_folder, entry.name)
|
|
dest_name = non_clobber_rename(dest_name)
|
|
shutil.move(entry.path, dest_name)
|
|
|
|
# Compress CbsPersist files with 7-Zip
|
|
cmd = [
|
|
global_vars['Tools']['SevenZip'],
|
|
'a', '-t7z', '-mx=3', '-bso0', '-bse0',
|
|
r'{}\CbsPersists.7z'.format(backup_folder),
|
|
r'{}\CbsPersist*'.format(temp_folder)]
|
|
run_program(cmd)
|
|
|
|
|
|
def cleanup_desktop():
|
|
"""Move known backup files and reports into the ClientDir."""
|
|
dest_folder = r'{LogDir}\Tools'.format(**global_vars)
|
|
os.makedirs(dest_folder, exist_ok=True)
|
|
|
|
desktop_path = r'{USERPROFILE}\Desktop'.format(**global_vars['Env'])
|
|
for entry in os.scandir(desktop_path):
|
|
if DESKTOP_ITEMS.search(entry.name):
|
|
dest_name = r'{}\{}'.format(dest_folder, entry.name)
|
|
dest_name = non_clobber_rename(dest_name)
|
|
shutil.move(entry.path, dest_name)
|
|
|
|
# Remove dir if empty
|
|
delete_empty_folders(dest_folder)
|
|
|
|
|
|
def delete_empty_folders(folder_path):
|
|
"""Delete all empty folders in path (depth first)."""
|
|
if not os.path.exists(folder_path) or not os.path.isdir(folder_path):
|
|
# Bail early (silently)
|
|
return
|
|
|
|
# Delete empty subfolders first
|
|
for item in os.scandir(folder_path):
|
|
if item.is_dir():
|
|
delete_empty_folders(item.path)
|
|
|
|
# Remove top folder
|
|
try:
|
|
os.rmdir(folder_path)
|
|
except OSError:
|
|
pass
|
|
|
|
|
|
def delete_registry_key(hive, key, recurse=False):
|
|
"""Delete a registry key and all it's subkeys."""
|
|
access = winreg.KEY_ALL_ACCESS
|
|
|
|
try:
|
|
if recurse:
|
|
# Delete all subkeys first
|
|
with winreg.OpenKeyEx(hive, key, 0, access) as k:
|
|
key_info = winreg.QueryInfoKey(k)
|
|
for x in range(key_info[0]):
|
|
subkey = r'{}\{}'.format(key, winreg.EnumKey(k, 0))
|
|
delete_registry_key(hive, subkey)
|
|
|
|
# Delete key
|
|
winreg.DeleteKey(hive, key)
|
|
except FileNotFoundError:
|
|
# Ignore
|
|
pass
|
|
|
|
|
|
def delete_registry_value(hive, key, value):
|
|
"""Delete a registry value."""
|
|
access = winreg.KEY_ALL_ACCESS
|
|
with winreg.OpenKeyEx(hive, key, 0, access) as k:
|
|
winreg.DeleteValue(k, value)
|
|
|
|
|
|
def restore_default_uac():
|
|
"""Restores default UAC settings via the registry."""
|
|
if global_vars['OS']['Version'] == '10':
|
|
write_registry_settings(UAC_DEFAULTS_WIN10, all_users=True)
|
|
else:
|
|
# Haven't checked Win8 settings, only applying minimum set
|
|
write_registry_settings(UAC_DEFAULTS_WIN7, all_users=True)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print("This file is not meant to be called directly.")
|