cleanup

ryan-willis · Apr 7, 2024 · 35487a0 · 35487a0
1 parent 08d2576
commit 35487a0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/Netivity.py b/Netivity.py
@@ -50,5 +50,9 @@ def tick(self, *_):
         self.app.title = bandwidth.retrieve(self.params)
 
 
-if __name__ == "__main__":
+def main():
     Netivity()
+
+
+if __name__ == "__main__":
+    main()
diff --git a/bandwidth.py b/bandwidth.py
@@ -11,6 +11,10 @@ class Params(Structure):
     ]
 
 
+get_bandwidth = CDLL("bandwidth.dylib").get_bandwidth
+get_bandwidth.argtypes = [POINTER(Params)]
+
+
 def human_rate(bytes):
     i = 0
     suffixes = ["B", "K", "M"]
@@ -27,11 +31,6 @@ def format_human(p: Params):
     return f"{iface}⬇{net_in}⬆{net_out}"
 
 
-get_bandwidth = CDLL("bandwidth.dylib").get_bandwidth
-get_bandwidth.argtypes = [POINTER(Params)]
-
-
 def retrieve(p: Params):
     get_bandwidth(byref(p))
-    print(p.last_in, p.last_out, p.net_in, p.net_out, p.ifa_name)
     return format_human(p)