Replace references to details with attributes

This commit is contained in:
2Shirt 2022-04-04 18:58:45 -06:00
parent 4647efb971
commit 5ffa6d8261
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
2 changed files with 11 additions and 14 deletions

View file

@ -28,9 +28,6 @@ from wk.cfg.ddrescue import (
DDRESCUE_SPECIFIC_PASS_SETTINGS,
)
from wk.hw import disk as hw_disk
from wk.hw import sensors as hw_sensors
from wk.hw import system as hw_system
from wk.hw.test import Test
# STATIC VARIABLES
@ -276,7 +273,7 @@ class BlockPair():
dest_size = -1
if self.destination.exists():
dest_obj = hw_disk.Disk(self.destination)
dest_size = dest_obj.details['size']
dest_size = dest_obj.size
del dest_obj
# Check destination size if cloning
@ -1674,7 +1671,7 @@ def get_object(path):
obj = hw_disk.Disk(path)
# Child/Parent check
parent = obj.details['parent']
parent = obj.raw_details['parent']
if parent:
std.print_warning(f'"{obj.path}" is a child device')
if std.ask(f'Use parent device "{parent}" instead?'):
@ -1836,7 +1833,7 @@ def source_or_destination_changed(state):
elif isinstance(obj, hw_disk.Disk):
compare_dev = hw_disk.Disk(obj.path)
for key in ('model', 'serial'):
changed = changed or obj.details[key] != compare_dev.details[key]
changed = changed or getattr(obj, key) != getattr(compare_dev, key)
# Update top panes
state.update_top_panes()

View file

@ -316,7 +316,7 @@ class State():
elif 'Disk' in name:
for disk in self.disks:
test_obj = Test(dev=disk, label=disk.path.name, name=name)
disk.test.append(test_obj)
disk.tests.append(test_obj)
self.tests[name]['Objects'].append(test_obj)
# Run safety checks
@ -592,7 +592,7 @@ def check_io_benchmark_results(test_obj, rate_list, graph_width):
avg_read = sum(rate_list) / len(rate_list)
min_read = min(rate_list)
max_read = max(rate_list)
if test_obj.dev.details['ssd']:
if test_obj.dev.ssd:
thresh_min = cfg.hw.THRESH_SSD_MIN
thresh_avg_high = cfg.hw.THRESH_SSD_AVG_HIGH
thresh_avg_low = cfg.hw.THRESH_SSD_AVG_LOW
@ -878,7 +878,7 @@ def disk_io_benchmark(state, test_objects, skip_usb=True):
# Get dd values or bail
try:
dd_values = calc_io_dd_values(test_obj.dev.details['size'])
dd_values = calc_io_dd_values(test_obj.dev.size)
except DeviceTooSmallError:
test_obj.set_status('N/A')
test_obj.report.append(
@ -951,7 +951,7 @@ def disk_io_benchmark(state, test_objects, skip_usb=True):
continue
# Skip USB devices if requested
if skip_usb and test.dev.details['bus'] == 'USB':
if skip_usb and test.dev.bus == 'USB':
test.set_status('Skipped')
continue
@ -1098,14 +1098,14 @@ def disk_surface_scan(state, test_objects):
test_obj.set_status('Working')
# Increase block size if necessary
if (dev.details['phy-sec'] == 4096
or dev.details['size'] >= cfg.hw.BADBLOCKS_LARGE_DISK):
if (dev.phy_sec == 4096
or dev.size >= cfg.hw.BADBLOCKS_LARGE_DISK):
block_size = '4096'
# Start scan
cmd = ['sudo', 'badblocks', '-sv', '-b', block_size, '-e', '1', dev_path]
with open(log_path, 'a', encoding='utf-8') as _f:
size_str = std.bytes_to_string(dev.details["size"], use_binary=False)
size_str = std.bytes_to_string(dev.size, use_binary=False)
_f.write(
std.color_string(
['[', dev.path.name, ' ', size_str, ']\n'],
@ -1156,7 +1156,7 @@ def disk_surface_scan(state, test_objects):
line for line in disk.generate_attribute_report() if 'failed' in line
]
if failed_attributes:
size_str = std.bytes_to_string(disk.details["size"], use_binary=False)
size_str = std.bytes_to_string(disk.size, use_binary=False)
std.print_colored(
['[', disk.path.name, ' ', size_str, ']'],
[None, 'BLUE', None, 'CYAN', None],