diff --git a/.bin/Scripts/functions/hw_diags.py b/.bin/Scripts/functions/hw_diags.py index 678e9d5d..046a3cf8 100644 --- a/.bin/Scripts/functions/hw_diags.py +++ b/.bin/Scripts/functions/hw_diags.py @@ -495,6 +495,7 @@ class State(): def __init__(self): self.cpu = None self.disks = [] + self.ost_integration = False self.panes = {} self.quick_mode = False self.tests = OrderedDict({ @@ -746,6 +747,7 @@ def menu_diags(state, args): {'Base Name': 'Full Diagnostic', 'Enabled': False}, {'Base Name': 'Disk Diagnostic', 'Enabled': False}, {'Base Name': 'Disk Diagnostic (Quick)', 'Enabled': False}, + {'Base Name': 'osTicket Integration', 'Enabled': True}, {'Base Name': 'Prime95', 'Enabled': False, 'CRLF': True}, {'Base Name': 'NVMe / SMART', 'Enabled': False}, {'Base Name': 'badblocks', 'Enabled': False}, @@ -777,11 +779,11 @@ def menu_diags(state, args): while True: # Set quick mode as necessary - if main_options[2]['Enabled'] and main_options[4]['Enabled']: + if main_options[2]['Enabled'] and main_options[5]['Enabled']: # Check if only Disk Diags (Quick) and NVMe/SMART are enabled # If so, verify no other tests are enabled and set quick_mode state.quick_mode = True - for opt in main_options[3:4] + main_options[5:]: + for opt in main_options[4:5] + main_options[6:]: state.quick_mode &= not opt['Enabled'] else: state.quick_mode = False @@ -795,13 +797,13 @@ def menu_diags(state, args): # Verify preset selections num_tests_selected = 0 - for opt in main_options[3:]: + for opt in main_options[4:]: if opt['Enabled']: num_tests_selected += 1 if num_tests_selected == 4: # Full main_options[0]['Enabled'] = True - elif num_tests_selected == 3 and not main_options[3]['Enabled']: + elif num_tests_selected == 3 and not main_options[4]['Enabled']: # Disk main_options[1]['Enabled'] = True @@ -1576,33 +1578,34 @@ def update_main_options(state, selection, main_options): if main_options[index]['Enabled']: for opt in main_options[1:3]: opt['Enabled'] = False - for opt in main_options[3:]: + for opt in main_options[4:]: opt['Enabled'] = True else: - for opt in main_options[3:]: + for opt in main_options[4:]: opt['Enabled'] = False elif index == 1: # Disk if main_options[index]['Enabled']: main_options[0]['Enabled'] = False - for opt in main_options[2:4]: - opt['Enabled'] = False - for opt in main_options[4:]: + main_options[2]['Enabled'] = False + main_options[4]['Enabled'] = False + for opt in main_options[5:]: opt['Enabled'] = True else: - for opt in main_options[4:]: + for opt in main_options[5:]: opt['Enabled'] = False elif index == 2: # Disk (Quick) if main_options[index]['Enabled']: - for opt in main_options[:2] + main_options[3:]: + for opt in main_options[:2] + main_options[4:]: opt['Enabled'] = False - main_options[4]['Enabled'] = True + main_options[5]['Enabled'] = True else: - main_options[4]['Enabled'] = False + main_options[5]['Enabled'] = False # Update state - for opt in main_options[3:]: + state.ost_integration = main_options[3]['Enabled'] + for opt in main_options[4:]: state.tests[opt['Base Name']]['Enabled'] = opt['Enabled'] # Done