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"})