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

Fix SMART attribute tracking

Browse source 
Since we've moved to delayed SMART attribute updates we need to set
initial_attributes after we first check the SMART data instead of at
object creation time.
This commit is contained in:
2Shirt 2023-07-08 18:07:55 -07:00
parent a78a077bdf
commit 8f14fd2442
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C
2 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
scripts/wk/hw/disk.py
View file

@ -1,7 +1,6 @@
"""WizardKit: Disk object and functions""" """WizardKit: Disk object and functions"""
# vim: sts=2 sw=2 ts=2 # vim: sts=2 sw=2 ts=2
import copy
import logging import logging
import pathlib import pathlib
import platform import platform
@ -39,7 +38,7 @@ class Disk:
children: list[dict] = field(init=False, default_factory=list) children: list[dict] = field(init=False, default_factory=list)
description: str = field(init=False) description: str = field(init=False)
filesystem: str = field(init=False) filesystem: str = field(init=False)
initial_attributes: dict[Any, dict] = 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) known_attributes: dict[Any, dict] = field(init=False, default_factory=dict)
log_sec: int = field(init=False) log_sec: int = field(init=False)
model: str = field(init=False) model: str = field(init=False)
@ -62,7 +61,6 @@ class Disk:
self.update_details() self.update_details()
self.set_description() self.set_description()
self.known_attributes = get_known_disk_attributes(self.model) self.known_attributes = get_known_disk_attributes(self.model)
self.initial_attributes = copy.deepcopy(self.attributes)
if not self.is_4k_aligned(): if not self.is_4k_aligned():
self.add_note('One or more partitions are not 4K aligned', 'YELLOW') self.add_note('One or more partitions are not 4K aligned', 'YELLOW')

4
scripts/wk/hw/smart.py
View file

@ -506,6 +506,10 @@ def update_smart_details(dev) -> None:
if not updated_attributes: if not updated_attributes:
dev.add_note('No NVMe or SMART data available', 'YELLOW') 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 # Done
dev.attributes.update(updated_attributes) dev.attributes.update(updated_attributes)