Cleanup unused keys from disk info for Windows OS

thevickypedia · Jan 2, 2025 · 818b5a3 · 818b5a3
1 parent 045dd92
commit 818b5a3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/pyarchitecture/__init__.py b/pyarchitecture/__init__.py
@@ -5,7 +5,7 @@
 
 from pyarchitecture import cpu, disks, gpu, memory
 
-version = "0.1.0a4"
+version = "0.1.0a5"
 
 
 def all_components() -> Dict[str, Any]:

diff --git a/pyarchitecture/disks/windows.py b/pyarchitecture/disks/windows.py
@@ -25,9 +25,11 @@ def reformat_windows(data: Dict[str, str | int | float]) -> Dict[str, str]:
     data["name"] = data["Model"]
     data["device_id"] = data["DeviceID"].replace("\\", "").replace(".", "")
     data["size"] = squire.size_converter(data["Size"])
-    del data["Caption"]
-    del data["Model"]
-    del data["DeviceID"]
+    data.pop("Caption", None)
+    data.pop("Model", None)
+    data.pop("DeviceID", None)
+    data.pop("Partitions", None)
+    data.pop("Size", None)
     return data