Merge pull request codehouseindia#346 from chiragarora01/master

codehouseindia · web-flow · commit 054888edebbc · 2020-10-05T23:31:30.000+05:30
Example of Socket Connection
diff --git a/Basic Socket/clint.py b/Basic Socket/clint.py
@@ -0,0 +1,33 @@
+from socket import AF_INET, socket, SOCK_STREAM
+from threading import Thread
+
+HOST = "localhost"
+PORT = 5500
+ADDR = (HOST, PORT)
+BUFSIZ = 1024
+messages = []
+
+client_socket = socket(AF_INET, SOCK_STREAM)
+client_socket.connect(ADDR)
+
+
+def receive_messages():
+    while True:
+        try:
+            msg = client_socket.recv(BUFSIZ).decode()
+            messages.append(msg)
+            print(msg)
+        except Exception as e:
+            print("[EXCEPTION]", e)
+            break
+
+
+def send_messages(msg):
+    client_socket.send(bytes(msg, "utf8"))
+    if msg == "{quit}":
+        client_socket.close()
+
+
+receive_thread = Thread(target=receive_messages)
+receive_thread.start()
+send_messages("chirag")
diff --git a/Basic Socket/person.py b/Basic Socket/person.py
@@ -0,0 +1,11 @@
+class Person:
+    def __int__(self, addr, client):
+        self.addr = addr
+        self.client = client
+        self.name = None
+
+    def set_name(self, name):
+        self.name = name
+
+    def __repr__(self):
+        return f"Person({self.addr},{self.name})"
diff --git a/Basic Socket/server.py b/Basic Socket/server.py
@@ -0,0 +1,75 @@
+from socket import AF_INET, socket, SOCK_STREAM
+from threading import Thread
+import time
+from person import Person
+
+HOST = "localhost"
+PORT = 5500
+ADDR = (HOST, PORT)
+MAX_CONNECTION = 1
+BUFFSIZ = 1024
+
+persons = []
+SERVER = socket(AF_INET, SOCK_STREAM)
+SERVER.bind(ADDR)
+
+
+def broadcast(msg, name):
+    for person in persons:
+        client = person.client
+        client.send(bytes(name, "utf8" + msg))
+
+
+def client_communication(person):
+    '''handle all messages from client'''
+    client = person.client
+
+    name = client.recv(BUFFSIZ).decode("utf8")
+    person.set_name(name)
+    msg = bytes(f"{name} has joined the chat", "utf8")
+    broadcast(msg, name)
+
+    while True:
+        try:
+            msg = client.recv(BUFFSIZ)
+            print(f"{name}:", msg.decode("utf8"))
+            if msg == bytes("{quit}", "utf8"):
+                broadcast(f"{name} has left the chat ...", "")
+                client.send(bytes("{quit}", "utf8"))
+                client.close()
+                persons.remove(person)
+                break
+            else:
+                broadcast(msg, name+": ")
+        except Exception as e:
+            print("[EXCEPTION]", e)
+            break
+
+
+def wait_for_connection():
+    '''wait for connection from new clients ,start new thread'''
+    run = True
+    while run:
+        try:
+            client, addr = SERVER.accept()
+            person = Person(addr, client)
+            persons.append(person)
+            print(f"[CONNECTION] {addr} connected to server at {time.time()} ")
+            Thread(target=client_communication, args=(person,)).start()
+        except Exception as e:
+            print("[EXCEPTION]", e)
+            run = False
+
+    print("server crash")
+
+
+SERVER = socket(AF_INET, SOCK_STREAM)
+SERVER.bind(ADDR)
+
+if __name__ == "__main__":
+    SERVER.listen(MAX_CONNECTION)
+    print("waiting for connection")
+    ACCEPT_THREAD = Thread(target=wait_for_connection)
+    ACCEPT_THREAD.start()
+    ACCEPT_THREAD.join()
+    SERVER.close()
