2Shirt/WizardKit
2Shirt/WizardKit
1
0
Fork 0
Code Issues 2 Pull requests Projects Releases 26 Packages Wiki Activity Actions

Drop extraneous SYSTEMDRIVE lookups

Browse source
This commit is contained in:
2Shirt 2021-05-01 17:00:48 -06:00
parent f706a48f60
commit cf8b600dd5
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
1 changed files with 18 additions and 14 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

32
scripts/wk/repairs/win.py
View file

@ -10,14 +10,19 @@ import re
import sys
import time
from subprocess import CalledProcessError, DEVNULL
from subprocess import CalledProcessError, DEVNULL
from wk.cfg.main import KIT_NAME_FULL
from wk.exe import get_procs, run_program, popen_program, wait_for_procs
from wk.io import delete_folder, rename_item
from wk.kit.tools import ARCH, get_tool_path, run_tool
from wk.log import format_log_path, update_log_path
from wk.os.win import (
from wk.cfg.main import KIT_NAME_FULL
from wk.exe import (
get_procs,
run_program,
popen_program,
wait_for_procs,
)
from wk.io import delete_folder, rename_item
from wk.kit.tools import ARCH, get_tool_path, run_tool
from wk.log import format_log_path, update_log_path
from wk.os.win import (
reg_delete_value,
reg_read_value,
reg_set_value,
@ -26,7 +31,7 @@ from wk.os.win import (
enable_service,
stop_service,
)
from wk.std import (
from wk.std import (
GenericError,
GenericWarning,
Menu,
@ -117,7 +122,7 @@ RKILL_WHITELIST = (
fr'{PROGRAMFILES_32}\TeamViewer\tv_x64.exe',
sys.executable,
)
SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE')
SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE', 'C:')
WIDTH = 50
TRY_PRINT = TryAndPrint()
TRY_PRINT.width = WIDTH
@ -553,14 +558,13 @@ def auto_bleachbit(group, name):
def auto_chkdsk(group, name):
"""Run CHKDSK repairs."""
needs_reboot = False
system_disk = os.environ.get('SYSTEMDRIVE', 'C:')
result = TRY_PRINT.run(f'CHKDSK ({system_disk})...', run_chkdsk_online)
result = TRY_PRINT.run(f'CHKDSK ({SYSTEMDRIVE})...', run_chkdsk_online)
# Run offline CHKDSK if required
if result['Failed'] and 'Repaired' not in result['Message']:
needs_reboot = True
result = TRY_PRINT.run(
f'Scheduling offline CHKDSK ({system_disk})...',
f'Scheduling offline CHKDSK ({SYSTEMDRIVE})...',
run_chkdsk_offline,
)
if not result['Failed']:
@ -918,7 +922,7 @@ def restore_uac_defaults():
def run_chkdsk_offline():
"""Set filesystem 'dirty bit' to force a CHKDSK during startup."""
cmd = ['fsutil', 'dirty', 'set', os.environ.get('SYSTEMDRIVE', 'C:')]
cmd = ['fsutil', 'dirty', 'set', SYSTEMDRIVE]
proc = run_program(cmd, check=False)
# Check result
@ -931,7 +935,7 @@ def run_chkdsk_online():
NOTE: If run on Windows 8+ online repairs are attempted.
"""
cmd = ['CHKDSK', os.environ.get('SYSTEMDRIVE', 'C:')]
cmd = ['CHKDSK', SYSTEMDRIVE]
if OS_VERSION >= 8:
cmd.extend(['/scan', '/perf'])
if IN_CONEMU: