增加返回时的type
This commit is contained in:
parent
2355c3845d
commit
4b358aa58e
10
CS3.1.py
10
CS3.1.py
@ -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)}
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user