D_7/API
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

增加返回时的type

Browse Source
This commit is contained in:
DZY 2025-06-02 15:36:14 +08:00
parent 2355c3845d
commit 4b358aa58e
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
CS3.1.py
View File

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