From 5fae374e95f2185d0e19902b47487600e4bc25a9 Mon Sep 17 00:00:00 2001
From: m1ngsama <contact@m1ng.space>
Date: Sun, 3 Nov 2024 14:30:00 +0800
Subject: [PATCH] Add network monitoring functionality

---
 tracker.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/tracker.py b/tracker.py
index 8da3cf2..25b3f6e 100644
--- a/tracker.py
+++ b/tracker.py
@@ -36,6 +36,16 @@ class SystemTracker:
             'percent': disk.percent
         }
 
+    def get_network_stats(self):
+        """Get network I/O statistics"""
+        net = psutil.net_io_counters()
+        return {
+            'bytes_sent': net.bytes_sent,
+            'bytes_recv': net.bytes_recv,
+            'packets_sent': net.packets_sent,
+            'packets_recv': net.packets_recv
+        }
+
     def display_stats(self):
         """Display all system statistics"""
         print(f"\n{'='*50}")
@@ -50,6 +60,9 @@ class SystemTracker:
         disk = self.get_disk_usage()
         print(f"Disk: {disk['percent']}% ({disk['used'] / (1024**3):.2f}GB / {disk['total'] / (1024**3):.2f}GB)")
 
+        net = self.get_network_stats()
+        print(f"Network: Sent {net['bytes_sent'] / (1024**2):.2f}MB | Recv {net['bytes_recv'] / (1024**2):.2f}MB")
+
 
 if __name__ == "__main__":
     tracker = SystemTracker()