Netwok bandwidth tracker

2025-02-11 20:22:35 +03:00
parent b4533d1dea
commit cedf9ee93a
3 changed files with 79 additions and 1 deletions
@@ -0,0 +1,39 @@
+#!/usr/bin/env -S\_/bin/sh\_-xc\_"source\_\$(eval\_echo\_\$ILLOGICAL_IMPULSE_VIRTUAL_ENV)/bin/activate&&exec\_python\_-E\_"\$0"\_"\$@""
+
+from time import sleep
+import sys
+import psutil
+
+try:
+    direction = sys.argv[1]
+except IndexError:
+    direction = "recv"
+
+init_bytes = final_bytes = 0
+
+match direction:
+    case "recv":
+        init_bytes = psutil.net_io_counters().bytes_recv
+        sleep(1)
+        final_bytes = psutil.net_io_counters().bytes_recv
+
+    case "sent":
+        init_bytes = psutil.net_io_counters().bytes_sent
+        sleep(1)
+        final_bytes = psutil.net_io_counters().bytes_sent
+
+    case _: 
+        print(f"wrong direction: {direction}")
+        sys.exit()
+
+i = 0
+divider = 1024
+bandwidth = int((final_bytes - init_bytes))
+units = ["B", "KB", "MB", "GB", "TB", "PB", "EB"]
+
+while bandwidth >= divider:
+    i += 1
+    bandwidth = bandwidth / divider
+
+print(f"{bandwidth:.1f}" + units[i] + "/s")
+