Update peer.py

MP2/peer.py

@@ -1,32 +1,26 @@
 from datetime import datetime, timezone
 import socket
 
-INTRODUCER_IP = "172.22.156.3"
-IP = socket.gethostbyname(socket.gethostname())
-SERVER_PORT = 6767
-INTRODUCER_FLAG = (INTRODUCER_IP == IP)
+def main():
+    INTRODUCER_IP = "172.22.156.3"
+    IP = socket.gethostbyname(socket.gethostname())
+    SERVER_PORT = 6767
+    INTRODUCER_FLAG = (INTRODUCER_IP == IP)
 
-#check if this VM is the introducer and add it self
-members_list = {}
-if INTRODUCER_FLAG:
-    members_list[(IP, PORT)] = {
-        "heart_beat": 1,
-        "last_seen": utc_time()
+    members_list = {
+        (IP, SERVER_PORT): {"heart_beat": 1, "last_seen": utc_time()}
     }
 
+    server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    server.bind(("", SERVER_PORT))
 
-#start UDP socket
-server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
-server.bind((IP, SERVER_PORT))
-
-if not INTRODUCER_FLAG:
+    if not INTRODUCER_FLAG:
         server.sendto(b"JOIN", (INTRODUCER_IP, SERVER_PORT))
-
-while True:
         data, addr = server.recvfrom(4096)
+        print(addr, data)
 
-#increment heart beat
-
-#utc time
 def utc_time():
     return datetime.now(timezone.utc)
+
+if __name__ == "__main__":
+    main()