178 lines
4 KiB
Python
178 lines
4 KiB
Python
"""WizardKit: Hardware diagnostics"""
|
|
# vim: sts=2 sw=2 ts=2
|
|
|
|
import logging
|
|
import pathlib
|
|
import platform
|
|
|
|
from collections import OrderedDict
|
|
from docopt import docopt
|
|
|
|
from wk.cfg.main import KIT_NAME_FULL
|
|
from wk.exe import run_program
|
|
from wk.net import (
|
|
connected_to_private_network,
|
|
ping,
|
|
show_valid_addresses,
|
|
speedtest,
|
|
)
|
|
from wk.std import (
|
|
Menu,
|
|
TryAndPrint,
|
|
clear_screen,
|
|
color_string,
|
|
pause,
|
|
print_error,
|
|
print_info,
|
|
print_standard,
|
|
print_warning,
|
|
sleep,
|
|
)
|
|
|
|
|
|
# STATIC VARIABLES
|
|
DOCSTRING = f'''{KIT_NAME_FULL}: Hardware Diagnostics
|
|
|
|
Usage:
|
|
hw-diags
|
|
hw-diags (-q | --quick)
|
|
hw-diags (-h | --help)
|
|
|
|
Options:
|
|
-h --help Show this page
|
|
-q --quick Skip menu and perform a quick check
|
|
'''
|
|
LOG = logging.getLogger(__name__)
|
|
MENU_ACTIONS = (
|
|
'Audio Test',
|
|
'Keyboard Test',
|
|
'Network Test',
|
|
'Start',
|
|
'Quit')
|
|
MENU_OPTIONS = (
|
|
'CPU & Cooling',
|
|
'Disk Attributes',
|
|
'Disk Self-Test',
|
|
'Disk Surface Scan',
|
|
'Disk I/O Benchmark',
|
|
)
|
|
MENU_OPTIONS_QUICK = ('Disk Attributes',)
|
|
MENU_SETS = {
|
|
'Full Diagnostic': (*MENU_OPTIONS,),
|
|
'Disk Diagnostic': (
|
|
'Disk Attributes',
|
|
'Disk Self-Test',
|
|
'Disk Surface Scan',
|
|
'Disk I/O Benchmark',
|
|
),
|
|
'Disk Diagnostic (Quick)': ('Disk Attributes',),
|
|
}
|
|
MENU_TOGGLES = []
|
|
|
|
|
|
# Classes
|
|
class State():
|
|
"""Object for tracking hardware diagnostic data."""
|
|
def __init__(self):
|
|
self.tests = OrderedDict()
|
|
|
|
|
|
# Functions
|
|
def audio_test():
|
|
"""Run an OS-specific audio test."""
|
|
if platform.system() == 'Linux':
|
|
audio_test_linux()
|
|
# TODO: Add tests for other OS
|
|
|
|
|
|
def audio_test_linux():
|
|
"""Run an audio test using amixer and speaker-test."""
|
|
clear_screen()
|
|
print_standard('Audio test')
|
|
print_standard('')
|
|
|
|
# Set volume
|
|
for source in ('Master', 'PCM'):
|
|
cmd = f'amixer -q set "{source}" 80% unmute'.split()
|
|
run_program(cmd, 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)
|
|
|
|
|
|
def main_menu():
|
|
"""Main menu for hardware diagnostics."""
|
|
args = docopt(DOCSTRING)
|
|
menu = Menu()
|
|
|
|
# Build menu
|
|
menu.title = color_string(
|
|
strings=['Hardware Diagnostics', 'Main Menu'],
|
|
colors=['GREEN', None],
|
|
sep='\n',
|
|
)
|
|
for action in MENU_ACTIONS:
|
|
menu.add_action(action)
|
|
for option in MENU_OPTIONS:
|
|
menu.add_option(option, {'Selected': True})
|
|
for toggle in MENU_TOGGLES:
|
|
menu.add_toggle(toggle, {'Selected': True})
|
|
for name, targets in MENU_SETS.items():
|
|
menu.add_set(name, {'Targets': targets})
|
|
menu.actions['Start']['Separator'] = True
|
|
|
|
# Check if running a quick check
|
|
if args['--quick']:
|
|
for name in menu.options.keys():
|
|
# Only select quick option(s)
|
|
menu.options[name]['Selected'] = name in MENU_OPTIONS_QUICK
|
|
|
|
# Compatibility checks
|
|
if platform.system() != 'Linux':
|
|
for name in ('Audio Test', 'Keyboard Test', 'Network Test'):
|
|
menu.actions[name]['Disabled'] = True
|
|
|
|
# Show menu
|
|
while True:
|
|
selection = menu.advanced_select()
|
|
if 'Audio Test' in selection:
|
|
audio_test()
|
|
if 'Network Test' in selection:
|
|
network_test()
|
|
elif 'Quit' in selection:
|
|
break
|
|
|
|
|
|
def network_test():
|
|
"""Run network tests."""
|
|
clear_screen()
|
|
try_and_print = TryAndPrint()
|
|
result = try_and_print.run(
|
|
'Network connection...', connected_to_private_network, msg_good='OK')
|
|
|
|
# Bail if not connected
|
|
if result['Failed']:
|
|
print_warning('Please connect to a network and try again')
|
|
pause('Press Enter to return to main menu...')
|
|
return
|
|
|
|
# Show IP address(es)
|
|
show_valid_addresses()
|
|
|
|
# Ping tests
|
|
try_and_print.run(
|
|
'Internet connection...', ping, msg_good='OK', addr='8.8.8.8')
|
|
try_and_print.run(
|
|
'DNS resolution...', ping, msg_good='OK', addr='google.com')
|
|
|
|
# Speedtest
|
|
try_and_print.run('Speedtest...', speedtest)
|
|
|
|
# Done
|
|
pause('Press Enter to return to main menu...')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
print("This file is not meant to be called directly.")
|