Get Pc information

When running in terminal, can pick a choice to get input devices plugged in, pc inoprmwtion, and network information through this terminal design

Author: Maday44

My-PC/info.py

@@ -1,8 +1,23 @@
 import win32com.client
 import wmi
+import socket
+import psutil
+
+
+# code source: https://thepythoncode.com/article/get-hardware-system-information-python
+def bytes_to_human(value, suffix="B"):
+    if value is None:
+        return "N/A"
+    value = float(value)
+    for unit in ("", "K", "M", "G", "T", "P"):
+        if abs(value) < 1024:
+            return f"{value:.2f}{unit}{suffix}"
+        value /= 1024
+    return f"{value:.2f}EB"
+
 
 def info_pc():
-    c = wmi.WMI()    
+    c = wmi.WMI()
     my_system = c.Win32_ComputerSystem()[0]
 
     print(f"Manufacturer: {my_system.Manufacturer}")
@@ -15,20 +30,17 @@ def info_pc():
 
 def list_plugged_in_devices():
     print("Loading getting machine devices\n")
-    
+
     # WMI API
     wmi = win32com.client.GetObject("winmgmts:")
 
     # query used to get the device information
-    devices = wmi.ExecQuery(
-    """
+    devices = wmi.ExecQuery("""
      SELECT Name, DeviceID, Manufacturer, PNPClass, Present, ConfigManagerErrorCode 
         FROM Win32_PnPEntity 
         WHERE Present = True AND Manufacturer != 'Microsoft'
-    """
-    )
+    """)
 
-    
     print(f"({len(devices)} devices found: \n")
     for dev in list(devices):
         print(f"Device Name: {dev.Name}")
@@ -39,28 +51,53 @@ def list_plugged_in_devices():
         print("-" * 100)
 
 
+def network_info():
+    for name, addresses in psutil.net_if_addrs().items():
+        print("Interface:", name)
+        for address in addresses:
+            if address.family == socket.AF_INET:
+                print("  IPv4:", address.address)
+                print("  Netmask:", address.netmask)
+            elif address.family == socket.AF_INET6:
+                print("  IPv6:", address.address)
+            elif hasattr(psutil, "AF_LINK") and address.family == psutil.AF_LINK:
+                print("  MAC:", address.address)
+
+    stats = psutil.net_if_stats()
+    for name, stat in stats.items():
+        print(name, "up=" + str(stat.isup), "speed=" + str(stat.speed), "Mbps")
+
+    io = psutil.net_io_counters()
+    print("Bytes sent:", bytes_to_human(io.bytes_sent))
+    print("Bytes received:", bytes_to_human(io.bytes_recv))
+
+
 def user_console():
     while True:
         print("\n See Pc information")
         print("1) Standard device information")
         print("2) Input devices")
-        print("3) Quit")
-        
+        print("3) Network Interface")
+        print("4) Quit")
+
         try:
-            option = int(input("What would you like to see about your device? (1-3): "))
-            
+            option = int(input("What would you like to see about your device? (1-4): "))
+
             if option == 1:
                 info_pc()
             elif option == 2:
                 list_plugged_in_devices()
             elif option == 3:
+                network_info()
+            elif option == 4:
                 print("Exiting console application. Goodbye!")
-                break  
+                break
             else:
                 print("Invalid choice! Please choose 1, 2, or 3.")
-                
+
         except ValueError:
             print("Error: Please enter a valid number (1, 2, or 3).")
 
+
 if __name__ == "__main__":
     user_console()