def login(acc, pwd):
pwd = hashlib.new('sha256', pwd.encode('UTF-8')).hexdigest()
UDP_SENIOR_IO.send_json(ADDR, {
"mode": "login",
"account": acc,
"pwd": pwd
})
def send_msg(addr: Tuple[str, int], msg: str, from_name: str, private: bool):
UDP_SENIOR_IO.send_json(addr, {
"mode": "getMsg",
"from": from_name,
"msg": msg,
"private": private
})
def sign_up(acc, pwd, name):
pwd = hashlib.new('sha256', pwd.encode('UTF-8')).hexdigest()
UDP_SENIOR_IO.send_json(ADDR, {
"mode": "signUp",
"account": acc,
"pwd": pwd,
"name": name
})
def reply_login(addr, args):
suc = args["success"]
if suc:
cp.blue_paint("登录成功")
cp.blue_paint(args["name"] + " 欢迎回来")
SETTING.set["cookie"] = args["cookie"]
UDP_SENIOR_IO.set_cookie(args["cookie"])
else:
cp.red_paint("登录失败")
def sign_up(addr, args):
suc = ACCOUNT.new_user(args["account"], args["pwd"], args["name"])
if suc:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replySignUp",
"type": True,
"hint": ""
})
else:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replySignUp",
"type": False,
"hint": "账号名或用户名重复"
})
def login(addr, args):
name, hash_d = ACCOUNT.get_user(args["account"])
if hash_d == args["pwd"]:
o: OnlineList = CACHE.get("onlineList")
cookie = o.add_user(name, addr)
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyLogin",
"success": True,
"cookie": cookie,
"name": name
})
else:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyLogin",
"success": False,
"cookie": "",
"name": ""
})
def logout(addr, args):
if "cookie" not in args:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyList",
"status": True,
"list": None
})
return
cookie = args["cookie"]
flag, name = verify_cookie(cookie)
if flag:
UDP_SENIOR_IO.send_json(addr, {"mode": "replyLogout", "type": True})
o: OnlineList = CACHE.get("onlineList")
o.remove_user(cookie)
else:
UDP_SENIOR_IO.send_json(addr, {"mode": "replyLogout", "type": False})
def msg(addr, args):
if "cookie" not in args:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyList",
"status": True,
"list": None
})
return
to, msg, cookie = args["to"], args["msg"], args["cookie"]
flag, name = verify_cookie(cookie)
if not flag:
UDP_SENIOR_IO.send_json(addr, {"mode": "replyMsg", "type": False})
else:
UDP_SENIOR_IO.send_json(addr, {"mode": "replyMsg", "type": True})
if to == "":
o: OnlineList = CACHE.get("onlineList")
for user in o.users:
send_msg(o.addrs[user], msg, name, False)
else:
o: OnlineList = CACHE.get("onlineList")
send_msg(o.addrs[to], msg, name, True)
def get_list(addr, args):
if "cookie" not in args:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyList",
"status": False,
"list": None
})
return
cookie = args["cookie"]
flag, name = verify_cookie(cookie)
o: OnlineList = CACHE.get("onlineList")
if flag:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyList",
"status": True,
"list": list(o.users)
})
else:
UDP_SENIOR_IO.send_json(addr, {
"mode": "replyList",
"status": False,
"list": None
})
def logout():
UDP_SENIOR_IO.send_json(ADDR, {
"mode": "logout",
})
UDP_SENIOR_IO.remove_cookie()
def msg(
msg,
to="",
):
UDP_SENIOR_IO.send_json(ADDR, {"mode": "msg", "msg": msg, "to": to})
def list_():
UDP_SENIOR_IO.send_json(ADDR, {
"mode": "getList",
})
import sys
sys.path.append('../')
from net_tool.SeniorNetIo import UDP_SENIOR_IO
addr = input("addr")
port = 23333
UDP_SENIOR_IO.set_cookie("test")
UDP_SENIOR_IO.send_json((addr, port), {"mode": "test", "aaa": "bbb"})