base code

Author: Vikranth3140

client.py

@@ -0,0 +1,34 @@
+import socket
+import threading
+
+# Choose a username
+username = input("Enter your username: ")
+
+# Client setup
+client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+client.connect(('localhost', 12345))
+
+# Listening to server and sending messages
+def receive_messages():
+ while True:
+ try:
+ message = client.recv(1024).decode('utf-8')
+ if message == 'USERNAME':
+ client.send(username.encode('utf-8'))
+ else:
+ print(message)
+ except:
+ print('An error occurred!')
+ client.close()
+ break
+
+def write_messages():
+ while True:
+ message = f'{username}: {input("")}'
+ client.send(message.encode('utf-8'))
+
+receive_thread = threading.Thread(target=receive_messages)
+receive_thread.start()
+
+write_thread = threading.Thread(target=write_messages)
+write_thread.start()

server.py

@@ -0,0 +1,51 @@
+import socket
+import threading
+
+# Server setup
+server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+server.bind(('localhost', 12345))
+server.listen()
+
+clients = []
+usernames = []
+
+# Broadcast messages to all clients
+def broadcast(message):
+ for client in clients:
+ client.send(message)
+
+# Handle messages from clients
+def handle_client(client):
+ while True:
+ try:
+ message = client.recv(1024)
+ broadcast(message)
+ except:
+ index = clients.index(client)
+ clients.remove(client)
+ client.close()
+ username = usernames[index]
+ broadcast(f'{username} has left the chat!'.encode('utf-8'))
+ usernames.remove(username)
+ break
+
+# Receive connections from clients
+def receive_connections():
+ while True:
+ client, address = server.accept()
+ print(f'Connected with {str(address)}')
+
+ client.send('USERNAME'.encode('utf-8'))
+ username = client.recv(1024).decode('utf-8')
+ usernames.append(username)
+ clients.append(client)
+
+ print(f'Username of the client is {username}')
+ broadcast(f'{username} has joined the chat!'.encode('utf-8'))
+ client.send('Connected to the server!'.encode('utf-8'))
+
+ thread = threading.Thread(target=handle_client, args=(client,))
+ thread.start()
+
+print('Server is listening...')
+receive_connections()