Compare commits
1 commit
dev
...
pulseaudio
| Author | SHA1 | Date | |
|---|---|---|---|
8d70c7bfc6
|
574 changed files with 9653 additions and 10387 deletions
|
|
@ -1,6 +0,0 @@
|
|||
repos:
|
||||
- repo: https://github.com/charliermarsh/ruff-pre-commit
|
||||
# Ruff version.
|
||||
rev: 'v0.0.257'
|
||||
hooks:
|
||||
- id: ruff
|
||||
|
|
@ -1,4 +1,4 @@
|
|||
Copyright (c) 2023 Alan Mason
|
||||
Copyright (c) 2021 Alan Mason
|
||||
|
||||
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
|
||||
|
||||
|
|
|
|||
|
|
@ -242,7 +242,7 @@ if defined L_NCMD (
|
|||
rem use Powershell's window instead of %CON%
|
||||
echo UAC.ShellExecute "%POWERSHELL%", "%ps_args% -File "%script%"", "", "runas", 3 >> "%bin%\tmp\Elevate.vbs"
|
||||
) else (
|
||||
echo UAC.ShellExecute "%CON%", "-run %POWERSHELL% %ps_args% -File "^"%script%^"" -new_console:n", "", "runas", 1 >> "%bin%\tmp\Elevate.vbs"
|
||||
echo UAC.ShellExecute "%CON%", "-run %POWERSHELL% %ps_args% -File "%script%" -new_console:n", "", "runas", 1 >> "%bin%\tmp\Elevate.vbs"
|
||||
)
|
||||
|
||||
rem Run
|
||||
|
|
|
|||
|
|
@ -1,2 +1,8 @@
|
|||
# WizardKit: Scripts #
|
||||
|
||||
## pylint ##
|
||||
|
||||
These scripts use two spaces per indent instead of the default four. As such you will need to update your pylintrc file or run like this:
|
||||
|
||||
`pylint --indent-after-paren=2 --indent-string=' ' wk`
|
||||
|
||||
|
|
|
|||
|
|
@ -1,25 +1,33 @@
|
|||
"""WizardKit: Auto Repair Tool"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from typing import Any
|
||||
import os
|
||||
import sys
|
||||
|
||||
import wk
|
||||
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
||||
sys.path.append(os.getcwd())
|
||||
import wk # pylint: disable=wrong-import-position
|
||||
|
||||
|
||||
# Classes
|
||||
REBOOT_STR = wk.ui.ansi.color_string('Reboot', 'YELLOW')
|
||||
class MenuEntry():
|
||||
# pylint: disable=too-few-public-methods
|
||||
"""Simple class to allow cleaner code below."""
|
||||
def __init__(
|
||||
self,
|
||||
name: str,
|
||||
function: str | None = None,
|
||||
selected: bool = True,
|
||||
**kwargs):
|
||||
self.name: str = name
|
||||
self.details: dict[str, Any] = {
|
||||
def __init__(self, name, function=None, **kwargs):
|
||||
self.name = name
|
||||
|
||||
# Color reboot entries
|
||||
if name == 'Reboot':
|
||||
self.name = wk.std.color_string(
|
||||
['Reboot', ' ', '(Forced)'],
|
||||
['YELLOW', None, 'ORANGE'],
|
||||
sep='',
|
||||
)
|
||||
|
||||
# Set details
|
||||
self.details = {
|
||||
'Function': function,
|
||||
'Selected': selected,
|
||||
'Selected': True,
|
||||
**kwargs,
|
||||
}
|
||||
|
||||
|
|
@ -41,7 +49,7 @@ BASE_MENUS = {
|
|||
'Windows Repairs': (
|
||||
MenuEntry('Disable Windows Updates', 'auto_windows_updates_disable'),
|
||||
MenuEntry('Reset Windows Updates', 'auto_windows_updates_reset'),
|
||||
MenuEntry(REBOOT_STR, 'auto_reboot'),
|
||||
MenuEntry('Reboot', 'auto_reboot'),
|
||||
MenuEntry('CHKDSK', 'auto_chkdsk'),
|
||||
MenuEntry('DISM RestoreHealth', 'auto_dism'),
|
||||
MenuEntry('SFC Scan', 'auto_sfc'),
|
||||
|
|
@ -57,44 +65,35 @@ BASE_MENUS = {
|
|||
MenuEntry('KVRT', 'auto_kvrt'),
|
||||
MenuEntry('Windows Defender', 'auto_microsoft_defender'),
|
||||
MenuEntry('Remove Custom Power Plan', 'auto_remove_power_plan'),
|
||||
MenuEntry(REBOOT_STR, 'auto_reboot'),
|
||||
MenuEntry('Reboot', 'auto_reboot'),
|
||||
),
|
||||
'Manual Steps': (
|
||||
MenuEntry('AdwCleaner', 'auto_adwcleaner'),
|
||||
MenuEntry('Bulk Crap Uninstaller', 'auto_bcuninstaller'),
|
||||
MenuEntry('IO Bit Uninstaller', 'auto_iobit_uninstaller'),
|
||||
MenuEntry('Enable Windows Updates', 'auto_windows_updates_enable'),
|
||||
),
|
||||
},
|
||||
'Options': (
|
||||
MenuEntry('Kill Explorer', selected=False),
|
||||
MenuEntry('Run AVRemover (once)'),
|
||||
MenuEntry('Kill Explorer'),
|
||||
MenuEntry('Run RKill'),
|
||||
MenuEntry('Run TDSSKiller (once)'),
|
||||
MenuEntry('Sync Clock'),
|
||||
MenuEntry('Use Autologon', selected=False),
|
||||
MenuEntry('Use Autologon'),
|
||||
),
|
||||
'Actions': (
|
||||
MenuEntry('Load Preset'),
|
||||
MenuEntry('Options'),
|
||||
MenuEntry('Start', Separator=True),
|
||||
MenuEntry('Quit'),
|
||||
),
|
||||
}
|
||||
PRESETS = {
|
||||
'Default': { # Will be expanded at runtime using BASE_MENUS
|
||||
'Options': (
|
||||
'Run RKill',
|
||||
'Sync Clock',
|
||||
),
|
||||
},
|
||||
'Custom': {}, # Will remain empty at runtime
|
||||
}
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
wk.repairs.win.run_auto_repairs(BASE_MENUS, PRESETS)
|
||||
wk.repairs.win.run_auto_repairs(BASE_MENUS)
|
||||
except KeyboardInterrupt:
|
||||
wk.ui.cli.abort()
|
||||
wk.std.abort()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,94 +1,20 @@
|
|||
"""WizardKit: Auto System Setup Tool"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from typing import Any
|
||||
import os
|
||||
import sys
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
# Classes
|
||||
class MenuEntry():
|
||||
"""Simple class to allow cleaner code below."""
|
||||
def __init__(
|
||||
self,
|
||||
name: str,
|
||||
function: str | None = None,
|
||||
selected: bool = True,
|
||||
**kwargs):
|
||||
self.name: str = name
|
||||
self.details: dict[str, Any] = {
|
||||
'Function': function,
|
||||
'Selected': selected,
|
||||
**kwargs,
|
||||
}
|
||||
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
||||
sys.path.append(os.getcwd())
|
||||
import wk # pylint: disable=wrong-import-position
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
BASE_MENUS = {
|
||||
'Groups': {
|
||||
'Backup Settings': (
|
||||
MenuEntry('Backup Browsers', 'auto_backup_browser_profiles'),
|
||||
MenuEntry('Backup Power Plans', 'auto_backup_power_plans'),
|
||||
MenuEntry('Reset Power Plans', 'auto_reset_power_plans'),
|
||||
MenuEntry('Set Custom Power Plan', 'auto_set_custom_power_plan'),
|
||||
),
|
||||
'Install Software': (
|
||||
MenuEntry('Winget', 'auto_install_winget'),
|
||||
MenuEntry('Firefox', 'auto_install_firefox'),
|
||||
MenuEntry('LibreOffice', 'auto_install_libreoffice', selected=False),
|
||||
MenuEntry('Open Shell', 'auto_install_open_shell'),
|
||||
MenuEntry('Software Bundle', 'auto_install_software_bundle'),
|
||||
MenuEntry('Software Upgrades', 'auto_install_software_upgrades'),
|
||||
MenuEntry('Visual C++ Runtimes', 'auto_install_vcredists'),
|
||||
),
|
||||
'Configure System': (
|
||||
MenuEntry('Open Shell', 'auto_config_open_shell'),
|
||||
MenuEntry('Disable Password Expiration', 'auto_disable_password_expiration'),
|
||||
MenuEntry('Enable BSoD MiniDumps', 'auto_enable_bsod_minidumps'),
|
||||
MenuEntry('Enable RegBack', 'auto_enable_regback'),
|
||||
MenuEntry('Enable System Restore', 'auto_system_restore_enable'),
|
||||
MenuEntry('Set System Restore Size', 'auto_system_restore_set_size'),
|
||||
MenuEntry('Enable Windows Updates', 'auto_windows_updates_enable'),
|
||||
MenuEntry('Windows Activation', 'auto_activate_windows'),
|
||||
MenuEntry('Windows Explorer', 'auto_config_explorer'),
|
||||
MenuEntry(r'Windows\Temp Fix', 'auto_windows_temp_fix'),
|
||||
MenuEntry('Configure Browsers', 'auto_config_browsers'),
|
||||
MenuEntry('Create System Restore', 'auto_system_restore_create'),
|
||||
),
|
||||
'System Information': (
|
||||
MenuEntry('AIDA64 Report', 'auto_export_aida64_report'),
|
||||
MenuEntry('Backup Registry', 'auto_backup_registry'),
|
||||
),
|
||||
'System Summary': (
|
||||
MenuEntry('Operating System', 'auto_show_os_name'),
|
||||
MenuEntry('Windows Activation', 'auto_show_os_activation'),
|
||||
MenuEntry('Secure Boot', 'auto_show_secure_boot_status'),
|
||||
MenuEntry('Installed RAM', 'auto_show_installed_ram'),
|
||||
MenuEntry('Storage Status', 'auto_show_storage_status'),
|
||||
MenuEntry('Virus Protection', 'auto_show_installed_antivirus'),
|
||||
MenuEntry('Partitions 4K Aligned', 'auto_show_4k_alignment_check'),
|
||||
),
|
||||
'Run Programs': (
|
||||
MenuEntry('Device Manager', 'auto_open_device_manager'),
|
||||
MenuEntry('HWiNFO Sensors', 'auto_open_hwinfo_sensors'),
|
||||
MenuEntry('Microsoft Store Updates', 'auto_open_microsoft_store_updates'),
|
||||
MenuEntry('Snappy Driver Installer', 'auto_open_snappy_driver_installer_origin'),
|
||||
MenuEntry('Windows Activation', 'auto_open_windows_activation'),
|
||||
MenuEntry('Windows Updates', 'auto_open_windows_updates'),
|
||||
MenuEntry('XMPlay', 'auto_open_xmplay'),
|
||||
),
|
||||
},
|
||||
'Actions': (
|
||||
MenuEntry('Load Preset'),
|
||||
MenuEntry('Start', Separator=True),
|
||||
MenuEntry('Quit'),
|
||||
),
|
||||
}
|
||||
PRESETS = {
|
||||
'Default': {}, # Will be built at runtime using BASE_MENUS
|
||||
'Additional User': {
|
||||
'Configure System': (
|
||||
'Configure Browsers',
|
||||
'Chrome Notifications',
|
||||
'Open Shell',
|
||||
'uBlock Origin',
|
||||
'Enable BSoD MiniDumps',
|
||||
|
|
@ -101,9 +27,6 @@ PRESETS = {
|
|||
'Install Software': (
|
||||
'Firefox', # Needed to handle profile upgrade nonsense
|
||||
),
|
||||
'Run Programs': (
|
||||
'Microsoft Store Updates',
|
||||
),
|
||||
'System Summary': (
|
||||
'Operating System',
|
||||
'Windows Activation',
|
||||
|
|
@ -163,13 +86,83 @@ PRESETS = {
|
|||
'Custom': {}, # Will remain empty at runtime
|
||||
}
|
||||
|
||||
# Classes
|
||||
class MenuEntry():
|
||||
# pylint: disable=too-few-public-methods
|
||||
"""Simple class to allow cleaner code below."""
|
||||
def __init__(self, name, function=None, selected=True, **kwargs):
|
||||
self.name = name
|
||||
self.details = {
|
||||
'Function': function,
|
||||
'Selected': selected,
|
||||
**kwargs,
|
||||
}
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
BASE_MENUS = {
|
||||
'Groups': {
|
||||
'Backup Settings': (
|
||||
MenuEntry('Backup Browsers', 'auto_backup_browser_profiles'),
|
||||
MenuEntry('Backup Power Plans', 'auto_backup_power_plans'),
|
||||
MenuEntry('Reset Power Plans', 'auto_reset_power_plans'),
|
||||
MenuEntry('Set Custom Power Plan', 'auto_set_custom_power_plan'),
|
||||
),
|
||||
'Install Software': (
|
||||
MenuEntry('Visual C++ Runtimes', 'auto_install_vcredists'),
|
||||
MenuEntry('Firefox', 'auto_install_firefox'),
|
||||
MenuEntry('LibreOffice', 'auto_install_libreoffice', selected=False),
|
||||
MenuEntry('Open Shell', 'auto_install_open_shell'),
|
||||
MenuEntry('Software Bundle', 'auto_install_software_bundle'),
|
||||
),
|
||||
'Configure System': (
|
||||
MenuEntry('Configure Browsers', 'auto_config_browsers'),
|
||||
MenuEntry('Open Shell', 'auto_config_open_shell'),
|
||||
MenuEntry('Enable BSoD MiniDumps', 'auto_enable_bsod_minidumps'),
|
||||
MenuEntry('Enable RegBack', 'auto_enable_regback'),
|
||||
MenuEntry('Enable System Restore', 'auto_system_restore_enable'),
|
||||
MenuEntry('Set System Restore Size', 'auto_system_restore_set_size'),
|
||||
MenuEntry('Enable Windows Updates', 'auto_windows_updates_enable'),
|
||||
MenuEntry('Windows Activation', 'auto_activate_windows'),
|
||||
MenuEntry('Windows Explorer', 'auto_config_explorer'),
|
||||
MenuEntry(r'Windows\Temp Fix', 'auto_windows_temp_fix'),
|
||||
MenuEntry('Create System Restore', 'auto_system_restore_create'),
|
||||
),
|
||||
'System Information': (
|
||||
MenuEntry('AIDA64 Report', 'auto_export_aida64_report'),
|
||||
MenuEntry('Backup Registry', 'auto_backup_registry'),
|
||||
),
|
||||
'System Summary': (
|
||||
MenuEntry('Operating System', 'auto_show_os_name'),
|
||||
MenuEntry('Windows Activation', 'auto_show_os_activation'),
|
||||
MenuEntry('Secure Boot', 'auto_show_secure_boot_status'),
|
||||
MenuEntry('Installed RAM', 'auto_show_installed_ram'),
|
||||
MenuEntry('Storage Status', 'auto_show_storage_status'),
|
||||
MenuEntry('Virus Protection', 'auto_show_installed_antivirus'),
|
||||
MenuEntry('Partitions 4K Aligned', 'auto_show_4k_alignment_check'),
|
||||
),
|
||||
'Run Programs': (
|
||||
MenuEntry('Device Manager', 'auto_open_device_manager'),
|
||||
MenuEntry('HWiNFO Sensors', 'auto_open_hwinfo_sensors'),
|
||||
MenuEntry('Windows Activation', 'auto_open_windows_activation'),
|
||||
MenuEntry('Windows Updates', 'auto_open_windows_updates'),
|
||||
MenuEntry('XMPlay', 'auto_open_xmplay'),
|
||||
),
|
||||
},
|
||||
'Actions': (
|
||||
MenuEntry('Load Preset'),
|
||||
MenuEntry('Start', Separator=True),
|
||||
MenuEntry('Quit'),
|
||||
),
|
||||
}
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
wk.setup.win.run_auto_setup(BASE_MENUS, PRESETS)
|
||||
except KeyboardInterrupt:
|
||||
wk.ui.cli.abort()
|
||||
wk.std.abort()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -10,5 +10,5 @@ if __name__ == '__main__':
|
|||
wk.kit.ufd.build_ufd()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,15 +1,20 @@
|
|||
"""WizardKit: Build Kit (Windows)."""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import wk
|
||||
import os
|
||||
import sys
|
||||
|
||||
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
||||
sys.path.append(os.getcwd())
|
||||
import wk # pylint: disable=wrong-import-position
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
wk.kit.build.build_kit()
|
||||
except KeyboardInterrupt:
|
||||
wk.ui.cli.abort()
|
||||
wk.std.abort()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,13 +0,0 @@
|
|||
# WizardKit: Check Antivirus
|
||||
|
||||
#Requires -Version 3.0
|
||||
if (Test-Path Env:\DEBUG) {
|
||||
Set-PSDebug -Trace 1
|
||||
}
|
||||
$Host.UI.RawUI.WindowTitle = "WizardKit: Check Antivirus"
|
||||
$Host.UI.RawUI.BackgroundColor = "black"
|
||||
$Host.UI.RawUI.ForegroundColor = "white"
|
||||
$ProgressPreference = "SilentlyContinue"
|
||||
|
||||
# Main
|
||||
Get-CimInstance -Namespace "root\SecurityCenter2" -ClassName AntivirusProduct | select displayName,productState | ConvertTo-Json
|
||||
|
|
@ -1,13 +0,0 @@
|
|||
# WizardKit: Check Partition Alignment
|
||||
|
||||
#Requires -Version 3.0
|
||||
if (Test-Path Env:\DEBUG) {
|
||||
Set-PSDebug -Trace 1
|
||||
}
|
||||
$Host.UI.RawUI.WindowTitle = "WizardKit: Check Partition Alignment"
|
||||
$Host.UI.RawUI.BackgroundColor = "black"
|
||||
$Host.UI.RawUI.ForegroundColor = "white"
|
||||
$ProgressPreference = "SilentlyContinue"
|
||||
|
||||
# Main
|
||||
Get-CimInstance -Query "Select * from Win32_DiskPartition" | select Name,Size,StartingOffset | ConvertTo-Json
|
||||
|
|
@ -1,14 +1,24 @@
|
|||
#!/usr/bin/env python3
|
||||
"""WizardKit: ddrescue TUI"""
|
||||
# pylint: disable=invalid-name
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from docopt import docopt
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
docopt(wk.clone.ddrescue.DOCSTRING)
|
||||
except SystemExit:
|
||||
print('')
|
||||
wk.std.pause('Press Enter to exit...')
|
||||
raise
|
||||
|
||||
try:
|
||||
wk.clone.ddrescue.main()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,13 +0,0 @@
|
|||
# WizardKit: Disable Password Expiration (Local Accounts)
|
||||
|
||||
#Requires -Version 3.0
|
||||
if (Test-Path Env:\DEBUG) {
|
||||
Set-PSDebug -Trace 1
|
||||
}
|
||||
$Host.UI.RawUI.WindowTitle = "Disable Password Expiration"
|
||||
$Host.UI.RawUI.BackgroundColor = "black"
|
||||
$Host.UI.RawUI.ForegroundColor = "white"
|
||||
$ProgressPreference = "SilentlyContinue"
|
||||
|
||||
# Main
|
||||
Get-LocalUser | Set-LocalUser -PasswordNeverExpires $true
|
||||
|
|
@ -5,18 +5,13 @@ python.exe -i embedded_python_env.py
|
|||
"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import pickle
|
||||
import wk
|
||||
import os
|
||||
import sys
|
||||
|
||||
|
||||
# Functions
|
||||
def load_state():
|
||||
with open('debug/state.pickle', 'rb') as f:
|
||||
return pickle.load(f)
|
||||
|
||||
|
||||
# Main
|
||||
wk.ui.cli.print_colored(
|
||||
os.chdir(os.path.dirname(os.path.realpath(__file__)))
|
||||
sys.path.append(os.getcwd())
|
||||
import wk # pylint: disable=wrong-import-position
|
||||
wk.std.print_colored(
|
||||
(wk.cfg.main.KIT_NAME_FULL, ': ', 'Debug Console'),
|
||||
('GREEN', None, 'YELLOW'),
|
||||
sep='',
|
||||
|
|
|
|||
|
|
@ -1,6 +0,0 @@
|
|||
"""WizardKit: Export Bitlocker Tool"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import wk
|
||||
|
||||
wk.os.win.export_bitlocker_info()
|
||||
|
|
@ -1,14 +1,24 @@
|
|||
#!/usr/bin/env python3
|
||||
"""WizardKit: Hardware Diagnostics"""
|
||||
# pylint: disable=invalid-name
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from docopt import docopt
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
try:
|
||||
docopt(wk.hw.diags.DOCSTRING)
|
||||
except SystemExit:
|
||||
print('')
|
||||
wk.std.pause('Press Enter to exit...')
|
||||
raise
|
||||
|
||||
try:
|
||||
wk.hw.diags.main()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -7,15 +7,15 @@ import platform
|
|||
import wk
|
||||
|
||||
|
||||
def main() -> None:
|
||||
def main():
|
||||
"""Show sensor data on screen."""
|
||||
sensors = wk.hw.sensors.Sensors()
|
||||
if platform.system() == 'Darwin':
|
||||
wk.ui.cli.clear_screen()
|
||||
wk.std.clear_screen()
|
||||
while True:
|
||||
print('\033[100A', end='')
|
||||
sensors.update_sensor_data()
|
||||
wk.ui.cli.print_report(sensors.generate_report('Current', 'Max'))
|
||||
wk.std.print_report(sensors.generate_report('Current', 'Max'))
|
||||
wk.std.sleep(1)
|
||||
elif platform.system() == 'Linux':
|
||||
proc = wk.exe.run_program(cmd=['mktemp'])
|
||||
|
|
@ -42,5 +42,5 @@ if __name__ == '__main__':
|
|||
pass
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,37 +0,0 @@
|
|||
# WizardKit: Install winget (if needed)
|
||||
|
||||
#Requires -Version 3.0
|
||||
if (Test-Path Env:\DEBUG) {
|
||||
Set-PSDebug -Trace 1
|
||||
}
|
||||
$Host.UI.RawUI.WindowTitle = "WizardKit: Winget installer"
|
||||
$Host.UI.RawUI.BackgroundColor = "black"
|
||||
$Host.UI.RawUI.ForegroundColor = "white"
|
||||
$ProgressPreference = "SilentlyContinue"
|
||||
|
||||
# STATIC VARIABLES
|
||||
$EXIT_OK = 0
|
||||
$EXIT_INSTALLED = 1
|
||||
$EXIT_FAILED_TO_INSTALL = 2
|
||||
|
||||
# Main
|
||||
$NeedsInstalled = $false
|
||||
try {
|
||||
$_ = $(winget --version)
|
||||
}
|
||||
catch {
|
||||
$NeedsInstalled = $true
|
||||
}
|
||||
|
||||
# Install
|
||||
if (! $NeedsInstalled) {
|
||||
exit $EXIT_INSTALLED
|
||||
}
|
||||
try {
|
||||
Add-AppxPackage -ErrorAction Stop -RegisterByFamilyName -MainPackage Microsoft.DesktopAppInstaller_8wekyb3d8bbwe
|
||||
}
|
||||
catch {
|
||||
exit $EXIT_FAILED_TO_INSTALL
|
||||
}
|
||||
|
||||
exit $EXIT_OK
|
||||
|
|
@ -1,7 +0,0 @@
|
|||
#!/bin/bash
|
||||
#
|
||||
## Monitor journal log for data recovery related events
|
||||
|
||||
echo -e 'Monitoring journal output...\n'
|
||||
journalctl -kf \
|
||||
| grep -Ei --color=always 'ata|nvme|scsi|sd[a..z]+|usb|comreset|critical|error'
|
||||
|
|
@ -1,99 +0,0 @@
|
|||
"""WizardKit: Launch Snappy Driver Installer Origin"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from subprocess import CompletedProcess
|
||||
|
||||
import wk
|
||||
from wk.cfg.net import SDIO_SERVER
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
MOUNT_EXCEPTIONS = (
|
||||
RuntimeError,
|
||||
wk.exe.subprocess.CalledProcessError,
|
||||
)
|
||||
SDIO_LOCAL_PATH = wk.kit.tools.get_tool_path("SDIO", "SDIO")
|
||||
SDIO_REMOTE_PATH = wk.io.get_path_obj(
|
||||
(
|
||||
fr'\\{SDIO_SERVER["Address"]}\{SDIO_SERVER["Share"]}\{SDIO_SERVER["Path"]}'
|
||||
fr'\SDIO{"64" if wk.os.win.ARCH == "64" else ""}.exe'
|
||||
),
|
||||
resolve=False,
|
||||
)
|
||||
|
||||
# Functions
|
||||
def try_again() -> bool:
|
||||
"""Ask to try again or quit."""
|
||||
if wk.ui.cli.ask(' Try again?'):
|
||||
return True
|
||||
if not wk.ui.cli.ask(' Use local version?'):
|
||||
wk.ui.cli.abort()
|
||||
return False
|
||||
|
||||
|
||||
def use_network_sdio() -> bool:
|
||||
"""Try to mount SDIO server."""
|
||||
use_network = False
|
||||
def _mount_server() -> CompletedProcess:
|
||||
print('Connecting to server... (Press CTRL+c to use local copy)')
|
||||
return wk.net.mount_network_share(SDIO_SERVER, read_write=False)
|
||||
|
||||
# Bail early
|
||||
if not SDIO_SERVER['Address']:
|
||||
return use_network
|
||||
|
||||
# Main loop
|
||||
while True:
|
||||
try:
|
||||
proc = _mount_server()
|
||||
except KeyboardInterrupt:
|
||||
break
|
||||
except MOUNT_EXCEPTIONS as err:
|
||||
wk.ui.cli.print_error(f' {err}')
|
||||
if not try_again():
|
||||
break
|
||||
else:
|
||||
if proc.returncode == 0:
|
||||
# Network copy available
|
||||
use_network = True
|
||||
break
|
||||
|
||||
# Failed to mount
|
||||
wk.ui.cli.print_error(' Failed to mount server')
|
||||
if not try_again():
|
||||
break
|
||||
|
||||
# Done
|
||||
return use_network
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
wk.ui.cli.set_title(
|
||||
f'{wk.cfg.main.KIT_NAME_FULL}: Snappy Driver Installer Origin Launcher',
|
||||
)
|
||||
log_dir = wk.log.format_log_path(tool=True).parent
|
||||
USE_NETWORK = False
|
||||
|
||||
# Windows 11 workaround
|
||||
if wk.os.win.OS_VERSION == 11:
|
||||
appid_services = ['appid', 'appidsvc', 'applockerfltr']
|
||||
for svc in appid_services:
|
||||
wk.os.win.stop_service(svc)
|
||||
if any([wk.os.win.get_service_status(s) != 'stopped' for s in appid_services]):
|
||||
raise wk.std.GenericWarning('Failed to stop AppID services')
|
||||
|
||||
# Try to mount server
|
||||
try:
|
||||
USE_NETWORK = use_network_sdio()
|
||||
except KeyboardInterrupt:
|
||||
wk.ui.cli.abort()
|
||||
|
||||
# Run SDIO
|
||||
EXE_PATH = SDIO_LOCAL_PATH
|
||||
if USE_NETWORK:
|
||||
EXE_PATH = SDIO_REMOTE_PATH
|
||||
print('Using network copy!')
|
||||
else:
|
||||
print('Using local copy!')
|
||||
cmd = [EXE_PATH, '-log_dir', log_dir]
|
||||
wk.exe.run_program(cmd, check=False, cwd=EXE_PATH.parent)
|
||||
|
|
@ -5,12 +5,10 @@ import json
|
|||
import re
|
||||
import subprocess
|
||||
|
||||
from typing import Any
|
||||
|
||||
CPU_REGEX = re.compile(r'(core|k\d+)temp', re.IGNORECASE)
|
||||
NON_TEMP_REGEX = re.compile(r'^(fan|in|curr)', re.IGNORECASE)
|
||||
|
||||
def get_data() -> dict[Any, Any]:
|
||||
def get_data():
|
||||
cmd = ('sensors', '-j')
|
||||
data = {}
|
||||
raw_data = []
|
||||
|
|
@ -40,7 +38,7 @@ def get_data() -> dict[Any, Any]:
|
|||
|
||||
return data
|
||||
|
||||
def get_max_temp(data) -> str:
|
||||
def get_max_temp(data):
|
||||
cpu_temps = []
|
||||
max_cpu_temp = '??° C'
|
||||
for adapter, sources in data.items():
|
||||
|
|
|
|||
|
|
@ -8,34 +8,34 @@ import wk
|
|||
|
||||
|
||||
# Functions
|
||||
def main() -> None:
|
||||
def main():
|
||||
"""Mount all volumes and show results."""
|
||||
wk.ui.cli.print_standard(f'{wk.cfg.main.KIT_NAME_FULL}: Volume mount tool')
|
||||
wk.ui.cli.print_standard(' ')
|
||||
wk.std.print_standard(f'{wk.cfg.main.KIT_NAME_FULL}: Volume mount tool')
|
||||
wk.std.print_standard(' ')
|
||||
|
||||
# Mount volumes and get report
|
||||
wk.ui.cli.print_standard('Mounting volumes...')
|
||||
wk.std.print_standard('Mounting volumes...')
|
||||
wk.os.linux.mount_volumes()
|
||||
report = wk.os.linux.build_volume_report()
|
||||
|
||||
# Show results
|
||||
wk.ui.cli.print_info('Results')
|
||||
wk.ui.cli.print_report(report)
|
||||
wk.std.print_info('Results')
|
||||
wk.std.print_report(report)
|
||||
|
||||
# GUI mode
|
||||
if 'gui' in sys.argv:
|
||||
wk.ui.cli.pause('Press Enter to exit...')
|
||||
wk.std.pause('Press Enter to exit...')
|
||||
wk.exe.popen_program(['nohup', 'thunar', '/media'])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
if wk.std.PLATFORM != 'Linux':
|
||||
os_name = wk.std.PLATFORM.replace('Darwin', 'macOS')
|
||||
wk.ui.cli.print_error(f'This script is not supported under {os_name}.')
|
||||
wk.ui.cli.abort()
|
||||
wk.std.print_error(f'This script is not supported under {os_name}.')
|
||||
wk.std.abort()
|
||||
try:
|
||||
main()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -1,14 +1,15 @@
|
|||
#!/usr/bin/env python3
|
||||
"""WizardKit: Mount Backup Shares"""
|
||||
# pylint: disable=invalid-name
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
# Functions
|
||||
def main() -> None:
|
||||
def main():
|
||||
"""Attempt to mount backup shares and print report."""
|
||||
wk.ui.cli.print_info('Mounting Backup Shares')
|
||||
wk.std.print_info('Mounting Backup Shares')
|
||||
report = wk.net.mount_backup_shares()
|
||||
for line in report:
|
||||
color = 'GREEN'
|
||||
|
|
@ -17,7 +18,7 @@ def main() -> None:
|
|||
color = 'RED'
|
||||
elif 'Already' in line:
|
||||
color = 'YELLOW'
|
||||
print(wk.ansi.color_string(line, color))
|
||||
print(wk.std.color_string(line, color))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
@ -25,5 +26,5 @@ if __name__ == '__main__':
|
|||
main()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -6,8 +6,6 @@ import os
|
|||
import re
|
||||
import sys
|
||||
|
||||
import wk
|
||||
|
||||
# STATIC VARIABLES
|
||||
SCANDIR = os.getcwd()
|
||||
USAGE = '''Usage: {script} <search-terms>...
|
||||
|
|
@ -15,6 +13,12 @@ USAGE = '''Usage: {script} <search-terms>...
|
|||
|
||||
This script will search all doc/docx files below the current directory for
|
||||
the search-terms provided (case-insensitive).'''.format(script=__file__)
|
||||
|
||||
# Init
|
||||
sys.path.append(os.path.dirname(os.path.realpath(__file__)))
|
||||
from functions.network import *
|
||||
init_global_vars()
|
||||
|
||||
REGEX_DOC_FILES = re.compile(r'\.docx?$', re.IGNORECASE)
|
||||
|
||||
|
||||
|
|
@ -30,10 +34,10 @@ def scan_file(file_path, search):
|
|||
match = False
|
||||
try:
|
||||
if entry.name.lower().endswith('.docx'):
|
||||
result = wk.exe.run_program(['unzip', '-p', entry.path])
|
||||
result = run_program(['unzip', '-p', entry.path])
|
||||
else:
|
||||
# Assuming .doc
|
||||
result = wk.exe.run_program(['antiword', entry.path])
|
||||
result = run_program(['antiword', entry.path])
|
||||
out = result.stdout.decode()
|
||||
match = re.search(search, out, re.IGNORECASE)
|
||||
except Exception:
|
||||
|
|
@ -46,13 +50,13 @@ def scan_file(file_path, search):
|
|||
if __name__ == '__main__':
|
||||
try:
|
||||
# Prep
|
||||
wk.ui.cli.clear_screen()
|
||||
clear_screen()
|
||||
terms = [re.sub(r'\s+', r'\s*', t) for t in sys.argv[1:]]
|
||||
search = '({})'.format('|'.join(terms))
|
||||
|
||||
if len(sys.argv) == 1:
|
||||
# Print usage
|
||||
wk.ui.cli.print_standard(USAGE)
|
||||
print_standard(USAGE)
|
||||
else:
|
||||
matches = []
|
||||
for entry in scan_for_docs(SCANDIR):
|
||||
|
|
@ -60,20 +64,21 @@ if __name__ == '__main__':
|
|||
# Strip None values (i.e. non-matching entries)
|
||||
matches = [m for m in matches if m]
|
||||
if matches:
|
||||
wk.ui.cli.print_success('Found {} {}:'.format(
|
||||
print_success('Found {} {}:'.format(
|
||||
len(matches),
|
||||
'Matches' if len(matches) > 1 else 'Match'))
|
||||
for match in matches:
|
||||
wk.ui.cli.print_standard(match)
|
||||
print_standard(match)
|
||||
else:
|
||||
wk.ui.cli.print_error('No matches found.')
|
||||
print_error('No matches found.')
|
||||
|
||||
# Done
|
||||
wk.ui.cli.print_standard('\nDone.')
|
||||
print_standard('\nDone.')
|
||||
#pause("Press Enter to exit...")
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
exit_script()
|
||||
except SystemExit as sys_exit:
|
||||
exit_script(sys_exit.code)
|
||||
except:
|
||||
major_exception()
|
||||
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
|
|
|||
|
|
@ -13,6 +13,6 @@ sudo sed -i -r "s/^SigLevel.*/SigLevel = Never/" /etc/pacman.conf
|
|||
# Init Pacman keyring
|
||||
sudo systemctl start pacman-init.service
|
||||
|
||||
# Refresh package databases and install packages (if provided)
|
||||
sudo pacman -Sy "$@"
|
||||
# Refresh package databases
|
||||
sudo pacman -Sy
|
||||
|
||||
|
|
|
|||
150
scripts/photorec-sort
Executable file
150
scripts/photorec-sort
Executable file
|
|
@ -0,0 +1,150 @@
|
|||
#!/bin/bash
|
||||
#
|
||||
## sort photorec results into something usefull
|
||||
|
||||
## Set paths
|
||||
recup_dir="${1%/}"
|
||||
[ -n "$recup_dir" ] || recup_dir="."
|
||||
recup_dir="$(realpath "$recup_dir")"
|
||||
out_dir="$recup_dir/Recovered"
|
||||
bad_dir="$recup_dir/Corrupt"
|
||||
|
||||
## Test path before starting (using current dir if not specified)
|
||||
for d in $recup_dir/recup*; do
|
||||
### Source: http://stackoverflow.com/a/6364244
|
||||
## Check if the glob gets expanded to existing files.
|
||||
## If not, f here will be exactly the pattern above
|
||||
## and the exists test will evaluate to false.
|
||||
[ -e "$d" ] && echo "Found recup folder(s)" || {
|
||||
echo "ERROR: No recup folders found"
|
||||
echo "Usage: $0 recup_dir"
|
||||
exit 1
|
||||
}
|
||||
|
||||
## This is all we needed to know, so we can break after the first iteration
|
||||
break
|
||||
done
|
||||
|
||||
# Hard link files into folders by type
|
||||
for d in $recup_dir/recup*; do
|
||||
if [ -d "$d" ]; then
|
||||
echo "Linking $d"
|
||||
pushd $d >/dev/null
|
||||
find -type f | while read k; do
|
||||
file="$(basename "$k")"
|
||||
src="$(realpath "$k")"
|
||||
ext="$(echo "${file##*.}" | tr '[:upper:]' '[:lower:]')"
|
||||
ext_dir="$out_dir/$ext"
|
||||
if [ "${file##*.}" = "$file" ]; then
|
||||
ext_dir="$out_dir/_MISC_"
|
||||
elif [ "$ext" = "jpg" ] && [ "${file:0:1}" = "t" ]; then
|
||||
ext_dir="$out_dir/jpg-thumbnail"
|
||||
fi
|
||||
#echo " $file -> $ext_dir"
|
||||
[ -d "$ext_dir" ] || mkdir -p "$ext_dir"
|
||||
ln "$src" "$ext_dir"
|
||||
done
|
||||
popd >/dev/null
|
||||
else
|
||||
echo "ERROR: '$d' not a directory"
|
||||
fi
|
||||
done
|
||||
|
||||
## Check the files output by photorec for corruption
|
||||
pushd "$out_dir" >/dev/null
|
||||
|
||||
# Check archives with 7-Zip
|
||||
#for d in 7z bz2 gz lzh lzo rar tar xz zip; do
|
||||
# if [ -d "$d" ]; then
|
||||
# echo "Checking $d files"
|
||||
# pushd "$d" >/dev/null
|
||||
# for f in *; do
|
||||
# if ! 7z t "$f" >/dev/null 2>&1; then
|
||||
# #echo " BAD: $f"
|
||||
# [ -d "$bad_dir/$d" ] || mkdir -p "$bad_dir/$d"
|
||||
# mv -n "$f" "$bad_dir/$d/$f"
|
||||
# fi
|
||||
# done
|
||||
# popd >/dev/null
|
||||
# fi
|
||||
#done
|
||||
|
||||
# Check Audio/Video files with ffprobe
|
||||
for d in avi flac flv m4a m4p m4v mkv mid mov mp2 mp3 mp4 mpg mpg2 ogg ts vob wav; do
|
||||
if [ -d "$d" ]; then
|
||||
echo "Checking $d files"
|
||||
pushd "$d" >/dev/null
|
||||
for f in *; do
|
||||
if ! ffprobe "$f" >/dev/null 2>&1; then
|
||||
#echo " BAD: $f"
|
||||
[ -d "$bad_dir/$d" ] || mkdir -p "$bad_dir/$d"
|
||||
mv -n "$f" "$bad_dir/$d/$f"
|
||||
fi
|
||||
done
|
||||
popd >/dev/null
|
||||
fi
|
||||
done
|
||||
|
||||
# Check .doc files with antiword
|
||||
if [ -d "doc" ]; then
|
||||
echo "Checking doc files"
|
||||
pushd "doc" >/dev/null
|
||||
for f in *doc; do
|
||||
if ! antiword "$f" >/dev/null 2>&1; then
|
||||
#echo " BAD: $f"
|
||||
[ -d "$bad_dir/doc" ] || mkdir -p "$bad_dir/doc"
|
||||
mv -n "$f" "$bad_dir/doc/$f"
|
||||
fi
|
||||
done
|
||||
popd >/dev/null
|
||||
fi
|
||||
|
||||
# Check .docx files with 7z and grep
|
||||
if [ -d "docx" ]; then
|
||||
echo "Checking docx files"
|
||||
pushd "docx" >/dev/null
|
||||
for f in *docx; do
|
||||
if ! 7z l "$f" | grep -q -s "word/document.xml"; then
|
||||
#echo " BAD: $f"
|
||||
[ -d "$bad_dir/docx" ] || mkdir -p "$bad_dir/docx"
|
||||
mv -n "$f" "$bad_dir/docx/$f"
|
||||
fi
|
||||
done
|
||||
popd >/dev/null
|
||||
fi
|
||||
|
||||
# Sort pictures by date (only for common camera formats)
|
||||
for d in jpg mrw orf raf raw rw2 tif x3f; do
|
||||
if [ -d "$d" ]; then
|
||||
echo "Sorting $d files by date"
|
||||
pushd "$d" >/dev/null
|
||||
for f in *; do
|
||||
date_dir="$(date -d "$(stat -c %y "$f")" +"%F")"
|
||||
[ -d "$date_dir" ] || mkdir "$date_dir"
|
||||
mv -n "$f" "$date_dir/"
|
||||
done
|
||||
popd >/dev/null
|
||||
fi
|
||||
done
|
||||
|
||||
# Sort mov files by encoded date
|
||||
if [ -d "mov" ]; then
|
||||
echo "Sorting mov files by date"
|
||||
pushd "mov" >/dev/null
|
||||
for f in *mov; do
|
||||
enc_date="$(mediainfo "$f" | grep -i "Encoded date" | head -1 | sed -r 's/.*: //')"
|
||||
date_dir="$(date -d "$enc_date" +"%F")"
|
||||
echo "$date_dir" | grep -E -q -s '^[0-9]{4}-[0-9]{2}-[0-9]{2}$' || date_dir="Unknown Date"
|
||||
[ -d "$date_dir" ] || mkdir "$date_dir"
|
||||
mv -n "$f" "$date_dir/"
|
||||
done
|
||||
popd >/dev/null
|
||||
fi
|
||||
|
||||
## sort audio files by tags
|
||||
|
||||
## sort matroska files by metadata
|
||||
|
||||
## return to original dir
|
||||
popd >/dev/null
|
||||
|
||||
|
|
@ -1,19 +0,0 @@
|
|||
[tool.ruff.per-file-ignores]
|
||||
# Init files
|
||||
"wk/__init__.py" = ["F401"]
|
||||
"wk/cfg/__init__.py" = ["F401"]
|
||||
"wk/clone/__init__.py" = ["F401"]
|
||||
"wk/hw/__init__.py" = ["F401"]
|
||||
"wk/kit/__init__.py" = ["F401"]
|
||||
"wk/os/__init__.py" = ["F401"]
|
||||
"wk/repairs/__init__.py" = ["F401"]
|
||||
"wk/setup/__init__.py" = ["F401"]
|
||||
"wk/ui/__init__.py" = ["F401"]
|
||||
|
||||
# Long lines
|
||||
"wk/borrowed/acpi.py" = ["E501", "F841"]
|
||||
"wk/cfg/ddrescue.py" = ["E501"]
|
||||
"wk/cfg/hw.py" = ["E501"]
|
||||
"wk/cfg/launchers.py" = ["E501"]
|
||||
"wk/cfg/setup.py" = ["E501"]
|
||||
"wk/cfg/sources.py" = ["E501"]
|
||||
|
|
@ -1,21 +1,22 @@
|
|||
#!/usr/bin/env python3
|
||||
"""WizardKit: Unmount Backup Shares"""
|
||||
# pylint: disable=invalid-name
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
# Functions
|
||||
def main() -> None:
|
||||
def main():
|
||||
"""Attempt to mount backup shares and print report."""
|
||||
wk.ui.cli.print_info('Unmounting Backup Shares')
|
||||
wk.std.print_info('Unmounting Backup Shares')
|
||||
report = wk.net.unmount_backup_shares()
|
||||
for line in report:
|
||||
color = 'GREEN'
|
||||
line = f' {line}'
|
||||
if 'Not mounted' in line:
|
||||
color = 'YELLOW'
|
||||
print(wk.ui.ansi.color_string(line, color))
|
||||
print(wk.std.color_string(line, color))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
@ -23,5 +24,5 @@ if __name__ == '__main__':
|
|||
main()
|
||||
except SystemExit:
|
||||
raise
|
||||
except: # noqa: E722
|
||||
wk.ui.cli.major_exception()
|
||||
except: #pylint: disable=bare-except
|
||||
wk.std.major_exception()
|
||||
|
|
|
|||
|
|
@ -25,24 +25,24 @@ if PLATFORM not in ('macOS', 'Linux'):
|
|||
|
||||
|
||||
# Functions
|
||||
def main() -> None:
|
||||
def main():
|
||||
"""Upload logs for review."""
|
||||
lines = []
|
||||
try_and_print = wk.ui.cli.TryAndPrint()
|
||||
try_and_print = wk.std.TryAndPrint()
|
||||
|
||||
# Set log
|
||||
wk.log.update_log_path(dest_name='Upload-Logs', timestamp=True)
|
||||
|
||||
# Instructions
|
||||
wk.ui.cli.print_success(f'{wk.cfg.main.KIT_NAME_FULL}: Upload Logs')
|
||||
wk.ui.cli.print_standard('')
|
||||
wk.ui.cli.print_standard('Please state the reason for the review.')
|
||||
wk.ui.cli.print_info(' End note with an empty line.')
|
||||
wk.ui.cli.print_standard('')
|
||||
wk.std.print_success(f'{wk.cfg.main.KIT_NAME_FULL}: Upload Logs')
|
||||
wk.std.print_standard('')
|
||||
wk.std.print_standard('Please state the reason for the review.')
|
||||
wk.std.print_info(' End note with an empty line.')
|
||||
wk.std.print_standard('')
|
||||
|
||||
# Get reason note
|
||||
while True:
|
||||
text = wk.ui.cli.input_text('> ')
|
||||
text = wk.std.input_text('> ')
|
||||
if not text:
|
||||
lines.append('')
|
||||
break
|
||||
|
|
@ -60,16 +60,17 @@ def main() -> None:
|
|||
raise SystemExit(1)
|
||||
|
||||
|
||||
def upload_log_dir(reason='Testing') -> None:
|
||||
def upload_log_dir(reason='Testing'):
|
||||
"""Upload compressed log_dir to the crash server."""
|
||||
server = wk.cfg.net.CRASH_SERVER
|
||||
dest = pathlib.Path(f'~/{reason}_{NOW.strftime("%Y-%m-%dT%H%M%S%z")}.txz')
|
||||
dest = dest.expanduser().resolve()
|
||||
data = None
|
||||
|
||||
# Compress LOG_DIR (relative to parent dir)
|
||||
os.chdir(LOG_DIR.parent)
|
||||
cmd = ['tar', 'caf', dest.name, LOG_DIR.name]
|
||||
wk.exe.run_program(cmd, check=False)
|
||||
proc = wk.exe.run_program(cmd, check=False)
|
||||
|
||||
# Upload compressed data
|
||||
url = f'{server["Url"]}/{dest.name}'
|
||||
|
|
|
|||
|
|
@ -1,5 +0,0 @@
|
|||
#!/bin/bash
|
||||
#
|
||||
## WizardKit: Debug Launcher
|
||||
|
||||
python3 -i /usr/local/bin/wk_debug.py
|
||||
|
|
@ -1,7 +1,7 @@
|
|||
"""WizardKit: wk module init"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from sys import stderr, version_info
|
||||
from sys import version_info as version
|
||||
|
||||
from . import cfg
|
||||
from . import clone
|
||||
|
|
@ -17,22 +17,21 @@ from . import os
|
|||
from . import repairs
|
||||
from . import setup
|
||||
from . import std
|
||||
from . import ui
|
||||
from . import tmux
|
||||
|
||||
|
||||
# Check env
|
||||
if version_info < (3, 10):
|
||||
if version < (3, 7):
|
||||
# Unsupported
|
||||
raise RuntimeError(
|
||||
'This package is unsupported on Python '
|
||||
f'{version_info.major}.{version_info.minor}'
|
||||
f'This package is unsupported on Python {version.major}.{version.minor}'
|
||||
)
|
||||
|
||||
# Init
|
||||
try:
|
||||
log.start()
|
||||
except UserWarning as err:
|
||||
print(err, file=stderr)
|
||||
std.print_warning(err)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -7,6 +7,7 @@ from . import log
|
|||
from . import main
|
||||
from . import music
|
||||
from . import net
|
||||
from . import python
|
||||
from . import repairs
|
||||
from . import setup
|
||||
from . import sources
|
||||
|
|
|
|||
|
|
@ -1,14 +1,17 @@
|
|||
"""WizardKit: Config - ddrescue"""
|
||||
# pylint: disable=line-too-long
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from collections import OrderedDict
|
||||
|
||||
|
||||
# Layout
|
||||
TMUX_SIDE_WIDTH = 21
|
||||
TMUX_LAYOUT = {
|
||||
TMUX_LAYOUT = OrderedDict({
|
||||
'Source': {'height': 2, 'Check': True},
|
||||
'Started': {'width': TMUX_SIDE_WIDTH, 'Check': True},
|
||||
'Progress': {'width': TMUX_SIDE_WIDTH, 'Check': True},
|
||||
}
|
||||
})
|
||||
|
||||
# ddrescue
|
||||
AUTO_PASS_THRESHOLDS = {
|
||||
|
|
@ -37,7 +40,7 @@ DDRESCUE_SETTINGS = {
|
|||
'--retry-passes': {'Selected': True, 'Value': '0', },
|
||||
'--reverse': {'Selected': False, },
|
||||
'--skip-size': {'Selected': True, 'Value': '0.001,0.02', }, # Percentages of source size
|
||||
'--test-mode': {'Selected': False, },
|
||||
'--test-mode': {'Selected': False, 'Value': 'test.map', },
|
||||
'--timeout': {'Selected': True, 'Value': '30m', },
|
||||
'-vvvv': {'Selected': True, 'Hidden': True, },
|
||||
},
|
||||
|
|
|
|||
|
|
@ -1,8 +1,11 @@
|
|||
"""WizardKit: Config - Hardware"""
|
||||
# pylint: disable=line-too-long
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import re
|
||||
|
||||
from collections import OrderedDict
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
ATTRIBUTE_COLORS = (
|
||||
|
|
@ -12,21 +15,14 @@ ATTRIBUTE_COLORS = (
|
|||
('Maximum', 'PURPLE'),
|
||||
)
|
||||
# NOTE: Force 4K read block size for disks >= 3TB
|
||||
BADBLOCKS_EXTRA_LARGE_DISK = 15 * 1024**4
|
||||
BADBLOCKS_LARGE_DISK = 3 * 1024**4
|
||||
BADBLOCKS_REGEX = re.compile(
|
||||
r'^Pass completed, (\d+) bad blocks found. .(\d+)/(\d+)/(\d+) errors',
|
||||
re.IGNORECASE,
|
||||
)
|
||||
BADBLOCKS_RESULTS_REGEX = re.compile(r'^(.*?)\x08.*\x08(.*)')
|
||||
BADBLOCKS_SKIP_REGEX = re.compile(r'^(Checking|\[)', re.IGNORECASE)
|
||||
CPU_TEMPS = {
|
||||
'Cooling Delta': 25,
|
||||
'Cooling Low Cutoff': 50,
|
||||
'Critical': 100,
|
||||
'Idle Delta': 25,
|
||||
'Idle High': 70,
|
||||
}
|
||||
CPU_CRITICAL_TEMP = 99
|
||||
CPU_FAILURE_TEMP = 90
|
||||
CPU_TEST_MINUTES = 7
|
||||
IO_GRAPH_WIDTH = 40
|
||||
IO_ALT_TEST_SIZE_FACTOR = 0.01
|
||||
|
|
@ -164,6 +160,18 @@ THRESH_HDD_AVG_LOW = 65 * 1024**2
|
|||
THRESH_SSD_MIN = 90 * 1024**2
|
||||
THRESH_SSD_AVG_HIGH = 135 * 1024**2
|
||||
THRESH_SSD_AVG_LOW = 100 * 1024**2
|
||||
TMUX_SIDE_WIDTH = 20
|
||||
TMUX_LAYOUT = OrderedDict({
|
||||
'Top': {'height': 2, 'Check': True},
|
||||
'Started': {'width': TMUX_SIDE_WIDTH, 'Check': True},
|
||||
'Progress': {'width': TMUX_SIDE_WIDTH, 'Check': True},
|
||||
# Testing panes
|
||||
'Temps': {'height': 1000, 'Check': False},
|
||||
'Prime95': {'height': 11, 'Check': False},
|
||||
'SMART': {'height': 4, 'Check': True},
|
||||
'badblocks': {'height': 5, 'Check': True},
|
||||
'I/O Benchmark': {'height': 1000, 'Check': False},
|
||||
})
|
||||
# VOLUME THRESHOLDS in percent
|
||||
VOLUME_WARNING_THRESHOLD = 70
|
||||
VOLUME_FAILURE_THRESHOLD = 85
|
||||
|
|
|
|||
|
|
@ -1,138 +1,45 @@
|
|||
"""WizardKit: Config - Launchers (Windows)"""
|
||||
# pylint: disable=line-too-long
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
LAUNCHERS = {
|
||||
r'': { # Root Dir
|
||||
'0) Export BitLocker': {
|
||||
'L_TYPE': 'PyScript',
|
||||
'L_PATH': 'Scripts',
|
||||
'L_ITEM': 'export_bitlocker.py',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'1) Auto Repairs': {
|
||||
'Auto Repairs': {
|
||||
'L_TYPE': 'PyScript',
|
||||
'L_PATH': 'Scripts',
|
||||
'L_ITEM': 'auto_repairs.py',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'2) Store & Windows Updates': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': r'%SystemRoot%\System32',
|
||||
'L_ITEM': 'control.exe',
|
||||
'L_ARGS': 'update',
|
||||
'Extra Code': ['explorer ms-windows-store:updates'],
|
||||
},
|
||||
'3) Snappy Driver Installer Origin': {
|
||||
'L_TYPE': 'PyScript',
|
||||
'L_PATH': 'Scripts',
|
||||
'L_ITEM': 'launch_sdio.py',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'4) Auto Setup': {
|
||||
'Auto Setup': {
|
||||
'L_TYPE': 'PyScript',
|
||||
'L_PATH': 'Scripts',
|
||||
'L_ITEM': 'auto_setup.py',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
},
|
||||
r'Tools': {
|
||||
'AIDA64': {
|
||||
r'Data Recovery': {
|
||||
'PhotoRec (CLI)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'AIDA64',
|
||||
'L_ITEM': 'aida64.exe',
|
||||
'L_PATH': 'TestDisk',
|
||||
'L_ITEM': 'photorec_win.exe',
|
||||
'L_ELEV': 'True',
|
||||
'L__CLI': 'True',
|
||||
},
|
||||
'Autoruns (with VirusTotal Scan)': {
|
||||
'PhotoRec': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'Sysinternals',
|
||||
'L_ITEM': 'Autoruns.exe',
|
||||
'L_ARGS': '-e',
|
||||
'Extra Code': [
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v checkvirustotal /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v shownomicrosoft /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v shownowindows /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v showonlyvirustotal /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v submitvirustotal /t REG_DWORD /d 0 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v verifysignatures /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\SigCheck /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\Streams /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\VirusTotal /v VirusTotalTermsAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
],
|
||||
},
|
||||
'BleachBit': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'BleachBit',
|
||||
'L_ITEM': 'bleachbit.exe',
|
||||
},
|
||||
'BlueScreenView': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'BlueScreenView',
|
||||
'L_ITEM': 'BlueScreenView.exe',
|
||||
},
|
||||
'BCUninstaller': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'BCUninstaller',
|
||||
'L_ITEM': 'BCUninstaller.exe',
|
||||
'L_PATH': 'TestDisk',
|
||||
'L_ITEM': 'qphotorec_win.exe',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'ConEmu (as ADMIN)': {
|
||||
'TestDisk': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'ConEmu': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
},
|
||||
'Debug Console (Command Prompt)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
'L_ARGS': r'-Dir %bin%\Scripts',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'Debug Console (Python)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
'L_ARGS': r'-Dir %bin%\Scripts -Run ..\Python\x%ARCH%\python.exe -i embedded_python_env.py',
|
||||
'L_ELEV': 'True',
|
||||
'Extra Code': [
|
||||
'set ARCH=32',
|
||||
'if /i "%PROCESSOR_ARCHITECTURE%" == "AMD64" set "ARCH=64"',
|
||||
],
|
||||
},
|
||||
'Device Cleanup': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'DeviceCleanup',
|
||||
'L_ITEM': 'DeviceCleanup.exe',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'Display Driver Uninstaller': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'DDU',
|
||||
'L_ITEM': 'Display Driver Uninstaller.exe',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'ERUNT': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'erunt',
|
||||
'L_ITEM': 'ERUNT.EXE',
|
||||
'L_ARGS': r'%client_dir%\Backups\Registry\%iso_date% sysreg curuser otherusers',
|
||||
'L_ELEV': 'True',
|
||||
'Extra Code': [
|
||||
r'call "%bin%\Scripts\init_client_dir.cmd" /Logs',
|
||||
],
|
||||
},
|
||||
'Everything': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'Everything',
|
||||
'L_ITEM': 'Everything.exe',
|
||||
'L_ARGS': '-nodb',
|
||||
'L_PATH': 'TestDisk',
|
||||
'L_ITEM': 'testdisk_win.exe',
|
||||
'L_ELEV': 'True',
|
||||
'L__CLI': 'True',
|
||||
},
|
||||
},
|
||||
r'Data Transfers': {
|
||||
'FastCopy (as ADMIN)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'FastCopy',
|
||||
|
|
@ -141,6 +48,7 @@ LAUNCHERS = {
|
|||
r' /logfile=%log_dir%\Tools\FastCopy.log'
|
||||
r' /acl'
|
||||
r' /cmd=noexist_only'
|
||||
r' /utf8'
|
||||
r' /skip_empty_dir'
|
||||
r' /linkdest'
|
||||
r' /exclude='
|
||||
|
|
@ -189,6 +97,7 @@ LAUNCHERS = {
|
|||
r' /logfile=%log_dir%\Tools\FastCopy.log'
|
||||
r' /acl'
|
||||
r' /cmd=noexist_only'
|
||||
r' /utf8'
|
||||
r' /skip_empty_dir'
|
||||
r' /linkdest'
|
||||
r' /exclude='
|
||||
|
|
@ -228,6 +137,51 @@ LAUNCHERS = {
|
|||
r'call "%bin%\Scripts\init_client_dir.cmd" /Logs /Transfer',
|
||||
],
|
||||
},
|
||||
},
|
||||
r'Diagnostics': {
|
||||
'AIDA64': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'AIDA64',
|
||||
'L_ITEM': 'aida64.exe',
|
||||
},
|
||||
'Autoruns (with VirusTotal Scan)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'Sysinternals',
|
||||
'L_ITEM': 'Autoruns.exe',
|
||||
'L_ARGS': '-e',
|
||||
'Extra Code': [
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v checkvirustotal /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v shownomicrosoft /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v shownowindows /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v showonlyvirustotal /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v submitvirustotal /t REG_DWORD /d 0 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns /v verifysignatures /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\SigCheck /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\Streams /v EulaAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
r'reg add HKCU\Software\Sysinternals\AutoRuns\VirusTotal /v VirusTotalTermsAccepted /t REG_DWORD /d 1 /f >nul',
|
||||
],
|
||||
},
|
||||
'BleachBit': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'BleachBit',
|
||||
'L_ITEM': 'bleachbit.exe',
|
||||
},
|
||||
'BlueScreenView': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'BlueScreenView',
|
||||
'L_ITEM': 'BlueScreenView.exe',
|
||||
},
|
||||
'ERUNT': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'erunt',
|
||||
'L_ITEM': 'ERUNT.EXE',
|
||||
'L_ARGS': r'%client_dir%\Backups\Registry\%iso_date% sysreg curuser otherusers',
|
||||
'L_ELEV': 'True',
|
||||
'Extra Code': [
|
||||
r'call "%bin%\Scripts\init_client_dir.cmd" /Logs',
|
||||
],
|
||||
},
|
||||
'FurMark': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'FurMark',
|
||||
|
|
@ -257,6 +211,31 @@ LAUNCHERS = {
|
|||
r')',
|
||||
],
|
||||
},
|
||||
'Snappy Driver Installer Origin': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'SDIO',
|
||||
'L_ITEM': 'SDIO.exe',
|
||||
},
|
||||
},
|
||||
r'Misc': {
|
||||
'ConEmu (as ADMIN)': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'ConEmu': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
},
|
||||
'Everything': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'Everything',
|
||||
'L_ITEM': 'Everything.exe',
|
||||
'L_ARGS': '-nodb',
|
||||
'L_ELEV': 'True',
|
||||
},
|
||||
'Notepad++': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'notepadplusplus',
|
||||
|
|
@ -267,6 +246,17 @@ LAUNCHERS = {
|
|||
'L_PATH': 'PuTTY',
|
||||
'L_ITEM': 'PUTTY.EXE',
|
||||
},
|
||||
'WizardKit Debug Console': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'ConEmu',
|
||||
'L_ITEM': 'ConEmu.exe',
|
||||
'L_ARGS': r'-Dir %bin%\Scripts -Run ..\Python\x%ARCH%\python.exe -i embedded_python_env.py',
|
||||
'L_ELEV': 'True',
|
||||
'Extra Code': [
|
||||
'set ARCH=32',
|
||||
'if /i "%PROCESSOR_ARCHITECTURE%" == "AMD64" set "ARCH=64"',
|
||||
],
|
||||
},
|
||||
'WizTree': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'WizTree',
|
||||
|
|
@ -280,6 +270,13 @@ LAUNCHERS = {
|
|||
'L_ARGS': r'"%bin%\XMPlay\music.7z"',
|
||||
},
|
||||
},
|
||||
r'Uninstallers': {
|
||||
'IObit Uninstaller': {
|
||||
'L_TYPE': 'Executable',
|
||||
'L_PATH': 'IObitUninstallerPortable',
|
||||
'L_ITEM': 'IObitUninstallerPortable.exe',
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -14,7 +14,7 @@ ENABLED_UPLOAD_DATA=False
|
|||
ARCHIVE_PASSWORD='Abracadabra'
|
||||
KIT_NAME_FULL='WizardKit'
|
||||
KIT_NAME_SHORT='WK'
|
||||
SUPPORT_MESSAGE='Please let 2Shirt know by opening an issue on Gitea'
|
||||
SUPPORT_MESSAGE='Please let 2Shirt know by opening an issue on GitHub'
|
||||
|
||||
# Text Formatting
|
||||
INDENT=4
|
||||
|
|
@ -27,8 +27,8 @@ TECH_PASSWORD='Abracadabra'
|
|||
# Time Zones
|
||||
## See 'timedatectl list-timezones' for valid Linux values
|
||||
## See 'tzutil /l' for valid Windows values
|
||||
LINUX_TIME_ZONE='America/Los_Angeles'
|
||||
WINDOWS_TIME_ZONE='Pacific Standard Time'
|
||||
LINUX_TIME_ZONE='America/Denver'
|
||||
WINDOWS_TIME_ZONE='Mountain Standard Time'
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -28,13 +28,6 @@ CRASH_SERVER = {
|
|||
#'Pass': '',
|
||||
#'Headers': {'X-Requested-With': 'XMLHttpRequest'},
|
||||
}
|
||||
SDIO_SERVER = {
|
||||
'Address': '',
|
||||
'Share': '',
|
||||
'Path': '',
|
||||
'RO-User': '',
|
||||
'RO-Pass': '',
|
||||
}
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
14
scripts/wk/cfg/python.py
Normal file
14
scripts/wk/cfg/python.py
Normal file
|
|
@ -0,0 +1,14 @@
|
|||
"""WizardKit: Config - Python"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from sys import version_info
|
||||
|
||||
DATACLASS_DECORATOR_KWARGS = {}
|
||||
if version_info.major >= 3 and version_info.minor >= 10:
|
||||
DATACLASS_DECORATOR_KWARGS['slots'] = True
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
|
@ -3,7 +3,7 @@
|
|||
|
||||
from wk.cfg.main import KIT_NAME_FULL
|
||||
|
||||
AUTO_REPAIR_DELAY_IN_SECONDS = 3
|
||||
AUTO_REPAIR_DELAY_IN_SECONDS = 30
|
||||
AUTO_REPAIR_KEY = fr'Software\{KIT_NAME_FULL}\Auto Repairs'
|
||||
BLEACH_BIT_CLEANERS = (
|
||||
# Applications
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
"""WizardKit: Config - Setup"""
|
||||
# pylint: disable=line-too-long
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
|
||||
|
|
@ -9,11 +10,6 @@ BROWSER_PATHS = {
|
|||
'Microsoft Edge': 'Microsoft/Edge/Application/msedge.exe',
|
||||
'Opera': 'Opera/launcher.exe',
|
||||
}
|
||||
DISABLED_ENTRIES_WINDOWS_11 = {
|
||||
# Group Name: Option Name
|
||||
'Install Software': 'Open Shell',
|
||||
'Configure System': 'Open Shell',
|
||||
}
|
||||
LIBREOFFICE_XCU_DATA = '''<?xml version="1.0" encoding="UTF-8"?>
|
||||
<oor:items xmlns:oor="http://openoffice.org/2001/registry" xmlns:xs="http://www.w3.org/2001/XMLSchema" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance">
|
||||
<item oor:path="/org.openoffice.Setup/Office/Factories/org.openoffice.Setup:Factory['com.sun.star.presentation.PresentationDocument']"><prop oor:name="ooSetupFactoryDefaultFilter" oor:op="fuse"><value>Impress MS PowerPoint 2007 XML</value></prop></item>
|
||||
|
|
@ -29,20 +25,9 @@ REG_CHROME_UBLOCK_ORIGIN = {
|
|||
)
|
||||
},
|
||||
}
|
||||
REG_WINDOWS_BSOD_MINIDUMPS = {
|
||||
'HKLM': {
|
||||
# Enable small memory dumps
|
||||
r'SYSTEM\CurrentControlSet\Control\CrashControl': (
|
||||
('CrashDumpEnabled', 3, 'DWORD'),
|
||||
)
|
||||
}
|
||||
}
|
||||
REG_WINDOWS_EXPLORER = {
|
||||
# pylint: disable=line-too-long
|
||||
'HKLM': {
|
||||
# Allow password sign-in for MS accounts
|
||||
r'Software\Microsoft\Windows NT\CurrentVersion\PasswordLess\Device': (
|
||||
('DevicePasswordLessBuildVersion', 0, 'DWORD'),
|
||||
),
|
||||
# Disable Location Tracking
|
||||
r'Software\Microsoft\Windows NT\CurrentVersion\Sensor\Overrides\{BFA794E4-F964-4FDB-90F6-51056BFE4B44}': (
|
||||
('SensorPermissionState', 0, 'DWORD'),
|
||||
|
|
@ -58,14 +43,6 @@ REG_WINDOWS_EXPLORER = {
|
|||
r'Software\Policies\Microsoft\Windows\DataCollection': (
|
||||
('AllowTelemetry', 0, 'DWORD'),
|
||||
),
|
||||
# Disable floating Bing search widget
|
||||
r'Software\Policies\Microsoft\Edge': (
|
||||
('WebWidgetAllowed', 0, 'DWORD'),
|
||||
),
|
||||
# Disable Edge first run screen
|
||||
r'Software\Policies\Microsoft\MicrosoftEdge\Main': (
|
||||
('PreventFirstRunPage', 1, 'DWORD'),
|
||||
),
|
||||
# Disable Wi-Fi Sense
|
||||
r'Software\Microsoft\PolicyManager\default\WiFi\AllowWiFiHotSpotReporting': (
|
||||
('Value', 0, 'DWORD'),
|
||||
|
|
@ -91,10 +68,6 @@ REG_WINDOWS_EXPLORER = {
|
|||
r'Software\Microsoft\Windows\CurrentVersion\Feeds': (
|
||||
('ShellFeedsTaskbarOpenOnHover', 0, 'DWORD'),
|
||||
),
|
||||
# Disable search highlights
|
||||
r'Software\Microsoft\Windows\CurrentVersion\Feeds\DSB': (
|
||||
('ShowDynamicContent', 0, 'DWORD'),
|
||||
),
|
||||
# File Explorer
|
||||
r'Software\Microsoft\Windows\CurrentVersion\Explorer\Advanced': (
|
||||
# Change default Explorer view to "Computer"
|
||||
|
|
@ -109,14 +82,6 @@ REG_WINDOWS_EXPLORER = {
|
|||
r'Software\Microsoft\Windows\CurrentVersion\Search': (
|
||||
('SearchboxTaskbarMode', 1, 'DWORD'),
|
||||
),
|
||||
# Disable search highlights from opening on hover
|
||||
r'Software\Microsoft\Windows\CurrentVersion\SearchSettings': (
|
||||
('IsDynamicSearchBoxEnabled', 0, 'DWORD'),
|
||||
),
|
||||
# Disable "Let's make Windows even better" screens
|
||||
r'Software\Microsoft\Windows\CurrentVersion\UserProfileEngagement': (
|
||||
('ScoobeSystemSettingEnabled', 0, 'DWORD'),
|
||||
),
|
||||
},
|
||||
}
|
||||
REG_OPEN_SHELL_SETTINGS = {
|
||||
|
|
@ -125,7 +90,6 @@ REG_OPEN_SHELL_SETTINGS = {
|
|||
('ShowedStyle2', 1, 'DWORD'),
|
||||
),
|
||||
r'Software\OpenShell\StartMenu\Settings': (
|
||||
('HighlightNew', 0, 'DWORD'),
|
||||
('MenuStyle', 'Win7', 'SZ'),
|
||||
('RecentPrograms', 'Recent', 'SZ'),
|
||||
('SkinW7', 'Fluent-Metro', 'SZ'),
|
||||
|
|
|
|||
|
|
@ -1,4 +1,5 @@
|
|||
"""WizardKit: Config - Tool Sources"""
|
||||
# pylint: disable=line-too-long
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
|
||||
|
|
@ -22,39 +23,47 @@ SOURCES = {
|
|||
'RKill': 'https://download.bleepingcomputer.com/grinler/rkill.exe',
|
||||
'RegDelNull': 'https://live.sysinternals.com/RegDelNull.exe',
|
||||
'RegDelNull64': 'https://live.sysinternals.com/RegDelNull64.exe',
|
||||
'Software Bundle': 'https://ninite.com/.net4.8-7zip-chrome-edge-vlc/ninite.exe',
|
||||
'TDSSKiller': 'https://media.kaspersky.com/utilities/VirusUtilities/EN/tdsskiller.exe',
|
||||
'VCRedist_2012_x32': 'https://download.microsoft.com/download/1/6/B/16B06F60-3B20-4FF2-B699-5E9B7962F9AE/VSU_4/vcredist_x86.exe',
|
||||
'VCRedist_2012_x64': 'https://download.microsoft.com/download/1/6/B/16B06F60-3B20-4FF2-B699-5E9B7962F9AE/VSU_4/vcredist_x64.exe',
|
||||
'VCRedist_2013_x32': 'https://aka.ms/highdpimfc2013x86enu',
|
||||
'VCRedist_2013_x64': 'https://aka.ms/highdpimfc2013x64enu',
|
||||
'VCRedist_2019_x32': 'https://aka.ms/vs/16/release/vc_redist.x86.exe',
|
||||
'VCRedist_2019_x64': 'https://aka.ms/vs/16/release/vc_redist.x64.exe',
|
||||
|
||||
# Build Kit
|
||||
'AIDA64': 'https://download.aida64.com/aida64engineer692.zip',
|
||||
'Adobe Reader DC': 'https://ardownload2.adobe.com/pub/adobe/reader/win/AcrobatDC/2300620360/AcroRdrDC2300620360_en_US.exe',
|
||||
'AIDA64': 'https://download.aida64.com/aida64engineer660.zip',
|
||||
'Adobe Reader DC': 'https://ardownload2.adobe.com/pub/adobe/reader/win/AcrobatDC/2101120039/AcroRdrDC2101120039_en_US.exe',
|
||||
'Aria2': 'https://github.com/aria2/aria2/releases/download/release-1.36.0/aria2-1.36.0-win-32bit-build1.zip',
|
||||
'Autoruns32': 'http://live.sysinternals.com/Autoruns.exe',
|
||||
'Autoruns64': 'http://live.sysinternals.com/Autoruns64.exe',
|
||||
'BleachBit': 'https://download.bleachbit.org/BleachBit-4.4.2-portable.zip',
|
||||
'BlueScreenView32': 'http://www.nirsoft.net/utils/bluescreenview.zip',
|
||||
'BlueScreenView64': 'http://www.nirsoft.net/utils/bluescreenview-x64.zip',
|
||||
'BCUninstaller': 'https://github.com/Klocman/Bulk-Crap-Uninstaller/releases/download/v5.7/BCUninstaller_5.7_portable.zip',
|
||||
'DDU': 'https://www.wagnardsoft.com/DDU/download/DDU%20v18.0.6.8.exe',
|
||||
'ERUNT': 'http://www.aumha.org/downloads/erunt.zip',
|
||||
'Everything32': 'https://www.voidtools.com/Everything-1.4.1.1024.x86.zip',
|
||||
'Everything64': 'https://www.voidtools.com/Everything-1.4.1.1024.x64.zip',
|
||||
'FastCopy': 'https://github.com/FastCopyLab/FastCopyDist2/raw/main/FastCopy5.4.2_installer.exe',
|
||||
'Fluent-Metro': 'https://github.com/bonzibudd/Fluent-Metro/releases/download/v1.5.3/Fluent-Metro_1.5.3.zip',
|
||||
'FurMark': 'https://geeks3d.com/dl/get/728',
|
||||
'HWiNFO': 'https://www.sac.sk/download/utildiag/hwi_764.zip',
|
||||
'LibreOffice32': 'https://download.documentfoundation.org/libreoffice/stable/7.6.2/win/x86/LibreOffice_7.6.2_Win_x86.msi',
|
||||
'LibreOffice64': 'https://download.documentfoundation.org/libreoffice/stable/7.6.2/win/x86_64/LibreOffice_7.6.2_Win_x86-64.msi',
|
||||
'Everything32': 'https://www.voidtools.com/Everything-1.4.1.1009.x86.en-US.zip',
|
||||
'Everything64': 'https://www.voidtools.com/Everything-1.4.1.1009.x64.en-US.zip',
|
||||
'FastCopy': 'https://ftp.vector.co.jp/73/10/2323/FastCopy392_installer.exe',
|
||||
'Fluent-Metro': 'https://github.com/bonzibudd/Fluent-Metro/releases/download/v1.5.2/Fluent-Metro_1.5.2.zip',
|
||||
'FurMark': 'https://geeks3d.com/dl/get/569',
|
||||
'HWiNFO': 'https://www.sac.sk/download/utildiag/hwi_712.zip',
|
||||
'IOBit Uninstaller': 'https://portableapps.com/redirect/?a=IObitUninstallerPortable&s=s&d=pa&f=IObitUninstallerPortable_7.5.0.7.paf.exe',
|
||||
'LibreOffice32': 'https://download.documentfoundation.org/libreoffice/stable/7.3.0/win/x86/LibreOffice_7.3.0_Win_x86.msi',
|
||||
'LibreOffice64': 'https://download.documentfoundation.org/libreoffice/stable/7.3.0/win/x86_64/LibreOffice_7.3.0_Win_x64.msi',
|
||||
'Macs Fan Control': 'https://www.crystalidea.com/downloads/macsfancontrol_setup.exe',
|
||||
'Neutron': 'http://keir.net/download/neutron.zip',
|
||||
'Notepad++': 'https://github.com/notepad-plus-plus/notepad-plus-plus/releases/download/v8.5.8/npp.8.5.8.portable.minimalist.7z',
|
||||
'OpenShell': 'https://github.com/Open-Shell/Open-Shell-Menu/releases/download/v4.4.191/OpenShellSetup_4_4_191.exe',
|
||||
'Notepad++': 'https://github.com/notepad-plus-plus/notepad-plus-plus/releases/download/v8.1.9.3/npp.8.1.9.3.portable.minimalist.7z',
|
||||
'OpenShell': 'https://github.com/Open-Shell/Open-Shell-Menu/releases/download/v4.4.160/OpenShellSetup_4_4_160.exe',
|
||||
'PuTTY': 'https://the.earth.li/~sgtatham/putty/latest/w32/putty.zip',
|
||||
'SDIO Torrent': 'https://www.glenn.delahoy.com/downloads/sdio/SDIO_Update.torrent',
|
||||
'WizTree': 'https://diskanalyzer.com/files/wiztree_4_15_portable.zip',
|
||||
'SDIO Torrent': 'https://www.snappy-driver-installer.org/downloads/SDIO_Update.torrent',
|
||||
'TestDisk': 'https://www.cgsecurity.org/testdisk-7.2-WIP.win.zip',
|
||||
'WizTree': 'https://diskanalyzer.com/files/wiztree_4_07_portable.zip',
|
||||
'XMPlay': 'https://support.xmplay.com/files/20/xmplay385.zip?v=47090',
|
||||
'XMPlay 7z': 'https://support.xmplay.com/files/16/xmp-7z.zip?v=800962',
|
||||
'XMPlay Game': 'https://support.xmplay.com/files/12/xmp-gme.zip?v=515637',
|
||||
'XMPlay RAR': 'https://support.xmplay.com/files/16/xmp-rar.zip?v=409646',
|
||||
'XMPlay Innocuous': 'https://support.xmplay.com/files/10/Innocuous%20(v1.7).zip?v=645281',
|
||||
'XMPlay Innocuous': 'https://support.xmplay.com/files/10/Innocuous%20(v1.5).zip?v=155959',
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -1,20 +1,24 @@
|
|||
"""WizardKit: Config - UFD"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from collections import OrderedDict
|
||||
|
||||
from wk.cfg.main import KIT_NAME_FULL
|
||||
|
||||
|
||||
# General
|
||||
SOURCES = {
|
||||
SOURCES = OrderedDict({
|
||||
'Linux': {'Arg': '--linux', 'Type': 'ISO'},
|
||||
'Linux (Minimal)': {'Arg': '--linux-minimal', 'Type': 'ISO'},
|
||||
'WinPE': {'Arg': '--winpe', 'Type': 'ISO'},
|
||||
'Main Kit': {'Arg': '--main-kit', 'Type': 'KIT'},
|
||||
'Extra Dir': {'Arg': '--extra-dir', 'Type': 'DIR'},
|
||||
}
|
||||
})
|
||||
|
||||
# Definitions: Boot entries
|
||||
BOOT_ENTRIES = {
|
||||
# Path to check: Comment to remove
|
||||
'/arch_minimal': 'UFD-MINIMAL',
|
||||
'/sources/boot.wim': 'UFD-WINPE',
|
||||
}
|
||||
BOOT_FILES = {
|
||||
|
|
@ -37,6 +41,14 @@ ITEMS = {
|
|||
),
|
||||
'Linux': (
|
||||
('/arch', '/'),
|
||||
('/EFI/boot', '/EFI/'),
|
||||
('/syslinux', '/'),
|
||||
),
|
||||
'Linux (Minimal)': (
|
||||
('/arch/boot/x86_64/initramfs-linux.img', '/arch_minimal/'),
|
||||
('/arch/boot/x86_64/vmlinuz-linux', '/arch_minimal/'),
|
||||
('/arch/pkglist.x86_64.txt', '/arch_minimal/'),
|
||||
('/arch/x86_64', '/arch_minimal/'),
|
||||
),
|
||||
'Main Kit': (
|
||||
('/', f'/{KIT_NAME_FULL}/'),
|
||||
|
|
@ -54,28 +66,10 @@ ITEMS = {
|
|||
('/sources/boot.wim', '/sources/'),
|
||||
),
|
||||
}
|
||||
ITEMS_FROM_LIVE = {
|
||||
'WizardKit UFD base': (
|
||||
('/usr/share/WizardKit/', '/'),
|
||||
),
|
||||
'rEFInd': (
|
||||
('/usr/share/refind/drivers_x64/', '/EFI/Boot/drivers_x64/'),
|
||||
('/usr/share/refind/icons/', '/EFI/Boot/icons/'),
|
||||
('/usr/share/refind/refind_x64.efi', '/EFI/Boot/'),
|
||||
),
|
||||
'Syslinux': (
|
||||
('/usr/lib/syslinux/bios/', '/syslinux/'),
|
||||
),
|
||||
'Memtest86': (
|
||||
('/usr/share/memtest86-efi/', '/EFI/Memtest86/'),
|
||||
),
|
||||
'Wimboot': (
|
||||
('/usr/share/wimboot/', '/syslinux/'),
|
||||
),
|
||||
}
|
||||
ITEMS_HIDDEN = (
|
||||
# Linux (all versions)
|
||||
'arch',
|
||||
'arch_minimal',
|
||||
'EFI',
|
||||
'syslinux',
|
||||
# Main Kit
|
||||
|
|
|
|||
|
|
@ -33,11 +33,4 @@ WINDOWS_BUILDS = {
|
|||
'10.0.19042': '20H2',
|
||||
'10.0.19043': '21H1',
|
||||
'10.0.19044': '21H2',
|
||||
'10.0.19045': '22H2',
|
||||
|
||||
# Windows 11
|
||||
'10.0.22000': '21H2',
|
||||
'10.0.22621': '22H2',
|
||||
'10.0.22631': '23H2',
|
||||
'10.0.26100': '24H2',
|
||||
}
|
||||
|
|
|
|||
|
|
@ -1,32 +0,0 @@
|
|||
{
|
||||
"$schema": "https://aka.ms/winget-packages.schema.2.0.json",
|
||||
"CreationDate": "2023-06-25T01:40:45.003-00:00",
|
||||
"Sources": [
|
||||
{
|
||||
"Packages": [
|
||||
{
|
||||
"PackageIdentifier": "7zip.7zip"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Google.Chrome"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Microsoft.Edge"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Mozilla.Firefox"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "VideoLAN.VLC"
|
||||
}
|
||||
],
|
||||
"SourceDetails": {
|
||||
"Argument": "https://cdn.winget.microsoft.com/cache",
|
||||
"Identifier": "Microsoft.Winget.Source_8wekyb3d8bbwe",
|
||||
"Name": "winget",
|
||||
"Type": "Microsoft.PreIndexed.Package"
|
||||
}
|
||||
}
|
||||
],
|
||||
"WinGetVersion": "1.4.11071"
|
||||
}
|
||||
|
|
@ -1,29 +0,0 @@
|
|||
{
|
||||
"$schema": "https://aka.ms/winget-packages.schema.2.0.json",
|
||||
"CreationDate": "2023-06-25T01:40:45.003-00:00",
|
||||
"Sources": [
|
||||
{
|
||||
"Packages": [
|
||||
{
|
||||
"PackageIdentifier": "Microsoft.VCRedist.2013.x64"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Microsoft.VCRedist.2013.x86"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Microsoft.VCRedist.2015+.x64"
|
||||
},
|
||||
{
|
||||
"PackageIdentifier": "Microsoft.VCRedist.2015+.x86"
|
||||
}
|
||||
],
|
||||
"SourceDetails": {
|
||||
"Argument": "https://cdn.winget.microsoft.com/cache",
|
||||
"Identifier": "Microsoft.Winget.Source_8wekyb3d8bbwe",
|
||||
"Name": "winget",
|
||||
"Type": "Microsoft.PreIndexed.Package"
|
||||
}
|
||||
}
|
||||
],
|
||||
"WinGetVersion": "1.4.11071"
|
||||
}
|
||||
|
|
@ -1,7 +1,3 @@
|
|||
"""WizardKit: ddrescue-tui module init"""
|
||||
|
||||
from . import block_pair
|
||||
from . import ddrescue
|
||||
from . import image
|
||||
from . import menus
|
||||
from . import state
|
||||
|
|
|
|||
|
|
@ -1,575 +0,0 @@
|
|||
"""WizardKit: ddrescue TUI - Block Pairs"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import logging
|
||||
import math
|
||||
import os
|
||||
import pathlib
|
||||
import plistlib
|
||||
import re
|
||||
import subprocess
|
||||
|
||||
from wk import cfg, exe, std
|
||||
from wk.clone import menus
|
||||
from wk.hw import disk as hw_disk
|
||||
from wk.ui import ansi, cli
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
DDRESCUE_LOG_REGEX = re.compile(
|
||||
r'^\s*(?P<key>\S+):\s+'
|
||||
r'(?P<size>\d+)\s+'
|
||||
r'(?P<unit>[PTGMKB]i?B?)'
|
||||
r'.*\(\s*(?P<percent>\d+\.?\d*)%\)$',
|
||||
re.IGNORECASE,
|
||||
)
|
||||
|
||||
|
||||
# Classes
|
||||
class BlockPair():
|
||||
"""Object for tracking source to dest recovery data."""
|
||||
def __init__(
|
||||
self,
|
||||
source_dev: hw_disk.Disk,
|
||||
destination: pathlib.Path,
|
||||
working_dir: pathlib.Path,
|
||||
):
|
||||
self.sector_size: int = source_dev.phy_sec
|
||||
self.source: pathlib.Path = pathlib.Path(source_dev.path)
|
||||
self.destination: pathlib.Path = destination
|
||||
self.map_data: dict[str, bool | int] = {}
|
||||
self.map_path: pathlib.Path = pathlib.Path()
|
||||
self.size: int = source_dev.size
|
||||
self.status: dict[str, float | int | str] = {
|
||||
'read-skip': 'Pending',
|
||||
'read-full': 'Pending',
|
||||
'trim': 'Pending',
|
||||
'scrape': 'Pending',
|
||||
}
|
||||
self.test_map: pathlib.Path | None = None
|
||||
self.view_map: bool = 'DISPLAY' in os.environ or 'WAYLAND_DISPLAY' in os.environ
|
||||
self.view_proc: subprocess.Popen | None = None
|
||||
|
||||
# Set map path
|
||||
# e.g. '(Clone|Image)_Model_Serial[_p#]_Size[_Label].map'
|
||||
map_name = f'{source_dev.model}_{source_dev.serial}'
|
||||
if source_dev.bus == 'Image':
|
||||
map_name = 'Image'
|
||||
if source_dev.parent:
|
||||
part_num = re.sub(r"^.*?(\d+)$", r"\1", self.source.name)
|
||||
map_name += f'_p{part_num}'
|
||||
size_str = std.bytes_to_string(
|
||||
size=self.size,
|
||||
use_binary=False,
|
||||
)
|
||||
map_name += f'_{size_str.replace(" ", "")}'
|
||||
if source_dev.raw_details.get('label', ''):
|
||||
map_name += f'_{source_dev.raw_details["label"]}'
|
||||
map_name = map_name.replace(' ', '_')
|
||||
map_name = map_name.replace('/', '_')
|
||||
map_name = map_name.replace('\\', '_')
|
||||
if destination.is_dir():
|
||||
# Imaging
|
||||
self.map_path = pathlib.Path(f'{destination}/Image_{map_name}.map')
|
||||
self.destination = self.map_path.with_suffix('.dd')
|
||||
self.destination.touch()
|
||||
else:
|
||||
# Cloning
|
||||
self.map_path = pathlib.Path(f'{working_dir}/Clone_{map_name}.map')
|
||||
|
||||
# Create map file if needed
|
||||
# NOTE: We need to set the domain size for --complete-only to work
|
||||
if not self.map_path.exists():
|
||||
self.map_path.write_text(
|
||||
data=cfg.ddrescue.DDRESCUE_MAP_TEMPLATE.format(
|
||||
name=cfg.main.KIT_NAME_FULL,
|
||||
size=self.size,
|
||||
),
|
||||
encoding='utf-8',
|
||||
)
|
||||
|
||||
# Set initial status
|
||||
self.set_initial_status()
|
||||
|
||||
def __getstate__(self):
|
||||
"""Override to allow pickling ddrescue.State() objects."""
|
||||
bp_state = self.__dict__.copy()
|
||||
del bp_state['view_proc']
|
||||
return bp_state
|
||||
|
||||
def get_error_size(self) -> int:
|
||||
"""Get error size in bytes, returns int."""
|
||||
return self.size - self.get_rescued_size()
|
||||
|
||||
def get_percent_recovered(self) -> float:
|
||||
"""Get percent rescued from map_data, returns float."""
|
||||
return 100 * self.map_data.get('rescued', 0) / self.size
|
||||
|
||||
def get_rescued_size(self) -> int:
|
||||
"""Get rescued size using map data.
|
||||
|
||||
NOTE: Returns 0 if no map data is available.
|
||||
"""
|
||||
self.load_map_data()
|
||||
return self.map_data.get('rescued', 0)
|
||||
|
||||
def load_map_data(self) -> None:
|
||||
"""Load map data from file.
|
||||
|
||||
NOTE: If the file is missing it is assumed that recovery hasn't
|
||||
started yet so default values will be returned instead.
|
||||
"""
|
||||
data: dict[str, bool | int] = {'full recovery': False, 'pass completed': False}
|
||||
|
||||
# Get output from ddrescuelog
|
||||
cmd = [
|
||||
'ddrescuelog',
|
||||
'--binary-prefixes',
|
||||
'--show-status',
|
||||
f'--size={self.size}',
|
||||
self.map_path,
|
||||
]
|
||||
proc = exe.run_program(cmd, check=False)
|
||||
|
||||
# Parse output
|
||||
for line in proc.stdout.splitlines():
|
||||
_r = DDRESCUE_LOG_REGEX.search(line)
|
||||
if _r:
|
||||
if _r.group('key') == 'rescued' and _r.group('percent') == '100':
|
||||
# Fix rounding errors from ddrescuelog output
|
||||
data['rescued'] = self.size
|
||||
else:
|
||||
data[_r.group('key')] = std.string_to_bytes(
|
||||
f'{_r.group("size")} {_r.group("unit")}',
|
||||
)
|
||||
data['pass completed'] = 'current status: finished' in line.lower()
|
||||
|
||||
# Check if 100% done (only if map is present and non-zero size
|
||||
# NOTE: ddrescuelog returns 0 (i.e. 100% done) for empty files
|
||||
if self.map_path.exists() and self.map_path.stat().st_size != 0:
|
||||
cmd = [
|
||||
'ddrescuelog',
|
||||
'--done-status',
|
||||
f'--size={self.size}',
|
||||
self.map_path,
|
||||
]
|
||||
proc = exe.run_program(cmd, check=False)
|
||||
data['full recovery'] = proc.returncode == 0
|
||||
|
||||
# Done
|
||||
self.map_data.update(data)
|
||||
|
||||
def pass_complete(self, pass_name) -> bool:
|
||||
"""Check if pass_name is complete based on map data, returns bool."""
|
||||
pending_size = self.map_data['non-tried']
|
||||
|
||||
# Full recovery
|
||||
if self.map_data.get('full recovery', False):
|
||||
return True
|
||||
|
||||
# New recovery
|
||||
if 'non-tried' not in self.map_data:
|
||||
return False
|
||||
|
||||
# Initial read skip pass
|
||||
if pass_name == 'read-skip':
|
||||
pass_threshold = cfg.ddrescue.AUTO_PASS_THRESHOLDS[pass_name]
|
||||
if self.get_percent_recovered() >= pass_threshold:
|
||||
return True
|
||||
|
||||
# Recovery in progress
|
||||
if pass_name in ('trim', 'scrape'):
|
||||
pending_size += self.map_data['non-trimmed']
|
||||
if pass_name == 'scrape':
|
||||
pending_size += self.map_data['non-scraped']
|
||||
if pending_size == 0:
|
||||
# This is true when the previous and current passes are complete
|
||||
return True
|
||||
|
||||
# This should never be reached
|
||||
return False
|
||||
|
||||
def safety_check(self) -> None:
|
||||
"""Run safety check and abort if necessary."""
|
||||
# TODO: Expand section to support non-Linux systems
|
||||
dest_size = -1
|
||||
if self.destination.is_block_device():
|
||||
cmd = [
|
||||
'lsblk', '--bytes', '--json',
|
||||
'--nodeps', '--noheadings', '--output=size',
|
||||
self.destination,
|
||||
]
|
||||
json_data = exe.get_json_from_command(cmd)
|
||||
dest_size = json_data['blockdevices'][0]['size']
|
||||
del json_data
|
||||
|
||||
# Check destination size if cloning
|
||||
if not self.destination.is_file() and dest_size < self.size:
|
||||
cli.print_error(f'Invalid destination: {self.destination}')
|
||||
raise std.GenericAbort()
|
||||
|
||||
def set_initial_status(self) -> None:
|
||||
"""Read map data and set initial statuses."""
|
||||
self.load_map_data()
|
||||
percent = self.get_percent_recovered()
|
||||
for name in self.status:
|
||||
if self.pass_complete(name):
|
||||
self.status[name] = percent
|
||||
else:
|
||||
# Stop checking
|
||||
if percent > 0:
|
||||
self.status[name] = percent
|
||||
break
|
||||
|
||||
def skip_pass(self, pass_name) -> None:
|
||||
"""Mark pass as skipped if applicable."""
|
||||
if self.status[pass_name] == 'Pending':
|
||||
self.status[pass_name] = 'Skipped'
|
||||
|
||||
def update_progress(self, pass_name) -> None:
|
||||
"""Update progress via map data."""
|
||||
self.load_map_data()
|
||||
|
||||
# Update status
|
||||
percent = self.get_percent_recovered()
|
||||
if percent > 0:
|
||||
self.status[pass_name] = percent
|
||||
|
||||
# Mark future passes as skipped if applicable
|
||||
if percent == 100:
|
||||
status_keys = list(self.status.keys())
|
||||
for pass_n in status_keys[status_keys.index(pass_name)+1:]:
|
||||
self.status[pass_n] = 'Skipped'
|
||||
|
||||
|
||||
# Functions
|
||||
def add_clone_block_pairs(state) -> list[hw_disk.Disk]:
|
||||
"""Add device to device block pairs and set settings if necessary."""
|
||||
source_sep = get_partition_separator(state.source.path.name)
|
||||
dest_sep = get_partition_separator(state.destination.path.name)
|
||||
settings = {}
|
||||
|
||||
# Clone settings
|
||||
settings = state.load_settings(discard_unused_settings=True)
|
||||
|
||||
# Add pairs from previous run
|
||||
if settings['Partition Mapping']:
|
||||
source_parts = []
|
||||
for part_map in settings['Partition Mapping']:
|
||||
bp_source = hw_disk.Disk(
|
||||
f'{state.source.path}{source_sep}{part_map[0]}',
|
||||
)
|
||||
bp_dest = pathlib.Path(
|
||||
f'{state.destination.path}{dest_sep}{part_map[1]}',
|
||||
)
|
||||
source_parts.append(bp_source)
|
||||
state.add_block_pair(bp_source, bp_dest)
|
||||
return source_parts
|
||||
|
||||
# Add pairs from selection
|
||||
source_parts = menus.select_disk_parts('Clone', state.source)
|
||||
if state.source.path.samefile(source_parts[0].path):
|
||||
# Whole disk (or single partition via args), skip settings
|
||||
bp_dest = state.destination.path
|
||||
state.add_block_pair(state.source, bp_dest)
|
||||
return source_parts
|
||||
|
||||
# New run, use new settings file
|
||||
settings['Needs Format'] = True
|
||||
offset = 0
|
||||
user_choice = cli.choice(
|
||||
'Format clone using GPT, MBR, or match Source type?',
|
||||
['G', 'M', 'S'],
|
||||
)
|
||||
if user_choice == 'G':
|
||||
settings['Table Type'] = 'GPT'
|
||||
elif user_choice == 'M':
|
||||
settings['Table Type'] = 'MBR'
|
||||
else:
|
||||
# Match source type
|
||||
settings['Table Type'] = get_table_type(state.source.path)
|
||||
if cli.ask('Create an empty Windows boot partition on the clone?'):
|
||||
settings['Create Boot Partition'] = True
|
||||
offset = 2 if settings['Table Type'] == 'GPT' else 1
|
||||
|
||||
# Add pairs
|
||||
for dest_num, part in enumerate(source_parts):
|
||||
dest_num += offset + 1
|
||||
bp_dest = pathlib.Path(
|
||||
f'{state.destination.path}{dest_sep}{dest_num}',
|
||||
)
|
||||
state.add_block_pair(part, bp_dest)
|
||||
|
||||
# Add to settings file
|
||||
source_num = re.sub(r'^.*?(\d+)$', r'\1', part.path.name)
|
||||
settings['Partition Mapping'].append([source_num, dest_num])
|
||||
|
||||
# Save settings
|
||||
state.save_settings(settings)
|
||||
|
||||
# Done
|
||||
return source_parts
|
||||
|
||||
|
||||
def add_image_block_pairs(state) -> list[hw_disk.Disk]:
|
||||
"""Add device to image file block pairs."""
|
||||
source_parts = menus.select_disk_parts(state.mode, state.source)
|
||||
for part in source_parts:
|
||||
state.add_block_pair(part, state.destination)
|
||||
|
||||
# Done
|
||||
return source_parts
|
||||
|
||||
|
||||
def build_block_pair_report(block_pairs, settings) -> list:
|
||||
"""Build block pair report, returns list."""
|
||||
report = []
|
||||
notes = []
|
||||
if block_pairs:
|
||||
report.append(ansi.color_string('Block Pairs', 'GREEN'))
|
||||
else:
|
||||
# Bail early
|
||||
return report
|
||||
|
||||
# Show block pair mapping
|
||||
if settings and settings['Create Boot Partition']:
|
||||
if settings['Table Type'] == 'GPT':
|
||||
report.append(f'{" —— ":<9} --> EFI System Partition')
|
||||
report.append(f'{" —— ":<9} --> Microsoft Reserved Partition')
|
||||
elif settings['Table Type'] == 'MBR':
|
||||
report.append(f'{" —— ":<9} --> System Reserved')
|
||||
for pair in block_pairs:
|
||||
report.append(f'{pair.source.name:<9} --> {pair.destination.name}')
|
||||
|
||||
# Show resume messages as necessary
|
||||
if settings:
|
||||
if not settings['First Run']:
|
||||
notes.append(
|
||||
ansi.color_string(
|
||||
['NOTE:', 'Clone settings loaded from previous run.'],
|
||||
['BLUE', None],
|
||||
),
|
||||
)
|
||||
if settings['Needs Format'] and settings['Table Type']:
|
||||
msg = f'Destination will be formatted using {settings["Table Type"]}'
|
||||
notes.append(
|
||||
ansi.color_string(
|
||||
['NOTE:', msg],
|
||||
['BLUE', None],
|
||||
),
|
||||
)
|
||||
if any(pair.get_rescued_size() > 0 for pair in block_pairs):
|
||||
notes.append(
|
||||
ansi.color_string(
|
||||
['NOTE:', 'Resume data loaded from map file(s).'],
|
||||
['BLUE', None],
|
||||
),
|
||||
)
|
||||
|
||||
# Add notes to report
|
||||
if notes:
|
||||
report.append(' ')
|
||||
report.extend(notes)
|
||||
|
||||
# Done
|
||||
return report
|
||||
|
||||
|
||||
def build_sfdisk_partition_line(table_type, dev_path, size, details) -> str:
|
||||
"""Build sfdisk partition line using passed details, returns str."""
|
||||
line = f'{dev_path} : size={size}'
|
||||
dest_type = ''
|
||||
source_filesystem = str(details.get('fstype', '')).upper()
|
||||
source_table_type = ''
|
||||
source_type = details.get('parttype', '')
|
||||
|
||||
# Set dest type
|
||||
if re.match(r'^0x\w+$', source_type):
|
||||
# Source is a MBR type
|
||||
source_table_type = 'MBR'
|
||||
if table_type == 'MBR':
|
||||
dest_type = source_type.replace('0x', '').lower()
|
||||
elif re.match(r'^\w{8}-\w{4}-\w{4}-\w{4}-\w{12}$', source_type):
|
||||
# Source is a GPT type
|
||||
source_table_type = 'GPT'
|
||||
if table_type == 'GPT':
|
||||
dest_type = source_type.upper()
|
||||
if not dest_type:
|
||||
# Assuming changing table types, set based on FS
|
||||
if source_filesystem in cfg.ddrescue.PARTITION_TYPES.get(table_type, {}):
|
||||
dest_type = cfg.ddrescue.PARTITION_TYPES[table_type][source_filesystem]
|
||||
line += f', type={dest_type}'
|
||||
|
||||
# Safety Check
|
||||
if not dest_type:
|
||||
cli.print_error(f'Failed to determine partition type for: {dev_path}')
|
||||
raise std.GenericAbort()
|
||||
|
||||
# Add extra details
|
||||
if details.get('partlabel', ''):
|
||||
line += f', name="{details["partlabel"]}"'
|
||||
if details.get('partuuid', '') and source_table_type == table_type:
|
||||
# Only add UUID if source/dest table types match
|
||||
line += f', uuid={details["partuuid"].upper()}'
|
||||
|
||||
# Done
|
||||
return line
|
||||
|
||||
|
||||
def get_partition_separator(name) -> str:
|
||||
"""Get partition separator based on device name, returns str."""
|
||||
separator = ''
|
||||
if re.search(r'(loop|mmc|nvme)', name, re.IGNORECASE):
|
||||
separator = 'p'
|
||||
|
||||
return separator
|
||||
|
||||
|
||||
def get_table_type(disk_path) -> str:
|
||||
"""Get disk partition table type, returns str.
|
||||
|
||||
NOTE: If resulting table type is not GPT or MBR
|
||||
then an exception is raised.
|
||||
"""
|
||||
disk_path = str(disk_path)
|
||||
table_type = None
|
||||
|
||||
# Linux
|
||||
if std.PLATFORM == 'Linux':
|
||||
cmd = f'lsblk --json --output=pttype --nodeps {disk_path}'.split()
|
||||
json_data = exe.get_json_from_command(cmd)
|
||||
table_type = json_data['blockdevices'][0].get('pttype', '').upper()
|
||||
table_type = table_type.replace('DOS', 'MBR')
|
||||
|
||||
# macOS
|
||||
if std.PLATFORM == 'Darwin':
|
||||
cmd = ['diskutil', 'list', '-plist', disk_path]
|
||||
proc = exe.run_program(cmd, check=False, encoding=None, errors=None)
|
||||
try:
|
||||
plist_data = plistlib.loads(proc.stdout)
|
||||
except (TypeError, ValueError):
|
||||
# Invalid / corrupt plist data? return empty dict to avoid crash
|
||||
pass
|
||||
else:
|
||||
disk_details = plist_data.get('AllDisksAndPartitions', [{}])[0]
|
||||
table_type = disk_details['Content']
|
||||
table_type = table_type.replace('FDisk_partition_scheme', 'MBR')
|
||||
table_type = table_type.replace('GUID_partition_scheme', 'GPT')
|
||||
|
||||
# Check type
|
||||
if table_type not in ('GPT', 'MBR'):
|
||||
cli.print_error(f'Unsupported partition table type: {table_type}')
|
||||
raise std.GenericAbort()
|
||||
|
||||
# Done
|
||||
return table_type
|
||||
|
||||
|
||||
def prep_destination(
|
||||
state,
|
||||
source_parts: list[hw_disk.Disk],
|
||||
dry_run: bool = True,
|
||||
) -> None:
|
||||
"""Prep destination as necessary."""
|
||||
# TODO: Split into Linux and macOS
|
||||
# logical sector size is not easily found under macOS
|
||||
# It might be easier to rewrite this section using macOS tools
|
||||
dest_prefix = str(state.destination.path)
|
||||
dest_prefix += get_partition_separator(state.destination.path.name)
|
||||
esp_type = 'C12A7328-F81F-11D2-BA4B-00A0C93EC93B'
|
||||
msr_type = 'E3C9E316-0B5C-4DB8-817D-F92DF00215AE'
|
||||
part_num = 0
|
||||
sfdisk_script = []
|
||||
settings = state.load_settings()
|
||||
|
||||
# Bail early
|
||||
if not settings['Needs Format']:
|
||||
return
|
||||
|
||||
# Add partition table settings
|
||||
if settings['Table Type'] == 'GPT':
|
||||
sfdisk_script.append('label: gpt')
|
||||
else:
|
||||
sfdisk_script.append('label: dos')
|
||||
sfdisk_script.append('unit: sectors')
|
||||
sfdisk_script.append('')
|
||||
|
||||
# Add boot partition if requested
|
||||
if settings['Create Boot Partition']:
|
||||
if settings['Table Type'] == 'GPT':
|
||||
part_num += 1
|
||||
sfdisk_script.append(
|
||||
build_sfdisk_partition_line(
|
||||
table_type='GPT',
|
||||
dev_path=f'{dest_prefix}{part_num}',
|
||||
size='260MiB',
|
||||
details={'parttype': esp_type, 'partlabel': 'EFI System'},
|
||||
),
|
||||
)
|
||||
part_num += 1
|
||||
sfdisk_script.append(
|
||||
build_sfdisk_partition_line(
|
||||
table_type=settings['Table Type'],
|
||||
dev_path=f'{dest_prefix}{part_num}',
|
||||
size='16MiB',
|
||||
details={'parttype': msr_type, 'partlabel': 'Microsoft Reserved'},
|
||||
),
|
||||
)
|
||||
elif settings['Table Type'] == 'MBR':
|
||||
part_num += 1
|
||||
sfdisk_script.append(
|
||||
build_sfdisk_partition_line(
|
||||
table_type='MBR',
|
||||
dev_path=f'{dest_prefix}{part_num}',
|
||||
size='100MiB',
|
||||
details={'parttype': '0x7', 'partlabel': 'System Reserved'},
|
||||
),
|
||||
)
|
||||
|
||||
# Add selected partition(s)
|
||||
for part in source_parts:
|
||||
num_sectors = part.size / state.destination.log_sec
|
||||
num_sectors = math.ceil(num_sectors)
|
||||
part_num += 1
|
||||
sfdisk_script.append(
|
||||
build_sfdisk_partition_line(
|
||||
table_type=settings['Table Type'],
|
||||
dev_path=f'{dest_prefix}{part_num}',
|
||||
size=num_sectors,
|
||||
details=part.raw_details,
|
||||
),
|
||||
)
|
||||
|
||||
# Save sfdisk script
|
||||
script_path = (
|
||||
f'{state.working_dir}/'
|
||||
f'sfdisk_{state.destination.path.name}.script'
|
||||
)
|
||||
with open(script_path, 'w', encoding='utf-8') as _f:
|
||||
_f.write('\n'.join(sfdisk_script))
|
||||
|
||||
# Skip real format for dry runs
|
||||
if dry_run:
|
||||
LOG.info('Dry run, refusing to format destination')
|
||||
return
|
||||
|
||||
# Format disk
|
||||
LOG.warning('Formatting destination: %s', state.destination.path)
|
||||
with open(script_path, 'r', encoding='utf-8') as _f:
|
||||
proc = exe.run_program(
|
||||
cmd=['sudo', 'sfdisk', state.destination.path],
|
||||
stdin=_f,
|
||||
check=False,
|
||||
)
|
||||
if proc.returncode != 0:
|
||||
cli.print_error('Error(s) encoundtered while formatting destination')
|
||||
raise std.GenericAbort()
|
||||
|
||||
# Update settings
|
||||
settings['Needs Format'] = False
|
||||
state.save_settings(settings)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
File diff suppressed because it is too large
Load diff
|
|
@ -1,109 +0,0 @@
|
|||
"""WizardKit: ddrescue TUI - State"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import atexit
|
||||
import logging
|
||||
import pathlib
|
||||
import plistlib
|
||||
import re
|
||||
|
||||
from wk import exe
|
||||
from wk.std import PLATFORM
|
||||
from wk.ui import cli
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
|
||||
|
||||
# Functions
|
||||
def mount_raw_image(path) -> pathlib.Path:
|
||||
"""Mount raw image using OS specific methods, returns pathlib.Path."""
|
||||
loopback_path = None
|
||||
|
||||
if PLATFORM == 'Darwin':
|
||||
loopback_path = mount_raw_image_macos(path)
|
||||
elif PLATFORM == 'Linux':
|
||||
loopback_path = mount_raw_image_linux(path)
|
||||
|
||||
# Check
|
||||
if not loopback_path:
|
||||
cli.print_error(f'Failed to mount image: {path}')
|
||||
|
||||
# Register unmount atexit
|
||||
atexit.register(unmount_loopback_device, loopback_path)
|
||||
|
||||
# Done
|
||||
return loopback_path
|
||||
|
||||
|
||||
def mount_raw_image_linux(path) -> pathlib.Path:
|
||||
"""Mount raw image using losetup, returns pathlib.Path."""
|
||||
loopback_path = None
|
||||
|
||||
# Mount using losetup
|
||||
cmd = [
|
||||
'sudo',
|
||||
'losetup',
|
||||
'--find',
|
||||
'--partscan',
|
||||
'--show',
|
||||
path,
|
||||
]
|
||||
proc = exe.run_program(cmd, check=False)
|
||||
|
||||
# Check result
|
||||
if proc.returncode == 0:
|
||||
loopback_path = proc.stdout.strip()
|
||||
|
||||
# Done
|
||||
return loopback_path
|
||||
|
||||
|
||||
def mount_raw_image_macos(path) -> pathlib.Path:
|
||||
"""Mount raw image using hdiutil, returns pathlib.Path."""
|
||||
loopback_path = None
|
||||
plist_data = {}
|
||||
|
||||
# Mount using hdiutil
|
||||
# plistdata['system-entities'][{}...]
|
||||
cmd = [
|
||||
'hdiutil', 'attach',
|
||||
'-imagekey', 'diskimage-class=CRawDiskImage',
|
||||
'-nomount',
|
||||
'-plist',
|
||||
'-readonly',
|
||||
path,
|
||||
]
|
||||
proc = exe.run_program(cmd, check=False, encoding=None, errors=None)
|
||||
|
||||
# Check result
|
||||
try:
|
||||
plist_data = plistlib.loads(proc.stdout)
|
||||
except plistlib.InvalidFileException:
|
||||
return None
|
||||
for dev in plist_data.get('system-entities', []):
|
||||
dev_path = dev.get('dev-entry', '')
|
||||
if re.match(r'^/dev/disk\d+$', dev_path):
|
||||
loopback_path = dev_path
|
||||
|
||||
# Done
|
||||
return loopback_path
|
||||
|
||||
|
||||
def unmount_loopback_device(path) -> None:
|
||||
"""Unmount loopback device using OS specific methods."""
|
||||
cmd = []
|
||||
|
||||
# Build OS specific cmd
|
||||
if PLATFORM == 'Darwin':
|
||||
cmd = ['hdiutil', 'detach', path]
|
||||
elif PLATFORM == 'Linux':
|
||||
cmd = ['sudo', 'losetup', '--detach', path]
|
||||
|
||||
# Unmount loopback device
|
||||
exe.run_program(cmd, check=False)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
@ -1,273 +0,0 @@
|
|||
"""WizardKit: ddrescue TUI - Menus"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import logging
|
||||
import pathlib
|
||||
|
||||
from wk.cfg.ddrescue import DDRESCUE_SETTINGS
|
||||
from wk.hw.disk import Disk, get_disks
|
||||
from wk.std import GenericAbort, PLATFORM, bytes_to_string
|
||||
from wk.ui import ansi, cli
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
CLONE_SETTINGS = {
|
||||
'Source': None,
|
||||
'Destination': None,
|
||||
'Create Boot Partition': False,
|
||||
'First Run': True,
|
||||
'Needs Format': False,
|
||||
'Table Type': None,
|
||||
'Partition Mapping': [
|
||||
# (5, 1) ## Clone source partition #5 to destination partition #1
|
||||
],
|
||||
}
|
||||
if PLATFORM == 'Darwin':
|
||||
# TODO: Direct I/O needs more testing under macOS
|
||||
DDRESCUE_SETTINGS['Default']['--idirect'] = {'Selected': False, 'Hidden': True}
|
||||
DDRESCUE_SETTINGS['Default']['--odirect'] = {'Selected': False, 'Hidden': True}
|
||||
MENU_ACTIONS = (
|
||||
'Start',
|
||||
f'Change settings {ansi.color_string("(experts only)", "YELLOW")}',
|
||||
f'Detect drives {ansi.color_string("(experts only)", "YELLOW")}',
|
||||
'Quit')
|
||||
MENU_TOGGLES = {
|
||||
'Auto continue (if recovery % over threshold)': True,
|
||||
'Retry (mark non-rescued sectors "non-tried")': False,
|
||||
}
|
||||
SETTING_PRESETS = (
|
||||
'Default',
|
||||
'Fast',
|
||||
'Safe',
|
||||
)
|
||||
|
||||
|
||||
# Functions
|
||||
def main() -> cli.Menu:
|
||||
"""Main menu, returns wk.ui.cli.Menu."""
|
||||
menu = cli.Menu(title=ansi.color_string('ddrescue TUI: Main Menu', 'GREEN'))
|
||||
menu.separator = ' '
|
||||
|
||||
# Add actions, options, etc
|
||||
for action in MENU_ACTIONS:
|
||||
if not (PLATFORM == 'Darwin' and 'Detect drives' in action):
|
||||
menu.add_action(action)
|
||||
for toggle, selected in MENU_TOGGLES.items():
|
||||
menu.add_toggle(toggle, {'Selected': selected})
|
||||
|
||||
# Done
|
||||
return menu
|
||||
|
||||
|
||||
def settings(mode: str, silent: bool = True) -> cli.Menu:
|
||||
"""Settings menu, returns wk.ui.cli.Menu."""
|
||||
title_text = [
|
||||
ansi.color_string('ddrescue TUI: Expert Settings', 'GREEN'),
|
||||
' ',
|
||||
ansi.color_string(
|
||||
['These settings can cause', 'MAJOR DAMAGE', 'to drives'],
|
||||
['YELLOW', 'RED', 'YELLOW'],
|
||||
),
|
||||
'Please read the manual before making changes',
|
||||
]
|
||||
menu = cli.Menu(title='\n'.join(title_text))
|
||||
menu.separator = ' '
|
||||
preset = 'Default'
|
||||
if not silent:
|
||||
# Ask which preset to use
|
||||
cli.print_standard(
|
||||
f'Available ddrescue presets: {" / ".join(SETTING_PRESETS)}'
|
||||
)
|
||||
preset = cli.choice('Please select a preset:', SETTING_PRESETS)
|
||||
|
||||
# Fix selection
|
||||
for _p in SETTING_PRESETS:
|
||||
if _p.startswith(preset):
|
||||
preset = _p
|
||||
|
||||
# Add default settings
|
||||
menu.add_action('Load Preset')
|
||||
menu.add_action('Main Menu')
|
||||
for name, details in DDRESCUE_SETTINGS['Default'].items():
|
||||
menu.add_option(name, details.copy())
|
||||
|
||||
# Update settings using preset
|
||||
if preset != 'Default':
|
||||
for name, details in DDRESCUE_SETTINGS[preset].items():
|
||||
menu.options[name].update(details.copy())
|
||||
|
||||
# Disable direct output when saving to an image
|
||||
if mode == 'Image':
|
||||
menu.options['--odirect']['Disabled'] = True
|
||||
menu.options['--odirect']['Selected'] = False
|
||||
|
||||
# Done
|
||||
return menu
|
||||
|
||||
|
||||
def disks() -> cli.Menu:
|
||||
"""Disk menu, returns wk.ui.cli.Menu()."""
|
||||
cli.print_info('Scanning disks...')
|
||||
available_disks = get_disks()
|
||||
menu = cli.Menu('ddrescue TUI: Disk selection')
|
||||
menu.disabled_str = 'Already selected'
|
||||
menu.separator = ' '
|
||||
menu.add_action('Quit')
|
||||
for disk in available_disks:
|
||||
menu.add_option(
|
||||
name=(
|
||||
f'{str(disk.path):<12} '
|
||||
f'{disk.bus:<5} '
|
||||
f'{bytes_to_string(disk.size, decimals=1, use_binary=False):<8} '
|
||||
f'{disk.model} '
|
||||
f'{disk.serial}'
|
||||
),
|
||||
details={'Object': disk},
|
||||
)
|
||||
|
||||
# Done
|
||||
return menu
|
||||
|
||||
|
||||
def select_disk(prompt_msg: str, menu: cli.Menu) -> Disk:
|
||||
"""Select disk from provided Menu, returns Disk()."""
|
||||
menu.title = ansi.color_string(
|
||||
f'ddrescue TUI: {prompt_msg} Selection', 'GREEN',
|
||||
)
|
||||
|
||||
# Get selection
|
||||
selection = menu.simple_select()
|
||||
if 'Quit' in selection:
|
||||
raise GenericAbort()
|
||||
|
||||
# Disable selected disk's menu entry
|
||||
menu.options[selection[0]]['Disabled'] = True
|
||||
|
||||
# Update details to include child devices
|
||||
selected_disk = selection[-1]['Object']
|
||||
selected_disk.update_details(skip_children=False)
|
||||
|
||||
# Done
|
||||
return selected_disk
|
||||
|
||||
|
||||
def select_disk_parts(prompt_msg, disk) -> list[Disk]:
|
||||
"""Select disk parts from list, returns list of Disk()."""
|
||||
title = ansi.color_string('ddrescue TUI: Partition Selection', 'GREEN')
|
||||
title += f'\n\nDisk: {disk.path} {disk.description}'
|
||||
menu = cli.Menu(title)
|
||||
menu.separator = ' '
|
||||
menu.add_action('All')
|
||||
menu.add_action('None')
|
||||
menu.add_action('Proceed', {'Separator': True})
|
||||
menu.add_action('Quit')
|
||||
object_list = []
|
||||
|
||||
def _select_parts(menu) -> None:
|
||||
"""Loop over selection menu until at least one partition selected."""
|
||||
while True:
|
||||
selection = menu.advanced_select(
|
||||
f'Please select the parts to {prompt_msg.lower()}: ',
|
||||
)
|
||||
if 'All' in selection:
|
||||
for option in menu.options.values():
|
||||
option['Selected'] = True
|
||||
elif 'None' in selection:
|
||||
for option in menu.options.values():
|
||||
option['Selected'] = False
|
||||
elif 'Proceed' in selection:
|
||||
if any(option['Selected'] for option in menu.options.values()):
|
||||
# At least one partition/device selected/device selected
|
||||
break
|
||||
elif 'Quit' in selection:
|
||||
raise GenericAbort()
|
||||
|
||||
# Bail early if running under macOS
|
||||
if PLATFORM == 'Darwin':
|
||||
return [disk]
|
||||
|
||||
# Bail early if child device selected
|
||||
if disk.parent:
|
||||
return [disk]
|
||||
|
||||
# Add parts
|
||||
whole_disk_str = f'{str(disk.path):<14} (Whole device)'
|
||||
for part in disk.children:
|
||||
fstype = part.get('fstype', '')
|
||||
fstype = str(fstype) if fstype else ''
|
||||
size = part["size"]
|
||||
name = (
|
||||
f'{str(part["path"]):<14} '
|
||||
f'{fstype.upper():<5} '
|
||||
f'({bytes_to_string(size, decimals=1, use_binary=True):>6})'
|
||||
)
|
||||
menu.add_option(name, details={'Selected': True, 'pathlib.Path': part['path']})
|
||||
|
||||
# Add whole disk if necessary
|
||||
if not menu.options:
|
||||
menu.add_option(whole_disk_str, {'Selected': True, 'pathlib.Path': disk.path})
|
||||
menu.title += '\n\n'
|
||||
menu.title += ansi.color_string(' No partitions detected.', 'YELLOW')
|
||||
|
||||
# Get selection
|
||||
_select_parts(menu)
|
||||
|
||||
# Build list of Disk() object_list
|
||||
for option in menu.options.values():
|
||||
if option['Selected']:
|
||||
object_list.append(option['pathlib.Path'])
|
||||
|
||||
# Check if whole disk selected
|
||||
if len(object_list) == len(disk.children):
|
||||
# NOTE: This is not true if the disk has no partitions
|
||||
msg = f'Preserve partition table and unused space in {prompt_msg.lower()}?'
|
||||
if cli.ask(msg):
|
||||
# Replace part list with whole disk obj
|
||||
object_list = [disk.path]
|
||||
|
||||
# Convert object_list to Disk() objects
|
||||
cli.print_standard(' ')
|
||||
cli.print_info('Getting disk/partition details...')
|
||||
object_list = [Disk(path) for path in object_list]
|
||||
|
||||
# Done
|
||||
return object_list
|
||||
|
||||
|
||||
def select_path(prompt_msg) -> pathlib.Path:
|
||||
"""Select path, returns pathlib.Path."""
|
||||
invalid = False
|
||||
menu = cli.Menu(
|
||||
title=ansi.color_string(f'ddrescue TUI: {prompt_msg} Path Selection', 'GREEN'),
|
||||
)
|
||||
menu.separator = ' '
|
||||
menu.add_action('Quit')
|
||||
menu.add_option('Current directory')
|
||||
menu.add_option('Enter manually')
|
||||
path = pathlib.Path.cwd()
|
||||
|
||||
# Make selection
|
||||
selection = menu.simple_select()
|
||||
if 'Current directory' in selection:
|
||||
pass
|
||||
elif 'Enter manually' in selection:
|
||||
path = pathlib.Path(cli.input_text('Please enter path: '))
|
||||
elif 'Quit' in selection:
|
||||
raise GenericAbort()
|
||||
|
||||
# Check
|
||||
try:
|
||||
path = path.resolve()
|
||||
except TypeError:
|
||||
invalid = True
|
||||
if invalid or not path.is_dir():
|
||||
cli.print_error(f'Invalid path: {path}')
|
||||
raise GenericAbort()
|
||||
|
||||
# Done
|
||||
return path
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
File diff suppressed because it is too large
Load diff
|
|
@ -1,87 +1,23 @@
|
|||
"""WizardKit: Debug Functions"""
|
||||
# pylint: disable=invalid-name
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import inspect
|
||||
import logging
|
||||
import lzma
|
||||
import os
|
||||
import pathlib
|
||||
import pickle
|
||||
import platform
|
||||
import re
|
||||
import socket
|
||||
import sys
|
||||
import time
|
||||
|
||||
from typing import Any
|
||||
|
||||
import requests
|
||||
|
||||
from wk.cfg.net import CRASH_SERVER
|
||||
from wk.log import get_root_logger_path
|
||||
|
||||
# Classes
|
||||
class Debug():
|
||||
# pylint: disable=too-few-public-methods
|
||||
"""Object used when dumping debug data."""
|
||||
def method(self) -> None:
|
||||
def method(self):
|
||||
"""Dummy method used to identify functions vs data."""
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
DEBUG_CLASS = Debug()
|
||||
METHOD_TYPE = type(DEBUG_CLASS.method)
|
||||
|
||||
|
||||
# Functions
|
||||
def generate_debug_report() -> str:
|
||||
"""Generate debug report, returns str."""
|
||||
platform_function_list = (
|
||||
'architecture',
|
||||
'machine',
|
||||
'platform',
|
||||
'python_version',
|
||||
)
|
||||
report = []
|
||||
|
||||
# Logging data
|
||||
try:
|
||||
log_path = get_root_logger_path()
|
||||
except RuntimeError:
|
||||
# Assuming logging wasn't started
|
||||
pass
|
||||
else:
|
||||
report.append('------ Start Log -------')
|
||||
report.append('')
|
||||
with open(log_path, 'r', encoding='utf-8') as log_file:
|
||||
report.extend(log_file.read().splitlines())
|
||||
report.append('')
|
||||
report.append('------- End Log --------')
|
||||
|
||||
# System
|
||||
report.append('--- Start debug info ---')
|
||||
report.append('')
|
||||
report.append('[System]')
|
||||
report.append(f' {"FQDN":<24} {socket.getfqdn()}')
|
||||
for func in platform_function_list:
|
||||
func_name = func.replace('_', ' ').capitalize()
|
||||
func_result = getattr(platform, func)()
|
||||
report.append(f' {func_name:<24} {func_result}')
|
||||
report.append(f' {"Python sys.argv":<24} {sys.argv}')
|
||||
report.append('')
|
||||
|
||||
# Environment
|
||||
report.append('[Environment Variables]')
|
||||
for key, value in sorted(os.environ.items()):
|
||||
report.append(f' {key:<24} {value}')
|
||||
report.append('')
|
||||
|
||||
# Done
|
||||
report.append('---- End debug info ----')
|
||||
return '\n'.join(report)
|
||||
|
||||
|
||||
def generate_object_report(obj: Any, indent: int = 0) -> list[str]:
|
||||
def generate_object_report(obj, indent=0):
|
||||
"""Generate debug report for obj, returns list."""
|
||||
report = []
|
||||
attr_list = []
|
||||
|
|
@ -103,7 +39,7 @@ def generate_object_report(obj: Any, indent: int = 0) -> list[str]:
|
|||
# Add attribute to report (expanded if necessary)
|
||||
if isinstance(attr, dict):
|
||||
report.append(f'{name}:')
|
||||
for key, value in attr.items():
|
||||
for key, value in sorted(attr.items()):
|
||||
report.append(f'{" "*(indent+1)}{key}: {str(value)}')
|
||||
else:
|
||||
report.append(f'{" "*indent}{name}: {str(attr)}')
|
||||
|
|
@ -112,78 +48,5 @@ def generate_object_report(obj: Any, indent: int = 0) -> list[str]:
|
|||
return report
|
||||
|
||||
|
||||
def save_pickles(
|
||||
obj_dict: dict[Any, Any],
|
||||
out_path: pathlib.Path | str | None = None,
|
||||
) -> None:
|
||||
"""Save dict of objects using pickle."""
|
||||
LOG.info('Saving pickles')
|
||||
|
||||
# Set path
|
||||
if not out_path:
|
||||
out_path = get_root_logger_path()
|
||||
out_path = out_path.parent.joinpath('../debug').resolve()
|
||||
|
||||
# Save pickles
|
||||
try:
|
||||
for name, obj in obj_dict.copy().items():
|
||||
if name.startswith('__') or inspect.ismodule(obj):
|
||||
continue
|
||||
with open(f'{out_path}/{name}.pickle', 'wb') as _f:
|
||||
pickle.dump(obj, _f, protocol=pickle.HIGHEST_PROTOCOL)
|
||||
except Exception:
|
||||
LOG.error('Failed to save all the pickles', exc_info=True)
|
||||
|
||||
|
||||
def upload_debug_report(
|
||||
report: str,
|
||||
compress: bool = True,
|
||||
reason: str = 'DEBUG',
|
||||
) -> None:
|
||||
"""Upload debug report to CRASH_SERVER as specified in wk.cfg.main."""
|
||||
LOG.info('Uploading debug report to %s', CRASH_SERVER.get('Name', '?'))
|
||||
headers = CRASH_SERVER.get('Headers', {'X-Requested-With': 'XMLHttpRequest'})
|
||||
if compress:
|
||||
headers['Content-Type'] = 'application/octet-stream'
|
||||
|
||||
# Check if the required server details are available
|
||||
if not all(CRASH_SERVER.get(key, False) for key in ('Name', 'Url', 'User')):
|
||||
msg = 'Server details missing, aborting upload.'
|
||||
print(msg)
|
||||
raise RuntimeError(msg)
|
||||
|
||||
# Set filename (based on the logging config if possible)
|
||||
filename = 'Unknown'
|
||||
try:
|
||||
log_path = get_root_logger_path()
|
||||
except RuntimeError:
|
||||
# Assuming logging wasn't started
|
||||
pass
|
||||
else:
|
||||
# Strip everything but the prefix
|
||||
filename = re.sub(r'^(.*)_(\d{4}-\d{2}-\d{2}.*)', r'\1', log_path.name)
|
||||
filename = f'{filename}_{reason}_{time.strftime("%Y-%m-%d_%H%M%S%z")}.log'
|
||||
LOG.debug('filename: %s', filename)
|
||||
|
||||
# Compress report
|
||||
if compress:
|
||||
filename += '.xz'
|
||||
xz_report = lzma.compress(report.encode('utf8'))
|
||||
|
||||
# Upload report
|
||||
url = f'{CRASH_SERVER["Url"]}/{filename}'
|
||||
response = requests.put(
|
||||
url,
|
||||
auth=(CRASH_SERVER['User'], CRASH_SERVER.get('Pass', '')),
|
||||
data=xz_report if compress else report,
|
||||
headers=headers,
|
||||
timeout=60,
|
||||
)
|
||||
|
||||
# Check response
|
||||
if not response.ok:
|
||||
raise RuntimeError('Failed to upload report')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
|
|||
|
|
@ -1,18 +1,15 @@
|
|||
"""WizardKit: Execution functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
#vim: sts=2 sw=2 ts=2
|
||||
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import pathlib
|
||||
import re
|
||||
import subprocess
|
||||
import time
|
||||
|
||||
from io import IOBase
|
||||
from queue import Queue, Empty
|
||||
from threading import Thread
|
||||
from typing import Any, Callable, Iterable
|
||||
from queue import Queue, Empty
|
||||
|
||||
import psutil
|
||||
|
||||
|
|
@ -24,15 +21,16 @@ LOG = logging.getLogger(__name__)
|
|||
# Classes
|
||||
class NonBlockingStreamReader():
|
||||
"""Class to allow non-blocking reads from a stream."""
|
||||
# pylint: disable=too-few-public-methods
|
||||
# Credits:
|
||||
## https://gist.github.com/EyalAr/7915597
|
||||
## https://stackoverflow.com/a/4896288
|
||||
|
||||
def __init__(self, stream: IOBase):
|
||||
self.stream: IOBase = stream
|
||||
self.queue: Queue = Queue()
|
||||
def __init__(self, stream):
|
||||
self.stream = stream
|
||||
self.queue = Queue()
|
||||
|
||||
def populate_queue(stream: IOBase, queue: Queue) -> None:
|
||||
def populate_queue(stream, queue):
|
||||
"""Collect lines from stream and put them in queue."""
|
||||
while not stream.closed:
|
||||
try:
|
||||
|
|
@ -48,18 +46,18 @@ class NonBlockingStreamReader():
|
|||
args=(self.stream, self.queue),
|
||||
)
|
||||
|
||||
def stop(self) -> None:
|
||||
def stop(self):
|
||||
"""Stop reading from input stream."""
|
||||
self.stream.close()
|
||||
|
||||
def read(self, timeout: float | int | None = None) -> Any:
|
||||
def read(self, timeout=None):
|
||||
"""Read from queue if possible, returns item from queue."""
|
||||
try:
|
||||
return self.queue.get(block=timeout is not None, timeout=timeout)
|
||||
except Empty:
|
||||
return None
|
||||
|
||||
def save_to_file(self, proc: subprocess.Popen, out_path: pathlib.Path | str) -> None:
|
||||
def save_to_file(self, proc, out_path):
|
||||
"""Continuously save output to file while proc is running."""
|
||||
LOG.debug('Saving process %s output to %s', proc, out_path)
|
||||
while proc.poll() is None:
|
||||
|
|
@ -77,12 +75,7 @@ class NonBlockingStreamReader():
|
|||
|
||||
|
||||
# Functions
|
||||
def build_cmd_kwargs(
|
||||
cmd: list[str],
|
||||
minimized: bool = False,
|
||||
pipe: bool = True,
|
||||
shell: bool = False,
|
||||
**kwargs) -> dict[str, Any]:
|
||||
def build_cmd_kwargs(cmd, minimized=False, pipe=True, shell=False, **kwargs):
|
||||
"""Build kwargs for use by subprocess functions, returns dict.
|
||||
|
||||
Specifically subprocess.run() and subprocess.Popen().
|
||||
|
|
@ -99,7 +92,7 @@ def build_cmd_kwargs(
|
|||
|
||||
# Strip sudo if appropriate
|
||||
if cmd[0] == 'sudo':
|
||||
if os.name == 'posix' and os.geteuid() == 0:
|
||||
if os.name == 'posix' and os.geteuid() == 0: # pylint: disable=no-member
|
||||
cmd.pop(0)
|
||||
|
||||
# Add additional kwargs if applicable
|
||||
|
|
@ -114,8 +107,8 @@ def build_cmd_kwargs(
|
|||
|
||||
# Start minimized
|
||||
if minimized:
|
||||
startupinfo = subprocess.STARTUPINFO() # type: ignore
|
||||
startupinfo.dwFlags = subprocess.STARTF_USESHOWWINDOW # type: ignore
|
||||
startupinfo = subprocess.STARTUPINFO()
|
||||
startupinfo.dwFlags = subprocess.STARTF_USESHOWWINDOW
|
||||
startupinfo.wShowWindow = 6
|
||||
cmd_kwargs['startupinfo'] = startupinfo
|
||||
|
||||
|
|
@ -130,12 +123,7 @@ def build_cmd_kwargs(
|
|||
return cmd_kwargs
|
||||
|
||||
|
||||
def get_json_from_command(
|
||||
cmd: list[str],
|
||||
check: bool = True,
|
||||
encoding: str = 'utf-8',
|
||||
errors: str = 'ignore',
|
||||
) -> dict[Any, Any]:
|
||||
def get_json_from_command(cmd, check=True, encoding='utf-8', errors='ignore'):
|
||||
"""Capture JSON content from cmd output, returns dict.
|
||||
|
||||
If the data can't be decoded then either an exception is raised
|
||||
|
|
@ -154,11 +142,7 @@ def get_json_from_command(
|
|||
return json_data
|
||||
|
||||
|
||||
def get_procs(
|
||||
name: str,
|
||||
exact: bool = True,
|
||||
try_again: bool = True,
|
||||
) -> list[psutil.Process]:
|
||||
def get_procs(name, exact=True, try_again=True):
|
||||
"""Get process object(s) based on name, returns list of proc objects."""
|
||||
LOG.debug('name: %s, exact: %s', name, exact)
|
||||
processes = []
|
||||
|
|
@ -178,12 +162,7 @@ def get_procs(
|
|||
return processes
|
||||
|
||||
|
||||
def kill_procs(
|
||||
name: str,
|
||||
exact: bool = True,
|
||||
force: bool = False,
|
||||
timeout: float | int = 30,
|
||||
) -> None:
|
||||
def kill_procs(name, exact=True, force=False, timeout=30):
|
||||
"""Kill all processes matching name (case-insensitively).
|
||||
|
||||
NOTE: Under Posix systems this will send SIGINT to allow processes
|
||||
|
|
@ -207,13 +186,7 @@ def kill_procs(
|
|||
proc.kill()
|
||||
|
||||
|
||||
def popen_program(
|
||||
cmd: list[str],
|
||||
minimized: bool = False,
|
||||
pipe: bool = False,
|
||||
shell: bool = False,
|
||||
**kwargs,
|
||||
) -> subprocess.Popen:
|
||||
def popen_program(cmd, minimized=False, pipe=False, shell=False, **kwargs):
|
||||
"""Run program and return a subprocess.Popen object."""
|
||||
LOG.debug(
|
||||
'cmd: %s, minimized: %s, pipe: %s, shell: %s',
|
||||
|
|
@ -227,6 +200,7 @@ def popen_program(
|
|||
shell=shell,
|
||||
**kwargs)
|
||||
try:
|
||||
# pylint: disable=consider-using-with
|
||||
proc = subprocess.Popen(**cmd_kwargs)
|
||||
except FileNotFoundError:
|
||||
LOG.error('Command not found: %s', cmd)
|
||||
|
|
@ -237,13 +211,8 @@ def popen_program(
|
|||
return proc
|
||||
|
||||
|
||||
def run_program(
|
||||
cmd: list[str],
|
||||
check: bool = True,
|
||||
pipe: bool = True,
|
||||
shell: bool = False,
|
||||
**kwargs,
|
||||
) -> subprocess.CompletedProcess:
|
||||
def run_program(cmd, check=True, pipe=True, shell=False, **kwargs):
|
||||
# pylint: disable=subprocess-run-check
|
||||
"""Run program and return a subprocess.CompletedProcess object."""
|
||||
LOG.debug(
|
||||
'cmd: %s, check: %s, pipe: %s, shell: %s',
|
||||
|
|
@ -256,9 +225,8 @@ def run_program(
|
|||
pipe=pipe,
|
||||
shell=shell,
|
||||
**kwargs)
|
||||
check = cmd_kwargs.pop('check', True) # Avoids linting warning
|
||||
try:
|
||||
proc = subprocess.run(check=check, **cmd_kwargs)
|
||||
proc = subprocess.run(**cmd_kwargs)
|
||||
except FileNotFoundError:
|
||||
LOG.error('Command not found: %s', cmd)
|
||||
raise
|
||||
|
|
@ -268,11 +236,7 @@ def run_program(
|
|||
return proc
|
||||
|
||||
|
||||
def start_thread(
|
||||
function: Callable,
|
||||
args: Iterable[Any] | None = None,
|
||||
daemon: bool = True,
|
||||
) -> Thread:
|
||||
def start_thread(function, args=None, daemon=True):
|
||||
"""Run function as thread in background, returns Thread object."""
|
||||
LOG.debug(
|
||||
'Starting background thread for function: %s, args: %s, daemon: %s',
|
||||
|
|
@ -284,7 +248,7 @@ def start_thread(
|
|||
return thread
|
||||
|
||||
|
||||
def stop_process(proc: subprocess.Popen, graceful: bool = True) -> None:
|
||||
def stop_process(proc, graceful=True):
|
||||
"""Stop process.
|
||||
|
||||
NOTES: proc should be a subprocess.Popen obj.
|
||||
|
|
@ -293,24 +257,20 @@ def stop_process(proc: subprocess.Popen, graceful: bool = True) -> None:
|
|||
|
||||
# Graceful exit
|
||||
if graceful:
|
||||
if os.name == 'posix' and os.geteuid() != 0:
|
||||
if os.name == 'posix' and os.geteuid() != 0: # pylint: disable=no-member
|
||||
run_program(['sudo', 'kill', str(proc.pid)], check=False)
|
||||
else:
|
||||
proc.terminate()
|
||||
time.sleep(2)
|
||||
|
||||
# Force exit
|
||||
if os.name == 'posix' and os.geteuid() != 0:
|
||||
if os.name == 'posix' and os.geteuid() != 0: # pylint: disable=no-member
|
||||
run_program(['sudo', 'kill', '-9', str(proc.pid)], check=False)
|
||||
else:
|
||||
proc.kill()
|
||||
|
||||
|
||||
def wait_for_procs(
|
||||
name: str,
|
||||
exact: bool = True,
|
||||
timeout: float | int | None = None,
|
||||
) -> None:
|
||||
def wait_for_procs(name, exact=True, timeout=None):
|
||||
"""Wait for all process matching name."""
|
||||
LOG.debug('name: %s, exact: %s, timeout: %s', name, exact, timeout)
|
||||
target_procs = get_procs(name, exact=exact)
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@
|
|||
|
||||
import logging
|
||||
|
||||
from wk.ui import ansi
|
||||
from wk.std import color_string
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -33,10 +33,7 @@ THRESH_GREAT = 750 * 1024**2
|
|||
|
||||
|
||||
# Functions
|
||||
def generate_horizontal_graph(
|
||||
rate_list: list[float],
|
||||
graph_width: int = 40,
|
||||
oneline: bool = False) -> list[str]:
|
||||
def generate_horizontal_graph(rate_list, graph_width=40, oneline=False):
|
||||
"""Generate horizontal graph from rate_list, returns list."""
|
||||
graph = ['', '', '', '']
|
||||
scale = 8 if oneline else 32
|
||||
|
|
@ -55,27 +52,27 @@ def generate_horizontal_graph(
|
|||
rate_color = 'GREEN'
|
||||
|
||||
# Build graph
|
||||
full_block = ansi.color_string((GRAPH_HORIZONTAL[-1],), (rate_color,))
|
||||
full_block = color_string((GRAPH_HORIZONTAL[-1],), (rate_color,))
|
||||
if step >= 24:
|
||||
graph[0] += ansi.color_string((GRAPH_HORIZONTAL[step-24],), (rate_color,))
|
||||
graph[0] += color_string((GRAPH_HORIZONTAL[step-24],), (rate_color,))
|
||||
graph[1] += full_block
|
||||
graph[2] += full_block
|
||||
graph[3] += full_block
|
||||
elif step >= 16:
|
||||
graph[0] += ' '
|
||||
graph[1] += ansi.color_string((GRAPH_HORIZONTAL[step-16],), (rate_color,))
|
||||
graph[1] += color_string((GRAPH_HORIZONTAL[step-16],), (rate_color,))
|
||||
graph[2] += full_block
|
||||
graph[3] += full_block
|
||||
elif step >= 8:
|
||||
graph[0] += ' '
|
||||
graph[1] += ' '
|
||||
graph[2] += ansi.color_string((GRAPH_HORIZONTAL[step-8],), (rate_color,))
|
||||
graph[2] += color_string((GRAPH_HORIZONTAL[step-8],), (rate_color,))
|
||||
graph[3] += full_block
|
||||
else:
|
||||
graph[0] += ' '
|
||||
graph[1] += ' '
|
||||
graph[2] += ' '
|
||||
graph[3] += ansi.color_string((GRAPH_HORIZONTAL[step],), (rate_color,))
|
||||
graph[3] += color_string((GRAPH_HORIZONTAL[step],), (rate_color,))
|
||||
|
||||
# Done
|
||||
if oneline:
|
||||
|
|
@ -83,7 +80,7 @@ def generate_horizontal_graph(
|
|||
return graph
|
||||
|
||||
|
||||
def get_graph_step(rate: float, scale: int = 16) -> int:
|
||||
def get_graph_step(rate, scale=16):
|
||||
"""Get graph step based on rate and scale, returns int."""
|
||||
rate_in_mb = rate / (1024**2)
|
||||
step = 0
|
||||
|
|
@ -98,17 +95,14 @@ def get_graph_step(rate: float, scale: int = 16) -> int:
|
|||
return step
|
||||
|
||||
|
||||
def merge_rates(
|
||||
rates: list[float],
|
||||
graph_width: int = 40,
|
||||
) -> list[int | float]:
|
||||
def merge_rates(rates, graph_width=40):
|
||||
"""Merge rates to have entries equal to the width, returns list."""
|
||||
merged_rates = []
|
||||
offset = 0
|
||||
slice_width = int(len(rates) / graph_width)
|
||||
|
||||
# Merge rates
|
||||
for _ in range(graph_width):
|
||||
for _i in range(graph_width):
|
||||
merged_rates.append(sum(rates[offset:offset+slice_width])/slice_width)
|
||||
offset += slice_width
|
||||
|
||||
|
|
@ -116,7 +110,7 @@ def merge_rates(
|
|||
return merged_rates
|
||||
|
||||
|
||||
def vertical_graph_line(percent: float, rate: float, scale: int = 32) -> str:
|
||||
def vertical_graph_line(percent, rate, scale=32):
|
||||
"""Build colored graph string using thresholds, returns str."""
|
||||
color_bar = None
|
||||
color_rate = None
|
||||
|
|
@ -134,7 +128,7 @@ def vertical_graph_line(percent: float, rate: float, scale: int = 32) -> str:
|
|||
color_rate = 'GREEN'
|
||||
|
||||
# Build string
|
||||
line = ansi.color_string(
|
||||
line = color_string(
|
||||
strings=(
|
||||
f'{percent:5.1f}%',
|
||||
f'{GRAPH_VERTICAL[step]:<4}',
|
||||
|
|
|
|||
|
|
@ -23,14 +23,17 @@ def audio_test_linux() -> None:
|
|||
LOG.info('Audio Test')
|
||||
|
||||
# Set volume
|
||||
for source in ('Master', 'PCM'):
|
||||
cmd = f'amixer -q set "{source}" 80% unmute'.split()
|
||||
run_program(cmd, check=False)
|
||||
commands = (
|
||||
'pactl set-sink-mute @DEFAULT_SINK@ off',
|
||||
'pactl set-sink-volume @DEFAULT_SINK@ 80%',
|
||||
)
|
||||
for cmd in commands:
|
||||
run_program(cmd.split(), check=False)
|
||||
|
||||
# Run audio tests
|
||||
for mode in ('pink', 'wav'):
|
||||
cmd = f'speaker-test -c 2 -l 1 -t {mode}'.split()
|
||||
run_program(cmd, check=False, pipe=False)
|
||||
cmd = f'speaker-test -c 2 -l 1 -t {mode}'
|
||||
run_program(cmd.split(), check=False, pipe=False)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -21,8 +21,11 @@ from wk.cfg.hw import (
|
|||
THRESH_SSD_MIN,
|
||||
)
|
||||
from wk.exe import run_program
|
||||
from wk.std import PLATFORM
|
||||
from wk.ui import ansi
|
||||
from wk.std import (
|
||||
PLATFORM,
|
||||
strip_colors,
|
||||
color_string,
|
||||
)
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -63,8 +66,8 @@ def calc_io_dd_values(dev_size, test_mode=False) -> dict[str, int]:
|
|||
|
||||
test_mode limits the benchmark to IO_MINIMUM_TEST_SIZE (if possible)
|
||||
"""
|
||||
if test_mode and dev_size > IO_MINIMUM_TEST_SIZE:
|
||||
dev_size = IO_MINIMUM_TEST_SIZE
|
||||
if test_mode:
|
||||
dev_size = min(IO_MINIMUM_TEST_SIZE, dev_size)
|
||||
read_total = min(IO_MINIMUM_TEST_SIZE, dev_size)
|
||||
read_total = max(read_total, dev_size*IO_ALT_TEST_SIZE_FACTOR)
|
||||
read_chunks = int(read_total // IO_CHUNK_SIZE)
|
||||
|
|
@ -82,12 +85,6 @@ def calc_io_dd_values(dev_size, test_mode=False) -> dict[str, int]:
|
|||
# skip_extra_rate == 0 is fine
|
||||
pass
|
||||
|
||||
# Test mode
|
||||
if test_mode:
|
||||
read_chunks_limit = int(read_chunks * 0.1)
|
||||
read_chunks_limit = (read_chunks_limit // IO_GRAPH_WIDTH) * IO_GRAPH_WIDTH
|
||||
read_chunks = max(IO_GRAPH_WIDTH, read_chunks_limit)
|
||||
|
||||
# Done
|
||||
return {
|
||||
'Read Chunks': read_chunks,
|
||||
|
|
@ -113,7 +110,7 @@ def check_io_results(test_obj, rate_list, graph_width) -> None:
|
|||
|
||||
# Add horizontal graph to report
|
||||
for line in graph.generate_horizontal_graph(rate_list, graph_width):
|
||||
if not ansi.strip_colors(line).strip():
|
||||
if not strip_colors(line).strip():
|
||||
# Skip empty lines
|
||||
continue
|
||||
test_obj.report.append(line)
|
||||
|
|
@ -151,7 +148,7 @@ def run_io_test(test_obj, log_path, test_mode=False) -> None:
|
|||
LOG.info('Using %s for better performance', dev_path)
|
||||
offset = 0
|
||||
read_rates = []
|
||||
test_obj.report.append(ansi.color_string('I/O Benchmark', 'BLUE'))
|
||||
test_obj.report.append(color_string('I/O Benchmark', 'BLUE'))
|
||||
|
||||
# Get dd values or bail
|
||||
try:
|
||||
|
|
@ -159,7 +156,7 @@ def run_io_test(test_obj, log_path, test_mode=False) -> None:
|
|||
except DeviceTooSmallError:
|
||||
test_obj.set_status('N/A')
|
||||
test_obj.report.append(
|
||||
ansi.color_string('Disk too small to test', 'YELLOW'),
|
||||
color_string('Disk too small to test', 'YELLOW'),
|
||||
)
|
||||
return
|
||||
|
||||
|
|
|
|||
|
|
@ -8,10 +8,15 @@ import subprocess
|
|||
from typing import TextIO
|
||||
|
||||
from wk import exe
|
||||
from wk.cfg.hw import CPU_TEMPS
|
||||
from wk.cfg.hw import CPU_FAILURE_TEMP
|
||||
from wk.os.mac import set_fans as macos_set_fans
|
||||
from wk.std import PLATFORM
|
||||
from wk.ui import ansi
|
||||
from wk.std import (
|
||||
PLATFORM,
|
||||
color_string,
|
||||
print_error,
|
||||
print_warning,
|
||||
)
|
||||
from wk.tmux import respawn_pane as tmux_respawn_pane
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -20,75 +25,32 @@ SysbenchType = tuple[subprocess.Popen, TextIO]
|
|||
|
||||
|
||||
# Functions
|
||||
def check_cooling_results(sensors, test_object) -> None:
|
||||
"""Check cooling result via sensor data."""
|
||||
idle_temp = sensors.get_cpu_temp('Idle')
|
||||
cooldown_temp = sensors.get_cpu_temp('Cooldown')
|
||||
max_temp = sensors.get_cpu_temp('Max')
|
||||
test_object.report.append(ansi.color_string('Temps', 'BLUE'))
|
||||
def check_cooling_results(test_obj, sensors, run_sysbench=False) -> None:
|
||||
"""Check cooling results and update test_obj."""
|
||||
max_temp = sensors.cpu_max_temp()
|
||||
temp_labels = ['Idle', 'Max', 'Cooldown']
|
||||
if run_sysbench:
|
||||
temp_labels.append('Sysbench')
|
||||
|
||||
# Check temps
|
||||
if max_temp > CPU_TEMPS['Critical']:
|
||||
test_object.failed = True
|
||||
test_object.set_status('Failed')
|
||||
test_object.report.extend([
|
||||
ansi.color_string(
|
||||
f' WARNING: Critical CPU temp of {CPU_TEMPS["Critical"]} exceeded.',
|
||||
'RED',
|
||||
),
|
||||
'',
|
||||
])
|
||||
elif idle_temp >= CPU_TEMPS['Idle High']:
|
||||
test_object.failed = True
|
||||
test_object.set_status('Failed')
|
||||
test_object.report.extend([
|
||||
ansi.color_string(
|
||||
f' WARNING: Max idle temp of {CPU_TEMPS["Idle High"]} exceeded.',
|
||||
'YELLOW',
|
||||
),
|
||||
'',
|
||||
])
|
||||
elif (
|
||||
cooldown_temp <= CPU_TEMPS['Cooling Low Cutoff']
|
||||
or max_temp - cooldown_temp >= CPU_TEMPS['Cooling Delta']
|
||||
):
|
||||
test_object.passed = True
|
||||
test_object.set_status('Passed')
|
||||
else:
|
||||
test_object.passed = False
|
||||
test_object.set_status('Unknown')
|
||||
if cooldown_temp - idle_temp >= CPU_TEMPS['Idle Delta']:
|
||||
test_object.report.extend([
|
||||
ansi.color_string(
|
||||
f' WARNING: Cooldown temp at least {CPU_TEMPS["Idle Delta"]}° over idle.',
|
||||
'YELLOW',
|
||||
),
|
||||
'',
|
||||
])
|
||||
if not max_temp:
|
||||
test_obj.set_status('Unknown')
|
||||
elif max_temp >= CPU_FAILURE_TEMP:
|
||||
test_obj.failed = True
|
||||
test_obj.set_status('Failed')
|
||||
elif 'Aborted' not in test_obj.status:
|
||||
test_obj.passed = True
|
||||
test_obj.set_status('Passed')
|
||||
|
||||
# Build report
|
||||
report_labels = ['Idle']
|
||||
average_labels = []
|
||||
if 'Sysbench' in sensors.temp_labels:
|
||||
average_labels.append('Sysbench')
|
||||
report_labels.extend(['Sysbench', 'Cooldown'])
|
||||
if 'Prime95' in sensors.temp_labels:
|
||||
average_labels.append('Prime95')
|
||||
report_labels.append('Prime95')
|
||||
if 'Cooldown' not in report_labels:
|
||||
report_labels.append('Cooldown')
|
||||
if len(sensors.temp_labels.intersection(['Prime95', 'Sysbench'])) < 1:
|
||||
# Include overall max temp if needed
|
||||
report_labels.append('Max')
|
||||
for line in sensors.generate_report(
|
||||
*report_labels, only_cpu=True, include_avg_for=average_labels):
|
||||
test_object.report.append(f' {line}')
|
||||
# Add temps to report
|
||||
for line in sensors.generate_report(*temp_labels, only_cpu=True):
|
||||
test_obj.report.append(f' {line}')
|
||||
|
||||
|
||||
def check_mprime_results(test_obj, working_dir) -> None:
|
||||
"""Check mprime log files and update test_obj."""
|
||||
passing_lines = set()
|
||||
warning_lines = set()
|
||||
passing_lines = {}
|
||||
warning_lines = {}
|
||||
|
||||
def _read_file(log_name) -> list[str]:
|
||||
"""Read file and split into lines, returns list."""
|
||||
|
|
@ -106,9 +68,9 @@ def check_mprime_results(test_obj, working_dir) -> None:
|
|||
for line in _read_file('results.txt'):
|
||||
line = line.strip()
|
||||
if re.search(r'(error|fail)', line, re.IGNORECASE):
|
||||
warning_lines.add(line)
|
||||
warning_lines[line] = None
|
||||
|
||||
# prime.log (check if passed)
|
||||
# print.log (check if passed)
|
||||
for line in _read_file('prime.log'):
|
||||
line = line.strip()
|
||||
match = re.search(
|
||||
|
|
@ -116,10 +78,10 @@ def check_mprime_results(test_obj, working_dir) -> None:
|
|||
if match:
|
||||
if int(match.group(2)) + int(match.group(3)) > 0:
|
||||
# Errors and/or warnings encountered
|
||||
warning_lines.add(match.group(1).capitalize())
|
||||
warning_lines[match.group(1).capitalize()] = None
|
||||
else:
|
||||
# No errors/warnings
|
||||
passing_lines.add(match.group(1).capitalize())
|
||||
passing_lines[match.group(1).capitalize()] = None
|
||||
|
||||
# Update status
|
||||
if warning_lines:
|
||||
|
|
@ -135,31 +97,29 @@ def check_mprime_results(test_obj, working_dir) -> None:
|
|||
for line in passing_lines:
|
||||
test_obj.report.append(f' {line}')
|
||||
for line in warning_lines:
|
||||
test_obj.report.append(ansi.color_string(f' {line}', 'YELLOW'))
|
||||
test_obj.report.append(color_string(f' {line}', 'YELLOW'))
|
||||
if not (passing_lines or warning_lines):
|
||||
test_obj.report.append(ansi.color_string(' Unknown result', 'YELLOW'))
|
||||
test_obj.report.append(color_string(' Unknown result', 'YELLOW'))
|
||||
|
||||
|
||||
def start_mprime(working_dir, log_path) -> subprocess.Popen:
|
||||
"""Start mprime and save filtered output to log, returns Popen object."""
|
||||
set_apple_fan_speed('max')
|
||||
proc_mprime = subprocess.Popen(
|
||||
proc_mprime = subprocess.Popen( # pylint: disable=consider-using-with
|
||||
['mprime', '-t'],
|
||||
cwd=working_dir,
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.STDOUT,
|
||||
)
|
||||
proc_grep = subprocess.Popen(
|
||||
proc_grep = subprocess.Popen( # pylint: disable=consider-using-with
|
||||
'grep --ignore-case --invert-match --line-buffered stress.txt'.split(),
|
||||
stdin=proc_mprime.stdout,
|
||||
stdout=subprocess.PIPE,
|
||||
)
|
||||
proc_mprime.stdout.close() # type: ignore[reportOptionalMemberAccess]
|
||||
save_nbsr = exe.NonBlockingStreamReader(
|
||||
proc_grep.stdout, # type: ignore[reportGeneralTypeIssues]
|
||||
)
|
||||
proc_mprime.stdout.close()
|
||||
save_nsbr = exe.NonBlockingStreamReader(proc_grep.stdout)
|
||||
exe.start_thread(
|
||||
save_nbsr.save_to_file,
|
||||
save_nsbr.save_to_file,
|
||||
args=(proc_grep, log_path),
|
||||
)
|
||||
|
||||
|
|
@ -167,6 +127,38 @@ def start_mprime(working_dir, log_path) -> subprocess.Popen:
|
|||
return proc_mprime
|
||||
|
||||
|
||||
def start_sysbench(sensors, sensors_out, log_path, pane) -> SysbenchType:
|
||||
"""Start sysbench, returns tuple with Popen object and file handle."""
|
||||
set_apple_fan_speed('max')
|
||||
sysbench_cmd = [
|
||||
'sysbench',
|
||||
f'--threads={exe.psutil.cpu_count()}',
|
||||
'--cpu-max-prime=1000000000',
|
||||
'cpu',
|
||||
'run',
|
||||
]
|
||||
|
||||
# Restart background monitor for Sysbench
|
||||
sensors.stop_background_monitor()
|
||||
sensors.start_background_monitor(
|
||||
sensors_out,
|
||||
alt_max='Sysbench',
|
||||
thermal_action=('killall', 'sysbench', '-INT'),
|
||||
)
|
||||
|
||||
# Update bottom pane
|
||||
tmux_respawn_pane(pane, watch_file=log_path, watch_cmd='tail')
|
||||
|
||||
# Start sysbench
|
||||
filehandle_sysbench = open( # pylint: disable=consider-using-with
|
||||
log_path, 'a', encoding='utf-8',
|
||||
)
|
||||
proc_sysbench = exe.popen_program(sysbench_cmd, stdout=filehandle_sysbench)
|
||||
|
||||
# Done
|
||||
return (proc_sysbench, filehandle_sysbench)
|
||||
|
||||
|
||||
def set_apple_fan_speed(speed) -> None:
|
||||
"""Set Apple fan speed."""
|
||||
cmd = None
|
||||
|
|
@ -182,35 +174,14 @@ def set_apple_fan_speed(speed) -> None:
|
|||
except (RuntimeError, ValueError, subprocess.CalledProcessError) as err:
|
||||
LOG.error('Failed to set fans to %s', speed)
|
||||
LOG.error('Error: %s', err)
|
||||
#ui.print_error(f'Failed to set fans to {speed}')
|
||||
#for line in str(err).splitlines():
|
||||
# ui.print_warning(f' {line.strip()}')
|
||||
print_error(f'Failed to set fans to {speed}')
|
||||
for line in str(err).splitlines():
|
||||
print_warning(f' {line.strip()}')
|
||||
elif PLATFORM == 'Linux':
|
||||
cmd = ['apple-fans', speed]
|
||||
exe.run_program(cmd, check=False)
|
||||
|
||||
|
||||
def start_sysbench(log_path) -> SysbenchType:
|
||||
"""Start sysbench, returns tuple with Popen object and file handle."""
|
||||
set_apple_fan_speed('max')
|
||||
cmd = [
|
||||
'sysbench',
|
||||
f'--threads={exe.psutil.cpu_count()}',
|
||||
'--cpu-max-prime=1000000000',
|
||||
'cpu',
|
||||
'run',
|
||||
]
|
||||
|
||||
# Start sysbench
|
||||
filehandle = open(
|
||||
log_path, 'a', encoding='utf-8',
|
||||
)
|
||||
proc = exe.popen_program(cmd, stdout=filehandle)
|
||||
|
||||
# Done
|
||||
return (proc, filehandle)
|
||||
|
||||
|
||||
def stop_mprime(proc_mprime) -> None:
|
||||
"""Stop mprime gracefully, then forcefully as needed."""
|
||||
proc_mprime.terminate()
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load diff
|
|
@ -8,17 +8,18 @@ import plistlib
|
|||
import re
|
||||
|
||||
from dataclasses import dataclass, field
|
||||
from typing import Any
|
||||
from typing import Any, Union
|
||||
|
||||
from wk.cfg.main import KIT_NAME_SHORT
|
||||
from wk.cfg.python import DATACLASS_DECORATOR_KWARGS
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.hw.test import Test
|
||||
from wk.hw.smart import (
|
||||
enable_smart,
|
||||
generate_attribute_report,
|
||||
get_known_disk_attributes,
|
||||
update_smart_details,
|
||||
)
|
||||
from wk.std import PLATFORM
|
||||
from wk.ui import ansi
|
||||
from wk.std import PLATFORM, bytes_to_string, color_string, strip_colors
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -30,44 +31,48 @@ WK_LABEL_REGEX = re.compile(
|
|||
|
||||
|
||||
# Classes
|
||||
@dataclass(slots=True)
|
||||
@dataclass(**DATACLASS_DECORATOR_KWARGS)
|
||||
class Disk:
|
||||
# pylint: disable=too-many-instance-attributes
|
||||
"""Object for tracking disk specific data."""
|
||||
attributes: dict[Any, dict] = field(init=False, default_factory=dict)
|
||||
bus: str = field(init=False)
|
||||
children: list[dict] = field(init=False, default_factory=list)
|
||||
description: str = field(init=False)
|
||||
filesystem: str = field(init=False)
|
||||
initial_attributes: dict[Any, dict] = field(init=False, default_factory=dict)
|
||||
known_attributes: dict[Any, dict] = field(init=False, default_factory=dict)
|
||||
log_sec: int = field(init=False)
|
||||
model: str = field(init=False)
|
||||
name: str = field(init=False)
|
||||
notes: list[str] = field(init=False, default_factory=list)
|
||||
path: pathlib.Path = field(init=False)
|
||||
path_str: pathlib.Path | str
|
||||
parent: str = field(init=False)
|
||||
phy_sec: int = field(init=False)
|
||||
raw_details: dict[str, Any] = field(init=False)
|
||||
raw_smartctl: dict[str, Any] = field(init=False, default_factory=dict)
|
||||
serial: str = field(init=False)
|
||||
size: int = field(init=False)
|
||||
ssd: bool = field(init=False)
|
||||
tests: list[Test] = field(init=False, default_factory=list)
|
||||
trim: bool = field(init=False)
|
||||
attributes: dict[Any, dict] = field(init=False, default_factory=dict)
|
||||
bus: str = field(init=False)
|
||||
children: list[dict] = field(init=False, default_factory=list)
|
||||
filesystem: str = field(init=False)
|
||||
log_sec: int = field(init=False)
|
||||
model: str = field(init=False)
|
||||
name: str = field(init=False)
|
||||
notes: list[str] = field(init=False, default_factory=list)
|
||||
path: Union[pathlib.Path, str]
|
||||
parent: str = field(init=False)
|
||||
phy_sec: int = field(init=False)
|
||||
raw_details: dict[str, Any] = field(init=False)
|
||||
raw_smartctl: dict[str, Any] = field(init=False)
|
||||
serial: str = field(init=False)
|
||||
size: int = field(init=False)
|
||||
ssd: bool = field(init=False)
|
||||
tests: list[Test] = field(init=False, default_factory=list)
|
||||
use_sat: bool = field(init=False, default=False)
|
||||
|
||||
def __post_init__(self):
|
||||
self.path = pathlib.Path(self.path_str).resolve()
|
||||
def __post_init__(self) -> None:
|
||||
self.path = pathlib.Path(self.path).resolve()
|
||||
self.update_details()
|
||||
self.set_description()
|
||||
self.known_attributes = get_known_disk_attributes(self.model)
|
||||
enable_smart(self)
|
||||
update_smart_details(self)
|
||||
if not self.attributes and self.bus == 'USB':
|
||||
# Try using SAT
|
||||
LOG.warning('Using SAT for smartctl for %s', self.path)
|
||||
self.notes = []
|
||||
self.use_sat = True
|
||||
enable_smart(self)
|
||||
update_smart_details(self)
|
||||
if not self.is_4k_aligned():
|
||||
self.add_note('One or more partitions are not 4K aligned', 'YELLOW')
|
||||
|
||||
def add_note(self, note, color=None) -> None:
|
||||
"""Add note that will be included in the disk report."""
|
||||
if color:
|
||||
note = ansi.color_string(note, color)
|
||||
note = color_string(note, color)
|
||||
if note not in self.notes:
|
||||
self.notes.append(note)
|
||||
self.notes.sort()
|
||||
|
|
@ -76,10 +81,18 @@ class Disk:
|
|||
"""Check if note is already present."""
|
||||
present = False
|
||||
for note in self.notes:
|
||||
if note_str == ansi.strip_colors(note):
|
||||
if note_str == strip_colors(note):
|
||||
present = True
|
||||
return present
|
||||
|
||||
@property
|
||||
def description(self) -> str:
|
||||
"""Get disk description from details."""
|
||||
return (
|
||||
f'{bytes_to_string(self.size, use_binary=False)}'
|
||||
f' ({self.bus}) {self.model} {self.serial}'
|
||||
)
|
||||
|
||||
def disable_disk_tests(self) -> None:
|
||||
"""Disable all tests."""
|
||||
LOG.warning('Disabling all tests for: %s', self.path)
|
||||
|
|
@ -92,18 +105,18 @@ class Disk:
|
|||
"""Generate Disk report, returns list."""
|
||||
report = []
|
||||
if header:
|
||||
report.append(ansi.color_string(f'Device ({self.path.name})', 'BLUE'))
|
||||
report.append(color_string(f'Device ({self.path.name})', 'BLUE'))
|
||||
report.append(f' {self.description}')
|
||||
|
||||
# Attributes
|
||||
if self.attributes:
|
||||
if header:
|
||||
report.append(ansi.color_string('Attributes', 'BLUE'))
|
||||
report.append(color_string('Attributes', 'BLUE'))
|
||||
report.extend(generate_attribute_report(self))
|
||||
|
||||
# Notes
|
||||
if self.notes:
|
||||
report.append(ansi.color_string('Notes', 'BLUE'))
|
||||
report.append(color_string('Notes', 'BLUE'))
|
||||
for note in self.notes:
|
||||
report.append(f' {note}')
|
||||
|
||||
|
|
@ -138,50 +151,12 @@ class Disk:
|
|||
|
||||
return aligned
|
||||
|
||||
@property
|
||||
def present(self) -> bool:
|
||||
"""Verify this device is still present, returns bool."""
|
||||
if not self.path.exists():
|
||||
self.add_note('Device disconnected', 'RED')
|
||||
return False
|
||||
return True
|
||||
|
||||
def set_description(self) -> None:
|
||||
"""Set disk description from details."""
|
||||
decimals = 1
|
||||
suffix = ' '
|
||||
|
||||
# Set size_str (try binary scale first)
|
||||
for scale in (1024, 1000):
|
||||
size = float(self.size)
|
||||
units = list('KMGTPEZY')
|
||||
while units:
|
||||
if abs(size) < scale:
|
||||
break
|
||||
size /= scale
|
||||
suffix = units.pop(0)
|
||||
size = ((size * 10) // 1) / 10
|
||||
if size % 1 == 0:
|
||||
# Found an exact whole number, drop the decimal
|
||||
decimals = 0
|
||||
break
|
||||
if size % 1 == 0.5:
|
||||
break
|
||||
|
||||
# Done
|
||||
self.description = (
|
||||
f'{size:0.{decimals}f} {suffix}B ({self.bus}) {self.model} {self.serial}'
|
||||
)
|
||||
|
||||
def update_details(self, skip_children=True) -> None:
|
||||
"""Update disk details using OS specific methods.
|
||||
|
||||
Required details default to generic descriptions
|
||||
and are converted to the correct type.
|
||||
"""
|
||||
if not self.present:
|
||||
return
|
||||
|
||||
if PLATFORM == 'Darwin':
|
||||
self.raw_details = get_disk_details_macos(
|
||||
self.path, skip_children=skip_children,
|
||||
|
|
@ -205,7 +180,6 @@ class Disk:
|
|||
self.serial = self.raw_details.get('serial', 'Unknown Serial')
|
||||
self.size = self.raw_details.get('size', -1)
|
||||
self.ssd = self.raw_details.get('ssd', False)
|
||||
self.trim = self.raw_details.get('trim', False)
|
||||
|
||||
# Ensure certain attributes types
|
||||
## NOTE: This is ugly, deal.
|
||||
|
|
@ -219,10 +193,6 @@ class Disk:
|
|||
if attr == 'size':
|
||||
setattr(self, attr, -1)
|
||||
|
||||
# Add TRIM note
|
||||
if self.trim:
|
||||
self.add_note('TRIM support detected', 'YELLOW')
|
||||
|
||||
|
||||
# Functions
|
||||
def get_disk_details_linux(disk_path, skip_children=True) -> dict[Any, Any]:
|
||||
|
|
@ -246,12 +216,10 @@ def get_disk_details_linux(disk_path, skip_children=True) -> dict[Any, Any]:
|
|||
dev['bus'] = dev.pop('tran', '???')
|
||||
dev['parent'] = dev.pop('pkname', None)
|
||||
dev['ssd'] = not dev.pop('rota', True)
|
||||
dev['trim'] = bool(dev.pop('disc-max', 0))
|
||||
if 'loop' in str(disk_path) and dev['bus'] is None:
|
||||
dev['bus'] = 'Image'
|
||||
dev['model'] = ''
|
||||
dev['serial'] = ''
|
||||
dev['trim'] = False # NOTE: This check is just for physical devices
|
||||
|
||||
# Convert to dict
|
||||
details = dev_list.pop(0)
|
||||
|
|
@ -309,7 +277,6 @@ def get_disk_details_macos(disk_path, skip_children=True) -> dict:
|
|||
dev['serial'] = get_disk_serial_macos(dev['path'])
|
||||
dev['size'] = dev.pop('Size', -1)
|
||||
dev['ssd'] = dev.pop('SolidState', False)
|
||||
dev['trim'] = False # TODO: ACtually check for TRIM
|
||||
dev['vendor'] = ''
|
||||
if dev.get('WholeDisk', True):
|
||||
dev['parent'] = None
|
||||
|
|
@ -344,8 +311,6 @@ def get_disks(skip_kits=False) -> list[Disk]:
|
|||
|
||||
# Skip WK disks
|
||||
if skip_kits:
|
||||
for disk in disks:
|
||||
disk.update_details(skip_children=False)
|
||||
disks = [
|
||||
disk_obj for disk_obj in disks
|
||||
if not any(
|
||||
|
|
@ -371,10 +336,6 @@ def get_disks_linux() -> list[Disk]:
|
|||
if disk_obj.raw_details.get('type', '???') != 'disk':
|
||||
continue
|
||||
|
||||
# Skip empty devices (usually card readers)
|
||||
if disk_obj.size <= 0:
|
||||
continue
|
||||
|
||||
# Add disk
|
||||
disks.append(disk_obj)
|
||||
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@
|
|||
import logging
|
||||
|
||||
from wk.exe import run_program
|
||||
from wk.std import PLATFORM
|
||||
from wk.std import PLATFORM, print_warning
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -17,8 +17,7 @@ def keyboard_test() -> None:
|
|||
if PLATFORM == 'Linux':
|
||||
run_xev()
|
||||
else:
|
||||
LOG.error('Not supported under this OS: %s', PLATFORM)
|
||||
raise NotImplementedError(f'Not supported under this OS: {PLATFORM}')
|
||||
print_warning(f'Not supported under this OS: {PLATFORM}')
|
||||
|
||||
|
||||
def run_xev() -> None:
|
||||
|
|
|
|||
|
|
@ -9,7 +9,11 @@ from wk.net import (
|
|||
show_valid_addresses,
|
||||
speedtest,
|
||||
)
|
||||
from wk.ui import cli as ui
|
||||
from wk.std import (
|
||||
TryAndPrint,
|
||||
pause,
|
||||
print_warning,
|
||||
)
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -20,7 +24,7 @@ LOG = logging.getLogger(__name__)
|
|||
def network_test() -> None:
|
||||
"""Run network tests."""
|
||||
LOG.info('Network Test')
|
||||
try_and_print = ui.TryAndPrint()
|
||||
try_and_print = TryAndPrint()
|
||||
result = try_and_print.run(
|
||||
message='Network connection...',
|
||||
function=connected_to_private_network,
|
||||
|
|
@ -30,8 +34,8 @@ def network_test() -> None:
|
|||
|
||||
# Bail if not connected
|
||||
if result['Failed']:
|
||||
ui.print_warning('Please connect to a network and try again')
|
||||
ui.pause('Press Enter to return to main menu...')
|
||||
print_warning('Please connect to a network and try again')
|
||||
pause('Press Enter to return to main menu...')
|
||||
return
|
||||
|
||||
# Show IP address(es)
|
||||
|
|
@ -47,7 +51,7 @@ def network_test() -> None:
|
|||
try_and_print.run('Speedtest...', speedtest)
|
||||
|
||||
# Done
|
||||
ui.pause('Press Enter to return to main menu...')
|
||||
pause('Press Enter to return to main menu...')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -6,7 +6,7 @@ import logging
|
|||
from subprocess import PIPE
|
||||
|
||||
from wk.exe import run_program
|
||||
from wk.ui import tmux
|
||||
from wk.tmux import zoom_pane as tmux_zoom_pane
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -31,9 +31,9 @@ def screensaver(name) -> None:
|
|||
]
|
||||
|
||||
# Switch pane to fullscreen and start screensaver
|
||||
tmux.zoom_pane()
|
||||
tmux_zoom_pane()
|
||||
run_program(cmd, check=False, pipe=False, stderr=PIPE)
|
||||
tmux.zoom_pane()
|
||||
tmux_zoom_pane()
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -6,16 +6,13 @@ import logging
|
|||
import pathlib
|
||||
import re
|
||||
|
||||
from copy import deepcopy
|
||||
from subprocess import CalledProcessError
|
||||
from threading import Thread
|
||||
from typing import Any
|
||||
|
||||
from wk.cfg.hw import CPU_TEMPS, SMC_IDS, TEMP_COLORS
|
||||
from wk.cfg.hw import CPU_CRITICAL_TEMP, SMC_IDS, TEMP_COLORS
|
||||
from wk.exe import run_program, start_thread
|
||||
from wk.io import non_clobber_path
|
||||
from wk.std import PLATFORM, sleep
|
||||
from wk.ui import ansi
|
||||
from wk.std import PLATFORM, color_string, sleep
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -37,83 +34,39 @@ class ThermalLimitReachedError(RuntimeError):
|
|||
|
||||
# Classes
|
||||
class Sensors():
|
||||
"""Class for holding sensor specific data.
|
||||
|
||||
# Sensor data structure
|
||||
#
|
||||
# Section # CPUTemps / Other
|
||||
# Adapters # coretemp / acpi / nvme / etc
|
||||
# Sources # Core 1 / SODIMM / Sensor X / etc
|
||||
# Label # temp1_input / etc (i.e. lm_sensor label)
|
||||
# Max # 99.0
|
||||
# X # 55.0 (where X is Idle/Current/Sysbench/etc)
|
||||
# Temps # [39.0, 38.0, 40.0, 39.0, 38.0, ...]
|
||||
#
|
||||
# e.g.
|
||||
# { 'CPUTemps': { 'coretemp-isa-0000': { 'Core 0': { 'Average': 44.5,
|
||||
# 'Current': 44.0,
|
||||
# 'Idle': 44.5,
|
||||
# 'Label': 'temp2_input',
|
||||
# 'Max': 45.0,
|
||||
# 'Temps': [ 45.0,
|
||||
# 45.0,
|
||||
# ...,
|
||||
# 42.0]}}}}
|
||||
#
|
||||
# Sensor history data structure
|
||||
# [ ('Name of "run"', sensor_data_structure_described_above), ]
|
||||
#
|
||||
# e.g.
|
||||
# [
|
||||
# ( 'Idle',
|
||||
# { 'CPUTemps': { 'coretemp-isa-0000': { 'Core 0': { 'Max': 45.0, ..., }}}}
|
||||
# ),
|
||||
# ( 'Sysbench',
|
||||
# { 'CPUTemps': { 'coretemp-isa-0000': { 'Core 0': { 'Max': 85.0, ..., }}}}
|
||||
# ),
|
||||
# ]
|
||||
"""
|
||||
"""Class for holding sensor specific data."""
|
||||
def __init__(self):
|
||||
self.background_thread: Thread | None = None
|
||||
self.data: dict[Any, Any] = get_sensor_data()
|
||||
self.history: list[tuple[str, dict]] = []
|
||||
self.history_index: dict[str, int] = {}
|
||||
self.history_next_label: str = 'Idle'
|
||||
self.out_path: pathlib.Path | str | None = None
|
||||
self.temp_labels: set = set(['Current', 'Max'])
|
||||
self.background_thread = None
|
||||
self.data = get_sensor_data()
|
||||
self.out_path = None
|
||||
|
||||
def clear_temps(self, next_label: str, save_history: bool = True) -> None:
|
||||
def clear_temps(self) -> None:
|
||||
"""Clear saved temps but keep structure"""
|
||||
prev_label = self.history_next_label
|
||||
self.history_next_label = next_label
|
||||
|
||||
# Save history
|
||||
if save_history:
|
||||
cur_data = deepcopy(self.data)
|
||||
|
||||
# Calculate averages
|
||||
for adapters in cur_data.values():
|
||||
for sources in adapters.values():
|
||||
for name in sources:
|
||||
temp_list = sources[name]['Temps']
|
||||
try:
|
||||
sources[name]['Average'] = sum(temp_list) / len(temp_list)
|
||||
except ZeroDivisionError:
|
||||
LOG.error('Failed to calculate averate temp for %s', name)
|
||||
sources[name]['Average'] = 0
|
||||
|
||||
# Add to history
|
||||
self.history.append((prev_label, cur_data))
|
||||
self.history_index[prev_label] = len(self.history) - 1
|
||||
|
||||
# Clear data
|
||||
for adapters in self.data.values():
|
||||
for sources in adapters.values():
|
||||
for source_data in sources.values():
|
||||
source_data['Temps'] = []
|
||||
|
||||
def cpu_max_temp(self) -> float:
|
||||
"""Get max temp from any CPU source, returns float.
|
||||
|
||||
NOTE: If no temps are found this returns zero.
|
||||
"""
|
||||
max_temp = 0.0
|
||||
|
||||
# Check all CPU Temps
|
||||
for section, adapters in self.data.items():
|
||||
if not section.startswith('CPU'):
|
||||
continue
|
||||
for sources in adapters.values():
|
||||
for source_data in sources.values():
|
||||
max_temp = max(max_temp, source_data.get('Max', 0))
|
||||
|
||||
# Done
|
||||
return max_temp
|
||||
|
||||
def cpu_reached_critical_temp(self) -> bool:
|
||||
"""Check if CPU exceeded critical temp, returns bool."""
|
||||
"""Check if CPU reached CPU_CRITICAL_TEMP, returns bool."""
|
||||
for section, adapters in self.data.items():
|
||||
if not section.startswith('CPU'):
|
||||
# Limit to CPU temps
|
||||
|
|
@ -122,22 +75,16 @@ class Sensors():
|
|||
# Ugly section
|
||||
for sources in adapters.values():
|
||||
for source_data in sources.values():
|
||||
if source_data.get('Max', -1) > CPU_TEMPS['Critical']:
|
||||
if source_data.get('Max', -1) >= CPU_CRITICAL_TEMP:
|
||||
return True
|
||||
|
||||
# Didn't return above so temps are within the threshold
|
||||
return False
|
||||
|
||||
def generate_report(
|
||||
self,
|
||||
*temp_labels: str,
|
||||
colored: bool = True,
|
||||
only_cpu: bool = False,
|
||||
include_avg_for: list[str] | None = None,
|
||||
) -> list[str]:
|
||||
self, *temp_labels, colored=True, only_cpu=False) -> list[str]:
|
||||
"""Generate report based on given temp_labels, returns list."""
|
||||
report = []
|
||||
include_avg_for = include_avg_for if include_avg_for else []
|
||||
|
||||
for section, adapters in sorted(self.data.items()):
|
||||
if only_cpu and not section.startswith('CPU'):
|
||||
|
|
@ -151,10 +98,6 @@ class Sensors():
|
|||
for label in temp_labels:
|
||||
if label != 'Current':
|
||||
line += f' {label.lower()}: '
|
||||
if label in include_avg_for:
|
||||
avg_temp = self.get_avg_temp(
|
||||
label, section, adapter, source, colored)
|
||||
line += f'{avg_temp} / '
|
||||
line += get_temp_str(
|
||||
source_data.get(label, '???'),
|
||||
colored=colored,
|
||||
|
|
@ -166,7 +109,7 @@ class Sensors():
|
|||
# Handle empty reports
|
||||
if not report:
|
||||
report = [
|
||||
ansi.color_string('WARNING: No sensors found', 'YELLOW'),
|
||||
color_string('WARNING: No sensors found', 'YELLOW'),
|
||||
'',
|
||||
'Please monitor temps manually',
|
||||
]
|
||||
|
|
@ -174,36 +117,11 @@ class Sensors():
|
|||
# Done
|
||||
return report
|
||||
|
||||
def get_avg_temp(self, label, section, adapter, source, colored) -> str:
|
||||
"""Get average temp from history, return str."""
|
||||
# NOTE: This is Super-ugly
|
||||
label_index = self.history_index[label]
|
||||
avg_temp = self.history[label_index][1][section][adapter][source]['Average']
|
||||
return get_temp_str(avg_temp, colored=colored)
|
||||
|
||||
def get_cpu_temp(self, label) -> float:
|
||||
"""Get temp for label from any CPU source, returns float.
|
||||
|
||||
NOTE: This returns the highest value for the label.
|
||||
NOTE 2: If no temps are found this returns zero.
|
||||
"""
|
||||
max_temp = 0.0
|
||||
|
||||
# Check all CPU Temps
|
||||
for section, adapters in self.data.items():
|
||||
if not section.startswith('CPU'):
|
||||
continue
|
||||
for sources in adapters.values():
|
||||
for source_data in sources.values():
|
||||
max_temp = max(max_temp, source_data.get(label, 0))
|
||||
|
||||
# Done
|
||||
return float(max_temp)
|
||||
|
||||
def monitor_to_file(
|
||||
self, out_path, alt_max=None,
|
||||
exit_on_thermal_limit=True, temp_labels=None,
|
||||
thermal_action=None) -> None:
|
||||
# pylint: disable=too-many-arguments
|
||||
"""Write report to path every second until stopped.
|
||||
|
||||
thermal_action is a cmd to run if ThermalLimitReachedError is caught.
|
||||
|
|
@ -217,7 +135,6 @@ class Sensors():
|
|||
temp_labels = ['Current', 'Max']
|
||||
if alt_max:
|
||||
temp_labels.append(alt_max)
|
||||
self.temp_labels.add(alt_max)
|
||||
|
||||
# Start loop
|
||||
while True:
|
||||
|
|
@ -237,15 +154,9 @@ class Sensors():
|
|||
# Sleep before next loop
|
||||
sleep(0.5)
|
||||
|
||||
def save_average_temps(
|
||||
self,
|
||||
temp_label: str,
|
||||
seconds: int = 10,
|
||||
save_history: bool = True,
|
||||
) -> None:
|
||||
def save_average_temps(self, temp_label, seconds=10) -> None:
|
||||
"""Save average temps under temp_label over provided seconds.."""
|
||||
self.clear_temps(next_label=temp_label, save_history=save_history)
|
||||
self.temp_labels.add(temp_label)
|
||||
self.clear_temps()
|
||||
|
||||
# Get temps
|
||||
for _ in range(seconds):
|
||||
|
|
@ -271,6 +182,7 @@ class Sensors():
|
|||
self, out_path, alt_max=None,
|
||||
exit_on_thermal_limit=True, temp_labels=None,
|
||||
thermal_action=None) -> None:
|
||||
# pylint: disable=too-many-arguments
|
||||
"""Start background thread to save report to file.
|
||||
|
||||
thermal_action is a cmd to run if ThermalLimitReachedError is caught.
|
||||
|
|
@ -288,10 +200,6 @@ class Sensors():
|
|||
|
||||
def stop_background_monitor(self) -> None:
|
||||
"""Stop background thread."""
|
||||
# Bail early
|
||||
if self.background_thread is None:
|
||||
return
|
||||
|
||||
self.out_path.with_suffix('.stop').touch()
|
||||
self.background_thread.join()
|
||||
|
||||
|
|
@ -302,8 +210,6 @@ class Sensors():
|
|||
def update_sensor_data(
|
||||
self, alt_max=None, exit_on_thermal_limit=True) -> None:
|
||||
"""Update sensor data via OS-specific means."""
|
||||
if alt_max:
|
||||
self.temp_labels.add(alt_max)
|
||||
if PLATFORM == 'Darwin':
|
||||
self.update_sensor_data_macos(alt_max, exit_on_thermal_limit)
|
||||
elif PLATFORM == 'Linux':
|
||||
|
|
@ -330,7 +236,7 @@ class Sensors():
|
|||
|
||||
# Raise exception if thermal limit reached
|
||||
if exit_on_thermal_limit and section == 'CPUTemps':
|
||||
if source_data['Current'] > CPU_TEMPS['Critical']:
|
||||
if source_data['Current'] >= CPU_CRITICAL_TEMP:
|
||||
raise ThermalLimitReachedError('CPU temps reached limit')
|
||||
|
||||
def update_sensor_data_macos(
|
||||
|
|
@ -357,7 +263,7 @@ class Sensors():
|
|||
|
||||
# Raise exception if thermal limit reached
|
||||
if exit_on_thermal_limit and section == 'CPUTemps':
|
||||
if source_data['Current'] > CPU_TEMPS['Critical']:
|
||||
if source_data['Current'] >= CPU_CRITICAL_TEMP:
|
||||
raise ThermalLimitReachedError('CPU temps reached limit')
|
||||
|
||||
|
||||
|
|
@ -409,11 +315,7 @@ def get_sensor_data_linux() -> dict[Any, Any]:
|
|||
## current temp is labeled xxxx_input
|
||||
for source, labels in sources.items():
|
||||
for label, temp in labels.items():
|
||||
if (
|
||||
label.startswith('fan')
|
||||
or label.startswith('in')
|
||||
or label.startswith('curr')
|
||||
):
|
||||
if label.startswith('fan') or label.startswith('in') or label.startswith('curr'):
|
||||
# Skip fan RPMs and voltages
|
||||
continue
|
||||
if 'input' in label:
|
||||
|
|
@ -514,14 +416,14 @@ def get_sensor_data_macos() -> dict[Any, Any]:
|
|||
|
||||
def get_temp_str(temp, colored=True) -> str:
|
||||
"""Get colored string based on temp, returns str."""
|
||||
temp_color = ''
|
||||
temp_color = None
|
||||
|
||||
# Safety check
|
||||
try:
|
||||
temp = float(temp)
|
||||
except (TypeError, ValueError):
|
||||
# Invalid temp?
|
||||
return ansi.color_string(temp, 'PURPLE')
|
||||
return color_string(temp, 'PURPLE')
|
||||
|
||||
# Determine color
|
||||
if colored:
|
||||
|
|
@ -531,7 +433,7 @@ def get_temp_str(temp, colored=True) -> str:
|
|||
break
|
||||
|
||||
# Done
|
||||
return ansi.color_string(f'{"-" if temp < 0 else ""}{temp:2.0f}°C', temp_color)
|
||||
return color_string(f'{"-" if temp < 0 else ""}{temp:2.0f}°C', temp_color)
|
||||
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -1,7 +1,6 @@
|
|||
"""WizardKit: SMART test functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import copy
|
||||
import logging
|
||||
import re
|
||||
|
||||
|
|
@ -18,8 +17,7 @@ from wk.cfg.hw import (
|
|||
SMART_SELF_TEST_START_TIMEOUT_IN_SECONDS,
|
||||
)
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.std import bytes_to_string, sleep
|
||||
from wk.ui import ansi
|
||||
from wk.std import bytes_to_string, color_string, sleep
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -41,25 +39,26 @@ def build_self_test_report(test_obj, aborted=False) -> None:
|
|||
For instance if the test was aborted the report should include the
|
||||
last known progress instead of just "was aborted by host."
|
||||
"""
|
||||
report = [ansi.color_string('Self-Test', 'BLUE')]
|
||||
test_result = get_smart_self_test_last_result(test_obj.dev)
|
||||
report = [color_string('Self-Test', 'BLUE')]
|
||||
test_details = get_smart_self_test_details(test_obj.dev)
|
||||
test_result = test_details.get('status', {}).get('string', 'Unknown')
|
||||
|
||||
# Build report
|
||||
if test_obj.disabled or test_obj.status == 'Denied':
|
||||
report.append(ansi.color_string(f' {test_obj.status}', 'RED'))
|
||||
report.append(color_string(f' {test_obj.status}', 'RED'))
|
||||
elif test_obj.status == 'N/A' or not test_obj.dev.attributes:
|
||||
report.append(ansi.color_string(f' {test_obj.status}', 'YELLOW'))
|
||||
report.append(color_string(f' {test_obj.status}', 'YELLOW'))
|
||||
elif test_obj.status == 'TestInProgress':
|
||||
report.append(color_string(' Failed to stop previous test', 'RED'))
|
||||
test_obj.set_status('Failed')
|
||||
else:
|
||||
# Other cases include self-test result string
|
||||
if test_obj.status == 'TestInProgress':
|
||||
report.append(ansi.color_string(' Failed to stop previous test', 'RED'))
|
||||
test_obj.set_status('Failed')
|
||||
elif test_obj.status == 'TimedOut':
|
||||
report.append(ansi.color_string(' TimedOut', 'YELLOW'))
|
||||
elif aborted and not (test_obj.passed or test_obj.failed):
|
||||
report.append(ansi.color_string(' Aborted', 'YELLOW'))
|
||||
report.append(f' {test_result.capitalize()}')
|
||||
if aborted and not (test_obj.passed or test_obj.failed):
|
||||
report.append(color_string(' Aborted', 'YELLOW'))
|
||||
test_obj.set_status('Aborted')
|
||||
report.append(f' {test_result}')
|
||||
elif test_obj.status == 'TimedOut':
|
||||
report.append(color_string(' TimedOut', 'YELLOW'))
|
||||
|
||||
# Done
|
||||
test_obj.report.extend(report)
|
||||
|
|
@ -68,15 +67,16 @@ def build_self_test_report(test_obj, aborted=False) -> None:
|
|||
def check_attributes(dev, only_blocking=False) -> bool:
|
||||
"""Check if any known attributes are failing, returns bool."""
|
||||
attributes_ok = True
|
||||
known_attributes = get_known_disk_attributes(dev.model)
|
||||
for attr, value in dev.attributes.items():
|
||||
# Skip unknown attributes
|
||||
if attr not in dev.known_attributes:
|
||||
if attr not in known_attributes:
|
||||
continue
|
||||
|
||||
# Get thresholds
|
||||
blocking_attribute = dev.known_attributes[attr].get('Blocking', False)
|
||||
err_thresh = dev.known_attributes[attr].get('Error', None)
|
||||
max_thresh = dev.known_attributes[attr].get('Maximum', None)
|
||||
blocking_attribute = known_attributes[attr].get('Blocking', False)
|
||||
err_thresh = known_attributes[attr].get('Error', None)
|
||||
max_thresh = known_attributes[attr].get('Maximum', None)
|
||||
if not max_thresh:
|
||||
max_thresh = float('inf')
|
||||
|
||||
|
|
@ -89,7 +89,7 @@ def check_attributes(dev, only_blocking=False) -> bool:
|
|||
continue
|
||||
|
||||
# Check attribute
|
||||
if dev.known_attributes[attr].get('PercentageLife', False):
|
||||
if known_attributes[attr].get('PercentageLife', False):
|
||||
if 0 <= value['raw'] <= err_thresh:
|
||||
attributes_ok = False
|
||||
elif err_thresh <= value['raw'] < max_thresh:
|
||||
|
|
@ -104,7 +104,7 @@ def enable_smart(dev) -> None:
|
|||
cmd = [
|
||||
'sudo',
|
||||
'smartctl',
|
||||
'--device=auto',
|
||||
f'--device={"sat,auto" if dev.use_sat else "auto"}',
|
||||
'--tolerance=permissive',
|
||||
'--smart=on',
|
||||
dev.path,
|
||||
|
|
@ -112,14 +112,21 @@ def enable_smart(dev) -> None:
|
|||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def generate_attribute_report(dev, only_failed=False) -> list[str]:
|
||||
def generate_attribute_report(dev) -> list[str]:
|
||||
"""Generate attribute report, returns list."""
|
||||
known_attributes = get_known_disk_attributes(dev.model)
|
||||
report = []
|
||||
for attr, value in sorted(dev.attributes.items()):
|
||||
note = ''
|
||||
value_color = 'GREEN'
|
||||
|
||||
# Skip attributes not in our list
|
||||
if attr not in dev.known_attributes:
|
||||
if attr not in known_attributes:
|
||||
continue
|
||||
|
||||
# Check for attribute note
|
||||
note = known_attributes[attr].get('Note', '')
|
||||
|
||||
# ID / Name
|
||||
label = f'{attr:>3}'
|
||||
if isinstance(attr, int):
|
||||
|
|
@ -127,17 +134,32 @@ def generate_attribute_report(dev, only_failed=False) -> list[str]:
|
|||
label += f' / {str(hex(attr))[2:].upper():0>2}: {value["name"]}'
|
||||
label = f' {label.replace("_", " "):38}'
|
||||
|
||||
# Color & Note
|
||||
value_color, note = get_attribute_value_color_and_note(dev, attr, value)
|
||||
# Value color
|
||||
if known_attributes[attr].get('PercentageLife', False):
|
||||
# PercentageLife values
|
||||
if 0 <= value['raw'] <= known_attributes[attr]['Error']:
|
||||
value_color = 'RED'
|
||||
note = '(failed, % life remaining)'
|
||||
elif value['raw'] < 0 or value['raw'] > 100:
|
||||
value_color = 'PURPLE'
|
||||
note = '(invalid?)'
|
||||
else:
|
||||
for threshold, color in ATTRIBUTE_COLORS:
|
||||
threshold_val = known_attributes[attr].get(threshold, None)
|
||||
if threshold_val and value['raw'] >= threshold_val:
|
||||
value_color = color
|
||||
if threshold == 'Error':
|
||||
note = '(failed)'
|
||||
elif threshold == 'Maximum':
|
||||
note = '(invalid?)'
|
||||
|
||||
# Skip non-failing attributes if requested
|
||||
## NOTE: This is a naive test and will include 'invalid' attributes
|
||||
if only_failed and not note:
|
||||
continue
|
||||
# 199/C7 warning
|
||||
if str(attr) == '199' and value['raw'] > 0:
|
||||
note = '(bad cable?)'
|
||||
|
||||
# Build colored string and append to report
|
||||
line = ansi.color_string(
|
||||
[label, get_attribute_value_string(dev, attr), note],
|
||||
line = color_string(
|
||||
[label, value['raw_str'], note],
|
||||
[None, value_color, 'YELLOW'],
|
||||
)
|
||||
report.append(line)
|
||||
|
|
@ -146,79 +168,24 @@ def generate_attribute_report(dev, only_failed=False) -> list[str]:
|
|||
return report
|
||||
|
||||
|
||||
def get_attribute_value_color_and_note(dev, attr, value) -> tuple[str, str]:
|
||||
"""Get attribute color and note based on SMART data."""
|
||||
value_color = 'GREEN'
|
||||
note = dev.known_attributes[attr].get('Note', '')
|
||||
|
||||
# Value value_color
|
||||
if dev.known_attributes[attr].get('PercentageLife', False):
|
||||
# PercentageLife values
|
||||
if 0 <= value['raw'] <= dev.known_attributes[attr]['Error']:
|
||||
value_color = 'RED'
|
||||
note = '(failed, % life remaining)'
|
||||
elif value['raw'] < 0 or value['raw'] > 100:
|
||||
value_color = 'PURPLE'
|
||||
note = '(invalid?)'
|
||||
else:
|
||||
for threshold, color in ATTRIBUTE_COLORS:
|
||||
threshold_val = dev.known_attributes[attr].get(threshold, None)
|
||||
if threshold_val and value['raw'] >= threshold_val:
|
||||
value_color = color
|
||||
if threshold == 'Error':
|
||||
note = '(failed)'
|
||||
elif threshold == 'Maximum':
|
||||
note = '(invalid?)'
|
||||
|
||||
# 199/C7 warning
|
||||
if str(attr) == '199' and value['raw'] > 0:
|
||||
note = '(bad cable?)'
|
||||
|
||||
# Done
|
||||
return (value_color, note)
|
||||
|
||||
|
||||
def get_attribute_value_string(dev, attr) -> str:
|
||||
"""Get attribute value string and report if it has changed."""
|
||||
current_value = dev.attributes.get(attr, {})
|
||||
initial_value = dev.initial_attributes.get(attr, {})
|
||||
value_str = current_value.get('raw_str', '')
|
||||
|
||||
# Compare current value against initial value
|
||||
if (
|
||||
current_value.get('raw', None) is None
|
||||
or initial_value.get('raw', None) is None
|
||||
):
|
||||
return value_str
|
||||
if current_value['raw'] != initial_value['raw']:
|
||||
value_str = (
|
||||
f'{initial_value.get("raw_str", "?")} --> '
|
||||
f'{current_value.get("raw_str", "?")}'
|
||||
)
|
||||
|
||||
# Done
|
||||
return value_str
|
||||
|
||||
|
||||
def get_known_disk_attributes(model) -> dict[str | int, dict[str, Any]]:
|
||||
"""Get known disk attributes based on the device model."""
|
||||
known_attributes = copy.deepcopy(KNOWN_DISK_ATTRIBUTES)
|
||||
def get_known_disk_attributes(model) -> dict[Any, dict]:
|
||||
"""Get known NVMe/SMART attributes (model specific), returns dict."""
|
||||
known_attributes = KNOWN_DISK_ATTRIBUTES.copy()
|
||||
|
||||
# Apply model-specific data
|
||||
for regex, data in KNOWN_DISK_MODELS.items():
|
||||
if not re.search(regex, model):
|
||||
continue
|
||||
for attr, thresholds in data.items():
|
||||
if attr in known_attributes:
|
||||
known_attributes[attr].update(thresholds)
|
||||
else:
|
||||
known_attributes[attr] = copy.deepcopy(thresholds)
|
||||
if re.search(regex, model):
|
||||
for attr, thresholds in data.items():
|
||||
if attr in known_attributes:
|
||||
known_attributes[attr].update(thresholds)
|
||||
else:
|
||||
known_attributes[attr] = thresholds
|
||||
|
||||
# Done
|
||||
return known_attributes
|
||||
|
||||
|
||||
def get_smart_self_test_details(dev) -> dict[str, Any]:
|
||||
def get_smart_self_test_details(dev) -> dict[Any, Any]:
|
||||
"""Shorthand to get deeply nested self-test details, returns dict."""
|
||||
details = {}
|
||||
try:
|
||||
|
|
@ -231,33 +198,6 @@ def get_smart_self_test_details(dev) -> dict[str, Any]:
|
|||
return details
|
||||
|
||||
|
||||
def get_smart_self_test_last_result(dev) -> str:
|
||||
"""Get last SMART self-test result, returns str."""
|
||||
result = 'Unknown'
|
||||
|
||||
# Parse SMART data
|
||||
data = dev.raw_smartctl.get(
|
||||
'ata_smart_self_test_log', {}).get(
|
||||
'standard', {}).get(
|
||||
'table', [])
|
||||
try:
|
||||
data = data[0]
|
||||
except IndexError:
|
||||
# No results found
|
||||
return result
|
||||
|
||||
# Build result string
|
||||
result = (
|
||||
f'Power-on hours: {data.get("lifetime_hours", "?")}'
|
||||
f', Type: {data.get("type", {}).get("string", "?")}'
|
||||
f', Passed: {data.get("status", {}).get("passed", "?")}'
|
||||
f', Result: {data.get("status", {}).get("string", "?")}'
|
||||
)
|
||||
|
||||
# Done
|
||||
return result
|
||||
|
||||
|
||||
def monitor_smart_self_test(test_obj, header_str, log_path) -> bool:
|
||||
"""Monitor SMART self-test status and update test_obj, returns bool."""
|
||||
started = False
|
||||
|
|
@ -289,9 +229,6 @@ def monitor_smart_self_test(test_obj, header_str, log_path) -> bool:
|
|||
if _i * 5 >= SMART_SELF_TEST_START_TIMEOUT_IN_SECONDS:
|
||||
# Test didn't start within limit, stop waiting
|
||||
abort_self_test(test_obj.dev)
|
||||
result = get_smart_self_test_last_result(test_obj.dev)
|
||||
if result == 'Unknown':
|
||||
result = 'SMART self-test failed to start'
|
||||
test_obj.failed = True
|
||||
test_obj.set_status('TimedOut')
|
||||
break
|
||||
|
|
@ -307,11 +244,6 @@ def monitor_smart_self_test(test_obj, header_str, log_path) -> bool:
|
|||
finished = True
|
||||
break
|
||||
|
||||
# Check if timed out
|
||||
if started and not finished:
|
||||
test_obj.failed = True
|
||||
test_obj.set_status('TimedOut')
|
||||
|
||||
# Done
|
||||
return finished
|
||||
|
||||
|
|
@ -325,15 +257,15 @@ def run_self_test(test_obj, log_path) -> None:
|
|||
run_smart_self_test(test_obj, log_path)
|
||||
|
||||
|
||||
def run_smart_self_test(test_obj, log_path) -> None:
|
||||
"""Run SMART self-test and check if it passed, returns None.
|
||||
def run_smart_self_test(test_obj, log_path) -> bool:
|
||||
"""Run SMART self-test and check if it passed, returns bool.
|
||||
|
||||
NOTE: An exception will be raised if the disk lacks SMART support.
|
||||
"""
|
||||
finished = False
|
||||
test_details = get_smart_self_test_details(test_obj.dev)
|
||||
size_str = bytes_to_string(test_obj.dev.size, use_binary=False)
|
||||
header_str = ansi.color_string(
|
||||
header_str = color_string(
|
||||
['[', test_obj.dev.path.name, ' ', size_str, ']'],
|
||||
[None, 'BLUE', None, 'CYAN', None],
|
||||
sep='',
|
||||
|
|
@ -383,15 +315,11 @@ def run_smart_self_test(test_obj, log_path) -> None:
|
|||
|
||||
# Check result
|
||||
if finished:
|
||||
test_details = get_smart_self_test_details(test_obj.dev)
|
||||
test_obj.passed = test_details.get('status', {}).get('passed', False)
|
||||
test_obj.failed = test_obj.failed or not test_obj.passed
|
||||
|
||||
# Set status
|
||||
if test_obj.status == 'TimedOut':
|
||||
# Preserve TimedOut status
|
||||
pass
|
||||
elif test_obj.failed:
|
||||
if test_obj.failed and test_obj.status != 'TimedOut':
|
||||
test_obj.set_status('Failed')
|
||||
elif test_obj.passed:
|
||||
test_obj.set_status('Passed')
|
||||
|
|
@ -452,16 +380,11 @@ def update_smart_details(dev) -> None:
|
|||
"""Update SMART details via smartctl."""
|
||||
updated_attributes = {}
|
||||
|
||||
# Bail if device was disconnected
|
||||
if not dev.present:
|
||||
dev.add_note('Device disconnected', 'RED')
|
||||
return
|
||||
|
||||
# Get SMART data
|
||||
cmd = [
|
||||
'sudo',
|
||||
'smartctl',
|
||||
'--device=auto',
|
||||
f'--device={"sat,auto" if dev.use_sat else "auto"}',
|
||||
'--tolerance=verypermissive',
|
||||
'--all',
|
||||
'--json',
|
||||
|
|
@ -506,10 +429,6 @@ def update_smart_details(dev) -> None:
|
|||
if not updated_attributes:
|
||||
dev.add_note('No NVMe or SMART data available', 'YELLOW')
|
||||
|
||||
# Update iniital_attributes if needed
|
||||
if not dev.initial_attributes:
|
||||
dev.initial_attributes = copy.deepcopy(updated_attributes)
|
||||
|
||||
# Done
|
||||
dev.attributes.update(updated_attributes)
|
||||
|
||||
|
|
|
|||
|
|
@ -6,16 +6,18 @@ import logging
|
|||
from subprocess import STDOUT
|
||||
|
||||
from wk.cfg.hw import (
|
||||
BADBLOCKS_EXTRA_LARGE_DISK,
|
||||
BADBLOCKS_LARGE_DISK,
|
||||
BADBLOCKS_REGEX,
|
||||
BADBLOCKS_RESULTS_REGEX,
|
||||
BADBLOCKS_SKIP_REGEX,
|
||||
TEST_MODE_BADBLOCKS_LIMIT,
|
||||
)
|
||||
from wk.exe import run_program
|
||||
from wk.std import PLATFORM, bytes_to_string
|
||||
from wk.ui import ansi
|
||||
from wk.std import (
|
||||
PLATFORM,
|
||||
bytes_to_string,
|
||||
color_string,
|
||||
strip_colors,
|
||||
)
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -27,15 +29,10 @@ def check_surface_scan_results(test_obj, log_path) -> None:
|
|||
"""Check results and set test status."""
|
||||
with open(log_path, 'r', encoding='utf-8') as _f:
|
||||
for line in _f.readlines():
|
||||
line = ansi.strip_colors(line.strip())
|
||||
line = strip_colors(line.strip())
|
||||
if not line or BADBLOCKS_SKIP_REGEX.match(line):
|
||||
# Skip
|
||||
continue
|
||||
|
||||
# Clean line by removing backspaces/etc
|
||||
line = BADBLOCKS_RESULTS_REGEX.sub(r'\1 \2', line)
|
||||
|
||||
# Add to report
|
||||
match = BADBLOCKS_REGEX.search(line)
|
||||
if match:
|
||||
if all(s == '0' for s in match.groups()):
|
||||
|
|
@ -44,10 +41,10 @@ def check_surface_scan_results(test_obj, log_path) -> None:
|
|||
test_obj.set_status('Passed')
|
||||
else:
|
||||
test_obj.failed = True
|
||||
test_obj.report.append(f' {ansi.color_string(line, "YELLOW")}')
|
||||
test_obj.report.append(f' {color_string(line, "YELLOW")}')
|
||||
test_obj.set_status('Failed')
|
||||
else:
|
||||
test_obj.report.append(f' {ansi.color_string(line, "YELLOW")}')
|
||||
test_obj.report.append(f' {color_string(line, "YELLOW")}')
|
||||
if not (test_obj.passed or test_obj.failed):
|
||||
test_obj.set_status('Unknown')
|
||||
|
||||
|
|
@ -61,12 +58,10 @@ def run_scan(test_obj, log_path, test_mode=False) -> None:
|
|||
# Use "RAW" disks under macOS
|
||||
dev_path = dev_path.with_name(f'r{dev_path.name}')
|
||||
LOG.info('Using %s for better performance', dev_path)
|
||||
test_obj.report.append(ansi.color_string('badblocks', 'BLUE'))
|
||||
test_obj.report.append(color_string('badblocks', 'BLUE'))
|
||||
test_obj.set_status('Working')
|
||||
|
||||
# Increase block size if necessary
|
||||
if dev.size >= BADBLOCKS_EXTRA_LARGE_DISK:
|
||||
block_size = '8192'
|
||||
if (dev.phy_sec == 4096
|
||||
or dev.size >= BADBLOCKS_LARGE_DISK):
|
||||
block_size = '4096'
|
||||
|
|
@ -80,7 +75,7 @@ def run_scan(test_obj, log_path, test_mode=False) -> None:
|
|||
with open(log_path, 'a', encoding='utf-8') as _f:
|
||||
size_str = bytes_to_string(dev.size, use_binary=False)
|
||||
_f.write(
|
||||
ansi.color_string(
|
||||
color_string(
|
||||
['[', dev.path.name, ' ', size_str, ']\n'],
|
||||
[None, 'BLUE', None, 'CYAN', None],
|
||||
sep='',
|
||||
|
|
|
|||
|
|
@ -9,17 +9,22 @@ from dataclasses import dataclass, field
|
|||
from typing import Any
|
||||
|
||||
from wk.cfg.hw import KNOWN_RAM_VENDOR_IDS
|
||||
from wk.cfg.python import DATACLASS_DECORATOR_KWARGS
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.hw.test import Test
|
||||
from wk.std import PLATFORM, bytes_to_string, string_to_bytes
|
||||
from wk.ui import ansi
|
||||
from wk.std import (
|
||||
PLATFORM,
|
||||
bytes_to_string,
|
||||
color_string,
|
||||
string_to_bytes,
|
||||
)
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
@dataclass(**DATACLASS_DECORATOR_KWARGS)
|
||||
class System:
|
||||
"""Object for tracking system specific hardware data."""
|
||||
cpu_description: str = field(init=False)
|
||||
|
|
@ -36,11 +41,11 @@ class System:
|
|||
def generate_report(self) -> list[str]:
|
||||
"""Generate CPU & RAM report, returns list."""
|
||||
report = []
|
||||
report.append(ansi.color_string('Device', 'BLUE'))
|
||||
report.append(color_string('Device', 'BLUE'))
|
||||
report.append(f' {self.cpu_description}')
|
||||
|
||||
# Include RAM details
|
||||
report.append(ansi.color_string('RAM', 'BLUE'))
|
||||
report.append(color_string('RAM', 'BLUE'))
|
||||
report.append(f' {self.ram_total} ({", ".join(self.ram_dimms)})')
|
||||
|
||||
# Tests
|
||||
|
|
@ -110,7 +115,6 @@ def get_ram_list_linux() -> list[list]:
|
|||
cmd = ['sudo', 'dmidecode', '--type', 'memory']
|
||||
dimm_list = []
|
||||
manufacturer = 'Unknown'
|
||||
part_number = 'Unknown'
|
||||
size = 0
|
||||
|
||||
# Get DMI data
|
||||
|
|
@ -121,12 +125,8 @@ def get_ram_list_linux() -> list[list]:
|
|||
for line in dmi_data:
|
||||
line = line.strip()
|
||||
if line == 'Memory Device':
|
||||
# Add to list
|
||||
if size and (manufacturer or part_number):
|
||||
dimm_list.append([size, manufacturer, part_number])
|
||||
# Reset vars
|
||||
manufacturer = 'Unknown'
|
||||
part_number = 'Unknown'
|
||||
size = 0
|
||||
elif line.startswith('Size:'):
|
||||
size = line.replace('Size: ', '')
|
||||
|
|
@ -137,18 +137,7 @@ def get_ram_list_linux() -> list[list]:
|
|||
size = 0
|
||||
elif line.startswith('Manufacturer:'):
|
||||
manufacturer = line.replace('Manufacturer: ', '')
|
||||
elif line.startswith('Part Number: '):
|
||||
part_number = line.replace('Part Number: ', '')
|
||||
|
||||
# Add last DIMM
|
||||
if size and (manufacturer or part_number):
|
||||
dimm_list.append([size, manufacturer, part_number])
|
||||
|
||||
# Cleanup list
|
||||
dimm_list = [
|
||||
[dimm[0], dimm[1] if dimm[1] != 'Unknown' else dimm[2]]
|
||||
for dimm in dimm_list
|
||||
]
|
||||
dimm_list.append([size, manufacturer])
|
||||
|
||||
# Save details
|
||||
return dimm_list
|
||||
|
|
|
|||
|
|
@ -4,14 +4,18 @@
|
|||
from dataclasses import dataclass, field
|
||||
from typing import Any, Callable
|
||||
|
||||
@dataclass(slots=True)
|
||||
from wk.cfg.python import DATACLASS_DECORATOR_KWARGS
|
||||
|
||||
@dataclass(**DATACLASS_DECORATOR_KWARGS)
|
||||
class Test:
|
||||
# pylint: disable=too-many-instance-attributes
|
||||
"""Object for tracking test specific data."""
|
||||
dev: Any
|
||||
label: str
|
||||
name: str
|
||||
disabled: bool = field(init=False, default=False)
|
||||
failed: bool = field(init=False, default=False)
|
||||
hidden: bool = False
|
||||
passed: bool = field(init=False, default=False)
|
||||
report: list[str] = field(init=False, default_factory=list)
|
||||
status: str = field(init=False, default='Pending')
|
||||
|
|
@ -25,7 +29,7 @@ class Test:
|
|||
self.status = status
|
||||
|
||||
|
||||
@dataclass(slots=True)
|
||||
@dataclass(**DATACLASS_DECORATOR_KWARGS)
|
||||
class TestGroup:
|
||||
"""Object for tracking groups of tests."""
|
||||
name: str
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ LOG = logging.getLogger(__name__)
|
|||
|
||||
|
||||
# Functions
|
||||
def case_insensitive_path(path: pathlib.Path | str) -> pathlib.Path:
|
||||
def case_insensitive_path(path):
|
||||
"""Find path case-insensitively, returns pathlib.Path obj."""
|
||||
given_path = pathlib.Path(path).resolve()
|
||||
real_path = None
|
||||
|
|
@ -37,13 +37,12 @@ def case_insensitive_path(path: pathlib.Path | str) -> pathlib.Path:
|
|||
return real_path
|
||||
|
||||
|
||||
def case_insensitive_search(
|
||||
path: pathlib.Path | str, item: str) -> pathlib.Path:
|
||||
def case_insensitive_search(path, item):
|
||||
"""Search path for item case insensitively, returns pathlib.Path obj."""
|
||||
path = pathlib.Path(path).resolve()
|
||||
given_path = path.joinpath(item)
|
||||
real_path = None
|
||||
regex = fr'^{item}$'
|
||||
regex = fr'^{item}'
|
||||
|
||||
# Quick check
|
||||
if given_path.exists():
|
||||
|
|
@ -62,10 +61,7 @@ def case_insensitive_search(
|
|||
return real_path
|
||||
|
||||
|
||||
def copy_file(
|
||||
source: pathlib.Path | str,
|
||||
dest: pathlib.Path | str,
|
||||
overwrite: bool = False) -> None:
|
||||
def copy_file(source, dest, overwrite=False):
|
||||
"""Copy file and optionally overwrite the destination."""
|
||||
source = case_insensitive_path(source)
|
||||
dest = pathlib.Path(dest).resolve()
|
||||
|
|
@ -76,7 +72,7 @@ def copy_file(
|
|||
shutil.copy2(source, dest)
|
||||
|
||||
|
||||
def delete_empty_folders(path: pathlib.Path | str) -> None:
|
||||
def delete_empty_folders(path):
|
||||
"""Recursively delete all empty folders in path."""
|
||||
LOG.debug('path: %s', path)
|
||||
|
||||
|
|
@ -93,11 +89,7 @@ def delete_empty_folders(path: pathlib.Path | str) -> None:
|
|||
pass
|
||||
|
||||
|
||||
def delete_folder(
|
||||
path: pathlib.Path | str,
|
||||
force: bool = False,
|
||||
ignore_errors: bool = False,
|
||||
) -> None:
|
||||
def delete_folder(path, force=False, ignore_errors=False):
|
||||
"""Delete folder if empty or if forced.
|
||||
|
||||
NOTE: Exceptions are not caught by this function,
|
||||
|
|
@ -114,11 +106,7 @@ def delete_folder(
|
|||
os.rmdir(path)
|
||||
|
||||
|
||||
def delete_item(
|
||||
path: pathlib.Path | str,
|
||||
force: bool = False,
|
||||
ignore_errors: bool = False,
|
||||
) -> None:
|
||||
def delete_item(path, force=False, ignore_errors=False):
|
||||
"""Delete file or folder, optionally recursively.
|
||||
|
||||
NOTE: Exceptions are not caught by this function,
|
||||
|
|
@ -136,11 +124,7 @@ def delete_item(
|
|||
os.remove(path)
|
||||
|
||||
|
||||
def get_path_obj(
|
||||
path: pathlib.Path | str,
|
||||
expanduser: bool = True,
|
||||
resolve: bool = True,
|
||||
) -> pathlib.Path:
|
||||
def get_path_obj(path, expanduser=True, resolve=True):
|
||||
"""Get based on path, returns pathlib.Path."""
|
||||
path = pathlib.Path(path)
|
||||
if expanduser:
|
||||
|
|
@ -150,7 +134,7 @@ def get_path_obj(
|
|||
return path
|
||||
|
||||
|
||||
def non_clobber_path(path: pathlib.Path | str) -> pathlib.Path:
|
||||
def non_clobber_path(path):
|
||||
"""Update path as needed to non-existing path, returns pathlib.Path."""
|
||||
LOG.debug('path: %s', path)
|
||||
path = pathlib.Path(path)
|
||||
|
|
@ -179,10 +163,7 @@ def non_clobber_path(path: pathlib.Path | str) -> pathlib.Path:
|
|||
return new_path
|
||||
|
||||
|
||||
def recursive_copy(
|
||||
source: pathlib.Path | str,
|
||||
dest: pathlib.Path | str,
|
||||
overwrite: bool = False) -> None:
|
||||
def recursive_copy(source, dest, overwrite=False):
|
||||
"""Copy source to dest recursively.
|
||||
|
||||
NOTE: This uses rsync style source/dest syntax.
|
||||
|
|
@ -232,10 +213,7 @@ def recursive_copy(
|
|||
raise FileExistsError(f'Refusing to delete file: {dest}')
|
||||
|
||||
|
||||
def rename_item(
|
||||
path: pathlib.Path | str,
|
||||
new_path: pathlib.Path | str,
|
||||
) -> pathlib.Path:
|
||||
def rename_item(path, new_path):
|
||||
"""Rename item, returns pathlib.Path."""
|
||||
path = pathlib.Path(path)
|
||||
return path.rename(new_path)
|
||||
|
|
|
|||
|
|
@ -6,7 +6,6 @@ NOTE: This script is meant to be called from within a new kit in ConEmu.
|
|||
|
||||
import logging
|
||||
import os
|
||||
import pathlib
|
||||
import re
|
||||
|
||||
from wk.cfg.launchers import LAUNCHERS
|
||||
|
|
@ -22,8 +21,16 @@ from wk.kit.tools import (
|
|||
get_tool_path,
|
||||
)
|
||||
from wk.log import update_log_path
|
||||
from wk.std import GenericError
|
||||
from wk.ui import cli as ui
|
||||
from wk.std import (
|
||||
GenericError,
|
||||
TryAndPrint,
|
||||
clear_screen,
|
||||
pause,
|
||||
print_info,
|
||||
print_success,
|
||||
set_title,
|
||||
sleep,
|
||||
)
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -45,7 +52,7 @@ WIDTH = 50
|
|||
|
||||
|
||||
# Functions
|
||||
def compress_cbin_dirs() -> None:
|
||||
def compress_cbin_dirs():
|
||||
"""Compress CBIN_DIR items using ARCHIVE_PASSWORD."""
|
||||
current_dir = os.getcwd()
|
||||
for item in CBIN_DIR.iterdir():
|
||||
|
|
@ -63,25 +70,25 @@ def compress_cbin_dirs() -> None:
|
|||
delete_item(item, force=True, ignore_errors=True)
|
||||
|
||||
|
||||
def delete_from_temp(item_path) -> None:
|
||||
def delete_from_temp(item_path):
|
||||
"""Delete item from temp."""
|
||||
delete_item(TMP_DIR.joinpath(item_path), force=True, ignore_errors=True)
|
||||
|
||||
|
||||
def download_to_temp(filename, source_url, referer=None) -> pathlib.Path:
|
||||
def download_to_temp(filename, source_url):
|
||||
"""Download file to temp dir, returns pathlib.Path."""
|
||||
out_path = TMP_DIR.joinpath(filename)
|
||||
download_file(out_path, source_url, referer=referer)
|
||||
download_file(out_path, source_url)
|
||||
return out_path
|
||||
|
||||
|
||||
def extract_to_bin(archive, folder) -> None:
|
||||
def extract_to_bin(archive, folder):
|
||||
"""Extract archive to folder under BIN_DIR."""
|
||||
out_path = BIN_DIR.joinpath(folder)
|
||||
extract_archive(archive, out_path)
|
||||
|
||||
|
||||
def generate_launcher(section, name, options) -> None:
|
||||
def generate_launcher(section, name, options):
|
||||
"""Generate launcher script."""
|
||||
dest = ROOT_DIR.joinpath(f'{section+"/" if section else ""}{name}.cmd')
|
||||
out_text = []
|
||||
|
|
@ -108,27 +115,27 @@ def generate_launcher(section, name, options) -> None:
|
|||
|
||||
|
||||
# Download functions
|
||||
def download_adobe_reader() -> None:
|
||||
def download_adobe_reader():
|
||||
"""Download Adobe Reader."""
|
||||
out_path = INSTALLERS_DIR.joinpath('Adobe Reader DC.exe')
|
||||
download_file(out_path, SOURCES['Adobe Reader DC'])
|
||||
|
||||
|
||||
def download_aida64() -> None:
|
||||
def download_aida64():
|
||||
"""Download AIDA64."""
|
||||
archive = download_to_temp('AIDA64.zip', SOURCES['AIDA64'])
|
||||
extract_to_bin(archive, 'AIDA64')
|
||||
delete_from_temp('AIDA64.zip')
|
||||
|
||||
|
||||
def download_autoruns() -> None:
|
||||
def download_autoruns():
|
||||
"""Download Autoruns."""
|
||||
for item in ('Autoruns32', 'Autoruns64'):
|
||||
out_path = BIN_DIR.joinpath(f'Sysinternals/{item}.exe')
|
||||
download_file(out_path, SOURCES[item])
|
||||
|
||||
|
||||
def download_bleachbit() -> None:
|
||||
def download_bleachbit():
|
||||
"""Download BleachBit."""
|
||||
out_path = BIN_DIR.joinpath('BleachBit')
|
||||
archive = download_to_temp('BleachBit.zip', SOURCES['BleachBit'])
|
||||
|
|
@ -143,7 +150,7 @@ def download_bleachbit() -> None:
|
|||
delete_from_temp('BleachBit.zip')
|
||||
|
||||
|
||||
def download_bluescreenview() -> None:
|
||||
def download_bluescreenview():
|
||||
"""Download BlueScreenView."""
|
||||
archive_32 = download_to_temp(
|
||||
'bluescreenview32.zip', SOURCES['BlueScreenView32'],
|
||||
|
|
@ -162,37 +169,14 @@ def download_bluescreenview() -> None:
|
|||
delete_from_temp('bluescreenview64.zip')
|
||||
|
||||
|
||||
def download_ddu() -> None:
|
||||
"""Download Display Driver Uninstaller."""
|
||||
archive = download_to_temp('DDU.exe', SOURCES['DDU'])
|
||||
out_path = BIN_DIR.joinpath('DDU')
|
||||
extract_archive(archive, out_path, 'DDU*/*.*', mode='e')
|
||||
out_path = out_path.joinpath('Settings')
|
||||
for item in ('AMD', 'INTEL', 'Languages', 'NVIDIA', 'REALTEK'):
|
||||
extract_archive(
|
||||
archive,
|
||||
out_path.joinpath(item),
|
||||
f'DDU*/Settings/{item}/*',
|
||||
mode='e',
|
||||
)
|
||||
delete_from_temp('DDU.exe')
|
||||
|
||||
|
||||
def download_bcuninstaller() -> None:
|
||||
"""Download Bulk Crap Uninstaller."""
|
||||
archive = download_to_temp('BCU.zip', SOURCES['BCUninstaller'])
|
||||
extract_to_bin(archive, 'BCUninstaller')
|
||||
delete_from_temp('BCU.zip')
|
||||
|
||||
|
||||
def download_erunt() -> None:
|
||||
def download_erunt():
|
||||
"""Download ERUNT."""
|
||||
archive = download_to_temp('erunt.zip', SOURCES['ERUNT'])
|
||||
extract_to_bin(archive, 'ERUNT')
|
||||
delete_from_temp('erunt.zip')
|
||||
|
||||
|
||||
def download_everything() -> None:
|
||||
def download_everything():
|
||||
"""Download Everything."""
|
||||
archive_32 = download_to_temp('everything32.zip', SOURCES['Everything32'])
|
||||
archive_64 = download_to_temp('everything64.zip', SOURCES['Everything64'])
|
||||
|
|
@ -207,7 +191,7 @@ def download_everything() -> None:
|
|||
delete_from_temp('everything64.zip')
|
||||
|
||||
|
||||
def download_fastcopy() -> None:
|
||||
def download_fastcopy():
|
||||
"""Download FastCopy."""
|
||||
installer = download_to_temp('FastCopyInstaller.exe', SOURCES['FastCopy'])
|
||||
out_path = BIN_DIR.joinpath('FastCopy')
|
||||
|
|
@ -223,13 +207,9 @@ def download_fastcopy() -> None:
|
|||
delete_item(BIN_DIR.joinpath('FastCopy/setup.exe'))
|
||||
|
||||
|
||||
def download_furmark() -> None:
|
||||
def download_furmark():
|
||||
"""Download FurMark."""
|
||||
installer = download_to_temp(
|
||||
'FurMark_Setup.exe',
|
||||
SOURCES['FurMark'],
|
||||
referer=SOURCES['FurMark'],
|
||||
)
|
||||
installer = download_to_temp('FurMark_Setup.exe', SOURCES['FurMark'])
|
||||
out_path = BIN_DIR.joinpath('FurMark')
|
||||
tmp_path = TMP_DIR.joinpath('FurMarkInstall')
|
||||
run_program([installer, f'/DIR={tmp_path}', '/SILENT'])
|
||||
|
|
@ -243,32 +223,37 @@ def download_furmark() -> None:
|
|||
delete_from_temp('FurMarkInstall')
|
||||
|
||||
|
||||
def download_hwinfo() -> None:
|
||||
def download_hwinfo():
|
||||
"""Download HWiNFO."""
|
||||
archive = download_to_temp('HWiNFO.zip', SOURCES['HWiNFO'])
|
||||
extract_to_bin(archive, 'HWiNFO')
|
||||
delete_from_temp('HWiNFO.zip')
|
||||
|
||||
|
||||
def download_macs_fan_control() -> None:
|
||||
def download_iobit_uninstaller():
|
||||
"""Download IOBit Uninstaller."""
|
||||
installer = CBIN_DIR.joinpath('IObitUninstallerPortable.exe')
|
||||
download_file(installer, SOURCES['IOBit Uninstaller'])
|
||||
popen_program([installer])
|
||||
sleep(1)
|
||||
wait_for_procs('IObitUninstallerPortable.exe')
|
||||
delete_item(installer)
|
||||
|
||||
|
||||
def download_macs_fan_control():
|
||||
"""Download Macs Fan Control."""
|
||||
out_path = INSTALLERS_DIR.joinpath('Macs Fan Control.exe')
|
||||
download_file(out_path, SOURCES['Macs Fan Control'])
|
||||
|
||||
|
||||
def download_libreoffice() -> None:
|
||||
def download_libreoffice():
|
||||
"""Download LibreOffice."""
|
||||
for arch in 32, 64:
|
||||
out_path = INSTALLERS_DIR.joinpath(f'LibreOffice{arch}.msi')
|
||||
download_file(
|
||||
out_path,
|
||||
SOURCES[f'LibreOffice{arch}'],
|
||||
referer='https://www.libreoffice.org/download/download-libreoffice/',
|
||||
)
|
||||
ui.sleep(1)
|
||||
download_file(out_path, SOURCES[f'LibreOffice{arch}'])
|
||||
|
||||
|
||||
def download_neutron() -> None:
|
||||
def download_neutron():
|
||||
"""Download Neutron."""
|
||||
archive = download_to_temp('neutron.zip', SOURCES['Neutron'])
|
||||
out_path = BIN_DIR.joinpath('Neutron')
|
||||
|
|
@ -276,7 +261,7 @@ def download_neutron() -> None:
|
|||
delete_from_temp('neutron.zip')
|
||||
|
||||
|
||||
def download_notepad_plus_plus() -> None:
|
||||
def download_notepad_plus_plus():
|
||||
"""Download Notepad++."""
|
||||
archive = download_to_temp('npp.7z', SOURCES['Notepad++'])
|
||||
extract_to_bin(archive, 'NotepadPlusPlus')
|
||||
|
|
@ -288,21 +273,21 @@ def download_notepad_plus_plus() -> None:
|
|||
delete_from_temp('npp.7z')
|
||||
|
||||
|
||||
def download_openshell() -> None:
|
||||
def download_openshell():
|
||||
"""Download OpenShell installer and Fluent-Metro skin."""
|
||||
for name in ('OpenShell.exe', 'Fluent-Metro.zip'):
|
||||
out_path = BIN_DIR.joinpath(f'OpenShell/{name}')
|
||||
download_file(out_path, SOURCES[name[:-4]])
|
||||
|
||||
|
||||
def download_putty() -> None:
|
||||
def download_putty():
|
||||
"""Download PuTTY."""
|
||||
archive = download_to_temp('putty.zip', SOURCES['PuTTY'])
|
||||
extract_to_bin(archive, 'PuTTY')
|
||||
delete_from_temp('putty.zip')
|
||||
|
||||
|
||||
def download_snappy_driver_installer_origin() -> None:
|
||||
def download_snappy_driver_installer_origin():
|
||||
"""Download Snappy Driver Installer Origin."""
|
||||
archive = download_to_temp('aria2.zip', SOURCES['Aria2'])
|
||||
aria2c = TMP_DIR.joinpath('aria2/aria2c.exe')
|
||||
|
|
@ -335,7 +320,7 @@ def download_snappy_driver_installer_origin() -> None:
|
|||
cmd.append('-new_console:n')
|
||||
cmd.append('-new_console:s33V')
|
||||
popen_program(cmd, cwd=aria2c.parent)
|
||||
ui.sleep(1)
|
||||
sleep(1)
|
||||
wait_for_procs('aria2c.exe')
|
||||
else:
|
||||
run_program(cmd)
|
||||
|
|
@ -372,14 +357,25 @@ def download_snappy_driver_installer_origin() -> None:
|
|||
delete_from_temp('fake.7z')
|
||||
|
||||
|
||||
def download_wiztree() -> None:
|
||||
def download_testdisk():
|
||||
"""Download TestDisk."""
|
||||
archive = download_to_temp('testdisk_wip.zip', SOURCES['TestDisk'])
|
||||
out_path = BIN_DIR.joinpath('TestDisk')
|
||||
tmp_path = TMP_DIR.joinpath('TestDisk')
|
||||
extract_archive(archive, tmp_path)
|
||||
rename_item(tmp_path.joinpath('testdisk-7.2-WIP'), out_path)
|
||||
delete_from_temp('TestDisk')
|
||||
delete_from_temp('testdisk_wip.zip')
|
||||
|
||||
|
||||
def download_wiztree():
|
||||
"""Download WizTree."""
|
||||
archive = download_to_temp('wiztree.zip', SOURCES['WizTree'])
|
||||
extract_to_bin(archive, 'WizTree')
|
||||
delete_from_temp('wiztree.zip')
|
||||
|
||||
|
||||
def download_xmplay() -> None:
|
||||
def download_xmplay():
|
||||
"""Download XMPlay."""
|
||||
archives = [
|
||||
download_to_temp('xmplay.zip', SOURCES['XMPlay']),
|
||||
|
|
@ -395,7 +391,7 @@ def download_xmplay() -> None:
|
|||
args = [archive, BIN_DIR.joinpath('XMPlay/plugins')]
|
||||
if archive.name == 'Innocuous.zip':
|
||||
args.append(
|
||||
'Innocuous (v1.7)/Innocuous (Hue Shifted)/'
|
||||
'Innocuous (v1.5)/Innocuous (Hue Shifted)/'
|
||||
'Innocuous (Dark Skies - Purple-80) [L1].xmpskin'
|
||||
)
|
||||
extract_archive(*args, mode='e')
|
||||
|
|
@ -407,7 +403,7 @@ def download_xmplay() -> None:
|
|||
delete_from_temp('xmp-rar.zip')
|
||||
delete_from_temp('Innocuous.zip')
|
||||
|
||||
def download_xmplay_music() -> None:
|
||||
def download_xmplay_music():
|
||||
"""Download XMPlay Music."""
|
||||
music_tmp = TMP_DIR.joinpath('music')
|
||||
music_tmp.mkdir(exist_ok=True)
|
||||
|
|
@ -460,17 +456,17 @@ def download_xmplay_music() -> None:
|
|||
|
||||
|
||||
# "Main" Function
|
||||
def build_kit() -> None:
|
||||
def build_kit():
|
||||
"""Build Kit."""
|
||||
update_log_path(dest_name='Build Tool', timestamp=True)
|
||||
title = f'{KIT_NAME_FULL}: Build Tool'
|
||||
ui.clear_screen()
|
||||
ui.set_title(title)
|
||||
ui.print_info(title)
|
||||
clear_screen()
|
||||
set_title(title)
|
||||
print_info(title)
|
||||
print('')
|
||||
|
||||
# Set up TryAndPrint
|
||||
try_print = ui.TryAndPrint()
|
||||
try_print = TryAndPrint()
|
||||
try_print.width = WIDTH
|
||||
try_print.verbose = True
|
||||
for error in ('CalledProcessError', 'FileNotFoundError'):
|
||||
|
|
@ -483,12 +479,11 @@ def build_kit() -> None:
|
|||
try_print.run('BleachBit...', download_bleachbit)
|
||||
try_print.run('BlueScreenView...', download_bluescreenview)
|
||||
try_print.run('ERUNT...', download_erunt)
|
||||
try_print.run('BulkCrapUninstaller...', download_bcuninstaller)
|
||||
try_print.run('DDU...', download_ddu)
|
||||
try_print.run('Everything...', download_everything)
|
||||
try_print.run('FastCopy...', download_fastcopy)
|
||||
try_print.run('FurMark...', download_furmark)
|
||||
try_print.run('HWiNFO...', download_hwinfo)
|
||||
try_print.run('IOBit Uninstaller...', download_iobit_uninstaller)
|
||||
try_print.run('LibreOffice...', download_libreoffice)
|
||||
try_print.run('Macs Fan Control...', download_macs_fan_control)
|
||||
try_print.run('Neutron...', download_neutron)
|
||||
|
|
@ -496,21 +491,22 @@ def build_kit() -> None:
|
|||
try_print.run('OpenShell...', download_openshell)
|
||||
try_print.run('PuTTY...', download_putty)
|
||||
try_print.run('Snappy Driver Installer...', download_snappy_driver_installer_origin)
|
||||
try_print.run('TestDisk...', download_testdisk)
|
||||
try_print.run('WizTree...', download_wiztree)
|
||||
try_print.run('XMPlay...', download_xmplay)
|
||||
try_print.run('XMPlay Music...', download_xmplay_music)
|
||||
|
||||
# Pause
|
||||
print('', flush=True)
|
||||
ui.pause('Please review and press Enter to continue...')
|
||||
pause('Please review and press Enter to continue...')
|
||||
|
||||
# Compress .cbin
|
||||
try_print.run('Compress cbin...', compress_cbin_dirs)
|
||||
|
||||
# Generate launcher scripts
|
||||
ui.print_success('Generating launchers')
|
||||
print_success('Generating launchers')
|
||||
for section, launchers in sorted(LAUNCHERS.items()):
|
||||
ui.print_info(f' {section if section else "(Root)"}')
|
||||
print_info(f' {section if section else "(Root)"}')
|
||||
for name, options in sorted(launchers.items()):
|
||||
try_print.run(
|
||||
f' {name}...', generate_launcher,
|
||||
|
|
@ -520,7 +516,7 @@ def build_kit() -> None:
|
|||
# Done
|
||||
print('')
|
||||
print('Done.')
|
||||
ui.pause('Press Enter to exit...')
|
||||
pause('Press Enter to exit...')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
|
|
|||
|
|
@ -1,19 +1,17 @@
|
|||
"""WizardKit: Tool Functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
import logging
|
||||
import pathlib
|
||||
import platform
|
||||
|
||||
from datetime import datetime, timedelta
|
||||
from subprocess import CompletedProcess, Popen
|
||||
|
||||
import requests
|
||||
|
||||
from wk.cfg.main import ARCHIVE_PASSWORD
|
||||
from wk.cfg.sources import DOWNLOAD_FREQUENCY, SOURCES
|
||||
from wk.exe import popen_program, run_program
|
||||
from wk.std import GenericError, sleep
|
||||
from wk.std import GenericError
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -32,9 +30,7 @@ CACHED_DIRS = {}
|
|||
|
||||
|
||||
# Functions
|
||||
def download_file(
|
||||
out_path, source_url,
|
||||
as_new=False, overwrite=False, referer=None) -> pathlib.Path:
|
||||
def download_file(out_path, source_url, as_new=False, overwrite=False):
|
||||
"""Download a file using requests, returns pathlib.Path."""
|
||||
out_path = pathlib.Path(out_path).resolve()
|
||||
name = out_path.name
|
||||
|
|
@ -42,7 +38,6 @@ def download_file(
|
|||
download_msg = f'Downloading {name}...'
|
||||
if as_new:
|
||||
out_path = out_path.with_suffix(f'{out_path.suffix}.new')
|
||||
overwrite = True
|
||||
print(download_msg, end='', flush=True)
|
||||
|
||||
# Avoid clobbering
|
||||
|
|
@ -52,34 +47,12 @@ def download_file(
|
|||
# Create destination directory
|
||||
out_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
# Update headers
|
||||
headers = HEADERS.copy()
|
||||
if referer:
|
||||
headers['referer'] = referer
|
||||
|
||||
# Request download
|
||||
with requests.Session() as session:
|
||||
try:
|
||||
response = session.get(
|
||||
source_url,
|
||||
allow_redirects=True,
|
||||
headers=headers,
|
||||
stream=True,
|
||||
)
|
||||
except requests.RequestException:
|
||||
try:
|
||||
sleep(1)
|
||||
response = session.get(
|
||||
source_url,
|
||||
allow_redirects=True,
|
||||
headers=headers,
|
||||
stream=True,
|
||||
)
|
||||
except requests.RequestException as _err:
|
||||
download_failed = _err
|
||||
else:
|
||||
if not response.ok:
|
||||
download_failed = response
|
||||
response = session.get(source_url, headers=HEADERS, stream=True)
|
||||
except requests.RequestException as _err:
|
||||
download_failed = _err
|
||||
else:
|
||||
if not response.ok:
|
||||
download_failed = response
|
||||
|
|
@ -99,7 +72,7 @@ def download_file(
|
|||
return out_path
|
||||
|
||||
|
||||
def download_tool(folder, name, suffix=None) -> None:
|
||||
def download_tool(folder, name, suffix=None):
|
||||
"""Download tool."""
|
||||
name_arch = f'{name}{ARCH}'
|
||||
out_path = get_tool_path(folder, name, check=False, suffix=suffix)
|
||||
|
|
@ -134,7 +107,7 @@ def download_tool(folder, name, suffix=None) -> None:
|
|||
raise
|
||||
|
||||
|
||||
def extract_archive(archive, out_path, *args, mode='x', silent=True) -> None:
|
||||
def extract_archive(archive, out_path, *args, mode='x', silent=True):
|
||||
"""Extract an archive to out_path."""
|
||||
out_path = pathlib.Path(out_path).resolve()
|
||||
out_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
|
@ -146,7 +119,7 @@ def extract_archive(archive, out_path, *args, mode='x', silent=True) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def extract_tool(folder) -> None:
|
||||
def extract_tool(folder):
|
||||
"""Extract tool."""
|
||||
extract_archive(
|
||||
find_kit_dir('.cbin').joinpath(folder).with_suffix('.7z'),
|
||||
|
|
@ -155,7 +128,7 @@ def extract_tool(folder) -> None:
|
|||
)
|
||||
|
||||
|
||||
def find_kit_dir(name=None) -> pathlib.Path:
|
||||
def find_kit_dir(name=None):
|
||||
"""Find folder in kit, returns pathlib.Path.
|
||||
|
||||
Search is performed in the script's path and then recursively upwards.
|
||||
|
|
@ -172,7 +145,7 @@ def find_kit_dir(name=None) -> pathlib.Path:
|
|||
cur_path = cur_path.parent
|
||||
|
||||
# Check
|
||||
if not cur_path.joinpath(search).exists():
|
||||
if cur_path.match(cur_path.anchor):
|
||||
raise FileNotFoundError(f'Failed to find kit dir, {name=}')
|
||||
if name:
|
||||
cur_path = cur_path.joinpath(name)
|
||||
|
|
@ -182,7 +155,7 @@ def find_kit_dir(name=None) -> pathlib.Path:
|
|||
return cur_path
|
||||
|
||||
|
||||
def get_tool_path(folder, name, check=True, suffix=None) -> pathlib.Path:
|
||||
def get_tool_path(folder, name, check=True, suffix=None):
|
||||
"""Get tool path, returns pathlib.Path"""
|
||||
bin_dir = find_kit_dir('.bin')
|
||||
if not suffix:
|
||||
|
|
@ -207,7 +180,7 @@ def run_tool(
|
|||
folder, name, *run_args,
|
||||
cbin=False, cwd=False, download=False, popen=False,
|
||||
**run_kwargs,
|
||||
) -> CompletedProcess | Popen:
|
||||
):
|
||||
"""Run tool from the kit or the Internet, returns proc obj.
|
||||
|
||||
proc will be either subprocess.CompletedProcess or subprocess.Popen."""
|
||||
|
|
|
|||
|
|
@ -1,33 +1,56 @@
|
|||
"""WizardKit: UFD Functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import argparse
|
||||
import logging
|
||||
import math
|
||||
import os
|
||||
import pathlib
|
||||
import re
|
||||
import shutil
|
||||
from subprocess import CalledProcessError
|
||||
|
||||
from wk import io, log
|
||||
from collections import OrderedDict
|
||||
from docopt import docopt
|
||||
|
||||
from wk import io, log, std
|
||||
from wk.cfg.main import KIT_NAME_FULL, KIT_NAME_SHORT
|
||||
from wk.cfg.ufd import (
|
||||
BOOT_ENTRIES,
|
||||
BOOT_FILES,
|
||||
IMAGE_BOOT_ENTRIES,
|
||||
ITEMS,
|
||||
ITEMS_FROM_LIVE,
|
||||
ITEMS_HIDDEN,
|
||||
SOURCES,
|
||||
)
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.os import linux
|
||||
|
||||
from wk.ui import cli as ui
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
DOCSTRING = '''WizardKit: Build UFD
|
||||
|
||||
Usage:
|
||||
build-ufd [options] --ufd-device PATH
|
||||
[--linux PATH]
|
||||
[--linux-minimal PATH]
|
||||
[--main-kit PATH]
|
||||
[--winpe PATH]
|
||||
[--extra-dir PATH]
|
||||
[EXTRA_IMAGES...]
|
||||
build-ufd (-h | --help)
|
||||
|
||||
Options:
|
||||
-e PATH, --extra-dir PATH
|
||||
-k PATH, --main-kit PATH
|
||||
-l PATH, --linux PATH
|
||||
-m PATH, --linux-minimal PATH
|
||||
-u PATH, --ufd-device PATH
|
||||
-w PATH, --winpe PATH
|
||||
|
||||
-d --debug Enable debug mode
|
||||
-h --help Show this page
|
||||
-M --use-mbr Use real MBR instead of GPT w/ Protective MBR
|
||||
-F --force Bypass all confirmation messages. USE WITH EXTREME CAUTION!
|
||||
-U --update Don't format device, just update
|
||||
'''
|
||||
LOG = logging.getLogger(__name__)
|
||||
EXTRA_IMAGES_LIST = '/mnt/UFD/arch/extra_images.list'
|
||||
MIB = 1024 ** 2
|
||||
|
|
@ -36,55 +59,7 @@ UFD_LABEL = f'{KIT_NAME_SHORT}_UFD'
|
|||
|
||||
|
||||
# Functions
|
||||
def argparse_helper() -> dict[str, None|bool|str]:
|
||||
"""Helper function to setup and return args, returns dict.
|
||||
|
||||
NOTE: A dict is used to match the legacy code.
|
||||
"""
|
||||
parser = argparse.ArgumentParser(
|
||||
prog='build-ufd',
|
||||
description=f'{KIT_NAME_FULL}: Build UFD',
|
||||
)
|
||||
parser.add_argument('-u', '--ufd-device', required=True)
|
||||
parser.add_argument('-l', '--linux', required=False)
|
||||
parser.add_argument('-e', '--extra-dir', required=False)
|
||||
parser.add_argument('-k', '--main-kit', required=False)
|
||||
parser.add_argument('-w', '--winpe', required=False)
|
||||
parser.add_argument(
|
||||
'-d', '--debug', action='store_true',
|
||||
help='Enable debug mode',
|
||||
)
|
||||
parser.add_argument(
|
||||
'-M', '--use-mbr', action='store_true',
|
||||
help='Use real MBR instead of GPT w/ Protective MBR',
|
||||
)
|
||||
parser.add_argument(
|
||||
'-F', '--force', action='store_true',
|
||||
help='Bypass all confirmation messages. USE WITH EXTREME CAUTION!',
|
||||
)
|
||||
parser.add_argument(
|
||||
'-U', '--update', action='store_true',
|
||||
help="Don't format device, just update",
|
||||
)
|
||||
parser.add_argument(
|
||||
'EXTRA_IMAGES', nargs='*',
|
||||
)
|
||||
args = parser.parse_args()
|
||||
legacy_args = {
|
||||
'--debug': args.debug,
|
||||
'--extra-dir': args.extra_dir,
|
||||
'--force': args.force,
|
||||
'--linux': args.linux,
|
||||
'--main-kit': args.main_kit,
|
||||
'--ufd-device': args.ufd_device,
|
||||
'--update': args.update,
|
||||
'--use-mbr': args.use_mbr,
|
||||
'--winpe': args.winpe,
|
||||
'EXTRA_IMAGES': args.EXTRA_IMAGES,
|
||||
}
|
||||
return legacy_args
|
||||
|
||||
def apply_image(part_path, image_path, hide_macos_boot=True) -> None:
|
||||
def apply_image(part_path, image_path, hide_macos_boot=True):
|
||||
"""Apply raw image to dev_path using dd."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -114,21 +89,17 @@ def apply_image(part_path, image_path, hide_macos_boot=True) -> None:
|
|||
linux.unmount(source_or_mountpoint='/mnt/TMP')
|
||||
|
||||
|
||||
def build_ufd() -> None:
|
||||
def build_ufd():
|
||||
# pylint: disable=too-many-statements
|
||||
"""Build UFD using selected sources."""
|
||||
try:
|
||||
args = argparse_helper()
|
||||
except SystemExit:
|
||||
print('')
|
||||
ui.pause('Press Enter to exit...')
|
||||
raise
|
||||
args = docopt(DOCSTRING)
|
||||
if args['--debug']:
|
||||
log.enable_debug_mode()
|
||||
if args['--update'] and args['EXTRA_IMAGES']:
|
||||
ui.print_warning('Extra images are ignored when updating')
|
||||
std.print_warning('Extra images are ignored when updating')
|
||||
args['EXTRA_IMAGES'] = []
|
||||
log.update_log_path(dest_name='build-ufd', timestamp=True)
|
||||
try_print = ui.TryAndPrint()
|
||||
try_print = std.TryAndPrint()
|
||||
try_print.add_error('FileNotFoundError')
|
||||
try_print.catch_all = False
|
||||
try_print.indent = 2
|
||||
|
|
@ -136,9 +107,9 @@ def build_ufd() -> None:
|
|||
try_print.width = 64
|
||||
|
||||
# Show header
|
||||
ui.print_success(KIT_NAME_FULL)
|
||||
ui.print_warning('UFD Build Tool')
|
||||
ui.print_warning(' ')
|
||||
std.print_success(KIT_NAME_FULL)
|
||||
std.print_warning('UFD Build Tool')
|
||||
std.print_warning(' ')
|
||||
|
||||
# Verify selections
|
||||
ufd_dev = verify_ufd(args['--ufd-device'])
|
||||
|
|
@ -150,9 +121,9 @@ def build_ufd() -> None:
|
|||
|
||||
# Prep UFD
|
||||
if not args['--update']:
|
||||
ui.print_info('Prep UFD')
|
||||
std.print_info('Prep UFD')
|
||||
try_print.run(
|
||||
message='Zeroing first 1MiB...',
|
||||
message='Zeroing first 64MiB...',
|
||||
function=zero_device,
|
||||
dev_path=ufd_dev,
|
||||
)
|
||||
|
|
@ -175,13 +146,6 @@ def build_ufd() -> None:
|
|||
dev_path=ufd_dev,
|
||||
label=UFD_LABEL,
|
||||
)
|
||||
try_print.run(
|
||||
message='Hiding extra partition(s)...',
|
||||
function=hide_extra_partitions,
|
||||
dev_path=ufd_dev,
|
||||
num_parts=len(extra_images),
|
||||
use_mbr=args['--use-mbr'],
|
||||
)
|
||||
ufd_dev_first_partition = find_first_partition(ufd_dev)
|
||||
|
||||
# Mount UFD
|
||||
|
|
@ -207,25 +171,14 @@ def build_ufd() -> None:
|
|||
message='Removing Linux...',
|
||||
function=remove_arch,
|
||||
)
|
||||
# Copy boot files
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Boot Files')
|
||||
for s_section, s_items in ITEMS_FROM_LIVE.items():
|
||||
s_section = pathlib.Path(s_section)
|
||||
try_print.run(
|
||||
message=f'Copying {s_section}...',
|
||||
function=copy_source,
|
||||
source=s_section,
|
||||
items=s_items,
|
||||
from_live=True,
|
||||
overwrite=True,
|
||||
)
|
||||
os.rename('/mnt/UFD/EFI/Boot/refind_x64.efi', '/mnt/UFD/EFI/Boot/bootx64.efi')
|
||||
|
||||
|
||||
# Copy sources
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Copy Sources')
|
||||
std.print_standard(' ')
|
||||
std.print_info('Copy Sources')
|
||||
try_print.run(
|
||||
'Copying Memtest86...', io.recursive_copy,
|
||||
'/usr/share/memtest86-efi/', '/mnt/UFD/EFI/Memtest86/', overwrite=True,
|
||||
)
|
||||
for s_label, s_path in sources.items():
|
||||
try_print.run(
|
||||
message=f'Copying {s_label}...',
|
||||
|
|
@ -237,8 +190,8 @@ def build_ufd() -> None:
|
|||
|
||||
# Apply extra images
|
||||
if not args['--update']:
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Apply Extra Images')
|
||||
std.print_standard(' ')
|
||||
std.print_info('Apply Extra Images')
|
||||
for part_num, image_path in enumerate(extra_images):
|
||||
try_print.run(
|
||||
message=f'Applying {image_path.name}...',
|
||||
|
|
@ -253,8 +206,8 @@ def build_ufd() -> None:
|
|||
_f.write('\n'.join([image.name for image in extra_images]))
|
||||
|
||||
# Update boot entries
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Boot Setup')
|
||||
std.print_standard(' ')
|
||||
std.print_info('Boot Setup')
|
||||
try_print.run(
|
||||
message='Updating boot entries...',
|
||||
function=update_boot_entries,
|
||||
|
|
@ -285,8 +238,8 @@ def build_ufd() -> None:
|
|||
)
|
||||
|
||||
# Hide items
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Final Touches')
|
||||
std.print_standard(' ')
|
||||
std.print_info('Final Touches')
|
||||
try_print.run(
|
||||
message='Hiding items...',
|
||||
function=hide_items,
|
||||
|
|
@ -295,35 +248,35 @@ def build_ufd() -> None:
|
|||
)
|
||||
|
||||
# Done
|
||||
ui.print_standard('\nDone.')
|
||||
std.print_standard('\nDone.')
|
||||
if not args['--force']:
|
||||
ui.pause('Press Enter to exit...')
|
||||
std.pause('Press Enter to exit...')
|
||||
|
||||
|
||||
def confirm_selections(update=False) -> None:
|
||||
def confirm_selections(update=False):
|
||||
"""Ask tech to confirm selections, twice if necessary."""
|
||||
if not ui.ask('Is the above information correct?'):
|
||||
ui.abort()
|
||||
if not std.ask('Is the above information correct?'):
|
||||
std.abort()
|
||||
|
||||
# Safety check
|
||||
if not update:
|
||||
ui.print_standard(' ')
|
||||
ui.print_warning('SAFETY CHECK')
|
||||
ui.print_standard(
|
||||
std.print_standard(' ')
|
||||
std.print_warning('SAFETY CHECK')
|
||||
std.print_standard(
|
||||
'All data will be DELETED from the disk and partition(s) listed above.')
|
||||
ui.print_colored(
|
||||
std.print_colored(
|
||||
['This is irreversible and will lead to', 'DATA LOSS'],
|
||||
[None, 'RED'],
|
||||
)
|
||||
if not ui.ask('Asking again to confirm, is this correct?'):
|
||||
ui.abort()
|
||||
if not std.ask('Asking again to confirm, is this correct?'):
|
||||
std.abort()
|
||||
|
||||
ui.print_standard(' ')
|
||||
std.print_standard(' ')
|
||||
|
||||
|
||||
def copy_source(source, items, from_live=False, overwrite=False) -> None:
|
||||
def copy_source(source, items, overwrite=False):
|
||||
"""Copy source items to /mnt/UFD."""
|
||||
is_image = not from_live and (source.is_file() or source.is_block_device())
|
||||
is_image = source.is_file()
|
||||
items_not_found = False
|
||||
|
||||
# Mount source if necessary
|
||||
|
|
@ -332,14 +285,7 @@ def copy_source(source, items, from_live=False, overwrite=False) -> None:
|
|||
|
||||
# Copy items
|
||||
for i_source, i_dest in items:
|
||||
if from_live:
|
||||
# Don't prepend source
|
||||
pass
|
||||
elif is_image:
|
||||
i_source = f'/mnt/Source{i_source}'
|
||||
else:
|
||||
# Prepend source
|
||||
i_source = f'{source}{i_source}'
|
||||
i_source = f'{"/mnt/Source" if is_image else source}{i_source}'
|
||||
i_dest = f'/mnt/UFD{i_dest}'
|
||||
try:
|
||||
io.recursive_copy(i_source, i_dest, overwrite=overwrite)
|
||||
|
|
@ -355,7 +301,7 @@ def copy_source(source, items, from_live=False, overwrite=False) -> None:
|
|||
raise FileNotFoundError('One or more items not found')
|
||||
|
||||
|
||||
def create_table(dev_path, use_mbr=False, images=None) -> None:
|
||||
def create_table(dev_path, use_mbr=False, images=None):
|
||||
"""Create GPT or DOS partition table."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -385,7 +331,7 @@ def create_table(dev_path, use_mbr=False, images=None) -> None:
|
|||
for part, real in zip(part_sizes, images):
|
||||
end = start + real
|
||||
cmd.append(
|
||||
f'mkpart primary "fat32" {start}B {end-1}B',
|
||||
f'mkpart primary {"fat32" if start==MIB else "hfs+"} {start}B {end-1}B',
|
||||
)
|
||||
start += part
|
||||
|
||||
|
|
@ -393,7 +339,7 @@ def create_table(dev_path, use_mbr=False, images=None) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def find_first_partition(dev_path) -> str:
|
||||
def find_first_partition(dev_path):
|
||||
"""Find path to first partition of dev, returns str."""
|
||||
cmd = [
|
||||
'lsblk',
|
||||
|
|
@ -412,7 +358,7 @@ def find_first_partition(dev_path) -> str:
|
|||
return part_path
|
||||
|
||||
|
||||
def format_partition(dev_path, label) -> None:
|
||||
def format_partition(dev_path, label):
|
||||
"""Format first partition on device FAT32."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -424,7 +370,7 @@ def format_partition(dev_path, label) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def get_block_device_size(dev_path) -> int:
|
||||
def get_block_device_size(dev_path):
|
||||
"""Get block device size via lsblk, returns int."""
|
||||
cmd = [
|
||||
'lsblk',
|
||||
|
|
@ -443,7 +389,7 @@ def get_block_device_size(dev_path) -> int:
|
|||
return int(proc.stdout.strip())
|
||||
|
||||
|
||||
def get_uuid(path) -> str:
|
||||
def get_uuid(path):
|
||||
"""Get filesystem UUID via findmnt, returns str."""
|
||||
cmd = [
|
||||
'findmnt',
|
||||
|
|
@ -459,7 +405,7 @@ def get_uuid(path) -> str:
|
|||
return proc.stdout.strip()
|
||||
|
||||
|
||||
def hide_items(ufd_dev_first_partition, items) -> None:
|
||||
def hide_items(ufd_dev_first_partition, items):
|
||||
"""Set FAT32 hidden flag for items."""
|
||||
with open('/root/.mtoolsrc', 'w', encoding='utf-8') as _f:
|
||||
_f.write(f'drive U: file="{ufd_dev_first_partition}"\n')
|
||||
|
|
@ -471,18 +417,7 @@ def hide_items(ufd_dev_first_partition, items) -> None:
|
|||
run_program(cmd, shell=True, check=False)
|
||||
|
||||
|
||||
def hide_extra_partitions(dev_path, num_parts, use_mbr) -> None:
|
||||
if use_mbr:
|
||||
# Bail early
|
||||
return
|
||||
|
||||
for part_id in range(num_parts):
|
||||
part_id += 2 # Extra partitions start at 2
|
||||
cmd = ['sfdisk', '--part-attrs', dev_path, str(part_id), 'RequiredPartition,62,63']
|
||||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def install_syslinux_to_dev(ufd_dev, use_mbr) -> None:
|
||||
def install_syslinux_to_dev(ufd_dev, use_mbr):
|
||||
"""Install Syslinux to UFD (dev)."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -495,7 +430,7 @@ def install_syslinux_to_dev(ufd_dev, use_mbr) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def install_syslinux_to_partition(partition) -> None:
|
||||
def install_syslinux_to_partition(partition):
|
||||
"""Install Syslinux to UFD (partition)."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -508,7 +443,7 @@ def install_syslinux_to_partition(partition) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def is_valid_path(path_obj, path_type) -> bool:
|
||||
def is_valid_path(path_obj, path_type):
|
||||
"""Verify path_obj is valid by type, returns bool."""
|
||||
valid_path = False
|
||||
if path_type == 'DIR':
|
||||
|
|
@ -519,14 +454,13 @@ def is_valid_path(path_obj, path_type) -> bool:
|
|||
valid_path = path_obj.is_file() and path_obj.suffix.lower() == '.img'
|
||||
elif path_type == 'ISO':
|
||||
valid_path = path_obj.is_file() and path_obj.suffix.lower() == '.iso'
|
||||
valid_path = valid_path or re.match(r'^/dev/sr\d+$', str(path_obj))
|
||||
elif path_type == 'UFD':
|
||||
valid_path = path_obj.is_block_device()
|
||||
|
||||
return valid_path
|
||||
|
||||
|
||||
def set_boot_flag(dev_path, use_mbr=False) -> None:
|
||||
def set_boot_flag(dev_path, use_mbr=False):
|
||||
"""Set modern or legacy boot flag."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
|
|
@ -538,7 +472,7 @@ def set_boot_flag(dev_path, use_mbr=False) -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def remove_arch() -> None:
|
||||
def remove_arch():
|
||||
"""Remove arch dir from UFD.
|
||||
|
||||
This ensures a clean installation to the UFD and resets the boot files
|
||||
|
|
@ -546,16 +480,16 @@ def remove_arch() -> None:
|
|||
shutil.rmtree(io.case_insensitive_path('/mnt/UFD/arch'))
|
||||
|
||||
|
||||
def show_selections(args, sources, ufd_dev, ufd_sources, extra_images) -> None:
|
||||
def show_selections(args, sources, ufd_dev, ufd_sources, extra_images):
|
||||
"""Show selections including non-specified options."""
|
||||
|
||||
# Sources
|
||||
ui.print_info('Sources')
|
||||
std.print_info('Sources')
|
||||
for label in ufd_sources.keys():
|
||||
if label in sources:
|
||||
ui.print_standard(f' {label+":":<18} {sources[label]}')
|
||||
std.print_standard(f' {label+":":<18} {sources[label]}')
|
||||
else:
|
||||
ui.print_colored(
|
||||
std.print_colored(
|
||||
[f' {label+":":<18}', 'Not Specified'],
|
||||
[None, 'YELLOW'],
|
||||
)
|
||||
|
|
@ -567,15 +501,15 @@ def show_selections(args, sources, ufd_dev, ufd_sources, extra_images) -> None:
|
|||
print(f' {" ":<18} {image}')
|
||||
|
||||
# Destination
|
||||
ui.print_standard(' ')
|
||||
ui.print_info('Destination')
|
||||
std.print_standard(' ')
|
||||
std.print_info('Destination')
|
||||
cmd = [
|
||||
'lsblk', '--nodeps', '--noheadings', '--paths',
|
||||
'--output', 'NAME,FSTYPE,TRAN,SIZE,VENDOR,MODEL,SERIAL',
|
||||
ufd_dev,
|
||||
]
|
||||
proc = run_program(cmd, check=False)
|
||||
ui.print_standard(proc.stdout.strip())
|
||||
std.print_standard(proc.stdout.strip())
|
||||
cmd = [
|
||||
'lsblk', '--noheadings', '--paths',
|
||||
'--output', 'NAME,SIZE,FSTYPE,LABEL,MOUNTPOINT',
|
||||
|
|
@ -583,17 +517,17 @@ def show_selections(args, sources, ufd_dev, ufd_sources, extra_images) -> None:
|
|||
]
|
||||
proc = run_program(cmd, check=False)
|
||||
for line in proc.stdout.splitlines()[1:]:
|
||||
ui.print_standard(line)
|
||||
std.print_standard(line)
|
||||
|
||||
# Notes
|
||||
if args['--update']:
|
||||
ui.print_warning('Updating kit in-place')
|
||||
std.print_warning('Updating kit in-place')
|
||||
elif args['--use-mbr']:
|
||||
ui.print_warning('Formatting using legacy MBR')
|
||||
ui.print_standard(' ')
|
||||
std.print_warning('Formatting using legacy MBR')
|
||||
std.print_standard(' ')
|
||||
|
||||
|
||||
def update_boot_entries(ufd_dev, images=None) -> None:
|
||||
def update_boot_entries(ufd_dev, images=None):
|
||||
"""Update boot files for UFD usage"""
|
||||
configs = []
|
||||
uuids = [get_uuid('/mnt/UFD')]
|
||||
|
|
@ -615,7 +549,7 @@ def update_boot_entries(ufd_dev, images=None) -> None:
|
|||
'sed',
|
||||
'--in-place',
|
||||
'--regexp-extended',
|
||||
f's/___+/{uuids[0]}/',
|
||||
f's#archisolabel={ISO_LABEL}#archisodevice=/dev/disk/by-uuid/{uuids[0]}#',
|
||||
*configs,
|
||||
]
|
||||
run_program(cmd)
|
||||
|
|
@ -680,9 +614,9 @@ def update_boot_entries(ufd_dev, images=None) -> None:
|
|||
break
|
||||
|
||||
|
||||
def verify_sources(args, ufd_sources) -> dict[str, pathlib.Path]:
|
||||
def verify_sources(args, ufd_sources):
|
||||
"""Check all sources and abort if necessary, returns dict."""
|
||||
sources = {}
|
||||
sources = OrderedDict()
|
||||
|
||||
for label, data in ufd_sources.items():
|
||||
s_path = args[data['Arg']]
|
||||
|
|
@ -690,41 +624,40 @@ def verify_sources(args, ufd_sources) -> dict[str, pathlib.Path]:
|
|||
try:
|
||||
s_path_obj = io.case_insensitive_path(s_path)
|
||||
except FileNotFoundError:
|
||||
ui.print_error(f'ERROR: {label} not found: {s_path}')
|
||||
ui.abort()
|
||||
else:
|
||||
if not is_valid_path(s_path_obj, data['Type']):
|
||||
ui.print_error(f'ERROR: Invalid {label} source: {s_path}')
|
||||
ui.abort()
|
||||
sources[label] = s_path_obj
|
||||
std.print_error(f'ERROR: {label} not found: {s_path}')
|
||||
std.abort()
|
||||
if not is_valid_path(s_path_obj, data['Type']):
|
||||
std.print_error(f'ERROR: Invalid {label} source: {s_path}')
|
||||
std.abort()
|
||||
sources[label] = s_path_obj
|
||||
|
||||
return sources
|
||||
|
||||
|
||||
def verify_ufd(dev_path) -> pathlib.Path:
|
||||
def verify_ufd(dev_path):
|
||||
"""Check that dev_path is a valid UFD, returns pathlib.Path obj."""
|
||||
ufd_dev = None
|
||||
|
||||
try:
|
||||
ufd_dev = io.case_insensitive_path(dev_path)
|
||||
except FileNotFoundError:
|
||||
ui.print_error(f'ERROR: UFD device not found: {dev_path}')
|
||||
ui.abort()
|
||||
std.print_error(f'ERROR: UFD device not found: {dev_path}')
|
||||
std.abort()
|
||||
|
||||
if not is_valid_path(ufd_dev, 'UFD'):
|
||||
ui.print_error(f'ERROR: Invalid UFD device: {ufd_dev}')
|
||||
ui.abort()
|
||||
std.print_error(f'ERROR: Invalid UFD device: {ufd_dev}')
|
||||
std.abort()
|
||||
|
||||
return ufd_dev # type: ignore[reportGeneralTypeIssues]
|
||||
return ufd_dev
|
||||
|
||||
|
||||
def zero_device(dev_path) -> None:
|
||||
"""Zero-out first 1MB of device."""
|
||||
def zero_device(dev_path):
|
||||
"""Zero-out first 64MB of device."""
|
||||
cmd = [
|
||||
'sudo',
|
||||
'dd',
|
||||
'bs=1M',
|
||||
'count=1',
|
||||
'bs=4M',
|
||||
'count=16',
|
||||
'if=/dev/zero',
|
||||
f'of={dev_path}',
|
||||
]
|
||||
|
|
|
|||
|
|
@ -26,7 +26,7 @@ DEFAULT_LOG_NAME = cfg.main.KIT_NAME_FULL
|
|||
|
||||
|
||||
# Functions
|
||||
def enable_debug_mode() -> None:
|
||||
def enable_debug_mode():
|
||||
"""Configures logging for better debugging."""
|
||||
root_logger = logging.getLogger()
|
||||
for handler in root_logger.handlers:
|
||||
|
|
@ -39,21 +39,13 @@ def enable_debug_mode() -> None:
|
|||
|
||||
|
||||
def format_log_path(
|
||||
log_dir: pathlib.Path | str | None = None,
|
||||
log_name: str | None = None,
|
||||
append: bool = False,
|
||||
kit: bool = False,
|
||||
sub_dir: str | None = None,
|
||||
timestamp: bool = False,
|
||||
tool: bool = False,
|
||||
) -> pathlib.Path:
|
||||
log_dir=None, log_name=None, timestamp=False,
|
||||
kit=False, tool=False):
|
||||
"""Format path based on args passed, returns pathlib.Path obj."""
|
||||
log_path = pathlib.Path(
|
||||
f'{log_dir if log_dir else DEFAULT_LOG_DIR}/'
|
||||
f'{cfg.main.KIT_NAME_FULL+"/" if kit else ""}'
|
||||
f'{"Tools/" if tool else ""}'
|
||||
f'{sub_dir+"_" if sub_dir else ""}'
|
||||
f'{time.strftime("%Y-%m-%d_%H%M%S%z") if sub_dir else ""}/'
|
||||
f'{log_name if log_name else DEFAULT_LOG_NAME}'
|
||||
f'{"_" if timestamp else ""}'
|
||||
f'{time.strftime("%Y-%m-%d_%H%M%S%z") if timestamp else ""}'
|
||||
|
|
@ -62,31 +54,28 @@ def format_log_path(
|
|||
log_path = log_path.resolve()
|
||||
|
||||
# Avoid clobbering
|
||||
if not append:
|
||||
log_path = non_clobber_path(log_path)
|
||||
log_path = non_clobber_path(log_path)
|
||||
|
||||
# Done
|
||||
return log_path
|
||||
|
||||
|
||||
def get_root_logger_path() -> pathlib.Path:
|
||||
"""Get the log filepath from the root logger, returns pathlib.Path obj.
|
||||
|
||||
NOTE: This will use the first handler baseFilename it finds (if any).
|
||||
"""
|
||||
def get_root_logger_path():
|
||||
"""Get path to log file from root logger, returns pathlib.Path obj."""
|
||||
log_path = None
|
||||
root_logger = logging.getLogger()
|
||||
|
||||
# Check handlers
|
||||
# Check all handlers and use the first fileHandler found
|
||||
for handler in root_logger.handlers:
|
||||
if hasattr(handler, 'baseFilename'):
|
||||
log_file = handler.baseFilename # type: ignore[reportGeneralTypeIssues]
|
||||
return pathlib.Path(log_file).resolve()
|
||||
if isinstance(handler, logging.FileHandler):
|
||||
log_path = pathlib.Path(handler.baseFilename).resolve()
|
||||
break
|
||||
|
||||
# No log file found
|
||||
raise RuntimeError('Log path not found.')
|
||||
# Done
|
||||
return log_path
|
||||
|
||||
|
||||
def remove_empty_log(log_path: None | pathlib.Path = None) -> None:
|
||||
def remove_empty_log(log_path=None):
|
||||
"""Remove log if empty.
|
||||
|
||||
NOTE: Under Windows an empty log is 2 bytes long.
|
||||
|
|
@ -109,7 +98,7 @@ def remove_empty_log(log_path: None | pathlib.Path = None) -> None:
|
|||
log_path.unlink()
|
||||
|
||||
|
||||
def start(config: dict[str, str] | None = None) -> None:
|
||||
def start(config=None):
|
||||
"""Configure and start logging using safe defaults."""
|
||||
log_path = format_log_path(timestamp=os.name != 'nt')
|
||||
root_logger = logging.getLogger()
|
||||
|
|
@ -132,15 +121,10 @@ def start(config: dict[str, str] | None = None) -> None:
|
|||
|
||||
|
||||
def update_log_path(
|
||||
dest_dir: None | pathlib.Path | str = None,
|
||||
dest_name: None | str = None,
|
||||
append: bool = False,
|
||||
keep_history: bool = True,
|
||||
timestamp: bool = True,
|
||||
) -> None:
|
||||
dest_dir=None, dest_name=None, keep_history=True, timestamp=True):
|
||||
"""Moves current log file to new path and updates the root logger."""
|
||||
root_logger = logging.getLogger()
|
||||
new_path = format_log_path(dest_dir, dest_name, timestamp=timestamp, append=append)
|
||||
new_path = format_log_path(dest_dir, dest_name, timestamp=timestamp)
|
||||
old_handler = None
|
||||
old_path = get_root_logger_path()
|
||||
os.makedirs(new_path.parent, exist_ok=True)
|
||||
|
|
@ -163,15 +147,10 @@ def update_log_path(
|
|||
new_handler = logging.FileHandler(new_path, mode='a')
|
||||
new_handler.setFormatter(old_handler.formatter)
|
||||
|
||||
# Remove old_handler
|
||||
# Remove old_handler and log if empty
|
||||
root_logger.removeHandler(old_handler)
|
||||
old_handler.close()
|
||||
|
||||
# Delete orignal log if needed
|
||||
if keep_history:
|
||||
remove_empty_log(old_path)
|
||||
else:
|
||||
old_path.unlink()
|
||||
remove_empty_log(old_path)
|
||||
|
||||
# Add new handler
|
||||
root_logger.addHandler(new_handler)
|
||||
|
|
|
|||
|
|
@ -5,16 +5,12 @@ import os
|
|||
import pathlib
|
||||
import re
|
||||
|
||||
from subprocess import CompletedProcess
|
||||
from typing import Any
|
||||
|
||||
import psutil
|
||||
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.std import PLATFORM, GenericError
|
||||
from wk.std import PLATFORM, GenericError, show_data
|
||||
|
||||
from wk.cfg.net import BACKUP_SERVERS
|
||||
from wk.ui import cli as ui
|
||||
|
||||
|
||||
# REGEX
|
||||
|
|
@ -26,7 +22,7 @@ REGEX_VALID_IP = re.compile(
|
|||
|
||||
|
||||
# Functions
|
||||
def connected_to_private_network(raise_on_error: bool = False) -> bool:
|
||||
def connected_to_private_network(raise_on_error=False):
|
||||
"""Check if connected to a private network, returns bool.
|
||||
|
||||
This checks for a valid private IP assigned to this system.
|
||||
|
|
@ -52,10 +48,12 @@ def connected_to_private_network(raise_on_error: bool = False) -> bool:
|
|||
raise GenericError('Not connected to a network')
|
||||
|
||||
# Done
|
||||
if raise_on_error:
|
||||
connected = None
|
||||
return connected
|
||||
|
||||
|
||||
def mount_backup_shares(read_write: bool = False) -> list[str]:
|
||||
def mount_backup_shares(read_write=False):
|
||||
"""Mount backup shares using OS specific methods."""
|
||||
report = []
|
||||
for name, details in BACKUP_SERVERS.items():
|
||||
|
|
@ -98,10 +96,7 @@ def mount_backup_shares(read_write: bool = False) -> list[str]:
|
|||
return report
|
||||
|
||||
|
||||
def mount_network_share(
|
||||
details: dict[str, Any],
|
||||
mount_point: None | pathlib.Path | str = None,
|
||||
read_write: bool = False) -> CompletedProcess:
|
||||
def mount_network_share(details, mount_point=None, read_write=False):
|
||||
"""Mount network share using OS specific methods."""
|
||||
cmd = None
|
||||
address = details['Address']
|
||||
|
|
@ -132,8 +127,8 @@ def mount_network_share(
|
|||
'-t', 'cifs',
|
||||
'-o', (
|
||||
f'{"rw" if read_write else "ro"}'
|
||||
f',uid={os.getuid()}'
|
||||
f',gid={os.getgid()}'
|
||||
f',uid={os.getuid()}' # pylint: disable=no-member
|
||||
f',gid={os.getgid()}' # pylint: disable=no-member
|
||||
f',username={username}'
|
||||
f',{"password=" if password else "guest"}{password}'
|
||||
),
|
||||
|
|
@ -152,7 +147,7 @@ def mount_network_share(
|
|||
return run_program(cmd, check=False)
|
||||
|
||||
|
||||
def ping(addr: str = 'google.com') -> None:
|
||||
def ping(addr='google.com'):
|
||||
"""Attempt to ping addr."""
|
||||
cmd = (
|
||||
'ping',
|
||||
|
|
@ -163,7 +158,7 @@ def ping(addr: str = 'google.com') -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def share_is_mounted(details: dict[str, Any]) -> bool:
|
||||
def share_is_mounted(details):
|
||||
"""Check if dev/share/etc is mounted, returns bool."""
|
||||
mounted = False
|
||||
|
||||
|
|
@ -197,20 +192,18 @@ def share_is_mounted(details: dict[str, Any]) -> bool:
|
|||
return mounted
|
||||
|
||||
|
||||
def show_valid_addresses() -> None:
|
||||
def show_valid_addresses():
|
||||
"""Show all valid private IP addresses assigned to the system."""
|
||||
# TODO: Refactor to remove ui dependancy
|
||||
devs = psutil.net_if_addrs()
|
||||
for dev, families in sorted(devs.items()):
|
||||
for family in families:
|
||||
if REGEX_VALID_IP.search(family.address):
|
||||
# Valid IP found
|
||||
ui.show_data(message=dev, data=family.address)
|
||||
show_data(message=dev, data=family.address)
|
||||
|
||||
|
||||
def speedtest() -> list[str]:
|
||||
def speedtest():
|
||||
"""Run a network speedtest using speedtest-cli."""
|
||||
# TODO: Refactor to use speedtest-cli's JSON output
|
||||
cmd = ['speedtest-cli', '--simple']
|
||||
proc = run_program(cmd, check=False)
|
||||
output = [line.strip() for line in proc.stdout.splitlines() if line.strip()]
|
||||
|
|
@ -219,7 +212,7 @@ def speedtest() -> list[str]:
|
|||
return [f'{a:<10}{b:6.2f} {c}' for a, b, c in output]
|
||||
|
||||
|
||||
def unmount_backup_shares() -> list[str]:
|
||||
def unmount_backup_shares():
|
||||
"""Unmount backup shares."""
|
||||
report = []
|
||||
for name, details in BACKUP_SERVERS.items():
|
||||
|
|
@ -248,10 +241,7 @@ def unmount_backup_shares() -> list[str]:
|
|||
return report
|
||||
|
||||
|
||||
def unmount_network_share(
|
||||
details: dict[str, Any] | None = None,
|
||||
mount_point: None | pathlib.Path | str = None,
|
||||
) -> CompletedProcess:
|
||||
def unmount_network_share(details=None, mount_point=None):
|
||||
"""Unmount network share"""
|
||||
cmd = []
|
||||
|
||||
|
|
|
|||
|
|
@ -10,8 +10,7 @@ import subprocess
|
|||
from wk.cfg.hw import VOLUME_FAILURE_THRESHOLD, VOLUME_WARNING_THRESHOLD
|
||||
from wk.exe import get_json_from_command, popen_program, run_program
|
||||
from wk.log import format_log_path
|
||||
from wk.std import bytes_to_string
|
||||
from wk.ui import ansi
|
||||
from wk.std import bytes_to_string, color_string
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -20,12 +19,12 @@ UUID_CORESTORAGE = '53746f72-6167-11aa-aa11-00306543ecac'
|
|||
|
||||
|
||||
# Functions
|
||||
def build_volume_report(device_path=None) -> list[str]:
|
||||
def build_volume_report(device_path=None) -> list:
|
||||
"""Build volume report using lsblk, returns list.
|
||||
|
||||
If device_path is provided the report is limited to that device.
|
||||
"""
|
||||
def _get_volumes(dev, indent=0) -> list[dict]:
|
||||
def _get_volumes(dev, indent=0) -> list:
|
||||
"""Convert lsblk JSON tree to a flat list of items, returns list."""
|
||||
dev['name'] = f'{" "*indent}{dev["name"]}'
|
||||
volumes = [dev]
|
||||
|
|
@ -83,20 +82,20 @@ def build_volume_report(device_path=None) -> list[str]:
|
|||
vol['mountpoint'] = f'Mounted on {vol["mountpoint"]}'
|
||||
|
||||
# Name and size
|
||||
line = ansi.color_string(
|
||||
line = color_string(
|
||||
[f'{vol["name"]:<20}', f'{vol["size"]:>9}'],
|
||||
[None, 'CYAN'],
|
||||
)
|
||||
|
||||
# Mountpoint and type
|
||||
line = ansi.color_string(
|
||||
line = color_string(
|
||||
[line, f'{vol["mountpoint"]:<{m_width}}', f'{vol["fstype"]:<11}'],
|
||||
[None, None, 'BLUE'],
|
||||
)
|
||||
|
||||
# Used and free
|
||||
if any([vol['fsused'], vol['fsavail']]):
|
||||
line = ansi.color_string(
|
||||
line = color_string(
|
||||
[line, f'({vol["fsused"]:>9} used, {vol["fsavail"]:>9} free)'],
|
||||
[None, size_color],
|
||||
)
|
||||
|
|
@ -108,7 +107,7 @@ def build_volume_report(device_path=None) -> list[str]:
|
|||
return report
|
||||
|
||||
|
||||
def get_user_home(user) -> pathlib.Path:
|
||||
def get_user_home(user):
|
||||
"""Get path to user's home dir, returns pathlib.Path obj."""
|
||||
home = None
|
||||
|
||||
|
|
@ -129,7 +128,7 @@ def get_user_home(user) -> pathlib.Path:
|
|||
return pathlib.Path(home)
|
||||
|
||||
|
||||
def get_user_name() -> str:
|
||||
def get_user_name():
|
||||
"""Get real user name, returns str."""
|
||||
user = None
|
||||
|
||||
|
|
@ -146,7 +145,7 @@ def get_user_name() -> str:
|
|||
return user
|
||||
|
||||
|
||||
def make_temp_file(suffix=None) -> pathlib.Path:
|
||||
def make_temp_file(suffix=None):
|
||||
"""Make temporary file, returns pathlib.Path() obj."""
|
||||
cmd = ['mktemp']
|
||||
if suffix:
|
||||
|
|
@ -155,7 +154,7 @@ def make_temp_file(suffix=None) -> pathlib.Path:
|
|||
return pathlib.Path(proc.stdout.strip())
|
||||
|
||||
|
||||
def mount(source, mount_point=None, read_write=False) -> None:
|
||||
def mount(source, mount_point=None, read_write=False):
|
||||
"""Mount source (on mount_point if provided).
|
||||
|
||||
NOTE: If not running_as_root() then udevil will be used.
|
||||
|
|
@ -178,13 +177,14 @@ def mount(source, mount_point=None, read_write=False) -> None:
|
|||
raise RuntimeError(f'Failed to mount: {source} on {mount_point}')
|
||||
|
||||
|
||||
def mount_volumes(device_path=None, read_write=False, scan_corestorage=False) -> None:
|
||||
def mount_volumes(device_path=None, read_write=False, scan_corestorage=False):
|
||||
# pylint: disable=too-many-branches
|
||||
"""Mount all detected volumes.
|
||||
|
||||
NOTE: If device_path is specified then only volumes
|
||||
under that path will be mounted.
|
||||
"""
|
||||
def _get_volumes(dev) -> list[dict]:
|
||||
def _get_volumes(dev) -> list:
|
||||
"""Convert lsblk JSON tree to a flat list of items, returns list."""
|
||||
volumes = [dev]
|
||||
for child in dev.get('children', []):
|
||||
|
|
@ -205,13 +205,9 @@ def mount_volumes(device_path=None, read_write=False, scan_corestorage=False) ->
|
|||
cmd.append(device_path)
|
||||
json_data = get_json_from_command(cmd)
|
||||
|
||||
# Bail if json_data is empty
|
||||
if not json_data:
|
||||
return
|
||||
|
||||
# Build list of volumes
|
||||
for dev in json_data.get('blockdevices', [{}]):
|
||||
volumes.extend(_get_volumes(dev))
|
||||
for dev in _get_volumes(json_data.get('blockdevices', [{}])[0]):
|
||||
volumes.append(dev)
|
||||
if dev.get('parttype', '') == UUID_CORESTORAGE:
|
||||
containers.append(dev['name'])
|
||||
|
||||
|
|
@ -233,12 +229,12 @@ def mount_volumes(device_path=None, read_write=False, scan_corestorage=False) ->
|
|||
pass
|
||||
|
||||
|
||||
def running_as_root() -> bool:
|
||||
def running_as_root():
|
||||
"""Check if running with effective UID of 0, returns bool."""
|
||||
return os.geteuid() == 0
|
||||
|
||||
|
||||
def scan_corestorage_container(container, timeout=300) -> list[dict]:
|
||||
def scan_corestorage_container(container, timeout=300):
|
||||
"""Scan CoreStorage container for inner volumes, returns list."""
|
||||
container_path = pathlib.Path(container)
|
||||
detected_volumes = {}
|
||||
|
|
@ -285,7 +281,7 @@ def scan_corestorage_container(container, timeout=300) -> list[dict]:
|
|||
return inner_volumes
|
||||
|
||||
|
||||
def unmount(source_or_mountpoint) -> None:
|
||||
def unmount(source_or_mountpoint):
|
||||
"""Unmount source_or_mountpoint.
|
||||
|
||||
NOTE: If not running_as_root() then udevil will be used.
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ REGEX_FANS = re.compile(r'^.*\(bytes (?P<bytes>.*)\)$')
|
|||
|
||||
|
||||
# Functions
|
||||
def decode_smc_bytes(text) -> int:
|
||||
def decode_smc_bytes(text):
|
||||
"""Decode SMC bytes, returns int."""
|
||||
result = None
|
||||
|
||||
|
|
@ -32,7 +32,7 @@ def decode_smc_bytes(text) -> int:
|
|||
return result
|
||||
|
||||
|
||||
def set_fans(mode) -> None:
|
||||
def set_fans(mode):
|
||||
"""Set fans to auto or max."""
|
||||
if mode == 'auto':
|
||||
set_fans_auto()
|
||||
|
|
@ -42,14 +42,14 @@ def set_fans(mode) -> None:
|
|||
raise RuntimeError(f'Invalid fan mode: {mode}')
|
||||
|
||||
|
||||
def set_fans_auto() -> None:
|
||||
def set_fans_auto():
|
||||
"""Set fans to auto."""
|
||||
LOG.info('Setting fans to auto')
|
||||
cmd = ['sudo', 'smc', '-k', 'FS! ', '-w', '0000']
|
||||
run_program(cmd)
|
||||
|
||||
|
||||
def set_fans_max() -> None:
|
||||
def set_fans_max():
|
||||
"""Set fans to their max speeds."""
|
||||
LOG.info('Setting fans to max')
|
||||
num_fans = 0
|
||||
|
|
|
|||
|
|
@ -8,8 +8,6 @@ import pathlib
|
|||
import platform
|
||||
|
||||
from contextlib import suppress
|
||||
from typing import Any
|
||||
|
||||
import psutil
|
||||
|
||||
try:
|
||||
|
|
@ -25,16 +23,15 @@ from wk.cfg.windows_builds import (
|
|||
OUTDATED_BUILD_NUMBERS,
|
||||
WINDOWS_BUILDS,
|
||||
)
|
||||
from wk.exe import get_json_from_command, run_program, wait_for_procs
|
||||
from wk.exe import get_json_from_command, run_program
|
||||
from wk.kit.tools import find_kit_dir
|
||||
from wk.std import (
|
||||
GenericError,
|
||||
GenericWarning,
|
||||
bytes_to_string,
|
||||
color_string,
|
||||
sleep,
|
||||
)
|
||||
from wk.ui import cli as ui
|
||||
from wk.ui import ansi
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -70,27 +67,16 @@ KNOWN_HIVE_NAMES = {
|
|||
winreg.HKEY_LOCAL_MACHINE: 'HKLM',
|
||||
winreg.HKEY_USERS: 'HKU',
|
||||
}
|
||||
OS_VERSION = platform.win32_ver()[0]
|
||||
OS_VERSION = 8.1 if OS_VERSION == '8.1' else int(OS_VERSION)
|
||||
RAM_OK = 5.5 * 1024**3 # ~6 GiB assuming a bit of shared memory
|
||||
RAM_WARNING = 3.5 * 1024**3 # ~4 GiB assuming a bit of shared memory
|
||||
REG_MSISERVER = r'HKLM\SYSTEM\CurrentControlSet\Control\SafeBoot\Network\MSIServer'
|
||||
SLMGR = pathlib.Path(f'{os.environ.get("SYSTEMROOT")}/System32/slmgr.vbs')
|
||||
SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE')
|
||||
|
||||
# STATIC OS VARIABLES
|
||||
WIN32_VER = platform.win32_ver()
|
||||
OS_BUILD_VERSION_FULL = WIN32_VER[1]
|
||||
OS_BUILD_VERSION = int(OS_BUILD_VERSION_FULL.split('.')[2])
|
||||
PLATFORM_VER = WIN32_VER[0]
|
||||
if PLATFORM_VER == '8.1':
|
||||
OS_VERSION = 8.1
|
||||
elif OS_BUILD_VERSION >= 22000:
|
||||
OS_VERSION = 11
|
||||
else:
|
||||
OS_VERSION = int(PLATFORM_VER)
|
||||
|
||||
|
||||
# Activation Functions
|
||||
def activate_with_bios() -> None:
|
||||
def activate_with_bios():
|
||||
"""Attempt to activate Windows with a key stored in the BIOS."""
|
||||
# Code borrowed from https://github.com/aeruder/get_win8key
|
||||
#####################################################
|
||||
|
|
@ -130,7 +116,7 @@ def activate_with_bios() -> None:
|
|||
raise GenericError('Activation Failed')
|
||||
|
||||
|
||||
def get_activation_string() -> str:
|
||||
def get_activation_string():
|
||||
"""Get activation status, returns str."""
|
||||
cmd = ['cscript', '//nologo', SLMGR, '/xpr']
|
||||
proc = run_program(cmd, check=False)
|
||||
|
|
@ -140,7 +126,7 @@ def get_activation_string() -> str:
|
|||
return act_str
|
||||
|
||||
|
||||
def is_activated() -> bool:
|
||||
def is_activated():
|
||||
"""Check if Windows is activated via slmgr.vbs and return bool."""
|
||||
act_str = get_activation_string()
|
||||
|
||||
|
|
@ -149,122 +135,80 @@ def is_activated() -> bool:
|
|||
|
||||
|
||||
# Date / Time functions
|
||||
def get_timezone() -> str:
|
||||
def get_timezone():
|
||||
"""Get current timezone using tzutil, returns str."""
|
||||
cmd = ['tzutil', '/g']
|
||||
proc = run_program(cmd, check=False)
|
||||
return proc.stdout
|
||||
|
||||
|
||||
def set_timezone(zone) -> None:
|
||||
def set_timezone(zone):
|
||||
"""Set current timezone using tzutil."""
|
||||
cmd = ['tzutil', '/s', zone]
|
||||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
# Info Functions
|
||||
def check_4k_alignment(show_alert=False) -> list[str]:
|
||||
"""Check if all partitions are 4K aligned, returns list."""
|
||||
script_path = find_kit_dir('Scripts').joinpath('check_partition_alignment.ps1')
|
||||
cmd = ['PowerShell', '-ExecutionPolicy', 'Bypass', '-File', script_path]
|
||||
json_data = get_json_from_command(cmd)
|
||||
report = []
|
||||
show_alert = False
|
||||
def check_4k_alignment(show_alert=False):
|
||||
"""Check if all partitions are 4K aligned, returns book."""
|
||||
cmd = ['WMIC', 'partition', 'get', 'StartingOffset']
|
||||
|
||||
# Check offsets
|
||||
for part in json_data:
|
||||
if part['StartingOffset'] % 4096 != 0:
|
||||
report.append(
|
||||
ansi.color_string(
|
||||
f'{part["Name"]}'
|
||||
f' ({bytes_to_string(part["Size"], decimals=1)})'
|
||||
,
|
||||
'RED'
|
||||
)
|
||||
)
|
||||
proc = run_program(cmd)
|
||||
for offset in proc.stdout.splitlines():
|
||||
offset = offset.strip()
|
||||
if not offset.isnumeric():
|
||||
continue
|
||||
if int(offset) % 4096 != 0:
|
||||
# Not aligned
|
||||
if show_alert:
|
||||
show_alert_box('One or more partitions are not 4K aligned')
|
||||
raise GenericError('One or more partitions are not 4K aligned')
|
||||
|
||||
# Show alert
|
||||
if show_alert:
|
||||
show_alert_box('One or more partitions not 4K aligned')
|
||||
|
||||
def get_installed_antivirus():
|
||||
"""Get list of installed antivirus programs, returns list."""
|
||||
cmd = [
|
||||
'WMIC', r'/namespace:\\root\SecurityCenter2',
|
||||
'path', 'AntivirusProduct',
|
||||
'get', 'displayName', '/value',
|
||||
]
|
||||
products = []
|
||||
report = []
|
||||
|
||||
# Get list of products
|
||||
proc = run_program(cmd)
|
||||
for line in proc.stdout.splitlines():
|
||||
line = line.strip()
|
||||
if '=' in line:
|
||||
products.append(line.split('=')[1])
|
||||
|
||||
# Check product(s) status
|
||||
for product in sorted(products):
|
||||
cmd = [
|
||||
'WMIC', r'/namespace:\\root\SecurityCenter2',
|
||||
'path', 'AntivirusProduct',
|
||||
'where', f'displayName="{product}"',
|
||||
'get', 'productState', '/value',
|
||||
]
|
||||
proc = run_program(cmd)
|
||||
state = proc.stdout.split('=')[1]
|
||||
state = hex(int(state))
|
||||
if str(state)[3:5] not in ['10', '11']:
|
||||
report.append(color_string(f'[Disabled] {product}', 'YELLOW'))
|
||||
else:
|
||||
report.append(product)
|
||||
|
||||
# Final check
|
||||
if not report:
|
||||
report.append(color_string('No products detected', 'RED'))
|
||||
|
||||
# Done
|
||||
if report:
|
||||
report.insert(
|
||||
0,
|
||||
ansi.color_string('One or more partitions not 4K aligned', 'YELLOW'),
|
||||
)
|
||||
report.sort()
|
||||
return report
|
||||
|
||||
|
||||
def export_bitlocker_info() -> None:
|
||||
"""Get Bitlocker info and save to the base directory of the kit."""
|
||||
commands = [
|
||||
['manage-bde', '-status', SYSTEMDRIVE],
|
||||
['manage-bde', '-protectors', '-get', SYSTEMDRIVE],
|
||||
]
|
||||
|
||||
# Get filename
|
||||
file_name = ui.input_text(prompt_msg='Enter filename')
|
||||
file_path = pathlib.Path(f'../../Bitlocker_{file_name}.txt').resolve()
|
||||
|
||||
# Save info
|
||||
with open(file_path, 'a', encoding='utf-8') as _f:
|
||||
for cmd in commands:
|
||||
proc = run_program(cmd, check=False)
|
||||
_f.write(f'{proc.stdout}\n\n')
|
||||
|
||||
|
||||
def get_installed_antivirus() -> dict[str, dict]:
|
||||
"""Get installed antivirus products and their status, returns dict."""
|
||||
script_path = find_kit_dir('Scripts').joinpath('check_av.ps1')
|
||||
cmd = ['PowerShell', '-ExecutionPolicy', 'Bypass', '-File', script_path]
|
||||
json_data = get_json_from_command(cmd)
|
||||
products = {}
|
||||
|
||||
# Check state and build dict
|
||||
for p in json_data:
|
||||
name = p['displayName']
|
||||
state = p['productState']
|
||||
enabled = ((state>>8) & 0x11) in (0x10, 0x11) # middle two hex digits
|
||||
outdated = (state & 0x11) != 0x00 # last two hex digits
|
||||
products[name] = {
|
||||
'Enabled': enabled,
|
||||
'Outdated': outdated,
|
||||
'State': state,
|
||||
}
|
||||
return products
|
||||
|
||||
|
||||
def list_installed_antivirus() -> list[str]:
|
||||
"""Get list of installed antivirus programs, returns list."""
|
||||
products = get_installed_antivirus()
|
||||
products_active = []
|
||||
products_inactive = []
|
||||
|
||||
# Check product(s) status
|
||||
for name, details in products.items():
|
||||
if details['Enabled']:
|
||||
if details['Outdated']:
|
||||
products_active.append(ansi.color_string(f'{name} [OUTDATED]', 'YELLOW'))
|
||||
else:
|
||||
products_active.append(name)
|
||||
else:
|
||||
# Disabled
|
||||
products_inactive.append(ansi.color_string(f'[Disabled] {name}', 'YELLOW'))
|
||||
|
||||
# Final check
|
||||
if not (products_active or products_inactive):
|
||||
products_inactive.append(ansi.color_string('No products detected', 'RED'))
|
||||
|
||||
# Done
|
||||
products_active.sort()
|
||||
products_inactive.sort()
|
||||
return products_active + products_inactive
|
||||
|
||||
|
||||
def get_installed_ram(as_list=False, raise_exceptions=False) -> list | str:
|
||||
"""Get installed RAM, returns list or str."""
|
||||
def get_installed_ram(as_list=False, raise_exceptions=False):
|
||||
"""Get installed RAM."""
|
||||
mem = psutil.virtual_memory()
|
||||
mem_str = bytes_to_string(mem.total, decimals=1)
|
||||
|
||||
|
|
@ -279,8 +223,8 @@ def get_installed_ram(as_list=False, raise_exceptions=False) -> list | str:
|
|||
return [mem_str] if as_list else mem_str
|
||||
|
||||
|
||||
def get_os_activation(as_list=False, check=True) -> list | str:
|
||||
"""Get OS activation status, returns list or str.
|
||||
def get_os_activation(as_list=False, check=True):
|
||||
"""Get OS activation status, returns str.
|
||||
|
||||
NOTE: If check=True then raise an exception if OS isn't activated.
|
||||
"""
|
||||
|
|
@ -296,33 +240,33 @@ def get_os_activation(as_list=False, check=True) -> list | str:
|
|||
return [act_str] if as_list else act_str
|
||||
|
||||
|
||||
def get_os_name(as_list=False, check=True) -> str:
|
||||
def get_os_name(as_list=False, check=True):
|
||||
"""Build OS display name, returns str.
|
||||
|
||||
NOTE: If check=True then an exception is raised if the OS version is
|
||||
outdated or unsupported.
|
||||
"""
|
||||
key = r'SOFTWARE\Microsoft\Windows NT\CurrentVersion'
|
||||
details = WINDOWS_BUILDS.get(OS_BUILD_VERSION_FULL, f'Build {OS_BUILD_VERSION}')
|
||||
build_version = int(reg_read_value("HKLM", key, "CurrentBuild"))
|
||||
build_version_full = platform.win32_ver()[1]
|
||||
details = WINDOWS_BUILDS.get(build_version_full, f'Build {build_version}')
|
||||
display_name = (
|
||||
f'{reg_read_value("HKLM", key, "ProductName")} {ARCH}-bit {details}'
|
||||
)
|
||||
if OS_BUILD_VERSION >= 22000:
|
||||
display_name = display_name.replace('Windows 10', 'Windows 11')
|
||||
|
||||
# Check for support issues
|
||||
if check:
|
||||
if OS_BUILD_VERSION in OUTDATED_BUILD_NUMBERS:
|
||||
if build_version in OUTDATED_BUILD_NUMBERS:
|
||||
raise GenericWarning(f'{display_name} (outdated)')
|
||||
|
||||
if OS_BUILD_VERSION < OLDEST_SUPPORTED_BUILD:
|
||||
if build_version < OLDEST_SUPPORTED_BUILD:
|
||||
raise GenericError(f'{display_name} (unsupported)')
|
||||
|
||||
# Done
|
||||
return [display_name] if as_list else display_name
|
||||
|
||||
|
||||
def get_raw_disks() -> list[str]:
|
||||
def get_raw_disks():
|
||||
"""Get all disks without a partiton table, returns list."""
|
||||
script_path = find_kit_dir('Scripts').joinpath('get_raw_disks.ps1')
|
||||
cmd = ['PowerShell', '-ExecutionPolicy', 'Bypass', '-File', script_path]
|
||||
|
|
@ -347,7 +291,7 @@ def get_raw_disks() -> list[str]:
|
|||
return raw_disks
|
||||
|
||||
|
||||
def get_volume_usage(use_colors=False) -> list[str]:
|
||||
def get_volume_usage(use_colors=False):
|
||||
"""Get space usage info for all fixed volumes, returns list."""
|
||||
report = []
|
||||
for disk in psutil.disk_partitions():
|
||||
|
|
@ -364,14 +308,14 @@ def get_volume_usage(use_colors=False) -> list[str]:
|
|||
f' ({bytes_to_string(free, 2):>10} / {bytes_to_string(total, 2):>10})'
|
||||
)
|
||||
if use_colors:
|
||||
display_str = ansi.color_string(display_str, color)
|
||||
display_str = color_string(display_str, color)
|
||||
report.append(f'{disk.device} {display_str}')
|
||||
|
||||
# Done
|
||||
return report
|
||||
|
||||
|
||||
def show_alert_box(message, title=None) -> None:
|
||||
def show_alert_box(message, title=None):
|
||||
"""Show Windows alert box with message."""
|
||||
title = title if title else f'{KIT_NAME_FULL} Warning'
|
||||
message_box = ctypes.windll.user32.MessageBoxW
|
||||
|
|
@ -379,7 +323,8 @@ def show_alert_box(message, title=None) -> None:
|
|||
|
||||
|
||||
# Registry Functions
|
||||
def reg_delete_key(hive, key, recurse=False) -> None:
|
||||
def reg_delete_key(hive, key, recurse=False):
|
||||
# pylint: disable=raise-missing-from
|
||||
"""Delete a key from the registry.
|
||||
|
||||
NOTE: If recurse is False then it will only work on empty keys.
|
||||
|
|
@ -401,7 +346,7 @@ def reg_delete_key(hive, key, recurse=False) -> None:
|
|||
except FileNotFoundError:
|
||||
# Ignore
|
||||
pass
|
||||
except PermissionError as _e:
|
||||
except PermissionError:
|
||||
LOG.error(r'Failed to delete registry key: %s\%s', hive_name, key)
|
||||
if recurse:
|
||||
# Re-raise exception
|
||||
|
|
@ -409,10 +354,10 @@ def reg_delete_key(hive, key, recurse=False) -> None:
|
|||
|
||||
# recurse is not True so assuming we tried to remove a non-empty key
|
||||
msg = fr'Refusing to remove non-empty key: {hive_name}\{key}'
|
||||
raise FileExistsError(msg) from _e
|
||||
raise FileExistsError(msg)
|
||||
|
||||
|
||||
def reg_delete_value(hive, key, value) -> None:
|
||||
def reg_delete_value(hive, key, value):
|
||||
"""Delete a value from the registry."""
|
||||
access = winreg.KEY_ALL_ACCESS
|
||||
hive = reg_get_hive(hive)
|
||||
|
|
@ -436,9 +381,8 @@ def reg_delete_value(hive, key, value) -> None:
|
|||
raise
|
||||
|
||||
|
||||
def reg_get_hive(hive) -> Any:
|
||||
def reg_get_hive(hive):
|
||||
"""Get winreg HKEY constant from string, returns HKEY constant."""
|
||||
# TODO: Fix type hint
|
||||
if isinstance(hive, int):
|
||||
# Assuming we're already a winreg HKEY constant
|
||||
pass
|
||||
|
|
@ -449,9 +393,8 @@ def reg_get_hive(hive) -> Any:
|
|||
return hive
|
||||
|
||||
|
||||
def reg_get_data_type(data_type) -> Any:
|
||||
def reg_get_data_type(data_type):
|
||||
"""Get registry data type from string, returns winreg constant."""
|
||||
# TODO: Fix type hint
|
||||
if isinstance(data_type, int):
|
||||
# Assuming we're already a winreg value type constant
|
||||
pass
|
||||
|
|
@ -462,7 +405,7 @@ def reg_get_data_type(data_type) -> Any:
|
|||
return data_type
|
||||
|
||||
|
||||
def reg_key_exists(hive, key) -> bool:
|
||||
def reg_key_exists(hive, key):
|
||||
"""Test if the specified hive/key exists, returns bool."""
|
||||
exists = False
|
||||
hive = reg_get_hive(hive)
|
||||
|
|
@ -480,7 +423,7 @@ def reg_key_exists(hive, key) -> bool:
|
|||
return exists
|
||||
|
||||
|
||||
def reg_read_value(hive, key, value, force_32=False, force_64=False) -> Any:
|
||||
def reg_read_value(hive, key, value, force_32=False, force_64=False):
|
||||
"""Query value from hive/hey, returns multiple types.
|
||||
|
||||
NOTE: Set value='' to read the default value.
|
||||
|
|
@ -504,7 +447,7 @@ def reg_read_value(hive, key, value, force_32=False, force_64=False) -> Any:
|
|||
return data
|
||||
|
||||
|
||||
def reg_write_settings(settings) -> None:
|
||||
def reg_write_settings(settings):
|
||||
"""Set registry values in bulk from a custom data structure.
|
||||
|
||||
Data structure should be as follows:
|
||||
|
|
@ -544,7 +487,8 @@ def reg_write_settings(settings) -> None:
|
|||
reg_set_value(hive, key, *value)
|
||||
|
||||
|
||||
def reg_set_value(hive, key, name, data, data_type, option=None) -> None:
|
||||
def reg_set_value(hive, key, name, data, data_type, option=None):
|
||||
# pylint: disable=too-many-arguments
|
||||
"""Set value for hive/key."""
|
||||
access = winreg.KEY_WRITE
|
||||
data_type = reg_get_data_type(data_type)
|
||||
|
|
@ -576,25 +520,25 @@ def reg_set_value(hive, key, name, data, data_type, option=None) -> None:
|
|||
|
||||
|
||||
# Safe Mode Functions
|
||||
def disable_safemode() -> None:
|
||||
def disable_safemode():
|
||||
"""Edit BCD to remove safeboot value."""
|
||||
cmd = ['bcdedit', '/deletevalue', '{default}', 'safeboot']
|
||||
run_program(cmd)
|
||||
|
||||
|
||||
def disable_safemode_msi() -> None:
|
||||
def disable_safemode_msi():
|
||||
"""Disable MSI access under safemode."""
|
||||
cmd = ['reg', 'delete', REG_MSISERVER, '/f']
|
||||
run_program(cmd)
|
||||
|
||||
|
||||
def enable_safemode() -> None:
|
||||
def enable_safemode():
|
||||
"""Edit BCD to set safeboot as default."""
|
||||
cmd = ['bcdedit', '/set', '{default}', 'safeboot', 'network']
|
||||
run_program(cmd)
|
||||
|
||||
|
||||
def enable_safemode_msi() -> None:
|
||||
def enable_safemode_msi():
|
||||
"""Enable MSI access under safemode."""
|
||||
cmd = ['reg', 'add', REG_MSISERVER, '/f']
|
||||
run_program(cmd)
|
||||
|
|
@ -607,7 +551,7 @@ def enable_safemode_msi() -> None:
|
|||
|
||||
|
||||
# Secure Boot Functions
|
||||
def is_booted_uefi() -> bool:
|
||||
def is_booted_uefi():
|
||||
"""Check if booted UEFI or legacy, returns bool."""
|
||||
kernel = ctypes.windll.kernel32
|
||||
firmware_type = ctypes.c_uint()
|
||||
|
|
@ -615,7 +559,7 @@ def is_booted_uefi() -> bool:
|
|||
# Get value from kernel32 API (firmware_type is updated by the call)
|
||||
try:
|
||||
kernel.GetFirmwareType(ctypes.byref(firmware_type))
|
||||
except Exception:
|
||||
except Exception: # pylint: disable=broad-except
|
||||
# Ignore and set firmware_type back to zero
|
||||
firmware_type = ctypes.c_uint(0)
|
||||
|
||||
|
|
@ -623,7 +567,7 @@ def is_booted_uefi() -> bool:
|
|||
return firmware_type.value == 2
|
||||
|
||||
|
||||
def is_secure_boot_enabled(raise_exceptions=False, show_alert=False) -> bool:
|
||||
def is_secure_boot_enabled(raise_exceptions=False, show_alert=False):
|
||||
"""Check if Secure Boot is enabled, returns bool.
|
||||
|
||||
If raise_exceptions is True then an exception is raised with details.
|
||||
|
|
@ -673,7 +617,7 @@ def is_secure_boot_enabled(raise_exceptions=False, show_alert=False) -> bool:
|
|||
|
||||
|
||||
# Service Functions
|
||||
def disable_service(service_name) -> None:
|
||||
def disable_service(service_name):
|
||||
"""Set service startup to disabled."""
|
||||
cmd = ['sc', 'config', service_name, 'start=', 'disabled']
|
||||
run_program(cmd, check=False)
|
||||
|
|
@ -683,7 +627,7 @@ def disable_service(service_name) -> None:
|
|||
raise GenericError(f'Failed to disable service {service_name}')
|
||||
|
||||
|
||||
def enable_service(service_name, start_type='auto') -> None:
|
||||
def enable_service(service_name, start_type='auto'):
|
||||
"""Enable service by setting start type."""
|
||||
cmd = ['sc', 'config', service_name, 'start=', start_type]
|
||||
psutil_type = 'automatic'
|
||||
|
|
@ -698,7 +642,7 @@ def enable_service(service_name, start_type='auto') -> None:
|
|||
raise GenericError(f'Failed to enable service {service_name}')
|
||||
|
||||
|
||||
def get_service_status(service_name) -> str:
|
||||
def get_service_status(service_name):
|
||||
"""Get service status using psutil, returns str."""
|
||||
status = 'unknown'
|
||||
try:
|
||||
|
|
@ -710,7 +654,7 @@ def get_service_status(service_name) -> str:
|
|||
return status
|
||||
|
||||
|
||||
def get_service_start_type(service_name) -> str:
|
||||
def get_service_start_type(service_name):
|
||||
"""Get service startup type using psutil, returns str."""
|
||||
start_type = 'unknown'
|
||||
try:
|
||||
|
|
@ -722,17 +666,17 @@ def get_service_start_type(service_name) -> str:
|
|||
return start_type
|
||||
|
||||
|
||||
def start_service(service_name) -> None:
|
||||
def start_service(service_name):
|
||||
"""Stop service."""
|
||||
cmd = ['net', 'start', service_name]
|
||||
run_program(cmd, check=False)
|
||||
|
||||
# Verify service was started
|
||||
if get_service_status(service_name) not in ('running', 'start_pending'):
|
||||
if not get_service_status(service_name) in ('running', 'start_pending'):
|
||||
raise GenericError(f'Failed to start service {service_name}')
|
||||
|
||||
|
||||
def stop_service(service_name) -> None:
|
||||
def stop_service(service_name):
|
||||
"""Stop service."""
|
||||
cmd = ['net', 'stop', service_name]
|
||||
run_program(cmd, check=False)
|
||||
|
|
@ -742,62 +686,5 @@ def stop_service(service_name) -> None:
|
|||
raise GenericError(f'Failed to stop service {service_name}')
|
||||
|
||||
|
||||
# Winget Functions
|
||||
def winget_check(raise_exceptions: bool = False) -> None:
|
||||
"""Check if winget is present, install if not."""
|
||||
cmd = [
|
||||
'powershell',
|
||||
'-ExecutionPolicy', 'bypass',
|
||||
'-File', find_kit_dir('Scripts').joinpath('install_winget.ps1'),
|
||||
]
|
||||
proc = run_program(cmd, check=False)
|
||||
|
||||
# Raise exception if requested
|
||||
if raise_exceptions:
|
||||
if proc.returncode == 1:
|
||||
raise GenericWarning('Already installed')
|
||||
if proc.returncode == 2:
|
||||
raise GenericError('Failed to install')
|
||||
|
||||
|
||||
def winget_import(group_name: str = 'default') -> None:
|
||||
"""Use winget to import a set of applications.
|
||||
|
||||
group_name should be the name of a JSON file exported from winget.
|
||||
|
||||
NOTE: The path is relative to .bin/Scripts/wk/cfg/winget/
|
||||
"""
|
||||
cmd = [
|
||||
'winget',
|
||||
'import', '--import-file',
|
||||
str(find_kit_dir('Scripts').joinpath(f'wk/cfg/winget/{group_name}.json')),
|
||||
]
|
||||
tmp_file = fr'{os.environ.get("TMP")}\run_winget.cmd'
|
||||
if CONEMU:
|
||||
with open(tmp_file, 'w', encoding='utf-8') as _f:
|
||||
_f.write('@echo off\n')
|
||||
_f.write(" ".join(cmd))
|
||||
cmd = ('cmd', '/c', tmp_file, '-new_console:n', '-new_console:s33V')
|
||||
run_program(cmd, check=False, pipe=False)
|
||||
sleep(1)
|
||||
wait_for_procs('winget.exe')
|
||||
|
||||
|
||||
def winget_upgrade() -> None:
|
||||
"""Upgrade all supported programs with winget, returns subprocess.Popen."""
|
||||
cmd = ['winget', 'upgrade', '--all']
|
||||
|
||||
# Adjust if running inside ConEmu
|
||||
tmp_file = fr'{os.environ.get("TMP")}\run_winget.cmd'
|
||||
if CONEMU:
|
||||
with open(tmp_file, 'w', encoding='utf-8') as _f:
|
||||
_f.write('@echo off\n')
|
||||
_f.write(" ".join(cmd))
|
||||
cmd = ('cmd', '/c', tmp_file, '-new_console:n', '-new_console:s33V')
|
||||
run_program(cmd, check=False, pipe=False)
|
||||
sleep(1)
|
||||
wait_for_procs('winget.exe')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load diff
|
|
@ -1,4 +1,5 @@
|
|||
"""WizardKit: Setup - Windows"""
|
||||
# pylint: disable=too-many-lines
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import configparser
|
||||
|
|
@ -8,24 +9,21 @@ import os
|
|||
import re
|
||||
import sys
|
||||
|
||||
from typing import Any
|
||||
|
||||
from wk.cfg.main import KIT_NAME_FULL
|
||||
from wk.cfg.setup import (
|
||||
BROWSER_PATHS,
|
||||
DISABLED_ENTRIES_WINDOWS_11,
|
||||
LIBREOFFICE_XCU_DATA,
|
||||
REG_CHROME_UBLOCK_ORIGIN,
|
||||
REG_WINDOWS_EXPLORER,
|
||||
REG_OPEN_SHELL_SETTINGS,
|
||||
REG_OPEN_SHELL_LOW_POWER_IDLE,
|
||||
REG_WINDOWS_BSOD_MINIDUMPS,
|
||||
UBLOCK_ORIGIN_URLS,
|
||||
)
|
||||
from wk.exe import kill_procs, run_program, popen_program
|
||||
from wk.io import case_insensitive_path, get_path_obj
|
||||
from wk.kit.tools import (
|
||||
ARCH,
|
||||
download_tool,
|
||||
extract_archive,
|
||||
extract_tool,
|
||||
find_kit_dir,
|
||||
|
|
@ -37,21 +35,16 @@ from wk.os.win import (
|
|||
OS_VERSION,
|
||||
activate_with_bios,
|
||||
check_4k_alignment,
|
||||
get_installed_antivirus,
|
||||
get_installed_ram,
|
||||
get_os_activation,
|
||||
get_os_name,
|
||||
get_raw_disks,
|
||||
get_service_status,
|
||||
get_volume_usage,
|
||||
is_activated,
|
||||
is_secure_boot_enabled,
|
||||
list_installed_antivirus,
|
||||
reg_set_value,
|
||||
reg_write_settings,
|
||||
stop_service,
|
||||
winget_check,
|
||||
winget_import,
|
||||
winget_upgrade,
|
||||
)
|
||||
from wk.repairs.win import (
|
||||
WIDTH,
|
||||
|
|
@ -67,10 +60,22 @@ from wk.repairs.win import (
|
|||
from wk.std import (
|
||||
GenericError,
|
||||
GenericWarning,
|
||||
Menu,
|
||||
TryAndPrint,
|
||||
abort,
|
||||
ask,
|
||||
clear_screen,
|
||||
color_string,
|
||||
pause,
|
||||
print_error,
|
||||
print_info,
|
||||
print_standard,
|
||||
print_warning,
|
||||
set_title,
|
||||
show_data,
|
||||
sleep,
|
||||
strip_colors,
|
||||
)
|
||||
from wk.ui import cli as ui
|
||||
from wk.ui import ansi
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
|
|
@ -86,7 +91,7 @@ KNOWN_ENCODINGS = (
|
|||
'utf-32-le',
|
||||
)
|
||||
IN_CONEMU = 'ConEmuPID' in os.environ
|
||||
MENU_PRESETS = ui.Menu()
|
||||
MENU_PRESETS = Menu()
|
||||
PROGRAMFILES_32 = os.environ.get(
|
||||
'PROGRAMFILES(X86)', os.environ.get(
|
||||
'PROGRAMFILES', r'C:\Program Files (x86)',
|
||||
|
|
@ -98,7 +103,7 @@ PROGRAMFILES_64 = os.environ.get(
|
|||
),
|
||||
)
|
||||
SYSTEMDRIVE = os.environ.get('SYSTEMDRIVE', 'C:')
|
||||
TRY_PRINT = ui.TryAndPrint()
|
||||
TRY_PRINT = TryAndPrint()
|
||||
TRY_PRINT.width = WIDTH
|
||||
TRY_PRINT.verbose = True
|
||||
for error in ('CalledProcessError', 'FileNotFoundError'):
|
||||
|
|
@ -106,10 +111,10 @@ for error in ('CalledProcessError', 'FileNotFoundError'):
|
|||
|
||||
|
||||
# Auto Setup
|
||||
def build_menus(base_menus, title, presets) -> dict[str, ui.Menu]:
|
||||
def build_menus(base_menus, title, presets):
|
||||
"""Build menus, returns dict."""
|
||||
menus = {}
|
||||
menus['Main'] = ui.Menu(title=f'{title}\n{ansi.color_string("Main Menu", "GREEN")}')
|
||||
menus['Main'] = Menu(title=f'{title}\n{color_string("Main Menu", "GREEN")}')
|
||||
|
||||
# Main Menu
|
||||
for entry in base_menus['Actions']:
|
||||
|
|
@ -119,7 +124,7 @@ def build_menus(base_menus, title, presets) -> dict[str, ui.Menu]:
|
|||
|
||||
# Run groups
|
||||
for group, entries in base_menus['Groups'].items():
|
||||
menus[group] = ui.Menu(title=f'{title}\n{ansi.color_string(group, "GREEN")}')
|
||||
menus[group] = Menu(title=f'{title}\n{color_string(group, "GREEN")}')
|
||||
for entry in entries:
|
||||
menus[group].add_option(entry.name, entry.details)
|
||||
menus[group].add_action('All')
|
||||
|
|
@ -148,7 +153,7 @@ def build_menus(base_menus, title, presets) -> dict[str, ui.Menu]:
|
|||
)
|
||||
|
||||
# Update presets Menu
|
||||
MENU_PRESETS.title = f'{title}\n{ansi.color_string("Load Preset", "GREEN")}'
|
||||
MENU_PRESETS.title = f'{title}\n{color_string("Load Preset", "GREEN")}'
|
||||
MENU_PRESETS.add_option('Default')
|
||||
for name in presets:
|
||||
MENU_PRESETS.add_option(name)
|
||||
|
|
@ -161,33 +166,33 @@ def build_menus(base_menus, title, presets) -> dict[str, ui.Menu]:
|
|||
return menus
|
||||
|
||||
|
||||
def check_os_and_set_menu_title(title) -> str:
|
||||
def check_os_and_set_menu_title(title):
|
||||
"""Check OS version and update title for menus, returns str."""
|
||||
color = None
|
||||
os_name = get_os_name(check=False)
|
||||
ui.print_standard(f'Operating System: {os_name}')
|
||||
print_standard(f'Operating System: {os_name}')
|
||||
|
||||
# Check support status and set color
|
||||
try:
|
||||
get_os_name()
|
||||
except GenericWarning:
|
||||
# Outdated version
|
||||
ui.print_warning('OS version is outdated, updating is recommended.')
|
||||
if not ui.ask('Continue anyway?'):
|
||||
ui.abort()
|
||||
print_warning('OS version is outdated, updating is recommended.')
|
||||
if not ask('Continue anyway?'):
|
||||
abort()
|
||||
color = 'YELLOW'
|
||||
except GenericError:
|
||||
# Unsupported version
|
||||
ui.print_error('OS version is unsupported, updating is recommended.')
|
||||
if not ui.ask('Continue anyway? (NOT RECOMMENDED)'):
|
||||
ui.abort()
|
||||
print_error('OS version is unsupported, updating is recommended.')
|
||||
if not ask('Continue anyway? (NOT RECOMMENDED)'):
|
||||
abort()
|
||||
color = 'RED'
|
||||
|
||||
# Done
|
||||
return f'{title} ({ansi.color_string(os_name, color)})'
|
||||
return f'{title} ({color_string(os_name, color)})'
|
||||
|
||||
|
||||
def load_preset(menus, presets, title, enable_menu_exit=True) -> None:
|
||||
def load_preset(menus, presets, title, enable_menu_exit=True):
|
||||
"""Load menu settings from preset and ask selection question(s)."""
|
||||
if not enable_menu_exit:
|
||||
MENU_PRESETS.actions['Main Menu'].update({'Disabled':True, 'Hidden':True})
|
||||
|
|
@ -210,31 +215,25 @@ def load_preset(menus, presets, title, enable_menu_exit=True) -> None:
|
|||
menu.options[name]['Selected'] = value
|
||||
|
||||
# Ask selection question(s)
|
||||
ui.clear_screen()
|
||||
ui.print_standard(f'{title}')
|
||||
clear_screen()
|
||||
print_standard(f'{title}')
|
||||
print('')
|
||||
if selection[0] == 'Default' and ui.ask('Install LibreOffice?'):
|
||||
if selection[0] == 'Default' and ask('Install LibreOffice?'):
|
||||
menus['Install Software'].options['LibreOffice']['Selected'] = True
|
||||
|
||||
# Re-enable Main Menu action if disabled
|
||||
MENU_PRESETS.actions['Main Menu'].update({'Disabled':False, 'Hidden':False})
|
||||
|
||||
# Disable entries incompatible with Windows 11
|
||||
if OS_VERSION == 11:
|
||||
for group_name, entry_name in DISABLED_ENTRIES_WINDOWS_11.items():
|
||||
menus[group_name].options[entry_name]['Disabled'] = True
|
||||
menus[group_name].options[entry_name]['Selected'] = False
|
||||
|
||||
|
||||
def run_auto_setup(base_menus, presets) -> None:
|
||||
def run_auto_setup(base_menus, presets):
|
||||
"""Run Auto Setup."""
|
||||
update_log_path(dest_name='Auto Setup', timestamp=True)
|
||||
title = f'{KIT_NAME_FULL}: Auto Setup'
|
||||
ui.clear_screen()
|
||||
ui.set_title(title)
|
||||
ui.print_info(title)
|
||||
clear_screen()
|
||||
set_title(title)
|
||||
print_info(title)
|
||||
print('')
|
||||
ui.print_standard('Initializing...')
|
||||
print_standard('Initializing...')
|
||||
|
||||
# Check OS and update title for menus
|
||||
title = check_os_and_set_menu_title(title)
|
||||
|
|
@ -249,10 +248,10 @@ def run_auto_setup(base_menus, presets) -> None:
|
|||
show_main_menu(base_menus, menus, presets, title)
|
||||
|
||||
# Start setup
|
||||
ui.clear_screen()
|
||||
ui.print_standard(title)
|
||||
clear_screen()
|
||||
print_standard(title)
|
||||
print('')
|
||||
ui.print_info('Running setup')
|
||||
print_info('Running setup')
|
||||
|
||||
# Run setup
|
||||
for group, menu in menus.items():
|
||||
|
|
@ -261,29 +260,29 @@ def run_auto_setup(base_menus, presets) -> None:
|
|||
try:
|
||||
run_group(group, menu)
|
||||
except KeyboardInterrupt:
|
||||
ui.abort()
|
||||
abort()
|
||||
|
||||
# Done
|
||||
ui.print_info('Done')
|
||||
ui.pause('Press Enter to exit...')
|
||||
print_info('Done')
|
||||
pause('Press Enter to exit...')
|
||||
|
||||
|
||||
def run_group(group, menu) -> None:
|
||||
def run_group(group, menu):
|
||||
"""Run entries in group if appropriate."""
|
||||
ui.print_info(f' {group}')
|
||||
print_info(f' {group}')
|
||||
for name, details in menu.options.items():
|
||||
name_str = ansi.strip_colors(name)
|
||||
name_str = strip_colors(name)
|
||||
|
||||
# Not selected
|
||||
if not details.get('Selected', False):
|
||||
ui.show_data(f'{name_str}...', 'Skipped', 'YELLOW', width=WIDTH)
|
||||
show_data(f'{name_str}...', 'Skipped', 'YELLOW', width=WIDTH)
|
||||
continue
|
||||
|
||||
# Selected
|
||||
details['Function']()
|
||||
|
||||
|
||||
def show_main_menu(base_menus, menus, presets, title) -> None:
|
||||
def show_main_menu(base_menus, menus, presets, title):
|
||||
"""Show main menu and handle actions."""
|
||||
while True:
|
||||
update_main_menu(menus)
|
||||
|
|
@ -298,7 +297,7 @@ def show_main_menu(base_menus, menus, presets, title) -> None:
|
|||
raise SystemExit
|
||||
|
||||
|
||||
def show_sub_menu(menu) -> None:
|
||||
def show_sub_menu(menu):
|
||||
"""Show sub-menu and handle sub-menu actions."""
|
||||
while True:
|
||||
selection = menu.advanced_select()
|
||||
|
|
@ -314,7 +313,7 @@ def show_sub_menu(menu) -> None:
|
|||
menu.options[name]['Selected'] = value
|
||||
|
||||
|
||||
def update_main_menu(menus) -> None:
|
||||
def update_main_menu(menus):
|
||||
"""Update main menu based on current selections."""
|
||||
index = 1
|
||||
skip = 'Reboot'
|
||||
|
|
@ -333,37 +332,37 @@ def update_main_menu(menus) -> None:
|
|||
|
||||
|
||||
# Auto Repairs: Wrapper Functions
|
||||
def auto_backup_registry() -> None:
|
||||
def auto_backup_registry():
|
||||
"""Backup registry."""
|
||||
TRY_PRINT.run('Backup Registry...', backup_registry)
|
||||
|
||||
|
||||
def auto_backup_browser_profiles() -> None:
|
||||
def auto_backup_browser_profiles():
|
||||
"""Backup browser profiles."""
|
||||
backup_all_browser_profiles(use_try_print=True)
|
||||
|
||||
|
||||
def auto_backup_power_plans() -> None:
|
||||
def auto_backup_power_plans():
|
||||
"""Backup power plans."""
|
||||
TRY_PRINT.run('Backup Power Plans...', export_power_plans)
|
||||
|
||||
|
||||
def auto_reset_power_plans() -> None:
|
||||
def auto_reset_power_plans():
|
||||
"""Reset power plans."""
|
||||
TRY_PRINT.run('Reset Power Plans...', reset_power_plans)
|
||||
|
||||
|
||||
def auto_set_custom_power_plan() -> None:
|
||||
def auto_set_custom_power_plan():
|
||||
"""Set custom power plan."""
|
||||
TRY_PRINT.run('Set Custom Power Plan...', create_custom_power_plan)
|
||||
|
||||
|
||||
def auto_enable_bsod_minidumps() -> None:
|
||||
def auto_enable_bsod_minidumps():
|
||||
"""Enable saving minidumps during BSoDs."""
|
||||
TRY_PRINT.run('Enable BSoD mini dumps...', enable_bsod_minidumps)
|
||||
|
||||
|
||||
def auto_enable_regback() -> None:
|
||||
def auto_enable_regback():
|
||||
"""Enable RegBack."""
|
||||
TRY_PRINT.run(
|
||||
'Enable RegBack...', reg_set_value, 'HKLM',
|
||||
|
|
@ -372,7 +371,7 @@ def auto_enable_regback() -> None:
|
|||
)
|
||||
|
||||
|
||||
def auto_system_restore_enable() -> None:
|
||||
def auto_system_restore_enable():
|
||||
"""Enable System Restore."""
|
||||
cmd = [
|
||||
'powershell', '-Command', 'Enable-ComputerRestore',
|
||||
|
|
@ -381,28 +380,28 @@ def auto_system_restore_enable() -> None:
|
|||
TRY_PRINT.run('Enable System Restore...', run_program, cmd=cmd)
|
||||
|
||||
|
||||
def auto_system_restore_set_size() -> None:
|
||||
def auto_system_restore_set_size():
|
||||
"""Set System Restore size."""
|
||||
TRY_PRINT.run('Set System Restore Size...', set_system_restore_size)
|
||||
|
||||
|
||||
def auto_system_restore_create() -> None:
|
||||
def auto_system_restore_create():
|
||||
"""Create System Restore point."""
|
||||
TRY_PRINT.run('Create System Restore...', create_system_restore_point)
|
||||
|
||||
|
||||
def auto_windows_updates_enable() -> None:
|
||||
def auto_windows_updates_enable():
|
||||
"""Enable Windows Updates."""
|
||||
TRY_PRINT.run('Enable Windows Updates...', enable_windows_updates)
|
||||
|
||||
|
||||
# Auto Setup: Wrapper Functions
|
||||
def auto_activate_windows() -> None:
|
||||
def auto_activate_windows():
|
||||
"""Attempt to activate Windows using BIOS key."""
|
||||
TRY_PRINT.run('Windows Activation...', activate_with_bios)
|
||||
|
||||
|
||||
def auto_config_browsers() -> None:
|
||||
def auto_config_browsers():
|
||||
"""Configure Browsers."""
|
||||
prompt = ' Press Enter to continue...'
|
||||
TRY_PRINT.run('Chrome Notifications...', disable_chrome_notifications)
|
||||
|
|
@ -413,38 +412,33 @@ def auto_config_browsers() -> None:
|
|||
'Set default browser...', set_default_browser, msg_good='STARTED',
|
||||
)
|
||||
print(prompt, end='', flush=True)
|
||||
ui.pause(' ')
|
||||
pause('')
|
||||
|
||||
# Move cursor to beginning of the previous line and clear prompt
|
||||
print(f'\033[F\r{" "*len(prompt)}\r', end='', flush=True)
|
||||
|
||||
|
||||
def auto_config_explorer() -> None:
|
||||
def auto_config_explorer():
|
||||
"""Configure Windows Explorer and restart the process."""
|
||||
TRY_PRINT.run('Windows Explorer...', config_explorer)
|
||||
|
||||
|
||||
def auto_config_open_shell() -> None:
|
||||
def auto_config_open_shell():
|
||||
"""Configure Open Shell."""
|
||||
TRY_PRINT.run('Open Shell...', config_open_shell)
|
||||
|
||||
|
||||
def auto_disable_password_expiration() -> None:
|
||||
"""Disable password expiration for all users."""
|
||||
TRY_PRINT.run('Disable password expiration...', disable_password_expiration)
|
||||
|
||||
|
||||
def auto_export_aida64_report() -> None:
|
||||
def auto_export_aida64_report():
|
||||
"""Export AIDA64 reports."""
|
||||
TRY_PRINT.run('AIDA64 Report...', export_aida64_report)
|
||||
|
||||
|
||||
def auto_install_firefox() -> None:
|
||||
def auto_install_firefox():
|
||||
"""Install Firefox."""
|
||||
TRY_PRINT.run('Firefox...', install_firefox)
|
||||
|
||||
|
||||
def auto_install_libreoffice() -> None:
|
||||
def auto_install_libreoffice():
|
||||
"""Install LibreOffice.
|
||||
|
||||
NOTE: It is assumed that auto_install_vcredists() will be run
|
||||
|
|
@ -453,120 +447,100 @@ def auto_install_libreoffice() -> None:
|
|||
TRY_PRINT.run('LibreOffice...', install_libreoffice, vcredist=False)
|
||||
|
||||
|
||||
def auto_install_open_shell() -> None:
|
||||
def auto_install_open_shell():
|
||||
"""Install Open Shell."""
|
||||
TRY_PRINT.run('Open Shell...', install_open_shell)
|
||||
|
||||
|
||||
def auto_install_software_bundle() -> None:
|
||||
def auto_install_software_bundle():
|
||||
"""Install standard software bundle."""
|
||||
TRY_PRINT.run('Software Bundle...', winget_import, group_name='default')
|
||||
TRY_PRINT.run('Software Bundle...', install_software_bundle)
|
||||
|
||||
|
||||
def auto_install_software_upgrades() -> None:
|
||||
"""Upgrade all supported installed software."""
|
||||
TRY_PRINT.run('Software Upgrades...', winget_upgrade)
|
||||
|
||||
|
||||
def auto_install_vcredists() -> None:
|
||||
def auto_install_vcredists():
|
||||
"""Install latest supported Visual C++ runtimes."""
|
||||
TRY_PRINT.run('Visual C++ Runtimes...', winget_import, group_name='vcredists')
|
||||
TRY_PRINT.run('Visual C++ Runtimes...', install_vcredists)
|
||||
|
||||
|
||||
def auto_install_winget() -> None:
|
||||
"""Install winget if needed."""
|
||||
TRY_PRINT.run('Winget...', winget_check, raise_exceptions=True)
|
||||
|
||||
|
||||
def auto_open_device_manager() -> None:
|
||||
def auto_open_device_manager():
|
||||
"""Open Device Manager."""
|
||||
TRY_PRINT.run('Device Manager...', open_device_manager)
|
||||
|
||||
|
||||
def auto_open_hwinfo_sensors() -> None:
|
||||
def auto_open_hwinfo_sensors():
|
||||
"""Open HWiNFO Sensors."""
|
||||
TRY_PRINT.run('HWiNFO Sensors...', open_hwinfo_sensors)
|
||||
|
||||
|
||||
def auto_open_microsoft_store_updates() -> None:
|
||||
"""Opem Microsoft Store Updates."""
|
||||
TRY_PRINT.run('Microsoft Store Updates...', open_microsoft_store_updates)
|
||||
|
||||
|
||||
def auto_open_snappy_driver_installer_origin() -> None:
|
||||
"""Open Snappy Driver Installer Origin."""
|
||||
TRY_PRINT.run('Snappy Driver Installer...', open_snappy_driver_installer_origin)
|
||||
|
||||
|
||||
def auto_open_windows_activation() -> None:
|
||||
def auto_open_windows_activation():
|
||||
"""Open Windows Activation."""
|
||||
if not is_activated():
|
||||
TRY_PRINT.run('Windows Activation...', open_windows_activation)
|
||||
|
||||
|
||||
def auto_open_windows_updates() -> None:
|
||||
def auto_open_windows_updates():
|
||||
"""Open Windows Updates."""
|
||||
TRY_PRINT.run('Windows Updates...', open_windows_updates)
|
||||
|
||||
|
||||
def auto_open_xmplay() -> None:
|
||||
def auto_open_xmplay():
|
||||
"""Open XMPlay."""
|
||||
TRY_PRINT.run('XMPlay...', open_xmplay)
|
||||
|
||||
|
||||
def auto_show_4k_alignment_check() -> None:
|
||||
def auto_show_4k_alignment_check():
|
||||
"""Display 4K alignment check."""
|
||||
TRY_PRINT.run('4K alignment Check...', check_4k_alignment, show_alert=True)
|
||||
|
||||
|
||||
def auto_show_installed_antivirus() -> None:
|
||||
def auto_show_installed_antivirus():
|
||||
"""Display installed antivirus."""
|
||||
TRY_PRINT.run('Virus Protection...', list_installed_antivirus)
|
||||
TRY_PRINT.run('Virus Protection...', get_installed_antivirus)
|
||||
|
||||
|
||||
def auto_show_installed_ram() -> None:
|
||||
def auto_show_installed_ram():
|
||||
"""Display installed RAM."""
|
||||
TRY_PRINT.run('Installed RAM...', get_installed_ram,
|
||||
as_list=True, raise_exceptions=True,
|
||||
)
|
||||
|
||||
|
||||
def auto_show_os_activation() -> None:
|
||||
def auto_show_os_activation():
|
||||
"""Display OS activation status."""
|
||||
TRY_PRINT.run('Activation...', get_os_activation, as_list=True)
|
||||
|
||||
|
||||
def auto_show_os_name() -> None:
|
||||
def auto_show_os_name():
|
||||
"""Display OS Name."""
|
||||
TRY_PRINT.run('Operating System...', get_os_name, as_list=True)
|
||||
|
||||
|
||||
def auto_show_secure_boot_status() -> None:
|
||||
def auto_show_secure_boot_status():
|
||||
"""Display Secure Boot status."""
|
||||
TRY_PRINT.run(
|
||||
'Secure Boot...', check_secure_boot_status, msg_good='Enabled',
|
||||
)
|
||||
|
||||
|
||||
def auto_show_storage_status() -> None:
|
||||
def auto_show_storage_status():
|
||||
"""Display storage status."""
|
||||
TRY_PRINT.run('Storage Status...', get_storage_status)
|
||||
|
||||
|
||||
def auto_windows_temp_fix() -> None:
|
||||
def auto_windows_temp_fix():
|
||||
"""Restore default ACLs for Windows\\Temp."""
|
||||
TRY_PRINT.run(r'Windows\Temp fix...', fix_windows_temp)
|
||||
|
||||
|
||||
# Configure Functions
|
||||
def config_explorer() -> None:
|
||||
def config_explorer():
|
||||
"""Configure Windows Explorer and restart the process."""
|
||||
reg_write_settings(REG_WINDOWS_EXPLORER)
|
||||
kill_procs('explorer.exe', force=True)
|
||||
popen_program(['explorer.exe'])
|
||||
|
||||
|
||||
def config_open_shell() -> None:
|
||||
def config_open_shell():
|
||||
"""Configure Open Shell."""
|
||||
has_low_power_idle = False
|
||||
|
||||
|
|
@ -586,7 +560,7 @@ def config_open_shell() -> None:
|
|||
reg_write_settings(REG_OPEN_SHELL_LOW_POWER_IDLE)
|
||||
|
||||
|
||||
def disable_chrome_notifications() -> None:
|
||||
def disable_chrome_notifications():
|
||||
"""Disable notifications in Google Chrome."""
|
||||
defaults_key = 'default_content_setting_values'
|
||||
profiles = []
|
||||
|
|
@ -628,19 +602,13 @@ def disable_chrome_notifications() -> None:
|
|||
pref_file.write_text(json.dumps(pref_data, separators=(',', ':')))
|
||||
|
||||
|
||||
def disable_password_expiration() -> None:
|
||||
"""Disable password expiration for all users."""
|
||||
script_path = find_kit_dir('Scripts').joinpath('disable_password_expiration.ps1')
|
||||
cmd = ['PowerShell', '-ExecutionPolicy', 'Bypass', '-File', script_path]
|
||||
def enable_bsod_minidumps():
|
||||
"""Enable saving minidumps during BSoDs."""
|
||||
cmd = ['wmic', 'RECOVEROS', 'set', 'DebugInfoType', '=', '3']
|
||||
run_program(cmd)
|
||||
|
||||
|
||||
def enable_bsod_minidumps() -> None:
|
||||
"""Enable saving minidumps during BSoDs."""
|
||||
reg_write_settings(REG_WINDOWS_BSOD_MINIDUMPS)
|
||||
|
||||
|
||||
def enable_ublock_origin() -> None:
|
||||
def enable_ublock_origin():
|
||||
"""Enable uBlock Origin in supported browsers."""
|
||||
base_paths = [
|
||||
PROGRAMFILES_64, PROGRAMFILES_32, os.environ.get('LOCALAPPDATA'),
|
||||
|
|
@ -667,10 +635,10 @@ def enable_ublock_origin() -> None:
|
|||
|
||||
# Open detected browsers
|
||||
for cmd in cmds:
|
||||
popen_program(cmd, pipe=True)
|
||||
popen_program(cmd)
|
||||
|
||||
|
||||
def fix_windows_temp() -> None:
|
||||
def fix_windows_temp():
|
||||
"""Restore default permissions for Windows\\Temp."""
|
||||
permissions = (
|
||||
'Users:(CI)(X,WD,AD)',
|
||||
|
|
@ -682,7 +650,7 @@ def fix_windows_temp() -> None:
|
|||
|
||||
|
||||
# Install Functions
|
||||
def install_firefox() -> None:
|
||||
def install_firefox():
|
||||
"""Install Firefox.
|
||||
|
||||
As far as I can tell if you use the EXE installers then it will use
|
||||
|
|
@ -787,12 +755,12 @@ def install_libreoffice(
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def install_open_shell() -> None:
|
||||
def install_open_shell():
|
||||
"""Install Open Shell (just the Start Menu)."""
|
||||
skin_zip = get_tool_path('OpenShell', 'Fluent-Metro', suffix='zip')
|
||||
|
||||
# Bail early
|
||||
if OS_VERSION < 10:
|
||||
if OS_VERSION != 10:
|
||||
raise GenericWarning('Unsupported OS')
|
||||
|
||||
# Install OpenShell
|
||||
|
|
@ -817,7 +785,49 @@ def install_open_shell() -> None:
|
|||
run_program(cmd)
|
||||
|
||||
|
||||
def uninstall_firefox() -> None:
|
||||
def install_software_bundle():
|
||||
"""Install standard software bundle."""
|
||||
download_tool('Ninite', 'Software Bundle')
|
||||
installer = get_tool_path('Ninite', 'Software Bundle')
|
||||
msg = 'Waiting for installations to finish...'
|
||||
warning = 'NOTE: Press CTRL+c to manually resume if it gets stuck...'
|
||||
|
||||
# Start installations and wait for them to finish
|
||||
print_standard(msg)
|
||||
print_warning(warning, end='', flush=True)
|
||||
proc = popen_program([installer])
|
||||
try:
|
||||
proc.wait()
|
||||
except KeyboardInterrupt:
|
||||
# Assuming user-forced continue
|
||||
pass
|
||||
|
||||
# Clear info lines
|
||||
print(
|
||||
'\r\033[0K' # Cursor to start of current line and clear to end of line
|
||||
'\033[F\033[54C' # Cursor to start of prev line and then move 54 right
|
||||
'\033[0K', # Clear from cursor to end of line
|
||||
end='', flush=True)
|
||||
|
||||
|
||||
def install_vcredists():
|
||||
"""Install latest supported Visual C++ runtimes."""
|
||||
for year in (2012, 2013, 2019):
|
||||
cmd_args = ['/install', '/passive', '/norestart']
|
||||
if year == 2012:
|
||||
cmd_args.pop(0)
|
||||
name = f'VCRedist_{year}_x32'
|
||||
download_tool('VCRedist', name)
|
||||
installer = get_tool_path('VCRedist', name)
|
||||
run_program([installer, *cmd_args])
|
||||
if ARCH == '64':
|
||||
name = f'{name[:-2]}64'
|
||||
download_tool('VCRedist', name)
|
||||
installer = get_tool_path('VCRedist', name)
|
||||
run_program([installer, *cmd_args])
|
||||
|
||||
|
||||
def uninstall_firefox():
|
||||
"""Uninstall all copies of Firefox."""
|
||||
json_file = format_log_path(log_name='Installed Programs', timestamp=True)
|
||||
json_file = json_file.with_name(f'{json_file.stem}.json')
|
||||
|
|
@ -838,14 +848,13 @@ def uninstall_firefox() -> None:
|
|||
|
||||
|
||||
# Misc Functions
|
||||
def check_secure_boot_status() -> None:
|
||||
def check_secure_boot_status():
|
||||
"""Check Secure Boot status."""
|
||||
is_secure_boot_enabled(raise_exceptions=True, show_alert=True)
|
||||
|
||||
|
||||
def get_firefox_default_profile(profiles_ini) -> Any:
|
||||
def get_firefox_default_profile(profiles_ini):
|
||||
"""Get Firefox default profile, returns(pathlib.Path, encoding) or None."""
|
||||
# TODO: Refactor to remove dependancy on Any
|
||||
default_profile = None
|
||||
encoding = None
|
||||
parser = None
|
||||
|
|
@ -882,24 +891,24 @@ def get_firefox_default_profile(profiles_ini) -> Any:
|
|||
return (default_profile, encoding)
|
||||
|
||||
|
||||
def get_storage_status() -> list[str]:
|
||||
def get_storage_status():
|
||||
"""Get storage status for fixed disks, returns list."""
|
||||
report = get_volume_usage(use_colors=True)
|
||||
for disk in get_raw_disks():
|
||||
report.append(ansi.color_string(f'Uninitialized Disk: {disk}', 'RED'))
|
||||
report.append(color_string(f'Uninitialized Disk: {disk}', 'RED'))
|
||||
|
||||
# Done
|
||||
return report
|
||||
|
||||
|
||||
def set_default_browser() -> None:
|
||||
def set_default_browser():
|
||||
"""Open Windows Settings to the default apps section."""
|
||||
cmd = ['start', '', 'ms-settings:defaultapps']
|
||||
popen_program(cmd, shell=True)
|
||||
|
||||
|
||||
# Tool Functions
|
||||
def export_aida64_report() -> None:
|
||||
def export_aida64_report():
|
||||
"""Export AIDA64 report."""
|
||||
report_path = format_log_path(
|
||||
log_name='AIDA64 System Report',
|
||||
|
|
@ -920,12 +929,12 @@ def export_aida64_report() -> None:
|
|||
raise GenericError('Error(s) encountered exporting report.')
|
||||
|
||||
|
||||
def open_device_manager() -> None:
|
||||
def open_device_manager():
|
||||
"""Open Device Manager."""
|
||||
popen_program(['mmc', 'devmgmt.msc'])
|
||||
|
||||
|
||||
def open_hwinfo_sensors() -> None:
|
||||
def open_hwinfo_sensors():
|
||||
"""Open HWiNFO sensors."""
|
||||
hwinfo_path = get_tool_path('HWiNFO', 'HWiNFO')
|
||||
base_config = hwinfo_path.with_name('general.ini')
|
||||
|
|
@ -941,33 +950,17 @@ def open_hwinfo_sensors() -> None:
|
|||
run_tool('HWiNFO', 'HWiNFO', popen=True)
|
||||
|
||||
|
||||
def open_microsoft_store_updates() -> None:
|
||||
"""Open Microsoft Store to the updates page."""
|
||||
popen_program(['explorer', 'ms-windows-store:updates'])
|
||||
|
||||
|
||||
def open_snappy_driver_installer_origin() -> None:
|
||||
"""Open Snappy Driver Installer Origin."""
|
||||
if OS_VERSION == 11:
|
||||
appid_services = ['appid', 'appidsvc', 'applockerfltr']
|
||||
for svc in appid_services:
|
||||
stop_service(svc)
|
||||
if any([get_service_status(s) != 'stopped' for s in appid_services]):
|
||||
raise GenericWarning('Failed to stop AppID services')
|
||||
run_tool('SDIO', 'SDIO', cwd=True, pipe=True, popen=True)
|
||||
|
||||
|
||||
def open_windows_activation() -> None:
|
||||
def open_windows_activation():
|
||||
"""Open Windows Activation."""
|
||||
popen_program(['slui'])
|
||||
|
||||
|
||||
def open_windows_updates() -> None:
|
||||
def open_windows_updates():
|
||||
"""Open Windows Updates."""
|
||||
popen_program(['control', '/name', 'Microsoft.WindowsUpdate'])
|
||||
|
||||
|
||||
def open_xmplay() -> None:
|
||||
def open_xmplay():
|
||||
"""Open XMPlay."""
|
||||
sleep(2)
|
||||
run_tool('XMPlay', 'XMPlay', 'music.7z', cwd=True, popen=True)
|
||||
|
|
|
|||
File diff suppressed because it is too large
Load diff
|
|
@ -4,8 +4,6 @@
|
|||
import logging
|
||||
import pathlib
|
||||
|
||||
from typing import Any
|
||||
|
||||
from wk.exe import run_program
|
||||
from wk.std import PLATFORM
|
||||
|
||||
|
|
@ -15,7 +13,7 @@ LOG = logging.getLogger(__name__)
|
|||
|
||||
|
||||
# Functions
|
||||
def capture_pane(pane_id: str | None = None) -> str:
|
||||
def capture_pane(pane_id=None):
|
||||
"""Capture text from current or target pane, returns str."""
|
||||
cmd = ['tmux', 'capture-pane', '-p']
|
||||
if pane_id:
|
||||
|
|
@ -26,122 +24,49 @@ def capture_pane(pane_id: str | None = None) -> str:
|
|||
return proc.stdout.strip()
|
||||
|
||||
|
||||
def clear_pane(pane_id: str | None = None) -> None:
|
||||
def clear_pane(pane_id=None):
|
||||
"""Clear pane buffer for current or target pane."""
|
||||
commands = [
|
||||
['tmux', 'send-keys', '-R'],
|
||||
['tmux', 'clear-history'],
|
||||
]
|
||||
cmd = ['tmux', 'send-keys', '-R']
|
||||
if pane_id:
|
||||
commands = [[*cmd, '-t', pane_id] for cmd in commands]
|
||||
cmd.extend(['-t', pane_id])
|
||||
|
||||
# Clear pane
|
||||
for cmd in commands:
|
||||
run_program(cmd, check=False)
|
||||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def fix_layout(
|
||||
layout: dict[str, dict[str, Any]],
|
||||
clear_on_resize: bool = False,
|
||||
forced: bool = False,
|
||||
) -> None:
|
||||
"""Fix pane sizes based on layout.
|
||||
|
||||
NOTE: The magic +/- 1 values are for the split rows/columns.
|
||||
"""
|
||||
resize_kwargs = []
|
||||
|
||||
# Bail early
|
||||
if not (forced or layout_needs_fixed(layout)):
|
||||
def fix_layout(panes, layout, forced=False):
|
||||
"""Fix pane sizes based on layout."""
|
||||
if not (forced or layout_needs_fixed(panes, layout)):
|
||||
# Layout should be fine
|
||||
return
|
||||
|
||||
# Clear current pane if needed
|
||||
if clear_on_resize:
|
||||
clear_pane()
|
||||
|
||||
# Remove closed panes
|
||||
for data in layout.values():
|
||||
data['Panes'] = [pane for pane in data['Panes'] if poll_pane(pane)]
|
||||
|
||||
# Calculate constraints
|
||||
avail_horizontal, avail_vertical = get_window_size()
|
||||
avail_vertical -= layout['Current'].get('height', 0)
|
||||
for group in ('Title', 'Info'):
|
||||
if not layout[group]['Panes']:
|
||||
# Update panes
|
||||
for name, data in layout.items():
|
||||
# Skip missing panes
|
||||
if name not in panes:
|
||||
continue
|
||||
avail_vertical -= layout[group].get('height', 0) + 1
|
||||
num_workers = len(layout['Workers']['Panes'])
|
||||
avail_vertical -= num_workers * (layout['Workers'].get('height', 0) + 1)
|
||||
avail_horizontal -= layout['Progress']['width'] + 1
|
||||
|
||||
# Fix heights
|
||||
for group, data in layout.items():
|
||||
if not data['Panes'] or group in ('Started', 'Progress'):
|
||||
continue
|
||||
resize_kwargs.append(
|
||||
{'pane_id': data['Panes'][0], 'height': data.get('height', avail_vertical)}
|
||||
)
|
||||
if group == 'Workers' and len(data['Panes']) > 1:
|
||||
for pane_id in data['Panes'][1:]:
|
||||
resize_kwargs.append(
|
||||
{'pane_id': pane_id, 'height': data.get('height', avail_vertical)}
|
||||
)
|
||||
|
||||
# Fix widths
|
||||
resize_kwargs.append(
|
||||
{'pane_id': layout['Progress']['Panes'][0], 'width': layout['Progress']['width']}
|
||||
)
|
||||
resize_kwargs.append(
|
||||
{'pane_id': layout['Started']['Panes'][0], 'height': layout['Started']['height']}
|
||||
)
|
||||
for group, data in layout.items():
|
||||
num_panes = len(data['Panes'])
|
||||
if num_panes < 2 or group not in ('Title', 'Info'):
|
||||
continue
|
||||
pane_width, remainder = divmod(avail_horizontal - (num_panes-1), num_panes)
|
||||
for pane_id in data['Panes']:
|
||||
new_width = pane_width
|
||||
if remainder > 0:
|
||||
new_width += 1
|
||||
remainder -= 1
|
||||
resize_kwargs.append({'pane_id': pane_id, 'width': new_width})
|
||||
|
||||
# Resize panes
|
||||
for kwargs in resize_kwargs:
|
||||
try:
|
||||
resize_pane(**kwargs)
|
||||
except RuntimeError:
|
||||
# Assuming pane was closed just before resizing
|
||||
pass
|
||||
# Resize pane(s)
|
||||
pane_list = panes[name]
|
||||
if isinstance(pane_list, str):
|
||||
pane_list = [pane_list]
|
||||
for pane_id in pane_list:
|
||||
if name == 'Current':
|
||||
pane_id = None
|
||||
try:
|
||||
resize_pane(pane_id, **data)
|
||||
except RuntimeError:
|
||||
# Assuming pane was closed just before resizing
|
||||
pass
|
||||
|
||||
|
||||
def get_pane_size(pane_id: str | None = None) -> tuple[int, int]:
|
||||
def get_pane_size(pane_id=None):
|
||||
"""Get current or target pane size, returns tuple."""
|
||||
cmd = ['tmux', 'display-message', '-p']
|
||||
cmd = ['tmux', 'display', '-p']
|
||||
if pane_id:
|
||||
cmd.extend(['-t', pane_id])
|
||||
cmd.append('#{pane_width} #{pane_height}')
|
||||
|
||||
# Get resolution
|
||||
proc = run_program(cmd, check=False)
|
||||
try:
|
||||
width, height = proc.stdout.strip().split()
|
||||
except ValueError:
|
||||
# Assuming this is a race condition as it usually happens inside the
|
||||
# background fix layout loop
|
||||
return 0, 0
|
||||
width = int(width)
|
||||
height = int(height)
|
||||
|
||||
# Done
|
||||
return (width, height)
|
||||
|
||||
|
||||
def get_window_size() -> tuple[int, int]:
|
||||
"""Get current window size, returns tuple."""
|
||||
cmd = ['tmux', 'display-message', '-p', '#{window_width} #{window_height}']
|
||||
|
||||
# Get resolution
|
||||
proc = run_program(cmd, check=False)
|
||||
width, height = proc.stdout.strip().split()
|
||||
|
|
@ -152,7 +77,7 @@ def get_window_size() -> tuple[int, int]:
|
|||
return (width, height)
|
||||
|
||||
|
||||
def kill_all_panes(pane_id: str | None = None) -> None:
|
||||
def kill_all_panes(pane_id=None):
|
||||
"""Kill all panes except for the current or target pane."""
|
||||
cmd = ['tmux', 'kill-pane', '-a']
|
||||
if pane_id:
|
||||
|
|
@ -162,7 +87,7 @@ def kill_all_panes(pane_id: str | None = None) -> None:
|
|||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def kill_pane(*pane_ids: str) -> None:
|
||||
def kill_pane(*pane_ids):
|
||||
"""Kill pane(s) by id."""
|
||||
cmd = ['tmux', 'kill-pane', '-t']
|
||||
|
||||
|
|
@ -171,28 +96,42 @@ def kill_pane(*pane_ids: str) -> None:
|
|||
run_program(cmd+[pane_id], check=False)
|
||||
|
||||
|
||||
def layout_needs_fixed(layout: dict[str, dict[str, Any]]) -> bool:
|
||||
def layout_needs_fixed(panes, layout):
|
||||
"""Check if layout needs fixed, returns bool."""
|
||||
needs_fixed = False
|
||||
|
||||
# Check panes
|
||||
for data in layout.values():
|
||||
if 'height' in data:
|
||||
needs_fixed = needs_fixed or any(
|
||||
get_pane_size(pane)[1] != data['height'] for pane in data['Panes']
|
||||
)
|
||||
if 'width' in data:
|
||||
needs_fixed = needs_fixed or any(
|
||||
get_pane_size(pane)[0] != data['width'] for pane in data['Panes']
|
||||
)
|
||||
for name, data in layout.items():
|
||||
# Skip unpredictably sized panes
|
||||
if not data.get('Check', False):
|
||||
continue
|
||||
|
||||
# Skip missing panes
|
||||
if name not in panes:
|
||||
continue
|
||||
|
||||
# Check pane size(s)
|
||||
pane_list = panes[name]
|
||||
if isinstance(pane_list, str):
|
||||
pane_list = [pane_list]
|
||||
for pane_id in pane_list:
|
||||
try:
|
||||
width, height = get_pane_size(pane_id)
|
||||
except ValueError:
|
||||
# Pane may have disappeared during this loop
|
||||
continue
|
||||
if data.get('width', False) and data['width'] != width:
|
||||
needs_fixed = True
|
||||
if data.get('height', False) and data['height'] != height:
|
||||
needs_fixed = True
|
||||
|
||||
# Done
|
||||
return needs_fixed
|
||||
|
||||
|
||||
def poll_pane(pane_id: str) -> bool:
|
||||
def poll_pane(pane_id):
|
||||
"""Check if pane exists, returns bool."""
|
||||
cmd = ['tmux', 'list-panes', '-F', '#{pane_id}']
|
||||
cmd = ['tmux', 'list-panes', '-F', '#D']
|
||||
|
||||
# Get list of panes
|
||||
proc = run_program(cmd, check=False)
|
||||
|
|
@ -203,12 +142,7 @@ def poll_pane(pane_id: str) -> bool:
|
|||
|
||||
|
||||
def prep_action(
|
||||
cmd: str | None = None,
|
||||
working_dir: pathlib.Path | str | None = None,
|
||||
text: str | None = None,
|
||||
watch_file: pathlib.Path | str | None = None,
|
||||
watch_cmd: str = 'cat',
|
||||
) -> list[str]:
|
||||
cmd=None, working_dir=None, text=None, watch_file=None, watch_cmd='cat'):
|
||||
"""Prep action to perform during a tmux call, returns list.
|
||||
|
||||
This will prep for running a basic command, displaying text on screen,
|
||||
|
|
@ -258,7 +192,7 @@ def prep_action(
|
|||
return action_cmd
|
||||
|
||||
|
||||
def prep_file(path: pathlib.Path | str) -> None:
|
||||
def prep_file(path):
|
||||
"""Check if file exists and create empty file if not."""
|
||||
path = pathlib.Path(path).resolve()
|
||||
try:
|
||||
|
|
@ -268,11 +202,8 @@ def prep_file(path: pathlib.Path | str) -> None:
|
|||
pass
|
||||
|
||||
|
||||
def resize_pane(
|
||||
pane_id: str | None = None,
|
||||
width: int | None = None,
|
||||
height: int | None = None,
|
||||
) -> None:
|
||||
def resize_pane(pane_id=None, width=None, height=None, **kwargs):
|
||||
# pylint: disable=unused-argument
|
||||
"""Resize current or target pane.
|
||||
|
||||
NOTE: kwargs is only here to make calling this function easier
|
||||
|
|
@ -297,24 +228,12 @@ def resize_pane(
|
|||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def respawn_pane(pane_id: str, **action) -> None:
|
||||
"""Respawn pane with action."""
|
||||
cmd = ['tmux', 'respawn-pane', '-k', '-t', pane_id]
|
||||
cmd.extend(prep_action(**action))
|
||||
|
||||
# Respawn
|
||||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def split_window(
|
||||
lines: int | None = None,
|
||||
percent: int | None = None,
|
||||
behind: bool = False,
|
||||
vertical: bool = False,
|
||||
target_id: str | None = None,
|
||||
**action) -> str:
|
||||
lines=None, percent=None,
|
||||
behind=False, vertical=False,
|
||||
target_id=None, **action):
|
||||
"""Split tmux window, run action, and return pane_id as str."""
|
||||
cmd = ['tmux', 'split-window', '-d', '-PF', '#{pane_id}']
|
||||
cmd = ['tmux', 'split-window', '-d', '-PF', '#D']
|
||||
|
||||
# Safety checks
|
||||
if not (lines or percent):
|
||||
|
|
@ -335,7 +254,7 @@ def split_window(
|
|||
if lines:
|
||||
cmd.extend(['-l', str(lines)])
|
||||
elif percent:
|
||||
cmd.extend(['-l', f'{percent}%'])
|
||||
cmd.extend(['-p', str(percent)])
|
||||
|
||||
# New pane action
|
||||
cmd.extend(prep_action(**action))
|
||||
|
|
@ -345,7 +264,16 @@ def split_window(
|
|||
return proc.stdout.strip()
|
||||
|
||||
|
||||
def zoom_pane(pane_id: str | None = None) -> None:
|
||||
def respawn_pane(pane_id, **action):
|
||||
"""Respawn pane with action."""
|
||||
cmd = ['tmux', 'respawn-pane', '-k', '-t', pane_id]
|
||||
cmd.extend(prep_action(**action))
|
||||
|
||||
# Respawn
|
||||
run_program(cmd, check=False)
|
||||
|
||||
|
||||
def zoom_pane(pane_id=None):
|
||||
"""Toggle zoom status for current or target pane."""
|
||||
cmd = ['tmux', 'resize-pane', '-Z']
|
||||
if pane_id:
|
||||
|
|
@ -1,6 +0,0 @@
|
|||
"""WizardKit: ui module init"""
|
||||
|
||||
from . import ansi
|
||||
from . import cli
|
||||
from . import tmux
|
||||
from . import tui
|
||||
|
|
@ -1,70 +0,0 @@
|
|||
"""WizardKit: ANSI control/escape functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import itertools
|
||||
import logging
|
||||
|
||||
from typing import Iterable
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
COLORS = {
|
||||
'CLEAR': '\033[0m',
|
||||
'RED': '\033[31m',
|
||||
'RED_BLINK': '\033[31;5m',
|
||||
'ORANGE': '\033[31;1m',
|
||||
'ORANGE_RED': '\033[1;31;41m',
|
||||
'GREEN': '\033[32m',
|
||||
'YELLOW': '\033[33m',
|
||||
'YELLOW_BLINK': '\033[33;5m',
|
||||
'BLUE': '\033[34m',
|
||||
'PURPLE': '\033[35m',
|
||||
'CYAN': '\033[36m',
|
||||
}
|
||||
|
||||
|
||||
# Functions
|
||||
def clear_screen() -> None:
|
||||
"""Clear screen using ANSI escape."""
|
||||
print('\033c', end='', flush=True)
|
||||
|
||||
|
||||
def color_string(
|
||||
strings: Iterable[str] | str,
|
||||
colors: Iterable[str | None] | str,
|
||||
sep=' ',
|
||||
) -> str:
|
||||
"""Build colored string using ANSI escapes, returns str."""
|
||||
data = {'strings': strings, 'colors': colors}
|
||||
msg = []
|
||||
|
||||
# Convert input to tuples of strings
|
||||
for k, v in data.items():
|
||||
if isinstance(v, str):
|
||||
# Avoid splitting string into a list of characters
|
||||
data[k] = (v,)
|
||||
try:
|
||||
iter(v)
|
||||
except TypeError:
|
||||
# Assuming single element passed, convert to string
|
||||
data[k] = (str(v),)
|
||||
|
||||
# Build new string with color escapes added
|
||||
for string, color in itertools.zip_longest(data['strings'], data['colors']):
|
||||
color_code = COLORS.get(str(color), COLORS['CLEAR'])
|
||||
msg.append(f'{color_code}{string}{COLORS["CLEAR"]}')
|
||||
|
||||
# Done
|
||||
return sep.join(msg)
|
||||
|
||||
|
||||
def strip_colors(string: str) -> str:
|
||||
"""Strip known ANSI color escapes from string, returns str."""
|
||||
LOG.debug('string: %s', string)
|
||||
for color in COLORS.values():
|
||||
string = string.replace(color, '')
|
||||
return string
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
@ -1,924 +0,0 @@
|
|||
"""WizardKit: CLI functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import logging
|
||||
import os
|
||||
import platform
|
||||
import re
|
||||
import subprocess
|
||||
import sys
|
||||
import traceback
|
||||
|
||||
from typing import Any, Callable, Iterable
|
||||
|
||||
from prompt_toolkit import prompt
|
||||
from prompt_toolkit.document import Document
|
||||
from prompt_toolkit.validation import Validator, ValidationError
|
||||
|
||||
try:
|
||||
from functools import cache
|
||||
except ImportError:
|
||||
# Assuming Python is < 3.9
|
||||
from functools import lru_cache as cache
|
||||
|
||||
from wk.cfg.main import (
|
||||
ENABLED_UPLOAD_DATA,
|
||||
INDENT,
|
||||
SUPPORT_MESSAGE,
|
||||
WIDTH,
|
||||
)
|
||||
from wk.std import (sleep, GenericWarning)
|
||||
from wk.ui.ansi import clear_screen, color_string, strip_colors
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
PLATFORM = platform.system()
|
||||
|
||||
|
||||
# Classes
|
||||
class InputChoiceValidator(Validator):
|
||||
"""Validate that input is one of the provided choices."""
|
||||
def __init__(self, choices: Iterable[str], allow_empty: bool = False):
|
||||
self.allow_empty: bool = allow_empty
|
||||
self.choices: list[str] = [str(c).upper() for c in choices]
|
||||
super().__init__()
|
||||
|
||||
def validate(self, document: Document) -> None:
|
||||
text = document.text
|
||||
if not (text or self.allow_empty):
|
||||
raise ValidationError(
|
||||
message='This input is required!',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
if text and text.upper() not in self.choices:
|
||||
raise ValidationError(
|
||||
message='Invalid selection',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
|
||||
class InputNotEmptyValidator(Validator):
|
||||
"""Validate that input is not empty."""
|
||||
def validate(self, document: Document) -> None:
|
||||
text = document.text
|
||||
if not text:
|
||||
raise ValidationError(
|
||||
message='This input is required!',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
|
||||
class InputTicketIDValidator(Validator):
|
||||
"""Validate that input resembles a ticket ID."""
|
||||
def __init__(self, allow_empty: bool = False):
|
||||
self.allow_empty: bool = allow_empty
|
||||
super().__init__()
|
||||
|
||||
def validate(self, document: Document) -> None:
|
||||
text = document.text
|
||||
if not (text or self.allow_empty):
|
||||
raise ValidationError(
|
||||
message='This input is required!',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
if text and not re.match(r'^\d', text):
|
||||
raise ValidationError(
|
||||
message='Ticket ID should start with a number!',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
|
||||
class InputYesNoValidator(Validator):
|
||||
"""Validate that input is a yes or no."""
|
||||
def __init__(self, allow_empty: bool = False):
|
||||
self.allow_empty: bool = allow_empty
|
||||
super().__init__()
|
||||
|
||||
def validate(self, document: Document) -> None:
|
||||
text = document.text
|
||||
if not (text or self.allow_empty):
|
||||
raise ValidationError(
|
||||
message='This input is required!',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
if text and not re.match(r'^(y(es|up|)|n(o|ope|))$', text, re.IGNORECASE):
|
||||
raise ValidationError(
|
||||
message='Please answer "yes" or "no"',
|
||||
cursor_position=len(text),
|
||||
)
|
||||
|
||||
class Menu():
|
||||
"""Object for tracking menu specific data and methods.
|
||||
|
||||
ASSUMPTIONS:
|
||||
1. All entry names are unique.
|
||||
2. All action entry names start with different letters.
|
||||
"""
|
||||
def __init__(self, title: str = '[Untitled Menu]'):
|
||||
self.actions: dict[str, dict[Any, Any]] = {}
|
||||
self.options: dict[str, dict[Any, Any]] = {}
|
||||
self.sets: dict[str, dict[Any, Any]] = {}
|
||||
self.toggles: dict[str, dict[Any, Any]] = {}
|
||||
self.disabled_str: str = 'Disabled'
|
||||
self.separator: str = '─'
|
||||
self.title: str = title
|
||||
|
||||
def _generate_menu_text(self) -> str:
|
||||
"""Generate menu text, returns str."""
|
||||
separator_string = self._get_separator_string()
|
||||
menu_lines = [self.title, separator_string] if self.title else []
|
||||
|
||||
# Sets & toggles
|
||||
for section in (self.sets, self.toggles):
|
||||
for details in section.values():
|
||||
if details.get('Hidden', False):
|
||||
continue
|
||||
if details.get('Separator', False):
|
||||
menu_lines.append(separator_string)
|
||||
menu_lines.append(details['Display Name'])
|
||||
if self.sets or self.toggles:
|
||||
menu_lines.append(separator_string)
|
||||
|
||||
# Options
|
||||
for details in self.options.values():
|
||||
if details.get('Hidden', False):
|
||||
continue
|
||||
if details.get('Separator', False):
|
||||
menu_lines.append(separator_string)
|
||||
menu_lines.append(details['Display Name'])
|
||||
if self.options:
|
||||
menu_lines.append(separator_string)
|
||||
|
||||
# Actions
|
||||
for details in self.actions.values():
|
||||
if details.get('Hidden', False):
|
||||
continue
|
||||
if details.get('Separator', False):
|
||||
menu_lines.append(separator_string)
|
||||
menu_lines.append(details['Display Name'])
|
||||
|
||||
# Show menu
|
||||
menu_lines.append('')
|
||||
menu_lines = [str(line) for line in menu_lines]
|
||||
return '\n'.join(menu_lines)
|
||||
|
||||
def _get_display_name(
|
||||
self, name, details,
|
||||
index=None, no_checkboxes=True, setting_item=False) -> str:
|
||||
"""Format display name based on details and args, returns str."""
|
||||
disabled = details.get('Disabled', False)
|
||||
if setting_item and not details['Selected']:
|
||||
# Display item in YELLOW
|
||||
disabled = True
|
||||
checkmark = '*'
|
||||
if 'CONEMUPID' in os.environ or 'DISPLAY' in os.environ or PLATFORM == 'Darwin':
|
||||
checkmark = '✓'
|
||||
display_name = f'{index if index else name[:1].upper()}: '
|
||||
if not (index and index >= 10):
|
||||
display_name = f' {display_name}'
|
||||
if setting_item and 'Value' in details:
|
||||
name = f'{name} = {details["Value"]}'
|
||||
|
||||
# Add enabled status if necessary
|
||||
if not no_checkboxes:
|
||||
display_name += f'[{checkmark if details["Selected"] else " "}] '
|
||||
|
||||
# Add name
|
||||
if disabled:
|
||||
display_name += color_string(f'{name} ({self.disabled_str})', 'YELLOW')
|
||||
else:
|
||||
display_name += name
|
||||
|
||||
# Done
|
||||
return display_name
|
||||
|
||||
def _get_separator_string(self) -> str:
|
||||
"""Format separator length based on name lengths, returns str."""
|
||||
separator_length = 0
|
||||
|
||||
# Check title line(s)
|
||||
if self.title:
|
||||
for line in self.title.split('\n'):
|
||||
separator_length = max(separator_length, len(strip_colors(line)))
|
||||
|
||||
# Loop over all item names
|
||||
for section in (self.actions, self.options, self.sets, self.toggles):
|
||||
for details in section.values():
|
||||
if details.get('Hidden', False):
|
||||
# Skip hidden lines
|
||||
continue
|
||||
line = strip_colors(details['Display Name'])
|
||||
separator_length = max(separator_length, len(line))
|
||||
separator_length += 1
|
||||
|
||||
# Done
|
||||
return self.separator * separator_length
|
||||
|
||||
def _get_valid_answers(self) -> list[str]:
|
||||
"""Get valid answers based on menu items, returns list."""
|
||||
valid_answers = []
|
||||
|
||||
# Numbered items
|
||||
index = 0
|
||||
for section in (self.sets, self.toggles, self.options):
|
||||
for details in section.values():
|
||||
if details.get('Hidden', False):
|
||||
# Don't increment index or add to valid_answers
|
||||
continue
|
||||
index += 1
|
||||
if not details.get('Disabled', False):
|
||||
valid_answers.append(str(index))
|
||||
|
||||
# Action items
|
||||
for name, details in self.actions.items():
|
||||
if not details.get('Disabled', False):
|
||||
valid_answers.append(name[:1].upper())
|
||||
|
||||
# Done
|
||||
return valid_answers
|
||||
|
||||
def _resolve_selection(self, selection: str) -> tuple[str, dict[Any, Any]]:
|
||||
"""Get menu item based on user selection, returns tuple."""
|
||||
offset = 1
|
||||
resolved_selection = tuple()
|
||||
if selection.isnumeric():
|
||||
# Enumerate over numbered entries
|
||||
entries = [
|
||||
*self.sets.items(),
|
||||
*self.toggles.items(),
|
||||
*self.options.items(),
|
||||
]
|
||||
for _i, details in enumerate(entries):
|
||||
if details[1].get('Hidden', False):
|
||||
offset -= 1
|
||||
elif str(_i+offset) == selection:
|
||||
# TODO: Fix this typo!
|
||||
# It was discovered after being in production for SEVERAL YEARS!
|
||||
# Extra testing is needed to verify any calls to this function still
|
||||
# depend on this functionality
|
||||
resolved_selection = (details)
|
||||
break
|
||||
else:
|
||||
# Just check actions
|
||||
for action, details in self.actions.items():
|
||||
if action.lower().startswith(selection.lower()):
|
||||
resolved_selection = (action, details)
|
||||
break
|
||||
|
||||
# Done
|
||||
return resolved_selection
|
||||
|
||||
def _update(self, single_selection: bool = True, settings_mode: bool = False) -> None:
|
||||
"""Update menu items in preparation for printing to screen."""
|
||||
index = 0
|
||||
|
||||
# Fix selection status for sets
|
||||
for set_details in self.sets.values():
|
||||
set_selected = True
|
||||
set_targets = set_details['Targets']
|
||||
for option, option_details in self.options.items():
|
||||
if option in set_targets and not option_details['Selected']:
|
||||
set_selected = False
|
||||
elif option not in set_targets and option_details['Selected']:
|
||||
set_selected = False
|
||||
set_details['Selected'] = set_selected
|
||||
|
||||
# Numbered sections
|
||||
for section in (self.sets, self.toggles, self.options):
|
||||
for name, details in section.items():
|
||||
if details.get('Hidden', False):
|
||||
# Skip hidden lines and don't increment index
|
||||
continue
|
||||
index += 1
|
||||
details['Display Name'] = self._get_display_name(
|
||||
name,
|
||||
details,
|
||||
index=index,
|
||||
no_checkboxes=single_selection,
|
||||
setting_item=settings_mode,
|
||||
)
|
||||
|
||||
# Actions
|
||||
for name, details in self.actions.items():
|
||||
details['Display Name'] = self._get_display_name(
|
||||
name,
|
||||
details,
|
||||
no_checkboxes=True,
|
||||
)
|
||||
|
||||
def _update_entry_selection_status(
|
||||
self, entry: str, toggle: bool = True, status: bool = False) -> None:
|
||||
"""Update entry selection status either directly or by toggling."""
|
||||
if entry in self.sets:
|
||||
# Update targets not the set itself
|
||||
new_status = not self.sets[entry]['Selected'] if toggle else status
|
||||
targets = self.sets[entry]['Targets']
|
||||
self._update_set_selection_status(targets, new_status)
|
||||
for section in (self.toggles, self.options, self.actions):
|
||||
if entry in section:
|
||||
if toggle:
|
||||
section[entry]['Selected'] = not section[entry]['Selected']
|
||||
else:
|
||||
section[entry]['Selected'] = status
|
||||
|
||||
def _update_set_selection_status(self, targets: Iterable[str], status: bool) -> None:
|
||||
"""Select or deselect options based on targets and status."""
|
||||
for option, details in self.options.items():
|
||||
# If (new) status is True and this option is a target then select
|
||||
# Otherwise deselect
|
||||
details['Selected'] = status and option in targets
|
||||
|
||||
def _user_select(self, prompt_msg: str) -> str:
|
||||
"""Show menu and select an entry, returns str."""
|
||||
menu_text = self._generate_menu_text()
|
||||
valid_answers = self._get_valid_answers()
|
||||
|
||||
# Menu loop
|
||||
while True:
|
||||
clear_screen()
|
||||
print(menu_text)
|
||||
sleep(0.01)
|
||||
answer = input_text(prompt_msg).strip()
|
||||
if answer.upper() in valid_answers:
|
||||
break
|
||||
|
||||
# Done
|
||||
return answer
|
||||
|
||||
def add_action(self, name: str, details: dict[Any, Any] | None = None) -> None:
|
||||
"""Add action to menu."""
|
||||
details = details if details else {}
|
||||
details['Selected'] = details.get('Selected', False)
|
||||
self.actions[name] = details
|
||||
|
||||
def add_option(self, name: str, details: dict[Any, Any] | None = None) -> None:
|
||||
"""Add option to menu."""
|
||||
details = details if details else {}
|
||||
details['Selected'] = details.get('Selected', False)
|
||||
self.options[name] = details
|
||||
|
||||
def add_set(self, name: str, details: dict[Any, Any] | None = None) -> None:
|
||||
"""Add set to menu."""
|
||||
details = details if details else {}
|
||||
details['Selected'] = details.get('Selected', False)
|
||||
|
||||
# Safety check
|
||||
if 'Targets' not in details:
|
||||
raise KeyError('Menu set has no targets')
|
||||
|
||||
# Add set
|
||||
self.sets[name] = details
|
||||
|
||||
def add_toggle(self, name: str, details: dict[Any, Any] | None = None) -> None:
|
||||
"""Add toggle to menu."""
|
||||
details = details if details else {}
|
||||
details['Selected'] = details.get('Selected', False)
|
||||
self.toggles[name] = details
|
||||
|
||||
def advanced_select(
|
||||
self,
|
||||
prompt_msg: str = 'Please make a selection: ',
|
||||
) -> tuple[str, dict[Any, Any]]:
|
||||
"""Display menu and make multiple selections, returns tuple.
|
||||
|
||||
NOTE: Menu is displayed until an action entry is selected.
|
||||
"""
|
||||
while True:
|
||||
self._update(single_selection=False)
|
||||
user_selection = self._user_select(prompt_msg)
|
||||
selected_entry = self._resolve_selection(user_selection)
|
||||
if user_selection.isnumeric():
|
||||
# Update selection(s)
|
||||
self._update_entry_selection_status(selected_entry[0])
|
||||
else:
|
||||
# Action selected
|
||||
break
|
||||
|
||||
# Done
|
||||
return selected_entry
|
||||
|
||||
def settings_select(
|
||||
self,
|
||||
prompt_msg: str = 'Please make a selection: ',
|
||||
) -> tuple[str, dict[Any, Any]]:
|
||||
"""Display menu and make multiple selections, returns tuple.
|
||||
|
||||
NOTE: Menu is displayed until an action entry is selected.
|
||||
"""
|
||||
choice_kwargs = {
|
||||
'prompt_msg': 'Toggle or change value?',
|
||||
'choices': ['T', 'C'],
|
||||
}
|
||||
|
||||
while True:
|
||||
self._update(single_selection=True, settings_mode=True)
|
||||
user_selection = self._user_select(prompt_msg)
|
||||
selected_entry = self._resolve_selection(user_selection)
|
||||
if user_selection.isnumeric():
|
||||
if 'Value' in selected_entry[-1] and choice(**choice_kwargs) == 'C':
|
||||
# Change
|
||||
selected_entry[-1]['Value'] = input_text('Enter new value: ')
|
||||
else:
|
||||
# Toggle
|
||||
self._update_entry_selection_status(selected_entry[0])
|
||||
else:
|
||||
# Action selected
|
||||
break
|
||||
|
||||
# Done
|
||||
return selected_entry
|
||||
|
||||
def simple_select(
|
||||
self,
|
||||
prompt_msg: str = 'Please make a selection: ',
|
||||
update: bool = True,
|
||||
) -> tuple[str, dict[Any, Any]]:
|
||||
"""Display menu and make a single selection, returns tuple."""
|
||||
if update:
|
||||
self._update()
|
||||
user_selection = self._user_select(prompt_msg)
|
||||
return self._resolve_selection(user_selection)
|
||||
|
||||
def update(self) -> None:
|
||||
"""Update menu with default settings."""
|
||||
self._update()
|
||||
|
||||
class TryAndPrint():
|
||||
"""Object used to standardize running functions and returning the result.
|
||||
|
||||
The errors and warning attributes are used to allow fine-tuned results
|
||||
based on exception names.
|
||||
"""
|
||||
def __init__(self, msg_bad: str = 'FAILED', msg_good: str = 'SUCCESS'):
|
||||
self.catch_all : bool = True
|
||||
self.indent: int = INDENT
|
||||
self.list_errors: list[str] = ['GenericError']
|
||||
self.list_warnings: list[str] = ['GenericWarning']
|
||||
self.msg_bad: str = msg_bad
|
||||
self.msg_good: str = msg_good
|
||||
self.verbose : bool = False
|
||||
self.width: int = WIDTH
|
||||
|
||||
def _format_exception_message(self, _exception: Exception) -> str:
|
||||
"""Format using the exception's args or name, returns str."""
|
||||
LOG.debug(
|
||||
'Formatting exception: %s, %s',
|
||||
_exception.__class__.__name__,
|
||||
_exception,
|
||||
)
|
||||
message = ''
|
||||
|
||||
# Format message string from _exception
|
||||
try:
|
||||
if isinstance(_exception, subprocess.CalledProcessError):
|
||||
message = _exception.stderr
|
||||
if not isinstance(message, str):
|
||||
message = message.decode('utf-8')
|
||||
message = message.strip()
|
||||
elif isinstance(_exception, ZeroDivisionError):
|
||||
# Skip and just use exception name below
|
||||
pass
|
||||
else:
|
||||
message = str(_exception)
|
||||
except Exception:
|
||||
# Just use the exception name instead
|
||||
pass
|
||||
|
||||
# Prepend exception name
|
||||
if _exception.__class__.__name__ not in ('GenericError', 'GenericWarning'):
|
||||
try:
|
||||
message = f'{_exception.__class__.__name__}: {message}'
|
||||
except Exception:
|
||||
message = f'UNKNOWN ERROR: {message}'
|
||||
|
||||
# Fix multi-line messages
|
||||
if '\n' in message:
|
||||
try:
|
||||
lines = [
|
||||
f'{" "*(self.indent+self.width)}{line.strip()}'
|
||||
for line in message.splitlines() if line.strip()
|
||||
]
|
||||
lines[0] = lines[0].strip()
|
||||
message = '\n'.join(lines)
|
||||
except Exception:
|
||||
pass
|
||||
|
||||
# Done
|
||||
return message
|
||||
|
||||
def _format_function_output(
|
||||
self,
|
||||
output: list | subprocess.CompletedProcess,
|
||||
msg_good: str,
|
||||
) -> str:
|
||||
"""Format function output for use in try_and_print(), returns str."""
|
||||
LOG.debug('Formatting output: %s', output)
|
||||
|
||||
if not output:
|
||||
raise GenericWarning('No output')
|
||||
|
||||
# Ensure we're working with a list
|
||||
if isinstance(output, subprocess.CompletedProcess):
|
||||
stdout = output.stdout
|
||||
if not isinstance(stdout, str):
|
||||
stdout = stdout.decode('utf8')
|
||||
output = stdout.strip().splitlines()
|
||||
if not output:
|
||||
# Going to treat these as successes (for now)
|
||||
LOG.warning('Program output was empty, assuming good result.')
|
||||
return color_string(msg_good, 'GREEN')
|
||||
else:
|
||||
try:
|
||||
output = list(output)
|
||||
except TypeError:
|
||||
output = [output]
|
||||
|
||||
# Safety check
|
||||
if not output:
|
||||
# Going to ignore empty function output for now
|
||||
LOG.error('Output is empty')
|
||||
return 'UNKNOWN'
|
||||
|
||||
# Build result_msg
|
||||
result_msg = f'{output.pop(0)}'
|
||||
if output:
|
||||
output = [f'{" "*(self.indent+self.width)}{line}' for line in output]
|
||||
result_msg += '\n' + '\n'.join(output)
|
||||
|
||||
# Done
|
||||
return result_msg
|
||||
|
||||
def _log_result(self, message: str, result_msg: str) -> None:
|
||||
"""Log result text without color formatting."""
|
||||
log_text = f'{" "*self.indent}{message:<{self.width}}{result_msg}'
|
||||
for line in log_text.splitlines():
|
||||
line = strip_colors(line)
|
||||
LOG.info(line)
|
||||
|
||||
def add_error(self, exception_name: str) -> None:
|
||||
"""Add exception name to error list."""
|
||||
if exception_name not in self.list_errors:
|
||||
self.list_errors.append(exception_name)
|
||||
|
||||
def add_warning(self, exception_name: str) -> None:
|
||||
"""Add exception name to warning list."""
|
||||
if exception_name not in self.list_warnings:
|
||||
self.list_warnings.append(exception_name)
|
||||
|
||||
def run(
|
||||
self,
|
||||
message: str,
|
||||
function: Callable,
|
||||
*args: Iterable[Any],
|
||||
catch_all: bool | None = None,
|
||||
msg_good: str | None = None,
|
||||
verbose: bool | None = None,
|
||||
**kwargs,
|
||||
) -> dict[str, Any]:
|
||||
"""Run a function and print the results, returns results as dict.
|
||||
|
||||
If catch_all is True then (nearly) all exceptions will be caught.
|
||||
Otherwise if an exception occurs that wasn't specified it will be
|
||||
re-raised.
|
||||
|
||||
If the function returns data it will be used instead of msg_good,
|
||||
msg_bad, or exception text.
|
||||
The output should be a list or a subprocess.CompletedProcess object.
|
||||
|
||||
If msg_good is passed it will override self.msg_good.
|
||||
|
||||
If verbose is True then exception names or messages will be used for
|
||||
the result message. Otherwise it will simply be set to result_bad.
|
||||
|
||||
If catch_all and/or verbose are passed it will override
|
||||
self.catch_all and/or self.verbose for this call.
|
||||
|
||||
args and kwargs are passed to the function.
|
||||
"""
|
||||
LOG.debug('function: %s.%s', function.__module__, function.__name__)
|
||||
LOG.debug('args: %s', args)
|
||||
LOG.debug('kwargs: %s', kwargs)
|
||||
LOG.debug(
|
||||
'catch_all: %s, msg_good: %s, verbose: %s',
|
||||
catch_all,
|
||||
msg_good,
|
||||
verbose,
|
||||
)
|
||||
f_exception = None
|
||||
catch_all = catch_all if catch_all is not None else self.catch_all
|
||||
msg_good = msg_good if msg_good is not None else self.msg_good
|
||||
output = None
|
||||
result_msg = 'UNKNOWN'
|
||||
verbose = verbose if verbose is not None else self.verbose
|
||||
|
||||
# Build exception tuples
|
||||
e_exceptions: tuple = tuple(get_exception(e) for e in self.list_errors)
|
||||
w_exceptions: tuple = tuple(get_exception(e) for e in self.list_warnings)
|
||||
|
||||
# Run function and catch exceptions
|
||||
print(f'{" "*self.indent}{message:<{self.width}}', end='', flush=True)
|
||||
LOG.debug('Running function: %s.%s', function.__module__, function.__name__)
|
||||
try:
|
||||
output = function(*args, **kwargs)
|
||||
except w_exceptions as _exception:
|
||||
# Warnings
|
||||
result_msg = self._format_exception_message(_exception)
|
||||
print_warning(result_msg, log=False)
|
||||
f_exception = _exception
|
||||
except e_exceptions as _exception:
|
||||
# Exceptions
|
||||
result_msg = self._format_exception_message(_exception)
|
||||
print_error(result_msg, log=False)
|
||||
f_exception = _exception
|
||||
except Exception as _exception:
|
||||
# Unexpected exceptions
|
||||
if verbose:
|
||||
result_msg = self._format_exception_message(_exception)
|
||||
else:
|
||||
result_msg = self.msg_bad
|
||||
print_error(result_msg, log=False)
|
||||
f_exception = _exception
|
||||
if not catch_all:
|
||||
# Re-raise error as necessary
|
||||
raise
|
||||
else:
|
||||
# Success
|
||||
if output:
|
||||
result_msg = self._format_function_output(output, msg_good)
|
||||
print(result_msg)
|
||||
else:
|
||||
result_msg = msg_good
|
||||
print_success(result_msg, log=False)
|
||||
|
||||
# Done
|
||||
self._log_result(message, result_msg)
|
||||
return {
|
||||
'Exception': f_exception,
|
||||
'Failed': bool(f_exception),
|
||||
'Message': result_msg,
|
||||
'Output': output,
|
||||
}
|
||||
|
||||
|
||||
# Functions
|
||||
def abort(
|
||||
prompt_msg: str = 'Aborted.',
|
||||
show_prompt_msg: bool = True,
|
||||
return_code: int = 1,
|
||||
) -> None:
|
||||
"""Abort script."""
|
||||
print_warning(prompt_msg)
|
||||
if show_prompt_msg:
|
||||
sleep(0.5)
|
||||
pause(prompt_msg='Press Enter to exit... ')
|
||||
sys.exit(return_code)
|
||||
|
||||
|
||||
def ask(prompt_msg: str) -> bool:
|
||||
"""Prompt the user with a Y/N question, returns bool."""
|
||||
validator = InputYesNoValidator()
|
||||
|
||||
# Show prompt
|
||||
response = input_text(f'{prompt_msg} [Y/N]: ', validator=validator)
|
||||
if response.upper().startswith('Y'):
|
||||
LOG.info('%s Yes', prompt_msg)
|
||||
return True
|
||||
if response.upper().startswith('N'):
|
||||
LOG.info('%s No', prompt_msg)
|
||||
return False
|
||||
|
||||
# This shouldn't ever be reached
|
||||
raise ValueError(f'Invalid answer given: {response}')
|
||||
|
||||
|
||||
def beep(repeat: int = 1) -> None:
|
||||
"""Play system bell with optional repeat."""
|
||||
while repeat >= 1:
|
||||
# Print bell char without a newline
|
||||
print('\a', end='', flush=True)
|
||||
sleep(0.5)
|
||||
repeat -= 1
|
||||
|
||||
|
||||
def choice(prompt_msg: str, choices: Iterable[str]) -> str:
|
||||
"""Choose an option from a provided list, returns str.
|
||||
|
||||
Choices provided will be converted to uppercase and returned as such.
|
||||
Similar to the commands choice (Windows) and select (Linux).
|
||||
"""
|
||||
LOG.debug('prompt_msg: %s, choices: %s', prompt_msg, choices)
|
||||
choices = [str(c).upper()[:1] for c in choices]
|
||||
prompt_msg = f'{prompt_msg} [{"/".join(choices)}]'
|
||||
|
||||
# Show prompt
|
||||
response = input_text(prompt_msg, validator=InputChoiceValidator(choices))
|
||||
|
||||
# Done
|
||||
LOG.info('%s %s', prompt_msg, response)
|
||||
return response.upper()
|
||||
|
||||
|
||||
def fix_prompt(message: str) -> str:
|
||||
"""Fix prompt, returns str."""
|
||||
if not message:
|
||||
message = 'Input text: '
|
||||
message = str(message)
|
||||
if message[-1:] != ' ':
|
||||
message += ' '
|
||||
return message
|
||||
|
||||
|
||||
@cache
|
||||
def get_exception(name: str) -> Exception:
|
||||
"""Get exception by name, returns exception object.
|
||||
|
||||
[Doctest]
|
||||
>>> t = TryAndPrint()
|
||||
>>> t._get_exception('AttributeError')
|
||||
<class 'AttributeError'>
|
||||
>>> t._get_exception('CalledProcessError')
|
||||
<class 'subprocess.CalledProcessError'>
|
||||
>>> t._get_exception('GenericError')
|
||||
<class 'wk.std.GenericError'>
|
||||
"""
|
||||
LOG.debug('Getting exception: %s', name)
|
||||
obj = getattr(sys.modules[__name__], name, None)
|
||||
if obj:
|
||||
return obj
|
||||
|
||||
# Try builtin classes
|
||||
obj = getattr(sys.modules['builtins'], name, None)
|
||||
if obj:
|
||||
return obj
|
||||
|
||||
# Try all modules
|
||||
for _mod in sys.modules.values():
|
||||
obj = getattr(_mod, name, None)
|
||||
if obj:
|
||||
break
|
||||
|
||||
# Check if not found
|
||||
if not obj:
|
||||
raise AttributeError(f'Failed to find exception: {name}')
|
||||
|
||||
# Done
|
||||
return obj
|
||||
|
||||
|
||||
def get_ticket_id() -> str:
|
||||
"""Get ticket ID, returns str."""
|
||||
prompt_msg = 'Please enter ticket ID:'
|
||||
validator = InputTicketIDValidator()
|
||||
|
||||
# Show prompt
|
||||
ticket_id = input_text(prompt_msg, validator=validator)
|
||||
|
||||
# Done
|
||||
return ticket_id
|
||||
|
||||
|
||||
def input_text(
|
||||
prompt_msg: str = 'Enter text: ',
|
||||
allow_empty: bool = False,
|
||||
validator: Validator | None = None,
|
||||
) -> str:
|
||||
"""Get input from user, returns str."""
|
||||
prompt_msg = fix_prompt(prompt_msg)
|
||||
|
||||
# Accept empty responses?
|
||||
if not (allow_empty or validator):
|
||||
validator = InputNotEmptyValidator()
|
||||
|
||||
# Show prompt
|
||||
result = None
|
||||
while result is None:
|
||||
try:
|
||||
result = prompt(prompt_msg, validator=validator)
|
||||
except KeyboardInterrupt:
|
||||
# Ignore CTRL+c
|
||||
pass
|
||||
|
||||
# Done
|
||||
return result
|
||||
|
||||
|
||||
def major_exception() -> None:
|
||||
"""Display traceback, optionally upload detailes, and exit."""
|
||||
LOG.critical('Major exception encountered', exc_info=True)
|
||||
print_error('Major exception', log=False)
|
||||
print_warning(SUPPORT_MESSAGE)
|
||||
if ENABLED_UPLOAD_DATA:
|
||||
print_warning('Also, please run upload-logs to help debugging!')
|
||||
print(traceback.format_exc())
|
||||
|
||||
# Done
|
||||
pause('Press Enter to exit... ')
|
||||
raise SystemExit(1)
|
||||
|
||||
|
||||
def pause(prompt_msg: str = 'Press Enter to continue... ') -> None:
|
||||
"""Simple pause implementation."""
|
||||
input_text(prompt_msg, allow_empty=True)
|
||||
|
||||
|
||||
def print_colored(
|
||||
strings: Iterable[str] | str,
|
||||
colors: Iterable[str | None] | str,
|
||||
log: bool = False,
|
||||
sep: str = ' ',
|
||||
**kwargs,
|
||||
) -> None:
|
||||
"""Prints strings in the colors specified."""
|
||||
LOG.debug(
|
||||
'strings: %s, colors: %s, sep: %s, kwargs: %s',
|
||||
strings, colors, sep, kwargs,
|
||||
)
|
||||
msg = color_string(strings, colors, sep=sep)
|
||||
print_options = {
|
||||
'end': kwargs.get('end', '\n'),
|
||||
'file': kwargs.get('file', sys.stdout),
|
||||
'flush': kwargs.get('flush', False),
|
||||
}
|
||||
|
||||
print(msg, **print_options)
|
||||
if log:
|
||||
LOG.info(strip_colors(msg))
|
||||
|
||||
|
||||
def print_error(msg: str, log: bool = True, **kwargs) -> None:
|
||||
"""Prints message in RED and log as ERROR."""
|
||||
if 'file' not in kwargs:
|
||||
# Only set if not specified
|
||||
kwargs['file'] = sys.stderr
|
||||
print_colored(msg, 'RED', **kwargs)
|
||||
if log:
|
||||
LOG.error(msg)
|
||||
|
||||
|
||||
def print_info(msg: str, log: bool = True, **kwargs) -> None:
|
||||
"""Prints message in BLUE and log as INFO."""
|
||||
print_colored(msg, 'BLUE', **kwargs)
|
||||
if log:
|
||||
LOG.info(msg)
|
||||
|
||||
|
||||
def print_report(report: list[str], indent=None, log: bool = True) -> None:
|
||||
"""Print report to screen and optionally to log."""
|
||||
for line in report:
|
||||
if indent:
|
||||
line = f'{" "*indent}{line}'
|
||||
print(line)
|
||||
if log:
|
||||
LOG.info(strip_colors(line))
|
||||
|
||||
|
||||
def print_standard(msg: str, log: bool = True, **kwargs) -> None:
|
||||
"""Prints message and log as INFO."""
|
||||
print(msg, **kwargs)
|
||||
if log:
|
||||
LOG.info(msg)
|
||||
|
||||
|
||||
def print_success(msg: str, log: bool = True, **kwargs) -> None:
|
||||
"""Prints message in GREEN and log as INFO."""
|
||||
print_colored(msg, 'GREEN', **kwargs)
|
||||
if log:
|
||||
LOG.info(msg)
|
||||
|
||||
|
||||
def print_warning(msg: str, log: bool = True, **kwargs) -> None:
|
||||
"""Prints message in YELLOW and log as WARNING."""
|
||||
if 'file' not in kwargs:
|
||||
# Only set if not specified
|
||||
kwargs['file'] = sys.stderr
|
||||
print_colored(msg, 'YELLOW', **kwargs)
|
||||
if log:
|
||||
LOG.warning(msg)
|
||||
|
||||
|
||||
def set_title(title: str) -> None:
|
||||
"""Set window title."""
|
||||
LOG.debug('title: %s', title)
|
||||
if os.name == 'nt':
|
||||
os.system(f'title {title}')
|
||||
else:
|
||||
print_error('Setting the title is only supported under Windows.')
|
||||
|
||||
|
||||
def show_data(
|
||||
message: str,
|
||||
data: Any,
|
||||
color: str | None = None,
|
||||
indent: int | None = None,
|
||||
width: int | None = None,
|
||||
) -> None:
|
||||
"""Display info using default or provided indent and width."""
|
||||
indent = INDENT if indent is None else indent
|
||||
width = WIDTH if width is None else width
|
||||
print_colored(
|
||||
(f'{" "*indent}{message:<{width}}', data),
|
||||
(None, color if color else None),
|
||||
log=True,
|
||||
sep='',
|
||||
)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
@ -1,382 +0,0 @@
|
|||
"""WizardKit: TUI functions"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import atexit
|
||||
import logging
|
||||
import time
|
||||
|
||||
from copy import deepcopy
|
||||
from os import environ
|
||||
from typing import Any
|
||||
|
||||
from wk.exe import start_thread
|
||||
from wk.std import sleep
|
||||
from wk.ui import ansi, tmux
|
||||
|
||||
# STATIC VARIABLES
|
||||
LOG = logging.getLogger(__name__)
|
||||
TMUX_SIDE_WIDTH = 21
|
||||
TMUX_TITLE_HEIGHT = 2
|
||||
TMUX_LAYOUT = { # NOTE: This needs to be in order from top to bottom
|
||||
'Title': {'Panes': [], 'height': TMUX_TITLE_HEIGHT},
|
||||
'Info': {'Panes': []},
|
||||
'Current': {'Panes': [environ.get('TMUX_PANE', None)]},
|
||||
'Workers': {'Panes': []},
|
||||
'Started': {'Panes': [], 'height': TMUX_TITLE_HEIGHT},
|
||||
'Progress': {'Panes': [], 'width': TMUX_SIDE_WIDTH},
|
||||
}
|
||||
|
||||
|
||||
# Classes
|
||||
class TUI():
|
||||
"""Object for tracking TUI elements."""
|
||||
def __init__(self, title_text: str | None = None):
|
||||
self.clear_on_resize = False
|
||||
self.layout: dict[str, dict[str, Any]] = deepcopy(TMUX_LAYOUT)
|
||||
self.side_width: int = TMUX_SIDE_WIDTH
|
||||
self.title_text: str = title_text if title_text else 'Title Text'
|
||||
self.title_text_line2: str = ''
|
||||
self.title_colors: list[str] = ['BLUE', '']
|
||||
|
||||
# Init tmux and start a background process to maintain layout
|
||||
self.init_tmux()
|
||||
start_thread(self.fix_layout_loop)
|
||||
|
||||
# Close all panes at exit
|
||||
atexit.register(tmux.kill_all_panes)
|
||||
|
||||
def add_info_pane(
|
||||
self,
|
||||
lines: int | None = None,
|
||||
percent: int = 0,
|
||||
update_layout: bool = True,
|
||||
**tmux_args,
|
||||
) -> None:
|
||||
"""Add info pane."""
|
||||
if not (lines or percent):
|
||||
# Bail early
|
||||
raise RuntimeError('Neither lines nor percent specified.')
|
||||
|
||||
# Calculate lines if needed
|
||||
if not lines:
|
||||
lines = int(tmux.get_pane_size()[1] * (percent/100))
|
||||
|
||||
# Set tmux split args
|
||||
tmux_args.update({
|
||||
'behind': True,
|
||||
'lines': lines,
|
||||
'target_id': None,
|
||||
'vertical': True,
|
||||
})
|
||||
if self.layout['Info']['Panes']:
|
||||
tmux_args.update({
|
||||
'behind': False,
|
||||
'percent': 50,
|
||||
'target_id': self.layout['Info']['Panes'][-1],
|
||||
'vertical': False,
|
||||
})
|
||||
tmux_args.pop('lines')
|
||||
|
||||
# Update layout
|
||||
if update_layout:
|
||||
self.layout['Info']['height'] = lines
|
||||
|
||||
# Add pane
|
||||
self.layout['Info']['Panes'].append(tmux.split_window(**tmux_args))
|
||||
|
||||
def add_title_pane(
|
||||
self,
|
||||
line1: str,
|
||||
line2: str | None = None,
|
||||
colors: list[str] | None = None,
|
||||
) -> None:
|
||||
"""Add pane to title row."""
|
||||
lines = [line1, line2]
|
||||
colors = colors if colors else self.title_colors.copy()
|
||||
if not line2:
|
||||
lines.pop()
|
||||
colors.pop()
|
||||
tmux_args = {
|
||||
'behind': True,
|
||||
'lines': TMUX_TITLE_HEIGHT,
|
||||
'target_id': None,
|
||||
'text': ansi.color_string(lines, colors, sep='\n'),
|
||||
'vertical': True,
|
||||
}
|
||||
if self.layout['Title']['Panes']:
|
||||
tmux_args.update({
|
||||
'behind': False,
|
||||
'percent': 50,
|
||||
'target_id': self.layout['Title']['Panes'][-1],
|
||||
'vertical': False,
|
||||
})
|
||||
tmux_args.pop('lines')
|
||||
|
||||
# Add pane
|
||||
self.layout['Title']['Panes'].append(tmux.split_window(**tmux_args))
|
||||
|
||||
def add_worker_pane(
|
||||
self,
|
||||
lines: int | None = None,
|
||||
percent: int = 0,
|
||||
update_layout: bool = True,
|
||||
**tmux_args,
|
||||
) -> None:
|
||||
"""Add worker pane."""
|
||||
height = lines
|
||||
|
||||
# Bail early
|
||||
if not (lines or percent):
|
||||
raise RuntimeError('Neither lines nor percent specified.')
|
||||
|
||||
# Calculate height if needed
|
||||
if not height:
|
||||
height = int(tmux.get_pane_size()[1] * (percent/100))
|
||||
|
||||
# Set tmux split args
|
||||
tmux_args.update({
|
||||
'behind': False,
|
||||
'lines': lines,
|
||||
'percent': percent if percent else None,
|
||||
'target_id': None,
|
||||
'vertical': True,
|
||||
})
|
||||
|
||||
# Update layout
|
||||
if update_layout:
|
||||
self.layout['Workers']['height'] = height
|
||||
|
||||
# Add pane (ensure panes are sorted top to bottom)
|
||||
self.layout['Workers']['Panes'].insert(0, tmux.split_window(**tmux_args))
|
||||
|
||||
def clear_current_pane(self) -> None:
|
||||
"""Clear screen and history for current pane."""
|
||||
tmux.clear_pane()
|
||||
|
||||
def clear_current_pane_height(self) -> None:
|
||||
"""Clear current pane height and update layout."""
|
||||
self.layout['Current'].pop('height', None)
|
||||
|
||||
def fix_layout(self, forced: bool = True) -> None:
|
||||
"""Fix tmux layout based on self.layout."""
|
||||
try:
|
||||
tmux.fix_layout(self.layout, clear_on_resize=self.clear_on_resize, forced=forced)
|
||||
except RuntimeError:
|
||||
# Assuming self.panes changed while running
|
||||
pass
|
||||
|
||||
def fix_layout_loop(self) -> None:
|
||||
"""Fix layout on a loop.
|
||||
|
||||
NOTE: This should be called as a thread.
|
||||
"""
|
||||
while True:
|
||||
self.fix_layout(forced=False)
|
||||
sleep(1)
|
||||
|
||||
def init_tmux(self) -> None:
|
||||
"""Initialize tmux layout."""
|
||||
tmux.kill_all_panes()
|
||||
self.layout.clear()
|
||||
self.layout.update(deepcopy(TMUX_LAYOUT))
|
||||
|
||||
# Progress
|
||||
self.layout['Progress']['Panes'].append(tmux.split_window(
|
||||
lines=TMUX_SIDE_WIDTH,
|
||||
text=' ',
|
||||
))
|
||||
|
||||
# Started
|
||||
self.layout['Started']['Panes'].append(tmux.split_window(
|
||||
behind=True,
|
||||
lines=2,
|
||||
target_id=self.layout['Progress']['Panes'][0],
|
||||
text=ansi.color_string(
|
||||
['Started', time.strftime("%Y-%m-%d %H:%M %Z")],
|
||||
['BLUE', None],
|
||||
sep='\n',
|
||||
),
|
||||
vertical=True,
|
||||
))
|
||||
|
||||
# Title
|
||||
self.layout['Title']['Panes'].append(tmux.split_window(
|
||||
behind=True,
|
||||
lines=2,
|
||||
vertical=True,
|
||||
text=ansi.color_string(
|
||||
[self.title_text, self.title_text_line2],
|
||||
self.title_colors,
|
||||
sep = '\n',
|
||||
),
|
||||
))
|
||||
|
||||
# Done
|
||||
sleep(0.2)
|
||||
|
||||
def remove_all_info_panes(self) -> None:
|
||||
"""Remove all info panes and update layout."""
|
||||
self.layout['Info'].pop('height', None)
|
||||
panes = self.layout['Info']['Panes'].copy()
|
||||
self.layout['Info']['Panes'].clear()
|
||||
tmux.kill_pane(*panes)
|
||||
|
||||
def remove_all_worker_panes(self) -> None:
|
||||
"""Remove all worker panes and update layout."""
|
||||
self.layout['Workers'].pop('height', None)
|
||||
panes = self.layout['Workers']['Panes'].copy()
|
||||
self.layout['Workers']['Panes'].clear()
|
||||
tmux.kill_pane(*panes)
|
||||
|
||||
def reset_title_pane(
|
||||
self,
|
||||
line1: str = 'Title Text',
|
||||
line2: str = '',
|
||||
colors: list[str] | None = None,
|
||||
) -> None:
|
||||
"""Remove all extra title panes, reset main title pane, and update layout."""
|
||||
colors = self.title_colors if colors is None else colors
|
||||
panes = self.layout['Title']['Panes'].copy()
|
||||
if len(panes) > 1:
|
||||
tmux.kill_pane(*panes[1:])
|
||||
self.layout['Title']['Panes'] = panes[:1]
|
||||
self.set_title(line1, line2, colors)
|
||||
|
||||
def set_current_pane_height(self, height: int) -> None:
|
||||
"""Set current pane height and update layout."""
|
||||
self.layout['Current']['height'] = height
|
||||
tmux.resize_pane(height=height)
|
||||
|
||||
def set_progress_file(self, progress_file: str) -> None:
|
||||
"""Set the file to use for the progresse pane."""
|
||||
tmux.respawn_pane(
|
||||
pane_id=self.layout['Progress']['Panes'][0],
|
||||
watch_file=progress_file,
|
||||
)
|
||||
|
||||
def set_title(
|
||||
self,
|
||||
line1: str,
|
||||
line2: str | None = None,
|
||||
colors: list[str] | None = None,
|
||||
) -> None:
|
||||
"""Set title text."""
|
||||
self.title_text = line1
|
||||
self.title_text_line2 = line2 if line2 else ''
|
||||
if colors:
|
||||
self.title_colors = colors
|
||||
|
||||
# Update pane (if present)
|
||||
if self.layout['Title']['Panes']:
|
||||
tmux.respawn_pane(
|
||||
pane_id=self.layout['Title']['Panes'][0],
|
||||
text=ansi.color_string(
|
||||
[self.title_text, self.title_text_line2],
|
||||
self.title_colors,
|
||||
sep = '\n',
|
||||
),
|
||||
)
|
||||
|
||||
def update_clock(self) -> None:
|
||||
"""Update 'Started' pane following clock sync."""
|
||||
tmux.respawn_pane(
|
||||
pane_id=self.layout['Started']['Panes'][0],
|
||||
text=ansi.color_string(
|
||||
['Started', time.strftime("%Y-%m-%d %H:%M %Z")],
|
||||
['BLUE', None],
|
||||
sep='\n',
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
# Functions
|
||||
def fix_layout(layout, forced: bool = False) -> None:
|
||||
"""Fix pane sizes based on layout."""
|
||||
resize_kwargs = []
|
||||
|
||||
# Bail early
|
||||
if not (forced or layout_needs_fixed(layout)):
|
||||
# Layout should be fine
|
||||
return
|
||||
|
||||
# Remove closed panes
|
||||
for data in layout.values():
|
||||
data['Panes'] = [pane for pane in data['Panes'] if tmux.poll_pane(pane)]
|
||||
|
||||
# Update main panes
|
||||
for section, data in layout.items():
|
||||
if section == 'Workers':
|
||||
# Skip for now
|
||||
continue
|
||||
|
||||
if 'height' in data:
|
||||
for pane_id in data['Panes']:
|
||||
resize_kwargs.append({'pane_id': pane_id, 'height': data['height']})
|
||||
if 'width' in data:
|
||||
for pane_id in data['Panes']:
|
||||
resize_kwargs.append({'pane_id': pane_id, 'width': data['width']})
|
||||
for kwargs in resize_kwargs:
|
||||
try:
|
||||
tmux.resize_pane(**kwargs)
|
||||
except RuntimeError:
|
||||
# Assuming pane was closed just before resizing
|
||||
pass
|
||||
|
||||
# Update "group" panes widths
|
||||
for group in ('Title', 'Info'):
|
||||
num_panes = len(layout[group]['Panes'])
|
||||
if num_panes <= 1:
|
||||
continue
|
||||
width = int( (tmux.get_pane_size()[0] - (1 - num_panes)) / num_panes )
|
||||
for pane_id in layout[group]['Panes']:
|
||||
tmux.resize_pane(pane_id, width=width)
|
||||
if group == 'Title':
|
||||
# (re)fix Started pane
|
||||
tmux.resize_pane(layout['Started']['Panes'][0], width=TMUX_SIDE_WIDTH)
|
||||
|
||||
# Bail early
|
||||
if not (layout['Workers']['Panes'] and layout['Workers']['height']):
|
||||
return
|
||||
|
||||
# Update worker heights
|
||||
worker_height = layout['Workers']['height']
|
||||
workers = layout['Workers']['Panes'].copy()
|
||||
num_workers = len(workers)
|
||||
avail_height = sum(tmux.get_pane_size(pane)[1] for pane in workers)
|
||||
avail_height += tmux.get_pane_size()[1] # Current pane
|
||||
# Check if window is too small
|
||||
if avail_height < (worker_height*num_workers) + 3:
|
||||
# Just leave things as-is
|
||||
return
|
||||
# Resize current pane
|
||||
tmux.resize_pane(height=avail_height-(worker_height*num_workers))
|
||||
# Resize bottom pane
|
||||
tmux.resize_pane(workers.pop(0), height=worker_height)
|
||||
# Resize the rest of the panes by adjusting the ones above them
|
||||
while len(workers) > 1:
|
||||
next_height = sum(tmux.get_pane_size(pane)[1] for pane in workers[:2])
|
||||
next_height -= worker_height
|
||||
tmux.resize_pane(workers[1], height=next_height)
|
||||
workers.pop(0)
|
||||
|
||||
def layout_needs_fixed(layout) -> bool:
|
||||
"""Check if layout needs fixed, returns bool."""
|
||||
needs_fixed = False
|
||||
|
||||
# Check panes
|
||||
for data in layout.values():
|
||||
if 'height' in data:
|
||||
needs_fixed = needs_fixed or any(
|
||||
tmux.get_pane_size(pane)[1] != data['height'] for pane in data['Panes']
|
||||
)
|
||||
if 'width' in data:
|
||||
needs_fixed = needs_fixed or any(
|
||||
tmux.get_pane_size(pane)[0] != data['width'] for pane in data['Panes']
|
||||
)
|
||||
|
||||
# Done
|
||||
return needs_fixed
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
print("This file is not meant to be called directly.")
|
||||
|
|
@ -1,68 +0,0 @@
|
|||
#!/usr/bin/env python3
|
||||
"""WizardKit: Debug Launcher"""
|
||||
# vim: sts=2 sw=2 ts=2
|
||||
|
||||
import pathlib
|
||||
import pickle
|
||||
|
||||
import wk
|
||||
|
||||
|
||||
# STATIC VARIABLES
|
||||
OPTIONS = {
|
||||
'Cloning / Imaging': 'ddrescue-TUI',
|
||||
'Hardware Diagnostics': 'Hardware-Diagnostics',
|
||||
}
|
||||
|
||||
|
||||
def get_debug_prefix() -> str:
|
||||
"""Ask what we're debugging, returns log dir prefix."""
|
||||
menu = wk.ui.cli.Menu(title=f'{wk.cfg.main.KIT_NAME_FULL}: Debugging Menu\n')
|
||||
for name, prefix in OPTIONS.items():
|
||||
menu.add_option(name, {'Prefix': prefix})
|
||||
selection = menu.simple_select()
|
||||
return selection[1]['Prefix']
|
||||
|
||||
def get_debug_path() -> pathlib.Path:
|
||||
"""Get debug path."""
|
||||
log_dir = pathlib.Path('~/Logs').expanduser().resolve()
|
||||
debug_paths = []
|
||||
prefix = get_debug_prefix()
|
||||
|
||||
# Build list of options
|
||||
for item in log_dir.iterdir():
|
||||
if item.is_dir() and item.name.startswith(prefix):
|
||||
debug_paths.append(item.joinpath('debug'))
|
||||
debug_paths = [item for item in debug_paths if item.exists()]
|
||||
debug_paths.sort()
|
||||
|
||||
# Safety check
|
||||
if not debug_paths:
|
||||
wk.ui.cli.abort('No logs found, aborting.')
|
||||
|
||||
# Use latest option
|
||||
if wk.ui.cli.ask('Use latest session?'):
|
||||
return debug_paths[-1]
|
||||
|
||||
# Select from list
|
||||
menu = wk.ui.cli.Menu(title=f'{wk.cfg.main.KIT_NAME_FULL}: Debugging Menu\n')
|
||||
for item in debug_paths:
|
||||
menu.add_option(item.parent.name, {'Path': item})
|
||||
selection = menu.simple_select()
|
||||
|
||||
# Done
|
||||
return selection[1]['Path']
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
# Leaving this at the global level
|
||||
try:
|
||||
debug_path = get_debug_path()
|
||||
except SystemExit:
|
||||
pass
|
||||
else:
|
||||
state = None
|
||||
|
||||
# Load pickle
|
||||
with open(debug_path.joinpath('state.pickle'), 'rb') as f:
|
||||
state = pickle.load(f)
|
||||
|
|
@ -35,9 +35,9 @@ fi
|
|||
function ask() {
|
||||
while :; do
|
||||
read -p "$1 [Y/N] " -r answer
|
||||
if echo "$answer" | grep -Eiq '^(y|yes|sure)$'; then
|
||||
if echo "$answer" | egrep -iq '^(y|yes|sure)$'; then
|
||||
return 0
|
||||
elif echo "$answer" | grep -Eiq '^(n|no|nope)$'; then
|
||||
elif echo "$answer" | egrep -iq '^(n|no|nope)$'; then
|
||||
return 1
|
||||
fi
|
||||
done
|
||||
|
|
@ -67,7 +67,7 @@ function fix_kit_permissions() {
|
|||
function load_settings() {
|
||||
dos2unix "$ROOT_DIR/scripts/wk/cfg/main.py"
|
||||
while read line; do
|
||||
if echo "$line" | grep -Eq "^\w+='"; then
|
||||
if echo "$line" | egrep -q "^\w+='"; then
|
||||
line="$(echo "$line" | sed -r 's/[\r\n]+//')"
|
||||
eval "$line"
|
||||
fi
|
||||
|
|
@ -76,25 +76,15 @@ function load_settings() {
|
|||
|
||||
function copy_live_env() {
|
||||
echo "Copying Archlinux files..."
|
||||
rsync -aI "$ROOT_DIR/setup/linux/profile/" "$PROFILE_DIR/"
|
||||
rsync -aI "$ROOT_DIR/setup/linux/profile_base/" "$PROFILE_DIR/"
|
||||
|
||||
# Add items
|
||||
if [[ "${1:-}" != "--minimal" ]]; then
|
||||
rsync -aI "$ROOT_DIR/setup/linux/profile_gui/" "$PROFILE_DIR/"
|
||||
fi
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/local/bin"
|
||||
rsync -aI "$ROOT_DIR/scripts/" "$PROFILE_DIR/airootfs/usr/local/bin/"
|
||||
|
||||
echo "Copying WizardKit UFD files..."
|
||||
rsync -aI --exclude="macOS-boot-icon.tar" "$ROOT_DIR/setup/ufd/" "$PROFILE_DIR/airootfs/usr/share/WizardKit/"
|
||||
tar xaf "$ROOT_DIR/setup/ufd/macOS-boot-icon.tar" -C "$PROFILE_DIR/airootfs/usr/share/WizardKit"
|
||||
cp "$ROOT_DIR/images/rEFInd.png" "$PROFILE_DIR/airootfs/usr/share/WizardKit/EFI/Boot/rEFInd.png"
|
||||
cp "$ROOT_DIR/images/Syslinux.png" "$PROFILE_DIR/airootfs/usr/share/WizardKit/syslinux/syslinux.png"
|
||||
|
||||
echo "Copying Memtest86+ files..."
|
||||
rsync -aI "/boot/memtest86+/memtest.bin" "$PROFILE_DIR/airootfs/usr/share/WizardKit/syslinux/"
|
||||
rsync -aI "/boot/memtest86+/memtest.efi" "$PROFILE_DIR/airootfs/usr/share/WizardKit/EFI/Memtest86+/"
|
||||
mv "$PROFILE_DIR/airootfs/usr/share/WizardKit/EFI/Memtest86+"/{memtest.efi,bootx64.efi}
|
||||
|
||||
# Pre-compile Python scripts
|
||||
unset PYTHONPYCACHEPREFIX
|
||||
python -m compileall "$PROFILE_DIR/airootfs/usr/local/bin/"
|
||||
|
||||
# Update profiledef.sh to set proper permissions for executable files
|
||||
for _file in $(find "$PROFILE_DIR/airootfs" -executable -type f | sed "s%$PROFILE_DIR/airootfs%%" | sort); do
|
||||
sed -i "\$i\ [\"$_file\"]=\"0:0:755\"" "$PROFILE_DIR/profiledef.sh"
|
||||
|
|
@ -127,21 +117,49 @@ function update_live_env() {
|
|||
username="tech"
|
||||
label="${KIT_NAME_SHORT}_LINUX"
|
||||
|
||||
# Boot config
|
||||
cp "$ROOT_DIR/images/Syslinux.png" "$PROFILE_DIR/syslinux/splash.png"
|
||||
sed -i -r "s/___+/${KIT_NAME_FULL}/" "$PROFILE_DIR/airootfs/usr/share/WizardKit/syslinux/syslinux.cfg"
|
||||
|
||||
# MOTD
|
||||
sed -i -r "s/KIT_NAME_SHORT/$KIT_NAME_SHORT/" "$PROFILE_DIR/profiledef.sh"
|
||||
sed -i -r "s/KIT_NAME_FULL/$KIT_NAME_SHORT/" "$PROFILE_DIR/profiledef.sh"
|
||||
sed -i -r "s/SUPPORT_URL/$KIT_NAME_SHORT/" "$PROFILE_DIR/profiledef.sh"
|
||||
|
||||
# Boot config (legacy)
|
||||
mkdir -p "$PROFILE_DIR/syslinux/wimboot"
|
||||
cp "$ROOT_DIR/images/Pxelinux.png" "$PROFILE_DIR/syslinux/pxelinux.png"
|
||||
cp "$ROOT_DIR/images/Syslinux.png" "$PROFILE_DIR/syslinux/syslinux.png"
|
||||
sed -i -r "s/__+/$KIT_NAME_FULL/" "$PROFILE_DIR/syslinux/syslinux.cfg"
|
||||
mkdir -p "$TEMP_DIR" 2>/dev/null
|
||||
curl -Lo "$TEMP_DIR/wimboot.zip" "http://git.ipxe.org/releases/wimboot/wimboot-latest.zip"
|
||||
7z e -aoa "$TEMP_DIR/wimboot.zip" -o"$PROFILE_DIR/syslinux/wimboot" 'wimboot*/LICENSE.txt' 'wimboot*/README.txt' 'wimboot*/wimboot'
|
||||
|
||||
# Boot config (UEFI)
|
||||
cp "/usr/share/refind/refind_x64.efi" "$PROFILE_DIR/EFI/boot/bootx64.efi"
|
||||
cp "$ROOT_DIR/images/rEFInd.png" "$PROFILE_DIR/EFI/boot/rEFInd.png"
|
||||
rsync -aI "/usr/share/refind/drivers_x64/" "$PROFILE_DIR/EFI/boot/drivers_x64/"
|
||||
rsync -aI "/usr/share/refind/icons/" "$PROFILE_DIR/EFI/boot/icons/" --exclude "/usr/share/refind/icons/svg"
|
||||
sed -i "s/%ARCHISO_LABEL%/${label}/" "$PROFILE_DIR/EFI/boot/refind.conf"
|
||||
|
||||
# Memtest86
|
||||
mkdir -p "$PROFILE_DIR/EFI/memtest86/Benchmark"
|
||||
mkdir -p "$TEMP_DIR/memtest86"
|
||||
curl -Lo "$TEMP_DIR/memtest86/memtest86-usb.zip" "https://www.memtest86.com/downloads/memtest86-usb.zip"
|
||||
7z e -aoa "$TEMP_DIR/memtest86/memtest86-usb.zip" -o"$TEMP_DIR/memtest86" "memtest86-usb.img"
|
||||
7z e -aoa "$TEMP_DIR/memtest86/memtest86-usb.img" -o"$TEMP_DIR/memtest86" "MemTest86.img"
|
||||
7z x -aoa "$TEMP_DIR/memtest86/MemTest86.img" -o"$TEMP_DIR/memtest86"
|
||||
rm "$TEMP_DIR/memtest86/EFI/BOOT/BOOTIA32.efi"
|
||||
mv "$TEMP_DIR/memtest86/EFI/BOOT/BOOTX64.efi" "$PROFILE_DIR/EFI/memtest86/memtestx64.efi"
|
||||
mv "$TEMP_DIR/memtest86/EFI/BOOT"/* "$PROFILE_DIR/EFI/memtest86"/
|
||||
mv "$TEMP_DIR/memtest86/help"/* "$PROFILE_DIR/EFI/memtest86"/
|
||||
mv "$TEMP_DIR/memtest86/license.rtf" "$PROFILE_DIR/EFI/memtest86"/
|
||||
|
||||
# Hostname
|
||||
echo "$hostname" > "$PROFILE_DIR/airootfs/etc/hostname"
|
||||
echo "127.0.1.1 $hostname.localdomain $hostname" >> "$PROFILE_DIR/airootfs/etc/hosts"
|
||||
|
||||
# Live packages
|
||||
cp "$ROOT_DIR/setup/linux/packages/base" "$PROFILE_DIR/packages.x86_64"
|
||||
if [[ "${1:-}" != "--minimal" ]]; then
|
||||
cat "$ROOT_DIR/setup/linux/packages/gui" >> "$PROFILE_DIR/packages.x86_64"
|
||||
fi
|
||||
echo "[custom]" >> "$PROFILE_DIR/pacman.conf"
|
||||
echo "SigLevel = Optional TrustAll" >> "$PROFILE_DIR/pacman.conf"
|
||||
echo "Server = file://$REPO_DIR" >> "$PROFILE_DIR/pacman.conf"
|
||||
|
|
@ -154,25 +172,30 @@ function update_live_env() {
|
|||
# MOTD
|
||||
sed -i -r "s/_+/$KIT_NAME_FULL Linux Environment/" "$PROFILE_DIR/airootfs/etc/motd"
|
||||
|
||||
# Network
|
||||
ln -s "/run/systemd/resolve/stub-resolv.conf" "$PROFILE_DIR/airootfs/etc/resolv.conf"
|
||||
|
||||
# Oh My ZSH
|
||||
git clone --depth=1 https://github.com/robbyrussell/oh-my-zsh.git "$SKEL_DIR/.oh-my-zsh"
|
||||
rm -Rf "$SKEL_DIR/.oh-my-zsh/.git"
|
||||
curl -o "$SKEL_DIR/.oh-my-zsh/themes/lean.zsh-theme" https://raw.githubusercontent.com/miekg/lean/master/prompt_lean_setup
|
||||
|
||||
# Openbox theme
|
||||
git clone --depth=1 https://github.com/addy-dclxvi/Openbox-Theme-Collections.git "$TEMP_DIR/ob-themes"
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/themes"
|
||||
cp -a "$TEMP_DIR/ob-themes/Triste-Orange" "$PROFILE_DIR/airootfs/usr/share/themes/"
|
||||
if [[ "${1:-}" != "--minimal" ]]; then
|
||||
# Openbox theme
|
||||
git clone --depth=1 https://github.com/addy-dclxvi/Openbox-Theme-Collections.git "$TEMP_DIR/ob-themes"
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/themes"
|
||||
cp -a "$TEMP_DIR/ob-themes/Triste-Orange" "$PROFILE_DIR/airootfs/usr/share/themes/"
|
||||
|
||||
# Rofi
|
||||
## Probably don't need the exact commit but it'll be fine
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/fonts/"
|
||||
curl -Lo \
|
||||
"$PROFILE_DIR/airootfs/usr/share/fonts/Fantasque-Sans-Mono-Nerd-Font.ttf" \
|
||||
"https://github.com/adi1090x/rofi/raw/9c4093c665326bb08d6affc7e16d18d8f25c4452/fonts/Fantasque-Sans-Mono-Nerd-Font.ttf"
|
||||
curl -Lo \
|
||||
"$PROFILE_DIR/airootfs/usr/share/fonts/Feather.ttf" \
|
||||
"https://github.com/adi1090x/rofi/raw/9c4093c665326bb08d6affc7e16d18d8f25c4452/fonts/Feather.ttf"
|
||||
# Rofi
|
||||
## Probably don't need the exact commit but it'll be fine
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/fonts/"
|
||||
curl -Lo \
|
||||
"$PROFILE_DIR/airootfs/usr/share/fonts/Fantasque-Sans-Mono-Nerd-Font.ttf" \
|
||||
"https://github.com/adi1090x/rofi/raw/9c4093c665326bb08d6affc7e16d18d8f25c4452/fonts/Fantasque-Sans-Mono-Nerd-Font.ttf"
|
||||
curl -Lo \
|
||||
"$PROFILE_DIR/airootfs/usr/share/fonts/Feather.ttf" \
|
||||
"https://github.com/adi1090x/rofi/raw/9c4093c665326bb08d6affc7e16d18d8f25c4452/fonts/Feather.ttf"
|
||||
fi
|
||||
|
||||
# SSH
|
||||
mkdir -p "$SKEL_DIR/.ssh"
|
||||
|
|
@ -191,13 +214,15 @@ function update_live_env() {
|
|||
# Timezone
|
||||
ln -sf "/usr/share/zoneinfo/$LINUX_TIME_ZONE" "$PROFILE_DIR/airootfs/etc/localtime"
|
||||
|
||||
# VNC password
|
||||
mkdir "$SKEL_DIR/.vnc"
|
||||
echo "$TECH_PASSWORD" | vncpasswd -f > "$SKEL_DIR/.vnc/passwd"
|
||||
if [[ "${1:-}" != "--minimal" ]]; then
|
||||
# VNC password
|
||||
mkdir "$SKEL_DIR/.vnc"
|
||||
echo "$TECH_PASSWORD" | vncpasswd -f > "$SKEL_DIR/.vnc/passwd"
|
||||
|
||||
# Wallpaper
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/wallpaper"
|
||||
cp "$ROOT_DIR/images/Linux.png" "$PROFILE_DIR/airootfs/usr/share/wallpaper/burned.in"
|
||||
# Wallpaper
|
||||
mkdir -p "$PROFILE_DIR/airootfs/usr/share/wallpaper"
|
||||
cp "$ROOT_DIR/images/Linux.png" "$PROFILE_DIR/airootfs/usr/share/wallpaper/burned.in"
|
||||
fi
|
||||
|
||||
# udevil
|
||||
mkdir -p "$PROFILE_DIR/airootfs/media"
|
||||
|
|
@ -275,9 +300,7 @@ function install_deps() {
|
|||
echo "Installing dependencies..."
|
||||
packages=
|
||||
while read -r line; do
|
||||
if ! echo "$line" | grep -Fq "#"; then
|
||||
packages="$packages $line"
|
||||
fi
|
||||
packages="$packages $line"
|
||||
done < "$ROOT_DIR/setup/linux/packages/dependencies"
|
||||
run_elevated pacman -Syu --needed --noconfirm $packages
|
||||
}
|
||||
|
|
@ -293,12 +316,21 @@ function build_linux() {
|
|||
update_repo
|
||||
fi
|
||||
|
||||
# Build live environment
|
||||
copy_live_env
|
||||
update_live_env
|
||||
|
||||
# Rerun script as root to start Archiso build process
|
||||
run_elevated "$(realpath "$0")" --build-iso
|
||||
# Build requested version(s)
|
||||
for version in "$@"; do
|
||||
if [[ "$version" == "Full" ]]; then
|
||||
copy_live_env
|
||||
update_live_env
|
||||
elif [[ "$version" == "Minimal" ]]; then
|
||||
copy_live_env --minimal
|
||||
update_live_env --minimal
|
||||
fi
|
||||
# Rerun script as root to start Archiso build process
|
||||
run_elevated "$(realpath "$0")" --build-iso
|
||||
# Cleanup
|
||||
mv -nv "$PROFILE_DIR" "${PROFILE_DIR}.${version}"
|
||||
perl-rename -v "s/(${KIT_NAME_SHORT}-Linux)-(${DATE}.*)/\1-${version}-\2/" "$OUT_DIR"/*
|
||||
done
|
||||
}
|
||||
|
||||
function build_iso() {
|
||||
|
|
@ -326,6 +358,18 @@ function build_iso() {
|
|||
-v "$PROFILE_DIR" \
|
||||
| tee -a "$LOG_DIR/$DATETIME.log"
|
||||
|
||||
# Build better ISO
|
||||
rm -r "${ISO_DIR:-safety}/EFI"
|
||||
rm -r "${ISO_DIR:-safety}/loader"
|
||||
rsync -aI "$PROFILE_DIR/EFI/" "${ISO_DIR:-safety}/EFI/"
|
||||
rsync -aI --ignore-existing "$PROFILE_DIR/syslinux/" "${ISO_DIR:-safety}/syslinux/"
|
||||
## Sketchy bit ##
|
||||
. /usr/bin/mkarchiso -o "${OUT_DIR}" -w "${WORK_DIR}" "${PROFILE_DIR}"
|
||||
isofs_dir="${ISO_DIR}"
|
||||
image_name="${KIT_NAME_SHORT}-Linux-${DATE}-x86_64.iso"
|
||||
rm "${OUT_DIR}/${image_name}"
|
||||
_build_iso_image
|
||||
|
||||
# Cleanup
|
||||
echo "Removing temp files..."
|
||||
rm "$TEMP_DIR/Linux" -Rf | tee -a "$LOG_DIR/$DATETIME.log"
|
||||
|
|
@ -339,8 +383,13 @@ function build_iso() {
|
|||
|
||||
# Check input
|
||||
case ${1:-} in
|
||||
-b|--build)
|
||||
build_linux
|
||||
-a|--build-all)
|
||||
build_linux Full Minimal
|
||||
echo Done
|
||||
;;
|
||||
|
||||
-b|--build-full)
|
||||
build_linux Full
|
||||
echo Done
|
||||
;;
|
||||
|
||||
|
|
@ -354,6 +403,18 @@ case ${1:-} in
|
|||
echo Done
|
||||
;;
|
||||
|
||||
-m|--build-minimal)
|
||||
build_linux Minimal
|
||||
echo Done
|
||||
;;
|
||||
|
||||
-n|--prep-minimal-env)
|
||||
load_settings --edit
|
||||
copy_live_env --minimal
|
||||
update_live_env --minimal
|
||||
echo Done
|
||||
;;
|
||||
|
||||
-o|--build-iso)
|
||||
load_settings
|
||||
build_iso
|
||||
|
|
@ -376,15 +437,18 @@ case ${1:-} in
|
|||
echo "Usage: $(basename "$0") [OPTIONS]"
|
||||
echo ""
|
||||
echo "Options:"
|
||||
echo " -b --build Perform all tasks to build the iso"
|
||||
echo " -h --help Show usage"
|
||||
echo " -a --build-all Perform all tasks to build all isos"
|
||||
echo " -b --build-full Perform all tasks to build the full iso"
|
||||
echo " -m --build-minimal Perform all tasks to build the minimal iso"
|
||||
echo " -h --help Show usage"
|
||||
echo ""
|
||||
echo "Advanced options:"
|
||||
echo " -f --fix-perms Fix folder permissions"
|
||||
echo " -i --install-deps Install build dependencies"
|
||||
echo " -o --build-iso Build ISO (using current setup)"
|
||||
echo " -p --prep-live-env Prep live & airootfs folders"
|
||||
echo " -u --update-repo Update custom pacman repo"
|
||||
echo " -f --fix-perms Fix folder permissions"
|
||||
echo " -i --install-deps Install build dependencies"
|
||||
echo " -n --prep-minimal-env Prep live & airootfs folders (minimal packages)"
|
||||
echo " -o --build-iso Build ISO (using current setup)"
|
||||
echo " -p --prep-live-env Prep live & airootfs folders"
|
||||
echo " -u --update-repo Update custom pacman repo"
|
||||
;;
|
||||
esac
|
||||
|
||||
|
|
|
|||
|
|
@ -6,6 +6,10 @@
|
|||
setlocal EnableDelayedExpansion
|
||||
title WizardKit: Build Tool
|
||||
call :CheckFlags %*
|
||||
rem TODO: Remove warning
|
||||
echo "Windows PE build is currently under development"
|
||||
echo " Proceeding will likely result in errors so be warned"
|
||||
pause
|
||||
call :CheckElevation || goto Exit
|
||||
call :FindKitsRoot || goto ErrorKitNotFound
|
||||
|
||||
|
|
@ -15,8 +19,14 @@ set "dandi_set_env=%adk_root%\Deployment Tools\DandISetEnv.bat"
|
|||
if not exist "%dandi_set_env%" (goto ErrorKitNotFound)
|
||||
call "%dandi_set_env%" || goto ErrorUnknown
|
||||
|
||||
:EnsureCRLF
|
||||
rem Rewrite main.py using PowerShell to have CRLF/`r`n lineendings
|
||||
set "script=%~dp0\.bin\Scripts\borrowed\set-eol.ps1"
|
||||
set "main=%~dp0\.bin\Scripts\settings\main.py"
|
||||
powershell -executionpolicy bypass -noprofile -file %script% -lineEnding win -file %main% || goto ErrorUnknown
|
||||
|
||||
:Launch
|
||||
set "script=%~dp0\pe\build_pe.ps1"
|
||||
set "script=%~dp0\.bin\Scripts\build_pe.ps1"
|
||||
powershell -executionpolicy bypass -noprofile -file %script% || goto ErrorUnknown
|
||||
goto Exit
|
||||
|
||||
|
|
|
|||
|
|
@ -1,3 +1,11 @@
|
|||
# WizardKit: Linux #
|
||||
|
||||
Files used to create the Linux build(s).
|
||||
|
||||
## Profiles ##
|
||||
|
||||
profile_base is used for both full and minimal Linux builds.
|
||||
|
||||
profile_gui is only used for full Linux builds.
|
||||
|
||||
NOTE: The Minimal Linux build is currently deprecated and may be removed from a future release.
|
||||
|
|
|
|||
|
|
@ -4,11 +4,11 @@ hardinfo-gtk3
|
|||
hfsprogs
|
||||
iwgtk
|
||||
memtest86-efi
|
||||
mprime-bin
|
||||
opensuperclone-git
|
||||
mprime
|
||||
openbox-patched
|
||||
pipes.sh
|
||||
smartmontools-svn
|
||||
testdisk-wip
|
||||
ttf-font-awesome-4
|
||||
udevil
|
||||
wd719x-firmware
|
||||
wimboot-bin
|
||||
|
|
|
|||
|
|
@ -2,188 +2,95 @@ aic94xx-firmware
|
|||
alsa-utils
|
||||
amd-ucode
|
||||
antiword
|
||||
arandr
|
||||
arc-gtk-theme
|
||||
arch-install-scripts
|
||||
base
|
||||
bc
|
||||
bind
|
||||
bluez
|
||||
bluez-utils
|
||||
bolt
|
||||
btrfs-progs
|
||||
cbatticon
|
||||
chntpw
|
||||
cmatrix
|
||||
colordiff
|
||||
conky
|
||||
cpio
|
||||
cryptsetup
|
||||
curl
|
||||
ddrescue
|
||||
ddrescueview-bin
|
||||
device-mapper
|
||||
diffutils
|
||||
dkms
|
||||
dmidecode
|
||||
dmraid
|
||||
dos2unix
|
||||
dosfstools
|
||||
dunst
|
||||
e2fsprogs
|
||||
edk2-shell
|
||||
efibootmgr
|
||||
evince
|
||||
exfatprogs
|
||||
f2fs-tools
|
||||
fatresize
|
||||
feh
|
||||
ffmpeg
|
||||
firefox
|
||||
foot-terminfo
|
||||
gnome-keyring
|
||||
gnu-netcat
|
||||
gparted
|
||||
gpicview
|
||||
gptfdisk
|
||||
grub
|
||||
gsmartcontrol
|
||||
hardinfo-gtk3
|
||||
hexedit
|
||||
hfsprogs
|
||||
htop
|
||||
inetutils
|
||||
intel-ucode
|
||||
iwd
|
||||
iwgtk
|
||||
jfsutils
|
||||
kitty-terminfo
|
||||
ldns
|
||||
leafpad
|
||||
less
|
||||
lha
|
||||
libewf
|
||||
libinput
|
||||
libldm
|
||||
libusb-compat
|
||||
libxft
|
||||
linux
|
||||
linux-firmware
|
||||
linux-firmware-marvell
|
||||
lm_sensors
|
||||
lsscsi
|
||||
lvm2
|
||||
lzip
|
||||
man-db
|
||||
man-pages
|
||||
mdadm
|
||||
mediainfo
|
||||
memtest86+
|
||||
memtest86-efi
|
||||
mesa-demos
|
||||
mesa-utils
|
||||
mkinitcpio
|
||||
mkinitcpio-archiso
|
||||
mkinitcpio-nfs-utils
|
||||
mkvtoolnix-cli
|
||||
mprime-bin
|
||||
mpv
|
||||
mprime
|
||||
mtools
|
||||
nano
|
||||
nbd
|
||||
ncdu
|
||||
ndisc6
|
||||
nfs-utils
|
||||
nmap
|
||||
noto-fonts
|
||||
noto-fonts-cjk
|
||||
ntfs-3g
|
||||
numlockx
|
||||
nvme-cli
|
||||
open-iscsi
|
||||
openbox
|
||||
openssh
|
||||
opensuperclone-git
|
||||
otf-font-awesome-4
|
||||
p7zip
|
||||
papirus-icon-theme
|
||||
parted
|
||||
perl
|
||||
picom
|
||||
pipes.sh
|
||||
reiserfsprogs
|
||||
pv
|
||||
python
|
||||
python-prompt_toolkit
|
||||
python-docopt
|
||||
python-psutil
|
||||
python-pytz
|
||||
python-requests
|
||||
qemu-guest-agent
|
||||
refind
|
||||
reiserfsprogs
|
||||
rfkill
|
||||
rofi
|
||||
rng-tools
|
||||
rsync
|
||||
rxvt-unicode
|
||||
rxvt-unicode-terminfo
|
||||
sdparm
|
||||
smartmontools-svn
|
||||
sof-firmware
|
||||
speedtest-cli
|
||||
spice-vdagent
|
||||
squashfs-tools
|
||||
st
|
||||
sudo
|
||||
sysbench
|
||||
sysfsutils
|
||||
syslinux
|
||||
systemd-resolvconf
|
||||
systemd-sysvcompat
|
||||
terminus-font
|
||||
testdisk
|
||||
testdisk-wip
|
||||
texinfo
|
||||
thunar
|
||||
tigervnc
|
||||
tint2
|
||||
tk
|
||||
tmux
|
||||
tpm2-tools
|
||||
tpm2-tss
|
||||
tree
|
||||
ttf-font-awesome-4
|
||||
ttf-hack
|
||||
ttf-inconsolata
|
||||
udevil
|
||||
udisks2
|
||||
ufw
|
||||
unarj
|
||||
unrar
|
||||
unzip
|
||||
usb_modeswitch
|
||||
usbmuxd
|
||||
usbutils
|
||||
util-linux
|
||||
veracrypt
|
||||
vim
|
||||
virtualbox-guest-utils
|
||||
volumeicon
|
||||
wd719x-firmware
|
||||
wezterm-terminfo
|
||||
which
|
||||
wimboot-bin
|
||||
wimlib
|
||||
wmctrl
|
||||
xarchiver
|
||||
xf86-input-libinput
|
||||
xf86-video-amdgpu
|
||||
xf86-video-fbdev
|
||||
xf86-video-nouveau
|
||||
xf86-video-qxl
|
||||
xf86-video-vesa
|
||||
xfsprogs
|
||||
xorg-server
|
||||
xorg-xdpyinfo
|
||||
xorg-xev
|
||||
xorg-xinit
|
||||
xorg-xinput
|
||||
xz
|
||||
zip
|
||||
zsh
|
||||
|
|
|
|||
|
|
@ -1,6 +1,6 @@
|
|||
pkgbase = st
|
||||
pkgdesc = A simple virtual terminal emulator for X.
|
||||
pkgver = 0.9
|
||||
pkgver = 0.8.5
|
||||
pkgrel = 1
|
||||
url = https://st.suckless.org
|
||||
arch = i686
|
||||
|
|
@ -9,10 +9,10 @@ pkgbase = st
|
|||
arch = aarch64
|
||||
license = MIT
|
||||
depends = libxft
|
||||
source = https://dl.suckless.org/st/st-0.9.tar.gz
|
||||
source = https://dl.suckless.org/st/st-0.8.5.tar.gz
|
||||
source = terminfo.patch
|
||||
source = README.terminfo.rst
|
||||
sha256sums = f36359799734eae785becb374063f0be833cf22f88b4f169cd251b99324e08e7
|
||||
sha256sums = ea6832203ed02ff74182bcb8adaa9ec454c8f989e79232cb859665e2f544ab37
|
||||
sha256sums = f9deea445a5c6203a0e8e699f3c3b55e27275f17fb408562c4dd5d649edeea23
|
||||
sha256sums = 0ebcbba881832adf9c98ce9fe7667c851d3cc3345077cb8ebe32702698665be2
|
||||
|
||||
|
|
|
|||
|
|
@ -1,11 +1,10 @@
|
|||
# Maintainer: Jose Riha <jose1711 gmail com>
|
||||
# Maintainer: Sebastian J. Bronner <waschtl@sbronner.com>
|
||||
# Maintainer: Kevin Stolp <kevinstolp@gmail.com>
|
||||
# Contributor: Patrick Jackson <PatrickSJackson gmail com>
|
||||
# Contributor: Christoph Vigano <mail@cvigano.de>
|
||||
|
||||
pkgname=st
|
||||
pkgver=0.9
|
||||
pkgver=0.8.5
|
||||
pkgrel=1
|
||||
pkgdesc='A simple virtual terminal emulator for X.'
|
||||
arch=('i686' 'x86_64' 'armv7h' 'aarch64')
|
||||
|
|
@ -24,7 +23,7 @@ source=(https://dl.suckless.org/$pkgname/$pkgname-$pkgver.tar.gz
|
|||
https://st.suckless.org/patches/scrollback/st-scrollback-mouse-altscreen-20220127-2c5edf2.diff
|
||||
st-scrollback-mouse-increment-0.8.2-fixed.diff
|
||||
)
|
||||
sha256sums=('f36359799734eae785becb374063f0be833cf22f88b4f169cd251b99324e08e7'
|
||||
sha256sums=('ea6832203ed02ff74182bcb8adaa9ec454c8f989e79232cb859665e2f544ab37'
|
||||
'f9deea445a5c6203a0e8e699f3c3b55e27275f17fb408562c4dd5d649edeea23'
|
||||
'0ebcbba881832adf9c98ce9fe7667c851d3cc3345077cb8ebe32702698665be2'
|
||||
'42e4803ce2a67835f7e533a707a8a28e3804a26ced163145108970b9aee5fb81'
|
||||
|
|
|
|||
482
setup/linux/packages/custom/st/config.def.h
Normal file
482
setup/linux/packages/custom/st/config.def.h
Normal file
|
|
@ -0,0 +1,482 @@
|
|||
/* See LICENSE file for copyright and license details. */
|
||||
|
||||
/*
|
||||
* appearance
|
||||
*
|
||||
* font: see http://freedesktop.org/software/fontconfig/fontconfig-user.html
|
||||
*/
|
||||
static char *font = "Liberation Mono:pixelsize=12:antialias=true:autohint=true";
|
||||
static int borderpx = 2;
|
||||
|
||||
/*
|
||||
* What program is execed by st depends of these precedence rules:
|
||||
* 1: program passed with -e
|
||||
* 2: scroll and/or utmp
|
||||
* 3: SHELL environment variable
|
||||
* 4: value of shell in /etc/passwd
|
||||
* 5: value of shell in config.h
|
||||
*/
|
||||
static char *shell = "/bin/sh";
|
||||
char *utmp = NULL;
|
||||
/* scroll program: to enable use a string like "scroll" */
|
||||
char *scroll = NULL;
|
||||
char *stty_args = "stty raw pass8 nl -echo -iexten -cstopb 38400";
|
||||
|
||||
/* identification sequence returned in DA and DECID */
|
||||
char *vtiden = "\033[?6c";
|
||||
|
||||
/* Kerning / character bounding-box multipliers */
|
||||
static float cwscale = 1.0;
|
||||
static float chscale = 1.0;
|
||||
|
||||
/*
|
||||
* word delimiter string
|
||||
*
|
||||
* More advanced example: L" `'\"()[]{}"
|
||||
*/
|
||||
wchar_t *worddelimiters = L" ";
|
||||
|
||||
/* selection timeouts (in milliseconds) */
|
||||
static unsigned int doubleclicktimeout = 300;
|
||||
static unsigned int tripleclicktimeout = 600;
|
||||
|
||||
/* alt screens */
|
||||
int allowaltscreen = 1;
|
||||
|
||||
/* allow certain non-interactive (insecure) window operations such as:
|
||||
setting the clipboard text */
|
||||
int allowwindowops = 0;
|
||||
|
||||
/*
|
||||
* draw latency range in ms - from new content/keypress/etc until drawing.
|
||||
* within this range, st draws when content stops arriving (idle). mostly it's
|
||||
* near minlatency, but it waits longer for slow updates to avoid partial draw.
|
||||
* low minlatency will tear/flicker more, as it can "detect" idle too early.
|
||||
*/
|
||||
static double minlatency = 8;
|
||||
static double maxlatency = 33;
|
||||
|
||||
/*
|
||||
* blinking timeout (set to 0 to disable blinking) for the terminal blinking
|
||||
* attribute.
|
||||
*/
|
||||
static unsigned int blinktimeout = 800;
|
||||
|
||||
/*
|
||||
* thickness of underline and bar cursors
|
||||
*/
|
||||
static unsigned int cursorthickness = 2;
|
||||
|
||||
/*
|
||||
* bell volume. It must be a value between -100 and 100. Use 0 for disabling
|
||||
* it
|
||||
*/
|
||||
static int bellvolume = 0;
|
||||
|
||||
/* default TERM value */
|
||||
char *termname = "st-256color";
|
||||
|
||||
/*
|
||||
* spaces per tab
|
||||
*
|
||||
* When you are changing this value, don't forget to adapt the »it« value in
|
||||
* the st.info and appropriately install the st.info in the environment where
|
||||
* you use this st version.
|
||||
*
|
||||
* it#$tabspaces,
|
||||
*
|
||||
* Secondly make sure your kernel is not expanding tabs. When running `stty
|
||||
* -a` »tab0« should appear. You can tell the terminal to not expand tabs by
|
||||
* running following command:
|
||||
*
|
||||
* stty tabs
|
||||
*/
|
||||
unsigned int tabspaces = 8;
|
||||
|
||||
/* bg opacity */
|
||||
float alpha = 0.8;
|
||||
|
||||
/* Terminal colors (16 first used in escape sequence) */
|
||||
static const char *colorname[] = {
|
||||
/* 8 normal colors */
|
||||
"#000000",
|
||||
"#ff0000",
|
||||
"#33ff00",
|
||||
"#ffd000",
|
||||
"#0066ff",
|
||||
"#cc00ff",
|
||||
"#00ffff",
|
||||
"#d0d0d0",
|
||||
|
||||
/* 8 bright colors */
|
||||
"#808080",
|
||||
"#ff9900",
|
||||
"#404040",
|
||||
"#606060",
|
||||
"#c0c0c0",
|
||||
"#e0e0e0",
|
||||
"#3300ff",
|
||||
"#ffffff",
|
||||
|
||||
[255] = 0,
|
||||
|
||||
/* more colors can be added after 255 to use with DefaultXX */
|
||||
"#cccccc",
|
||||
"#555555",
|
||||
"gray90", /* default foreground colour */
|
||||
"black", /* default background colour */
|
||||
};
|
||||
|
||||
|
||||
/*
|
||||
* Default colors (colorname index)
|
||||
* foreground, background, cursor, reverse cursor
|
||||
*/
|
||||
unsigned int defaultfg = 258;
|
||||
unsigned int defaultbg = 259;
|
||||
unsigned int defaultcs = 256;
|
||||
static unsigned int defaultrcs = 257;
|
||||
|
||||
/*
|
||||
* Default shape of cursor
|
||||
* 2: Block ("█")
|
||||
* 4: Underline ("_")
|
||||
* 6: Bar ("|")
|
||||
* 7: Snowman ("☃")
|
||||
*/
|
||||
static unsigned int cursorshape = 2;
|
||||
|
||||
/*
|
||||
* Default columns and rows numbers
|
||||
*/
|
||||
|
||||
static unsigned int cols = 80;
|
||||
static unsigned int rows = 24;
|
||||
|
||||
/*
|
||||
* Default colour and shape of the mouse cursor
|
||||
*/
|
||||
static unsigned int mouseshape = XC_xterm;
|
||||
static unsigned int mousefg = 7;
|
||||
static unsigned int mousebg = 0;
|
||||
|
||||
/*
|
||||
* Color used to display font attributes when fontconfig selected a font which
|
||||
* doesn't match the ones requested.
|
||||
*/
|
||||
static unsigned int defaultattr = 11;
|
||||
|
||||
/*
|
||||
* Force mouse select/shortcuts while mask is active (when MODE_MOUSE is set).
|
||||
* Note that if you want to use ShiftMask with selmasks, set this to an other
|
||||
* modifier, set to 0 to not use it.
|
||||
*/
|
||||
static uint forcemousemod = ShiftMask;
|
||||
|
||||
/*
|
||||
* Internal mouse shortcuts.
|
||||
* Beware that overloading Button1 will disable the selection.
|
||||
*/
|
||||
const unsigned int mousescrollincrement = 4;
|
||||
static MouseShortcut mshortcuts[] = {
|
||||
/* mask button function argument release */
|
||||
{ XK_ANY_MOD, Button4, kscrollup, {.i = mousescrollincrement}, 0, /* !alt */ -1 },
|
||||
{ XK_ANY_MOD, Button5, kscrolldown, {.i = mousescrollincrement}, 0, /* !alt */ -1 },
|
||||
{ XK_ANY_MOD, Button2, selpaste, {.i = 0}, 1 },
|
||||
{ ShiftMask, Button4, ttysend, {.s = "\033[5;2~"} },
|
||||
{ XK_ANY_MOD, Button4, ttysend, {.s = "\031"} },
|
||||
{ ShiftMask, Button5, ttysend, {.s = "\033[6;2~"} },
|
||||
{ XK_ANY_MOD, Button5, ttysend, {.s = "\005"} },
|
||||
};
|
||||
|
||||
/* Internal keyboard shortcuts. */
|
||||
#define MODKEY Mod1Mask
|
||||
#define TERMMOD (ControlMask|ShiftMask)
|
||||
|
||||
static Shortcut shortcuts[] = {
|
||||
/* mask keysym function argument */
|
||||
{ XK_ANY_MOD, XK_Break, sendbreak, {.i = 0} },
|
||||
{ ControlMask, XK_Print, toggleprinter, {.i = 0} },
|
||||
{ ShiftMask, XK_Print, printscreen, {.i = 0} },
|
||||
{ XK_ANY_MOD, XK_Print, printsel, {.i = 0} },
|
||||
{ TERMMOD, XK_Prior, zoom, {.f = +1} },
|
||||
{ TERMMOD, XK_Next, zoom, {.f = -1} },
|
||||
{ TERMMOD, XK_Home, zoomreset, {.f = 0} },
|
||||
{ TERMMOD, XK_C, clipcopy, {.i = 0} },
|
||||
{ TERMMOD, XK_V, clippaste, {.i = 0} },
|
||||
{ TERMMOD, XK_Y, selpaste, {.i = 0} },
|
||||
{ ShiftMask, XK_Insert, selpaste, {.i = 0} },
|
||||
{ TERMMOD, XK_Num_Lock, numlock, {.i = 0} },
|
||||
{ ShiftMask, XK_Page_Up, kscrollup, {.i = -1} },
|
||||
{ ShiftMask, XK_Page_Down, kscrolldown, {.i = -1} },
|
||||
};
|
||||
|
||||
/*
|
||||
* Special keys (change & recompile st.info accordingly)
|
||||
*
|
||||
* Mask value:
|
||||
* * Use XK_ANY_MOD to match the key no matter modifiers state
|
||||
* * Use XK_NO_MOD to match the key alone (no modifiers)
|
||||
* appkey value:
|
||||
* * 0: no value
|
||||
* * > 0: keypad application mode enabled
|
||||
* * = 2: term.numlock = 1
|
||||
* * < 0: keypad application mode disabled
|
||||
* appcursor value:
|
||||
* * 0: no value
|
||||
* * > 0: cursor application mode enabled
|
||||
* * < 0: cursor application mode disabled
|
||||
*
|
||||
* Be careful with the order of the definitions because st searches in
|
||||
* this table sequentially, so any XK_ANY_MOD must be in the last
|
||||
* position for a key.
|
||||
*/
|
||||
|
||||
/*
|
||||
* If you want keys other than the X11 function keys (0xFD00 - 0xFFFF)
|
||||
* to be mapped below, add them to this array.
|
||||
*/
|
||||
static KeySym mappedkeys[] = { -1 };
|
||||
|
||||
/*
|
||||
* State bits to ignore when matching key or button events. By default,
|
||||
* numlock (Mod2Mask) and keyboard layout (XK_SWITCH_MOD) are ignored.
|
||||
*/
|
||||
static uint ignoremod = Mod2Mask|XK_SWITCH_MOD;
|
||||
|
||||
/*
|
||||
* This is the huge key array which defines all compatibility to the Linux
|
||||
* world. Please decide about changes wisely.
|
||||
*/
|
||||
static Key key[] = {
|
||||
/* keysym mask string appkey appcursor */
|
||||
{ XK_KP_Home, ShiftMask, "\033[2J", 0, -1},
|
||||
{ XK_KP_Home, ShiftMask, "\033[1;2H", 0, +1},
|
||||
{ XK_KP_Home, XK_ANY_MOD, "\033[H", 0, -1},
|
||||
{ XK_KP_Home, XK_ANY_MOD, "\033[1~", 0, +1},
|
||||
{ XK_KP_Up, XK_ANY_MOD, "\033Ox", +1, 0},
|
||||
{ XK_KP_Up, XK_ANY_MOD, "\033[A", 0, -1},
|
||||
{ XK_KP_Up, XK_ANY_MOD, "\033OA", 0, +1},
|
||||
{ XK_KP_Down, XK_ANY_MOD, "\033Or", +1, 0},
|
||||
{ XK_KP_Down, XK_ANY_MOD, "\033[B", 0, -1},
|
||||
{ XK_KP_Down, XK_ANY_MOD, "\033OB", 0, +1},
|
||||
{ XK_KP_Left, XK_ANY_MOD, "\033Ot", +1, 0},
|
||||
{ XK_KP_Left, XK_ANY_MOD, "\033[D", 0, -1},
|
||||
{ XK_KP_Left, XK_ANY_MOD, "\033OD", 0, +1},
|
||||
{ XK_KP_Right, XK_ANY_MOD, "\033Ov", +1, 0},
|
||||
{ XK_KP_Right, XK_ANY_MOD, "\033[C", 0, -1},
|
||||
{ XK_KP_Right, XK_ANY_MOD, "\033OC", 0, +1},
|
||||
{ XK_KP_Prior, ShiftMask, "\033[5;2~", 0, 0},
|
||||
{ XK_KP_Prior, XK_ANY_MOD, "\033[5~", 0, 0},
|
||||
{ XK_KP_Begin, XK_ANY_MOD, "\033[E", 0, 0},
|
||||
{ XK_KP_End, ControlMask, "\033[J", -1, 0},
|
||||
{ XK_KP_End, ControlMask, "\033[1;5F", +1, 0},
|
||||
{ XK_KP_End, ShiftMask, "\033[K", -1, 0},
|
||||
{ XK_KP_End, ShiftMask, "\033[1;2F", +1, 0},
|
||||
{ XK_KP_End, XK_ANY_MOD, "\033[4~", 0, 0},
|
||||
{ XK_KP_Next, ShiftMask, "\033[6;2~", 0, 0},
|
||||
{ XK_KP_Next, XK_ANY_MOD, "\033[6~", 0, 0},
|
||||
{ XK_KP_Insert, ShiftMask, "\033[2;2~", +1, 0},
|
||||
{ XK_KP_Insert, ShiftMask, "\033[4l", -1, 0},
|
||||
{ XK_KP_Insert, ControlMask, "\033[L", -1, 0},
|
||||
{ XK_KP_Insert, ControlMask, "\033[2;5~", +1, 0},
|
||||
{ XK_KP_Insert, XK_ANY_MOD, "\033[4h", -1, 0},
|
||||
{ XK_KP_Insert, XK_ANY_MOD, "\033[2~", +1, 0},
|
||||
{ XK_KP_Delete, ControlMask, "\033[M", -1, 0},
|
||||
{ XK_KP_Delete, ControlMask, "\033[3;5~", +1, 0},
|
||||
{ XK_KP_Delete, ShiftMask, "\033[2K", -1, 0},
|
||||
{ XK_KP_Delete, ShiftMask, "\033[3;2~", +1, 0},
|
||||
{ XK_KP_Delete, XK_ANY_MOD, "\033[3~", -1, 0},
|
||||
{ XK_KP_Delete, XK_ANY_MOD, "\033[3~", +1, 0},
|
||||
{ XK_KP_Multiply, XK_ANY_MOD, "\033Oj", +2, 0},
|
||||
{ XK_KP_Add, XK_ANY_MOD, "\033Ok", +2, 0},
|
||||
{ XK_KP_Enter, XK_ANY_MOD, "\033OM", +2, 0},
|
||||
{ XK_KP_Enter, XK_ANY_MOD, "\r", -1, 0},
|
||||
{ XK_KP_Subtract, XK_ANY_MOD, "\033Om", +2, 0},
|
||||
{ XK_KP_Decimal, XK_ANY_MOD, "\033On", +2, 0},
|
||||
{ XK_KP_Divide, XK_ANY_MOD, "\033Oo", +2, 0},
|
||||
{ XK_KP_0, XK_ANY_MOD, "\033Op", +2, 0},
|
||||
{ XK_KP_1, XK_ANY_MOD, "\033Oq", +2, 0},
|
||||
{ XK_KP_2, XK_ANY_MOD, "\033Or", +2, 0},
|
||||
{ XK_KP_3, XK_ANY_MOD, "\033Os", +2, 0},
|
||||
{ XK_KP_4, XK_ANY_MOD, "\033Ot", +2, 0},
|
||||
{ XK_KP_5, XK_ANY_MOD, "\033Ou", +2, 0},
|
||||
{ XK_KP_6, XK_ANY_MOD, "\033Ov", +2, 0},
|
||||
{ XK_KP_7, XK_ANY_MOD, "\033Ow", +2, 0},
|
||||
{ XK_KP_8, XK_ANY_MOD, "\033Ox", +2, 0},
|
||||
{ XK_KP_9, XK_ANY_MOD, "\033Oy", +2, 0},
|
||||
{ XK_Up, ShiftMask, "\033[1;2A", 0, 0},
|
||||
{ XK_Up, Mod1Mask, "\033[1;3A", 0, 0},
|
||||
{ XK_Up, ShiftMask|Mod1Mask,"\033[1;4A", 0, 0},
|
||||
{ XK_Up, ControlMask, "\033[1;5A", 0, 0},
|
||||
{ XK_Up, ShiftMask|ControlMask,"\033[1;6A", 0, 0},
|
||||
{ XK_Up, ControlMask|Mod1Mask,"\033[1;7A", 0, 0},
|
||||
{ XK_Up,ShiftMask|ControlMask|Mod1Mask,"\033[1;8A", 0, 0},
|
||||
{ XK_Up, XK_ANY_MOD, "\033[A", 0, -1},
|
||||
{ XK_Up, XK_ANY_MOD, "\033OA", 0, +1},
|
||||
{ XK_Down, ShiftMask, "\033[1;2B", 0, 0},
|
||||
{ XK_Down, Mod1Mask, "\033[1;3B", 0, 0},
|
||||
{ XK_Down, ShiftMask|Mod1Mask,"\033[1;4B", 0, 0},
|
||||
{ XK_Down, ControlMask, "\033[1;5B", 0, 0},
|
||||
{ XK_Down, ShiftMask|ControlMask,"\033[1;6B", 0, 0},
|
||||
{ XK_Down, ControlMask|Mod1Mask,"\033[1;7B", 0, 0},
|
||||
{ XK_Down,ShiftMask|ControlMask|Mod1Mask,"\033[1;8B",0, 0},
|
||||
{ XK_Down, XK_ANY_MOD, "\033[B", 0, -1},
|
||||
{ XK_Down, XK_ANY_MOD, "\033OB", 0, +1},
|
||||
{ XK_Left, ShiftMask, "\033[1;2D", 0, 0},
|
||||
{ XK_Left, Mod1Mask, "\033[1;3D", 0, 0},
|
||||
{ XK_Left, ShiftMask|Mod1Mask,"\033[1;4D", 0, 0},
|
||||
{ XK_Left, ControlMask, "\033[1;5D", 0, 0},
|
||||
{ XK_Left, ShiftMask|ControlMask,"\033[1;6D", 0, 0},
|
||||
{ XK_Left, ControlMask|Mod1Mask,"\033[1;7D", 0, 0},
|
||||
{ XK_Left,ShiftMask|ControlMask|Mod1Mask,"\033[1;8D",0, 0},
|
||||
{ XK_Left, XK_ANY_MOD, "\033[D", 0, -1},
|
||||
{ XK_Left, XK_ANY_MOD, "\033OD", 0, +1},
|
||||
{ XK_Right, ShiftMask, "\033[1;2C", 0, 0},
|
||||
{ XK_Right, Mod1Mask, "\033[1;3C", 0, 0},
|
||||
{ XK_Right, ShiftMask|Mod1Mask,"\033[1;4C", 0, 0},
|
||||
{ XK_Right, ControlMask, "\033[1;5C", 0, 0},
|
||||
{ XK_Right, ShiftMask|ControlMask,"\033[1;6C", 0, 0},
|
||||
{ XK_Right, ControlMask|Mod1Mask,"\033[1;7C", 0, 0},
|
||||
{ XK_Right,ShiftMask|ControlMask|Mod1Mask,"\033[1;8C",0, 0},
|
||||
{ XK_Right, XK_ANY_MOD, "\033[C", 0, -1},
|
||||
{ XK_Right, XK_ANY_MOD, "\033OC", 0, +1},
|
||||
{ XK_ISO_Left_Tab, ShiftMask, "\033[Z", 0, 0},
|
||||
{ XK_Return, Mod1Mask, "\033\r", 0, 0},
|
||||
{ XK_Return, XK_ANY_MOD, "\r", 0, 0},
|
||||
{ XK_Insert, ShiftMask, "\033[4l", -1, 0},
|
||||
{ XK_Insert, ShiftMask, "\033[2;2~", +1, 0},
|
||||
{ XK_Insert, ControlMask, "\033[L", -1, 0},
|
||||
{ XK_Insert, ControlMask, "\033[2;5~", +1, 0},
|
||||
{ XK_Insert, XK_ANY_MOD, "\033[4h", -1, 0},
|
||||
{ XK_Insert, XK_ANY_MOD, "\033[2~", +1, 0},
|
||||
{ XK_Delete, ControlMask, "\033[M", -1, 0},
|
||||
{ XK_Delete, ControlMask, "\033[3;5~", +1, 0},
|
||||
{ XK_Delete, ShiftMask, "\033[2K", -1, 0},
|
||||
{ XK_Delete, ShiftMask, "\033[3;2~", +1, 0},
|
||||
{ XK_Delete, XK_ANY_MOD, "\033[3~", -1, 0},
|
||||
{ XK_Delete, XK_ANY_MOD, "\033[3~", +1, 0},
|
||||
{ XK_BackSpace, XK_NO_MOD, "\177", 0, 0},
|
||||
{ XK_BackSpace, Mod1Mask, "\033\177", 0, 0},
|
||||
{ XK_Home, ShiftMask, "\033[2J", 0, -1},
|
||||
{ XK_Home, ShiftMask, "\033[1;2H", 0, +1},
|
||||
{ XK_Home, XK_ANY_MOD, "\033[H", 0, -1},
|
||||
{ XK_Home, XK_ANY_MOD, "\033[1~", 0, +1},
|
||||
{ XK_End, ControlMask, "\033[J", -1, 0},
|
||||
{ XK_End, ControlMask, "\033[1;5F", +1, 0},
|
||||
{ XK_End, ShiftMask, "\033[K", -1, 0},
|
||||
{ XK_End, ShiftMask, "\033[1;2F", +1, 0},
|
||||
{ XK_End, XK_ANY_MOD, "\033[4~", 0, 0},
|
||||
{ XK_Prior, ControlMask, "\033[5;5~", 0, 0},
|
||||
{ XK_Prior, ShiftMask, "\033[5;2~", 0, 0},
|
||||
{ XK_Prior, XK_ANY_MOD, "\033[5~", 0, 0},
|
||||
{ XK_Next, ControlMask, "\033[6;5~", 0, 0},
|
||||
{ XK_Next, ShiftMask, "\033[6;2~", 0, 0},
|
||||
{ XK_Next, XK_ANY_MOD, "\033[6~", 0, 0},
|
||||
{ XK_F1, XK_NO_MOD, "\033OP" , 0, 0},
|
||||
{ XK_F1, /* F13 */ ShiftMask, "\033[1;2P", 0, 0},
|
||||
{ XK_F1, /* F25 */ ControlMask, "\033[1;5P", 0, 0},
|
||||
{ XK_F1, /* F37 */ Mod4Mask, "\033[1;6P", 0, 0},
|
||||
{ XK_F1, /* F49 */ Mod1Mask, "\033[1;3P", 0, 0},
|
||||
{ XK_F1, /* F61 */ Mod3Mask, "\033[1;4P", 0, 0},
|
||||
{ XK_F2, XK_NO_MOD, "\033OQ" , 0, 0},
|
||||
{ XK_F2, /* F14 */ ShiftMask, "\033[1;2Q", 0, 0},
|
||||
{ XK_F2, /* F26 */ ControlMask, "\033[1;5Q", 0, 0},
|
||||
{ XK_F2, /* F38 */ Mod4Mask, "\033[1;6Q", 0, 0},
|
||||
{ XK_F2, /* F50 */ Mod1Mask, "\033[1;3Q", 0, 0},
|
||||
{ XK_F2, /* F62 */ Mod3Mask, "\033[1;4Q", 0, 0},
|
||||
{ XK_F3, XK_NO_MOD, "\033OR" , 0, 0},
|
||||
{ XK_F3, /* F15 */ ShiftMask, "\033[1;2R", 0, 0},
|
||||
{ XK_F3, /* F27 */ ControlMask, "\033[1;5R", 0, 0},
|
||||
{ XK_F3, /* F39 */ Mod4Mask, "\033[1;6R", 0, 0},
|
||||
{ XK_F3, /* F51 */ Mod1Mask, "\033[1;3R", 0, 0},
|
||||
{ XK_F3, /* F63 */ Mod3Mask, "\033[1;4R", 0, 0},
|
||||
{ XK_F4, XK_NO_MOD, "\033OS" , 0, 0},
|
||||
{ XK_F4, /* F16 */ ShiftMask, "\033[1;2S", 0, 0},
|
||||
{ XK_F4, /* F28 */ ControlMask, "\033[1;5S", 0, 0},
|
||||
{ XK_F4, /* F40 */ Mod4Mask, "\033[1;6S", 0, 0},
|
||||
{ XK_F4, /* F52 */ Mod1Mask, "\033[1;3S", 0, 0},
|
||||
{ XK_F5, XK_NO_MOD, "\033[15~", 0, 0},
|
||||
{ XK_F5, /* F17 */ ShiftMask, "\033[15;2~", 0, 0},
|
||||
{ XK_F5, /* F29 */ ControlMask, "\033[15;5~", 0, 0},
|
||||
{ XK_F5, /* F41 */ Mod4Mask, "\033[15;6~", 0, 0},
|
||||
{ XK_F5, /* F53 */ Mod1Mask, "\033[15;3~", 0, 0},
|
||||
{ XK_F6, XK_NO_MOD, "\033[17~", 0, 0},
|
||||
{ XK_F6, /* F18 */ ShiftMask, "\033[17;2~", 0, 0},
|
||||
{ XK_F6, /* F30 */ ControlMask, "\033[17;5~", 0, 0},
|
||||
{ XK_F6, /* F42 */ Mod4Mask, "\033[17;6~", 0, 0},
|
||||
{ XK_F6, /* F54 */ Mod1Mask, "\033[17;3~", 0, 0},
|
||||
{ XK_F7, XK_NO_MOD, "\033[18~", 0, 0},
|
||||
{ XK_F7, /* F19 */ ShiftMask, "\033[18;2~", 0, 0},
|
||||
{ XK_F7, /* F31 */ ControlMask, "\033[18;5~", 0, 0},
|
||||
{ XK_F7, /* F43 */ Mod4Mask, "\033[18;6~", 0, 0},
|
||||
{ XK_F7, /* F55 */ Mod1Mask, "\033[18;3~", 0, 0},
|
||||
{ XK_F8, XK_NO_MOD, "\033[19~", 0, 0},
|
||||
{ XK_F8, /* F20 */ ShiftMask, "\033[19;2~", 0, 0},
|
||||
{ XK_F8, /* F32 */ ControlMask, "\033[19;5~", 0, 0},
|
||||
{ XK_F8, /* F44 */ Mod4Mask, "\033[19;6~", 0, 0},
|
||||
{ XK_F8, /* F56 */ Mod1Mask, "\033[19;3~", 0, 0},
|
||||
{ XK_F9, XK_NO_MOD, "\033[20~", 0, 0},
|
||||
{ XK_F9, /* F21 */ ShiftMask, "\033[20;2~", 0, 0},
|
||||
{ XK_F9, /* F33 */ ControlMask, "\033[20;5~", 0, 0},
|
||||
{ XK_F9, /* F45 */ Mod4Mask, "\033[20;6~", 0, 0},
|
||||
{ XK_F9, /* F57 */ Mod1Mask, "\033[20;3~", 0, 0},
|
||||
{ XK_F10, XK_NO_MOD, "\033[21~", 0, 0},
|
||||
{ XK_F10, /* F22 */ ShiftMask, "\033[21;2~", 0, 0},
|
||||
{ XK_F10, /* F34 */ ControlMask, "\033[21;5~", 0, 0},
|
||||
{ XK_F10, /* F46 */ Mod4Mask, "\033[21;6~", 0, 0},
|
||||
{ XK_F10, /* F58 */ Mod1Mask, "\033[21;3~", 0, 0},
|
||||
{ XK_F11, XK_NO_MOD, "\033[23~", 0, 0},
|
||||
{ XK_F11, /* F23 */ ShiftMask, "\033[23;2~", 0, 0},
|
||||
{ XK_F11, /* F35 */ ControlMask, "\033[23;5~", 0, 0},
|
||||
{ XK_F11, /* F47 */ Mod4Mask, "\033[23;6~", 0, 0},
|
||||
{ XK_F11, /* F59 */ Mod1Mask, "\033[23;3~", 0, 0},
|
||||
{ XK_F12, XK_NO_MOD, "\033[24~", 0, 0},
|
||||
{ XK_F12, /* F24 */ ShiftMask, "\033[24;2~", 0, 0},
|
||||
{ XK_F12, /* F36 */ ControlMask, "\033[24;5~", 0, 0},
|
||||
{ XK_F12, /* F48 */ Mod4Mask, "\033[24;6~", 0, 0},
|
||||
{ XK_F12, /* F60 */ Mod1Mask, "\033[24;3~", 0, 0},
|
||||
{ XK_F13, XK_NO_MOD, "\033[1;2P", 0, 0},
|
||||
{ XK_F14, XK_NO_MOD, "\033[1;2Q", 0, 0},
|
||||
{ XK_F15, XK_NO_MOD, "\033[1;2R", 0, 0},
|
||||
{ XK_F16, XK_NO_MOD, "\033[1;2S", 0, 0},
|
||||
{ XK_F17, XK_NO_MOD, "\033[15;2~", 0, 0},
|
||||
{ XK_F18, XK_NO_MOD, "\033[17;2~", 0, 0},
|
||||
{ XK_F19, XK_NO_MOD, "\033[18;2~", 0, 0},
|
||||
{ XK_F20, XK_NO_MOD, "\033[19;2~", 0, 0},
|
||||
{ XK_F21, XK_NO_MOD, "\033[20;2~", 0, 0},
|
||||
{ XK_F22, XK_NO_MOD, "\033[21;2~", 0, 0},
|
||||
{ XK_F23, XK_NO_MOD, "\033[23;2~", 0, 0},
|
||||
{ XK_F24, XK_NO_MOD, "\033[24;2~", 0, 0},
|
||||
{ XK_F25, XK_NO_MOD, "\033[1;5P", 0, 0},
|
||||
{ XK_F26, XK_NO_MOD, "\033[1;5Q", 0, 0},
|
||||
{ XK_F27, XK_NO_MOD, "\033[1;5R", 0, 0},
|
||||
{ XK_F28, XK_NO_MOD, "\033[1;5S", 0, 0},
|
||||
{ XK_F29, XK_NO_MOD, "\033[15;5~", 0, 0},
|
||||
{ XK_F30, XK_NO_MOD, "\033[17;5~", 0, 0},
|
||||
{ XK_F31, XK_NO_MOD, "\033[18;5~", 0, 0},
|
||||
{ XK_F32, XK_NO_MOD, "\033[19;5~", 0, 0},
|
||||
{ XK_F33, XK_NO_MOD, "\033[20;5~", 0, 0},
|
||||
{ XK_F34, XK_NO_MOD, "\033[21;5~", 0, 0},
|
||||
{ XK_F35, XK_NO_MOD, "\033[23;5~", 0, 0},
|
||||
};
|
||||
|
||||
/*
|
||||
* Selection types' masks.
|
||||
* Use the same masks as usual.
|
||||
* Button1Mask is always unset, to make masks match between ButtonPress.
|
||||
* ButtonRelease and MotionNotify.
|
||||
* If no match is found, regular selection is used.
|
||||
*/
|
||||
static uint selmasks[] = {
|
||||
[SEL_RECTANGULAR] = Mod1Mask,
|
||||
};
|
||||
|
||||
/*
|
||||
* Printable characters in ASCII, used to estimate the advance width
|
||||
* of single wide characters.
|
||||
*/
|
||||
static char ascii_printable[] =
|
||||
" !\"#$%&'()*+,-./0123456789:;<=>?"
|
||||
"@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_"
|
||||
"`abcdefghijklmnopqrstuvwxyz{|}~";
|
||||
43
setup/linux/packages/custom/st/st-colors.diff
Normal file
43
setup/linux/packages/custom/st/st-colors.diff
Normal file
|
|
@ -0,0 +1,43 @@
|
|||
--- a/config.def.h 2021-12-20 21:35:22.214525009 -0700
|
||||
+++ b/config.def.h 2021-12-20 21:38:26.933659040 -0700
|
||||
@@ -99,24 +99,24 @@
|
||||
/* Terminal colors (16 first used in escape sequence) */
|
||||
static const char *colorname[] = {
|
||||
/* 8 normal colors */
|
||||
- "black",
|
||||
- "red3",
|
||||
- "green3",
|
||||
- "yellow3",
|
||||
- "blue2",
|
||||
- "magenta3",
|
||||
- "cyan3",
|
||||
- "gray90",
|
||||
+ "#000000",
|
||||
+ "#ff0000",
|
||||
+ "#33ff00",
|
||||
+ "#ffd000",
|
||||
+ "#0066ff",
|
||||
+ "#cc00ff",
|
||||
+ "#00ffff",
|
||||
+ "#d0d0d0",
|
||||
|
||||
/* 8 bright colors */
|
||||
- "gray50",
|
||||
- "red",
|
||||
- "green",
|
||||
- "yellow",
|
||||
- "#5c5cff",
|
||||
- "magenta",
|
||||
- "cyan",
|
||||
- "white",
|
||||
+ "#808080",
|
||||
+ "#ff9900",
|
||||
+ "#404040",
|
||||
+ "#606060",
|
||||
+ "#c0c0c0",
|
||||
+ "#e0e0e0",
|
||||
+ "#3300ff",
|
||||
+ "#ffffff",
|
||||
|
||||
[255] = 0,
|
||||
|
||||
|
|
@ -1,35 +1,39 @@
|
|||
# WizardKit: Package dependencies
|
||||
archiso
|
||||
attr
|
||||
base-devel
|
||||
boost
|
||||
curl
|
||||
dos2unix
|
||||
erofs-utils
|
||||
git
|
||||
gtk-doc
|
||||
gtk3
|
||||
memtest86+
|
||||
memtest86+-efi
|
||||
hwloc
|
||||
imlib2
|
||||
iwd
|
||||
json-glib
|
||||
lhasa
|
||||
libbsd
|
||||
libewf
|
||||
librsvg
|
||||
libsm
|
||||
libxcursor
|
||||
libxft
|
||||
libxinerama
|
||||
libxml2
|
||||
libxrandr
|
||||
ntfs-3g
|
||||
openssh
|
||||
p7zip
|
||||
pango
|
||||
perl-rename
|
||||
reiserfsprogs
|
||||
pv
|
||||
refind
|
||||
rsync
|
||||
startup-notification
|
||||
subversion
|
||||
syslinux
|
||||
tigervnc
|
||||
unzip
|
||||
wpa_supplicant
|
||||
|
||||
# hardinfo-gtk3 / opensuperclone-git
|
||||
cmake
|
||||
|
||||
# iwgtk
|
||||
gtk4
|
||||
meson
|
||||
qrencode
|
||||
scdoc
|
||||
|
||||
# smartmontools-svn
|
||||
subversion
|
||||
|
||||
# udevil
|
||||
gettext
|
||||
intltool
|
||||
|
||||
# wd719x-firmware
|
||||
lha
|
||||
|
|
|
|||
56
setup/linux/packages/gui
Normal file
56
setup/linux/packages/gui
Normal file
|
|
@ -0,0 +1,56 @@
|
|||
arandr
|
||||
arc-gtk-theme
|
||||
cbatticon
|
||||
compton
|
||||
conky
|
||||
ddrescueview-bin
|
||||
dunst
|
||||
evince
|
||||
feh
|
||||
ffmpeg
|
||||
firefox
|
||||
gnome-keyring
|
||||
gparted
|
||||
gpicview
|
||||
gsmartcontrol
|
||||
hardinfo-gtk3
|
||||
iwgtk
|
||||
leafpad
|
||||
libinput
|
||||
libxft
|
||||
mesa-demos
|
||||
mkvtoolnix-cli
|
||||
mpv
|
||||
noto-fonts
|
||||
noto-fonts-cjk
|
||||
numlockx
|
||||
openbox-patched
|
||||
otf-font-awesome-4
|
||||
papirus-icon-theme
|
||||
qemu-guest-agent
|
||||
rofi
|
||||
rxvt-unicode
|
||||
spice-vdagent
|
||||
st
|
||||
thunar
|
||||
tigervnc
|
||||
tint2
|
||||
tk
|
||||
ttf-font-awesome-4
|
||||
ttf-hack
|
||||
ttf-inconsolata
|
||||
veracrypt
|
||||
virtualbox-guest-utils
|
||||
volumeicon
|
||||
wmctrl
|
||||
xarchiver
|
||||
xf86-input-libinput
|
||||
xf86-video-amdgpu
|
||||
xf86-video-fbdev
|
||||
xf86-video-nouveau
|
||||
xf86-video-vesa
|
||||
xorg-server
|
||||
xorg-xdpyinfo
|
||||
xorg-xev
|
||||
xorg-xinit
|
||||
xorg-xinput
|
||||
|
|
@ -1 +0,0 @@
|
|||
LANG=C.UTF-8
|
||||
|
|
@ -1,2 +0,0 @@
|
|||
HOOKS=(base udev modconf kms memdisk archiso archiso_loop_mnt archiso_pxe_common archiso_pxe_nbd archiso_pxe_http archiso_pxe_nfs block filesystems keyboard)
|
||||
COMPRESSION="xz"
|
||||
|
|
@ -1,23 +0,0 @@
|
|||
# This is /run/systemd/resolve/stub-resolv.conf managed by man:systemd-resolved(8).
|
||||
# Do not edit.
|
||||
#
|
||||
# This file might be symlinked as /etc/resolv.conf. If you're looking at
|
||||
# /etc/resolv.conf and seeing this text, you have followed the symlink.
|
||||
#
|
||||
# This is a dynamic resolv.conf file for connecting local clients to the
|
||||
# internal DNS stub resolver of systemd-resolved. This file lists all
|
||||
# configured search domains.
|
||||
#
|
||||
# Run "resolvectl status" to see details about the uplink DNS servers
|
||||
# currently in use.
|
||||
#
|
||||
# Third party programs should typically not access this file directly, but only
|
||||
# through the symlink at /etc/resolv.conf. To manage man:resolv.conf(5) in a
|
||||
# different way, replace this symlink by a static file or a different symlink.
|
||||
#
|
||||
# See man:systemd-resolved.service(8) for details about the supported modes of
|
||||
# operation for /etc/resolv.conf.
|
||||
|
||||
nameserver 127.0.0.53
|
||||
options edns0 trust-ad
|
||||
search .
|
||||
|
|
@ -1,16 +0,0 @@
|
|||
#!/bin/env bash
|
||||
#
|
||||
## Set up teststation settings if detected
|
||||
|
||||
# Hostname
|
||||
if [[ -s "/run/archiso/bootmnt/teststation.name" ]]; then
|
||||
NEW_HOSTNAME="$(head -1 "/run/archiso/bootmnt/teststation.name" \
|
||||
| sed -r 's/\s//g')"
|
||||
sudo hostnamectl set-hostname "${NEW_HOSTNAME}"
|
||||
fi
|
||||
|
||||
# Wallpaper
|
||||
if [[ -s "/run/archiso/bootmnt/teststation.wall" ]]; then
|
||||
rm "${HOME}/.wallpaper" >/dev/null 2>&1
|
||||
ln -s "/run/archiso/bootmnt/teststation.wall" "${HOME}/.wallpaper"
|
||||
fi
|
||||
|
|
@ -1,2 +0,0 @@
|
|||
[Network]
|
||||
IPv6PrivacyExtensions=yes
|
||||
Some files were not shown because too many files have changed in this diff Show more
Loading…
Reference in a new issue