changed to the use of the curses library

Author: Hillary520

main.py

@@ -1,18 +1,16 @@
+import curses
 import time
 from cpu_monitor import CPUMonitor
 from memory_monitor import MemoryMonitor
 from network_monitor import NetworkMonitor
 from system_info import SystemInfo
 from processes import ProcessMonitor
 
-# 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
-COLOR_CYAN = "\033[96m"    # Cyan for Headers
+def main(stdscr):
+    # Clear screen
+    stdscr.clear()
+    curses.curs_set(0)  # Hide the cursor
 
-def main():
     cpu_monitor = CPUMonitor()
     memory_monitor = MemoryMonitor()
     network_monitor = NetworkMonitor()
@@ -21,46 +19,58 @@ def main():
 
     try:
         while True:
+            # Get screen dimensions
+            height, width = stdscr.getmaxyx()
+            half_width = width // 2
+
             # Get real-time data
             cpu_output = cpu_monitor.get_cpu_usage()
             memory_output = memory_monitor.get_memory_usage()
             bandwidth_output = network_monitor.get_bandwidth_usage()
             sys_info_output = system_info.get_system_info()
             processes_output = process_monitor.get_running_processes(num_processes=10)
 
-            # 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.replace('Received', '↓').replace('Sent', '↑')}{COLOR_RESET}"
-            sys_info_output = f"{COLOR_CYAN}{sys_info_output}{COLOR_RESET}"
-            processes_output = "\n".join([f"{COLOR_CYAN}Running Processes{COLOR_RESET}", "-----------------"] + processes_output)
-
             # Clear screen
-            print("\033c", end='')
+            stdscr.clear()
 
-            # Print headers
-            print(f"{COLOR_CYAN}CPU & Memory      Network{COLOR_RESET}")
-            print(f"{COLOR_CYAN}-------------      -------{COLOR_RESET}")
+            # Define color pairs
+            curses.start_color()
+            curses.init_pair(1, curses.COLOR_YELLOW, curses.COLOR_BLACK)
+            curses.init_pair(2, curses.COLOR_GREEN, curses.COLOR_BLACK)
+            curses.init_pair(3, curses.COLOR_BLUE, curses.COLOR_BLACK)
+            curses.init_pair(4, curses.COLOR_CYAN, curses.COLOR_BLACK)
+            curses.init_pair(5, curses.COLOR_RED, curses.COLOR_BLACK)
 
-            # Print meters and system info
-            print(f"{cpu_output}      {bandwidth_output}")
-            print(f"{memory_output}")
-            print(f"{COLOR_CYAN}System Info{COLOR_RESET}")
-            print(f"{COLOR_CYAN}-----------{COLOR_RESET}")
-            sys_info_lines = [line[:40] for line in sys_info_output.split("\n")]
-            for line in sys_info_lines:
-                print(line)
+            # Print CPU and Memory Usage
+            stdscr.addstr(0, 0, "CPU Usage", curses.color_pair(4))
+            stdscr.addstr(1, 0, cpu_output[:half_width], curses.color_pair(1))
+            stdscr.addstr(2, 0, "Memory Usage", curses.color_pair(4))
+            stdscr.addstr(3, 0, memory_output[:half_width], curses.color_pair(2))
 
-            # Print Running Processes
-            print(processes_output)
+            # Print Bandwidth Usage
+            stdscr.addstr(0, half_width, "Network Usage", curses.color_pair(4))
+            stdscr.addstr(1, half_width, bandwidth_output.replace('Received', '↓').replace('Sent', '↑')[:half_width], curses.color_pair(3))
 
-            # Move cursor to the top
-            print("\033[6A", end='')
+            # Print System Info
+            stdscr.addstr(5, 0, "System Info", curses.color_pair(4))
+            stdscr.addstr(6, 0, "-----------", curses.color_pair(4))
+            sys_info_lines = [line[:half_width] for line in sys_info_output.split("\n")]
+            for idx, line in enumerate(sys_info_lines):
+                if 7 + idx < height:
+                    stdscr.addstr(7 + idx, 0, line, curses.color_pair(4))
+
+            # Print Running Processes
+            stdscr.addstr(5, half_width, "Running Processes", curses.color_pair(4))
+            stdscr.addstr(6, half_width, "-----------------", curses.color_pair(4))
+            for idx, process in enumerate(processes_output):
+                if 7 + idx < height:
+                    stdscr.addstr(7 + idx, half_width, process[:half_width], curses.color_pair(5))
 
+            stdscr.refresh()
             time.sleep(0.5)  # Wait for 0.5 seconds before updating again
 
     except KeyboardInterrupt:
         print("\nExiting...")
 
 if __name__ == "__main__":
-    main()
+    curses.wrapper(main)

memory_monitor.py

@@ -12,6 +12,3 @@ 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

@@ -15,8 +15,8 @@ def get_bandwidth_usage(self):
         self.last_received = bytes_received
         self.last_sent = bytes_sent
 
-        received = new_received / 1024
-        sent = new_sent / 1024
-        total = (new_received + new_sent) / 1024
+        received_kb = new_received / 1024
+        sent_kb = new_sent / 1024
+        total_kb = (new_received + new_sent) / 1024
 
-        return f"Received: {received:.2f} KB | Sent: {sent:.2f} KB | Total: {total:.2f} KB"
+        return f"Received: {received_kb:.2f} KB | Sent: {sent_kb:.2f} KB | Total: {total_kb:.2f} KB"

processes.py

@@ -0,0 +1,24 @@
+import psutil
+
+class ProcessMonitor:
+    def get_running_processes(self, num_processes=10):
+        processes = psutil.process_iter(attrs=['name', 'pid', 'username'])
+        process_info = []
+        for process in processes:
+            try:
+                name = process.info['name']
+                pid = process.info['pid']
+                username = process.info['username']
+                process_info.append(f"{name} (PID: {pid}, User: {username})")
+            except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
+                continue
+            if len(process_info) >= num_processes:
+                break
+        return process_info
+
+# Example usage:
+if __name__ == "__main__":
+    process_monitor = ProcessMonitor()
+    processes = process_monitor.get_running_processes()
+    for process in processes:
+        print(process)

system_info.py

@@ -2,6 +2,7 @@
 import psutil
 import time
 import socket
+import subprocess
 
 class SystemInfo:
     def get_system_info(self):
@@ -21,7 +22,6 @@ def get_system_info(self):
         battery_status = self._get_battery_status()
         battery_health = self._get_battery_health()
 
-        # Apply color to system information output
         sys_info_output = (
             f"OS: {os_name}\n"
             f"Python: {python_version}\n"
@@ -84,8 +84,8 @@ def _get_battery_health(self):
                     ["ioreg", "-r", "-c", "AppleSmartBattery", "-k", "DesignCapacity", "-k", "MaxCapacity"],
                     universal_newlines=True
                 )
-                design_capacity = int(health_info.split('DesignCapacity')[1].split('= ')[1].split('\n')[0])
-                max_capacity = int(health_info.split('MaxCapacity')[1].split('= ')[1].split('\n')[0])
+                design_capacity = int(self._parse_health_info(health_info, 'DesignCapacity'))
+                max_capacity = int(self._parse_health_info(health_info, 'MaxCapacity'))
                 health_percent = (max_capacity / design_capacity) * 100
                 return f"{health_percent:.2f}%"
             elif os_name == 'Windows':  # Windows
@@ -96,8 +96,8 @@ def _get_battery_health(self):
                 # Read and parse the battery_report.html to get battery health information
                 with open("battery_report.html", "r") as file:
                     report = file.read()
-                design_capacity = int(report.split('DESIGN CAPACITY')[1].split('<td>')[1].split(' mWh')[0].replace(',', ''))
-                full_charge_capacity = int(report.split('FULL CHARGE CAPACITY')[1].split('<td>')[1].split(' mWh')[0].replace(',', ''))
+                design_capacity = int(self._parse_battery_report(report, 'DESIGN CAPACITY'))
+                full_charge_capacity = int(self._parse_battery_report(report, 'FULL CHARGE CAPACITY'))
                 health_percent = (full_charge_capacity / design_capacity) * 100
                 return f"{health_percent:.2f}%"
             elif os_name == 'Linux':  # Linux
@@ -111,3 +111,10 @@ def _get_battery_health(self):
                 return "Unsupported OS"
         except Exception as e:
             return f"Error retrieving battery health: {e}"
+
+    def _parse_health_info(self, health_info, key):
+        return health_info.split(f"{key} = ")[1].split('\n')[0].strip()
+
+    def _parse_battery_report(self, report, key):
+        return report.split(f'{key}')[1].split('<td>')[1].split(' mWh')[0].replace(',', '')
+