41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
import json
|
|
import socket
|
|
|
|
# 创建socket
|
|
sk = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sk.bind(("0.0.0.0", 9000))
|
|
active = []
|
|
db = {}
|
|
chat_msg = {}
|
|
|
|
while 1:
|
|
print(f"当前活跃用户：{active} 当前注册用户：{db}")
|
|
data, addr = sk.recvfrom(2048)
|
|
data = json.loads(data)
|
|
res = {"status": 0}
|
|
if data["option"] == "login":
|
|
username = data["username"]
|
|
password = data["password"]
|
|
if username in db:
|
|
if db[username] == password:
|
|
res = {"status": 1}
|
|
active.append(data["username"])
|
|
else:
|
|
res = {"status": 0}
|
|
else:
|
|
res = {"status": 0}
|
|
elif data["option"] == "register":
|
|
username = data["username"]
|
|
password = data["password"]
|
|
if username in db:
|
|
res = {"status": 0, "msg": "用户名已存在"}
|
|
else:
|
|
db[username] = password
|
|
res = {"status": 1, "msg": "注册成功"}
|
|
elif data["option"] == "chat":
|
|
# 接受聊天信息，并且存入聊天临时字典
|
|
pass
|
|
elif data["option"] == "hello":
|
|
# 接受客户端的存活状态，如果发现临时聊天字典有该用户的未收消息，发送过去
|
|
pass
|
|
sk.sendto(json.dumps(res).encode("utf-8"), addr) |