implemented OOP and added colours

Author: Hillary520

cpu_monitor.py

@@ -0,0 +1,14 @@
+import psutil
+
+class CPUMonitor:
+    def __init__(self, bars=50):
+        self.bars = bars
+
+    def get_cpu_usage(self):
+        cpu = psutil.cpu_percent(interval=1)
+        return self._format_usage(cpu)
+
+    def _format_usage(self, cpu):
+        cpu_percent = cpu / 100.0
+        cpu_visual = '#' * int(cpu_percent * self.bars) + '-' * (self.bars - int(cpu_percent * self.bars))
+        return f"CPU: [{cpu_visual}] {cpu:.2f}%"

main.py

@@ -1,49 +1,34 @@
 import time
-import psutil
+from cpu_monitor import CPUMonitor
+from memory_monitor import MemoryMonitor
+from network_monitor import NetworkMonitor
 
-# Function to display CPU and memory usage in a visual format
-def cpu_and_memory_usage(cpu, memory, bars=50):
-    cpu_percent = (cpu / 100.0)
-    cpu_visual = '#' * int(cpu_percent * bars) + '-' * (bars - int(cpu_percent * bars))
-
-    memory_percent = (memory / 100.0)
-    memory_visual = '#' * int(memory_percent * bars) + '-' * (bars - int(memory_percent * bars))
-
-    return f"CPU: [{cpu_visual}] {cpu:.2f}% | Memory: [{memory_visual}] {memory:.2f}%"
-
-# Function to monitor network bandwidth usage
-def bandwidth_monitor(last_received, last_sent):
-    # Get the current number of bytes received and sent
-    bytes_received = psutil.net_io_counters().bytes_recv
-    bytes_sent = psutil.net_io_counters().bytes_sent
-
-    # Calculate the difference from the previous values
-    new_received = bytes_received - last_received
-    new_sent = bytes_sent - last_sent
-
-    # Convert to kilobytes
-    received = new_received / 1024
-    sent = new_sent / 1024
-    total = (new_received + new_sent) / 1024
-
-    return f"Received: {received:.2f} KB | Sent: {sent:.2f} KB | Total: {total:.2f} KB", bytes_received, bytes_sent
+# ANSI color escape codes
+COLOR_RESET = "\033[0m"
+COLOR_YELLOW = "\033[93m"  # Yellow for CPU
+COLOR_GREEN = "\033[92m"   # Green for Memory
+COLOR_BLUE = "\033[94m"    # Blue for Bandwidth
 
 def main():
-    # Get the initial values for bytes received and sent
-    last_received = psutil.net_io_counters().bytes_recv
-    last_sent = psutil.net_io_counters().bytes_sent
+    cpu_monitor = CPUMonitor()
+    memory_monitor = MemoryMonitor()
+    network_monitor = NetworkMonitor()
 
-    while True:
-        # Get CPU, memory, and bandwidth usage
-        cpu_memory_output = cpu_and_memory_usage(psutil.cpu_percent(), psutil.virtual_memory().percent)
-        bandwidth_output, last_received, last_sent = bandwidth_monitor(last_received, last_sent)
-
-        # Clear the screen and print the updated output
-        print(f"\033[H\033[J{cpu_memory_output}\n{bandwidth_output}", end='')
-        time.sleep(0.5)  # Wait for 0.5 seconds before updating again
-
-if __name__ == "__main__":
     try:
-        main()
+        while True:
+            cpu_output = cpu_monitor.get_cpu_usage()
+            memory_output = memory_monitor.get_memory_usage()
+            bandwidth_output = network_monitor.get_bandwidth_usage()
+
+            # Apply colors to entire output strings
+            cpu_output = f"{COLOR_YELLOW}{cpu_output}{COLOR_RESET}"
+            memory_output = f"{COLOR_GREEN}{memory_output}{COLOR_RESET}"
+            bandwidth_output = f"{COLOR_BLUE}{bandwidth_output}{COLOR_RESET}"
+
+            print(f"\033[H\033[J{cpu_output} | {memory_output}\n{bandwidth_output}", end='\r')
+            time.sleep(0.5)  # Wait for 0.5 seconds before updating again
     except KeyboardInterrupt:
-        print("\nExiting...")
+        print("\nExiting...")
+
+if __name__ == "__main__":
+    main()

memory_monitor.py

@@ -0,0 +1,17 @@
+import psutil
+
+class MemoryMonitor:
+    def __init__(self, bars=50):
+        self.bars = bars
+
+    def get_memory_usage(self):
+        memory = psutil.virtual_memory().percent
+        return self._format_usage(memory)
+
+    def _format_usage(self, memory):
+        memory_percent = memory / 100.0
+        memory_visual = '#' * int(memory_percent * self.bars) + '-' * (self.bars - int(memory_percent * self.bars))
+        return f"Memory: [{memory_visual}] {memory:.2f}%"
+
+    def get_memory_percentage(self):
+        return psutil.virtual_memory().percent

network_monitor.py

@@ -0,0 +1,22 @@
+import psutil
+
+class NetworkMonitor:
+    def __init__(self):
+        self.last_received = psutil.net_io_counters().bytes_recv
+        self.last_sent = psutil.net_io_counters().bytes_sent
+
+    def get_bandwidth_usage(self):
+        bytes_received = psutil.net_io_counters().bytes_recv
+        bytes_sent = psutil.net_io_counters().bytes_sent
+
+        new_received = bytes_received - self.last_received
+        new_sent = bytes_sent - self.last_sent
+
+        self.last_received = bytes_received
+        self.last_sent = bytes_sent
+
+        received = new_received / 1024
+        sent = new_sent / 1024
+        total = (new_received + new_sent) / 1024
+
+        return f"Received: {received:.2f} KB | Sent: {sent:.2f} KB | Total: {total:.2f} KB"