mirror of
https://github.com/Xevion/tcp-chat.git
synced 2025-12-06 11:16:39 -06:00
add custom logging with clients marked via thread name
This commit is contained in:
34
oserver.py
34
oserver.py
@@ -5,6 +5,7 @@ import threading
|
|||||||
import time
|
import time
|
||||||
import traceback
|
import traceback
|
||||||
import uuid
|
import uuid
|
||||||
|
import logging
|
||||||
from pprint import pprint
|
from pprint import pprint
|
||||||
|
|
||||||
import constants
|
import constants
|
||||||
@@ -20,6 +21,10 @@ server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|||||||
server.bind((host, port))
|
server.bind((host, port))
|
||||||
server.listen()
|
server.listen()
|
||||||
|
|
||||||
|
logging.basicConfig(level=logging.DEBUG,
|
||||||
|
format='[%(asctime)s] [%(name)s] [%(threadName)s] %(message)s')
|
||||||
|
logger = logging.getLogger('server')
|
||||||
|
|
||||||
# Lists For Clients and Their Nicknames
|
# Lists For Clients and Their Nicknames
|
||||||
clients = {}
|
clients = {}
|
||||||
|
|
||||||
@@ -31,6 +36,7 @@ def broadcast_data(data):
|
|||||||
|
|
||||||
# Handling Messages From Clients
|
# Handling Messages From Clients
|
||||||
def handle(client_id):
|
def handle(client_id):
|
||||||
|
logger.info(f'Beginning handling of {client_id}')
|
||||||
client = clients[client_id]['client']
|
client = clients[client_id]['client']
|
||||||
nickname = clients[client_id]['nickname']
|
nickname = clients[client_id]['nickname']
|
||||||
|
|
||||||
@@ -38,7 +44,9 @@ def handle(client_id):
|
|||||||
try:
|
try:
|
||||||
# Broadcasting Messages
|
# Broadcasting Messages
|
||||||
length = int(client.recv(HEADER_LENGTH).decode('utf-8'))
|
length = int(client.recv(HEADER_LENGTH).decode('utf-8'))
|
||||||
|
logger.debug(f'Header received - Length {length}')
|
||||||
message = json.loads(client.recv(length).decode('utf-8'))
|
message = json.loads(client.recv(length).decode('utf-8'))
|
||||||
|
logger.info(f'Data received/parsed, type: {message["type"]}')
|
||||||
|
|
||||||
if message['type'] == constants.Types.REQUEST:
|
if message['type'] == constants.Types.REQUEST:
|
||||||
if message['request'] == constants.Requests.REFRESH_CONNECTIONS_LIST:
|
if message['request'] == constants.Requests.REFRESH_CONNECTIONS_LIST:
|
||||||
@@ -56,7 +64,7 @@ def handle(client_id):
|
|||||||
elif message['type'] == constants.Types.NICKNAME:
|
elif message['type'] == constants.Types.NICKNAME:
|
||||||
nickname = message['nickname']
|
nickname = message['nickname']
|
||||||
if not clients[client_id]['has_nickname']:
|
if not clients[client_id]['has_nickname']:
|
||||||
print("Nickname is {}".format(nickname))
|
logger.info("Nickname is {}".format(nickname))
|
||||||
broadcast_data(helpers.prepare_server_message(
|
broadcast_data(helpers.prepare_server_message(
|
||||||
nickname='Server',
|
nickname='Server',
|
||||||
message=f'{nickname} joined!',
|
message=f'{nickname} joined!',
|
||||||
@@ -64,19 +72,13 @@ def handle(client_id):
|
|||||||
))
|
))
|
||||||
clients[client_id]['has_nickname'] = True
|
clients[client_id]['has_nickname'] = True
|
||||||
else:
|
else:
|
||||||
print(f'{clients[client_id]["nickname"]} changed their name to {nickname}')
|
logger.info(f'{clients[client_id]["nickname"]} changed their name to {nickname}')
|
||||||
clients[client_id]['nickname'] = nickname
|
clients[client_id]['nickname'] = nickname
|
||||||
|
|
||||||
elif message['type'] == constants.Types.MESSAGE:
|
elif message['type'] == constants.Types.MESSAGE:
|
||||||
# Echo message back to all other users
|
broadcast_data(helpers.prepare_server_message(
|
||||||
broadcast_data(helpers.prepare_json(
|
nickname=nickname,
|
||||||
{
|
message=message['content'],
|
||||||
'type': constants.Types.MESSAGE,
|
color=clients[client_id]['color']
|
||||||
'nickname': nickname,
|
|
||||||
'content': message["content"],
|
|
||||||
'color': clients[client_id]['color'],
|
|
||||||
'time': int(time.time())
|
|
||||||
}
|
|
||||||
))
|
))
|
||||||
|
|
||||||
# Basic command processing
|
# Basic command processing
|
||||||
@@ -91,7 +93,7 @@ def handle(client_id):
|
|||||||
))
|
))
|
||||||
except:
|
except:
|
||||||
traceback.print_exc()
|
traceback.print_exc()
|
||||||
print(f'Closing Client {clients[client_id]["nickname"]}')
|
logger.info(f'Client {client_id} closed. ({clients[client_id]["nickname"]})')
|
||||||
client.close()
|
client.close()
|
||||||
del clients[client_id]
|
del clients[client_id]
|
||||||
broadcast_data(helpers.prepare_server_message(
|
broadcast_data(helpers.prepare_server_message(
|
||||||
@@ -107,7 +109,7 @@ def receive():
|
|||||||
while True:
|
while True:
|
||||||
# Accept Connection
|
# Accept Connection
|
||||||
client, address = server.accept()
|
client, address = server.accept()
|
||||||
print("New Client from {}".format(str(address)))
|
logger.info("New Client from {}".format(str(address)))
|
||||||
|
|
||||||
# Request And Store Nickname
|
# Request And Store Nickname
|
||||||
client_id = str(uuid.uuid4())
|
client_id = str(uuid.uuid4())
|
||||||
@@ -126,7 +128,7 @@ def receive():
|
|||||||
'has_nickname': False,
|
'has_nickname': False,
|
||||||
'color': random.choice(constants.Colors.ALL)
|
'color': random.choice(constants.Colors.ALL)
|
||||||
}
|
}
|
||||||
pprint(clients[client_id])
|
# pprint(clients[client_id])
|
||||||
|
|
||||||
client.send(helpers.prepare_json(
|
client.send(helpers.prepare_json(
|
||||||
{
|
{
|
||||||
@@ -136,7 +138,7 @@ def receive():
|
|||||||
))
|
))
|
||||||
|
|
||||||
# Start Handling Thread For Client
|
# Start Handling Thread For Client
|
||||||
thread = threading.Thread(target=handle, args=(client_id,))
|
thread = threading.Thread(target=handle, args=(client_id,), name=client_id[:8])
|
||||||
thread.start()
|
thread.start()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user