Files
tcp-chat/server/main.py
2022-06-12 16:11:14 -05:00

62 lines
1.7 KiB
Python

import logging
import socket
import sys
import threading
from server import handler
from shared import constants
host = constants.DEFAULT_IP
port = constants.DEFAULT_PORT
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((host, port))
server.listen(1)
server.settimeout(0.5)
logger = logging.getLogger('server')
logger.setLevel(logging.DEBUG)
clients = []
# Receiving / Listening Function
def receive():
stop_flag: bool = False
try:
logger.debug('Waiting for connections...')
while True:
try:
# Accept Connection
conn, address = server.accept()
logger.info(f"New connection from {address}")
client = handler.Client(conn, address, clients, lambda: stop_flag)
clients.append(client)
client.request_nickname()
# Inform all clients of new client, give new client connections list
if len(clients) > 0:
logger.debug('Informing all connected clients of incoming connection.')
for client in clients:
client.send_connections_list()
# Start Handling Thread For Client
thread = threading.Thread(target=client.handle, name=client.id[:8])
thread.start()
except socket.timeout:
pass
except KeyboardInterrupt as e:
raise e
except Exception as e:
logger.critical(e, exc_info=e)
break
except KeyboardInterrupt:
logger.info('User stopped server manually. Enabling stop flag.')
stop_flag = True
sys.exit()
if __name__ == '__main__':
receive()