66 lines
2 KiB
Python
66 lines
2 KiB
Python
from threading import Thread
|
|
from SecureSocketService import Socket
|
|
|
|
|
|
class Server(Socket):
|
|
def __init__(self, host: str = "localhost", port: int = 3621, service_id: int = 2):
|
|
super().__init__()
|
|
self.socket.bind((host, port))
|
|
self.socket.listen(5)
|
|
self.service_id = service_id
|
|
self.clients = dict()
|
|
Thread(target=self.connexion).start()
|
|
|
|
def connexion(self):
|
|
while True:
|
|
c, adress = self.connect_client(self.socket)
|
|
name = self.client_name(c)
|
|
if name:
|
|
self.clients[name] = c
|
|
self.broadcast(f"{name} is online !")
|
|
Thread(target=self.listen_client, args=(name,)).start()
|
|
|
|
def client_name(self, sock):
|
|
while True:
|
|
try:
|
|
self.send(sock, "Your name ?")
|
|
name = self.receive(sock)
|
|
if name in self.clients:
|
|
self.send(sock, "Name already taken !")
|
|
elif name.lower() == "quit":
|
|
sock.close()
|
|
name = None
|
|
else:
|
|
break
|
|
except:
|
|
name = None
|
|
break
|
|
return name
|
|
|
|
def listen_client(self, name):
|
|
while True:
|
|
try:
|
|
data = self.receive(self.clients[name])
|
|
assert data.lower() != "quit"
|
|
except:
|
|
self.send(self.clients[name], "quit")
|
|
self.clients[name].close()
|
|
del self.clients[name]
|
|
self.broadcast(f"{name} is offline !")
|
|
break
|
|
else:
|
|
Thread(target=self.broadcast, args=(f"{name}: {data}", name)).start()
|
|
|
|
def broadcast(self, message, author = None):
|
|
print(message)
|
|
for i in self.clients:
|
|
if i == author:
|
|
continue
|
|
try:
|
|
self.send(self.clients[i], message)
|
|
except:
|
|
pass
|
|
|
|
|
|
if __name__ == "__main__":
|
|
Server()
|