增加返回时的type

This commit is contained in:
DZY 2025-06-02 15:36:14 +08:00
parent 2355c3845d
commit 4b358aa58e

View File

@ -83,14 +83,14 @@ def handle_socket_message(data, addr, conn):
action = data.get('type')
if action == 'register':
result = register_user(data.get('username'), data.get('password'))
return {"status": "success" if result['success'] else "error", "message": result['message']}
return {"type": "register","status": "success" if result['success'] else "error", "message": result['message']}
elif action == 'login':
if isuserxist(data['username']):
if ispsswdright(data['username'], data['password']):
active_users[addr[0]] = data['username']
chat_connections.append(conn)
return {"status": "success", "message": "Login successful"}
return {"status": "error", "message": "Invalid credentials"}
return {"type": "login", "status": "success", "message": "Login successful"}
return {"type": "login", "status": "error", "message": "Invalid credentials"}
elif action == 'chat':
if addr[0] in active_users:
message = {
@ -99,8 +99,8 @@ def handle_socket_message(data, addr, conn):
"message": data['message']
}
broadcast_message(message)
return {"status": "success"}
return {"status": "error", "message": "Not logged in"}
return {"type": "chat", "status": "success", }
return {"type": "chat", "status": "error", "message": "Not logged in"}
except Exception as e:
return {"status": "error", "message": str(e)}