Add dev type

lukemartinlogan · Jan 28, 2025 · df44f21 · df44f21
1 parent 14a1f99
commit df44f21
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/jarvis_util/introspect/system_info.py b/jarvis_util/introspect/system_info.py
@@ -225,10 +225,22 @@ def wait(self):
             for dev in lsblk_data:
                 if dev['tran'] == 'pcie':
                     dev['tran'] = 'nvme'
+                dev['dev_type'] = self.GetDevType(dev)
                 dev['host'] = host
                 total.append(dev)
         self.df = sdf.SmallDf(rows=total, columns=self.columns)
 
+    def GetDevType(self, dev):
+        if dev['tran'] == 'sata':
+            if dev['rota']:
+                return str(StorageDeviceType.HDD)
+            else:
+                return str(StorageDeviceType.SSD)
+        elif dev['tran'] == 'nvme':
+            return str(StorageDeviceType.NVME)
+        elif dev['tran'] == 'dimm':
+            return str(StorageDeviceType.PMEM)
+
 
 class Blkid(Exec):
     """