pypong/networking/host.py

@@ -0,0 +1,36 @@
+import psutil
+
+
+class InterfaceInfo:
+    """
+    Class used for storing interface information to make it easier to use
+    """
+    def __init__(self, interface_name, address, subnet_mask):
+        self.interface_name = interface_name
+        self.address = address
+        self.subnet_mask = subnet_mask
+
+    def __str__(self):
+        return '(Interface:{0}, Address:{1}, Subnet Mask:{2})'.format(self.interface_name,
+                                                                      self.address, self.subnet_mask)
+
+    def __repr__(self):
+        return '(Interface:{0}, Address:{1}, Subnet Mask:{2})'.format(self.interface_name,
+                                                                      self.address, self.subnet_mask)
+
+
+def main():
+    raw_interface_data = psutil.net_if_addrs()
+    interface_data = []
+
+    # Extract needed information from psutil and put it into a InterfaceInfo object
+    for interface in raw_interface_data:
+        address = (raw_interface_data.get(interface)[0]).address
+        subnet_mask = (raw_interface_data.get(interface)[0]).netmask
+        interface_data.append(InterfaceInfo(interface, address, subnet_mask))
+
+    print(interface_data)
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -1 +1,2 @@
 pygame~=2.1.2
+psutil~=5.9.4