first commit

client1.py

@@ -0,0 +1,59 @@
+import json
+import socket
+
+
+def chat(username, chat_obj, content):
+    '''
+    发送聊天信息
+    '''
+    pass
+
+
+# 创建socket
+ip_port = ("127.0.0.1", 9000)
+sk = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+
+# 选择菜单
+while 1:
+    print("menu".center(30, "="))
+    print("1.login".center(30, " "))
+    print("2.register".center(30, " "))
+    print("=".center(30, "="))
+    choice = input(">>>")
+
+    if choice == "1":
+        username = input("用户名：")
+        password = input("密码：")
+        data = {
+            "option": "login",
+            "username": username,
+            "password": password
+        }
+        sk.sendto(json.dumps(data).encode("utf-8"), ip_port)
+        msg, addr = sk.recvfrom(1024)
+        # {"status": 1}
+        if json.loads(msg.decode("utf-8"))["status"]:
+            chat_obj = input("输入收件人：")
+            content = input("输入内容：")
+            chat(username, chat_obj, content)
+            continue
+        else:
+            print("登录失败")
+            continue
+    elif choice == "2":
+        username = input("用户名：")
+        password = input("密码：")
+        data = {
+            "option": "register",
+            "username": username,
+            "password": password
+        }
+        sk.sendto(json.dumps(data).encode("utf-8"), ip_port)
+        msg, addr = sk.recvfrom(1024)
+        # {"status": 1}
+        if json.loads(msg.decode("utf-8"))["status"]:
+            print("注册成功")
+            continue
+        else:
+            print(json.loads(msg.decode("utf-8"))["msg"])
+            continue