Refactored system status into a class. Added charts

src/modules/SystemStatus.py

@@ -3,7 +3,25 @@ from asyncio.subprocess import Process
 import psutil
 
 class SystemStatus:
-    pass
+    def __init__(self):
+        self.CPU = CPU()
+        self.MemoryVirtual = Memory('virtual')
+        self.MemorySwap = Memory('swap')
+        self.Disks = Disks()
+        self.NetworkInterfaces = NetworkInterfaces()
+        self.Processes = Processes()
+    def toJson(self):
+        return {
+            "CPU": self.CPU,
+            "Memory": {
+                "VirtualMemory": self.MemoryVirtual,
+                "SwapMemory": self.MemorySwap
+            },
+            "Disks": self.Disks,
+            "NetworkInterfaces": self.NetworkInterfaces,
+            "Processes": self.Processes
+        }
+        
 
 class CPU:
     def __init__(self):
@@ -11,8 +29,8 @@ class CPU:
         self.cpu_percent_per_cpu: list[float] = []
     def getData(self):
         try:
-            self.cpu_percent = psutil.cpu_percent(interval=0.5, percpu=True)
-            self.cpu_percent_per_cpu = psutil.cpu_percent(interval=0.5)
+            self.cpu_percent_per_cpu = psutil.cpu_percent(interval=0.5, percpu=True)
+            self.cpu_percent = psutil.cpu_percent(interval=0.5)
         except Exception as e:
             pass
     def toJson(self):
@@ -21,22 +39,37 @@ class CPU:
 
 class Memory:
     def __init__(self, memoryType: str):
-        self.__memoryType = memoryType
+        self.__memoryType__ = memoryType
         self.total = 0
         self.available = 0
         self.percent = 0
     def getData(self):
-        if self.__memoryType == "virtual":
-            memory = psutil.virtual_memory()
-        else:
-            memory = psutil.swap_memory()
-        self.total = memory.total
-        self.available = memory.available
-        self.percent = memory.percent
+        try:
+            if self.__memoryType__ == "virtual":
+                memory = psutil.virtual_memory()
+            else:
+                memory = psutil.swap_memory()
+            self.total = memory.total
+            self.available = memory.available
+            self.percent = memory.percent
+        except Exception as e:
+            pass
     def toJson(self):
         self.getData()
         return self.__dict__
-    
+
+class Disks:
+    def __init__(self):
+        self.disks : list[Disk] = []
+    def getData(self):
+        try:
+            self.disks = list(map(lambda x : Disk(x.mountpoint), psutil.disk_partitions()))
+        except Exception as e:
+            pass
+    def toJson(self):
+        self.getData()
+        return self.disks
+
 class Disk:
     def __init__(self, mountPoint: str):
         self.total = 0
@@ -45,11 +78,14 @@ class Disk:
         self.percent = 0
         self.mountPoint = mountPoint
     def getData(self):
-        disk = psutil.disk_usage(self.mountPoint)
-        self.total = disk.total
-        self.free = disk.free
-        self.used = disk.used
-        self.percent = disk.percent
+        try:
+            disk = psutil.disk_usage(self.mountPoint)
+            self.total = disk.total
+            self.free = disk.free
+            self.used = disk.used
+            self.percent = disk.percent
+        except Exception as e:
+            pass
     def toJson(self):
         self.getData()
         return self.__dict__
@@ -58,9 +94,12 @@ class NetworkInterfaces:
     def __init__(self):
         self.interfaces = {}
     def getData(self):
-        network = psutil.net_io_counters(pernic=True, nowrap=True)
-        for i in network.keys():
-            self.interfaces[i] = network[i]._asdict()
+        try:
+            network = psutil.net_io_counters(pernic=True, nowrap=True)
+            for i in network.keys():
+                self.interfaces[i] = network[i]._asdict()
+        except Exception as e:
+            pass
     def toJson(self):
         self.getData()
         return self.interfaces
@@ -78,19 +117,21 @@ class Processes:
         self.CPU_Top_10_Processes: list[Processes.Process] = []
         self.Memory_Top_10_Processes: list[Processes.Process] = []
     def getData(self):
-        processes = list(psutil.process_iter())
-        self.CPU_Top_10_Processes = sorted(list(map(lambda x : 
-                    Processes.Process(x.name(), " ".join(x.cmdline()), x.pid, x.cpu_percent()), processes)),
-                    key=lambda x : x.percent, reverse=True)[:10]
-        self.Memory_Top_10_Processes = sorted(list(map(lambda x :
-                    Processes.Process(x.name(), " ".join(x.cmdline()), x.pid, x.memory_percent()), processes)),
-                    key=lambda x : x.percent, reverse=True)[:10]
+        while True:
+            try:
+                processes = list(psutil.process_iter())
+                self.CPU_Top_10_Processes = sorted(
+                    list(map(lambda x : Processes.Process(x.name(), " ".join(x.cmdline()), x.pid, x.cpu_percent()), processes)),
+                    key=lambda x : x.percent, reverse=True)[:20]
+                self.Memory_Top_10_Processes = sorted(
+                    list(map(lambda x : Processes.Process(x.name(), " ".join(x.cmdline()), x.pid, x.memory_percent()), processes)),
+                    key=lambda x : x.percent, reverse=True)[:20]
+                break
+            except Exception as e:
+                continue
     def toJson(self):
         self.getData()
         return {
             "cpu_top_10": self.CPU_Top_10_Processes,
             "memory_top_10": self.Memory_Top_10_Processes
-        }
-
-p = Processes()
-print(p.toJson())
+        }