Upload via file

desktop/main.py

@@ -0,0 +1,129 @@
+import json
+import threading
+from socket import *
+
+is_login = False
+is_broadcast = True
+
+# 接受消息类
+class ClientReceiveThread(threading.Thread):
+    __buf = 1024
+    # 初始化
+    def __init__(self, cs):
+        super(ClientReceiveThread, self).__init__()
+        self.__cs = cs
+
+    def run(self):
+        self.receive_msg()
+
+    def receive_msg(self):
+        while True:
+            msg = self.__cs.recv(self.__buf).decode('utf-8')
+            if not msg:
+                break
+            js = json.loads(msg)
+            if js['type'] == "login":
+                if js['success']:
+                    global is_login
+                    is_login = True
+                print(js['msg'])
+            elif js['type'] == "ignore":
+                if js['success']:
+                    global is_broadcast
+                    if is_broadcast:
+                        is_broadcast = False
+                    else:
+                        is_broadcast = True
+                print(js['msg'])
+            else:
+                if not is_broadcast:
+                    print("[现在处于屏蔽模式]")
+                print(js['msg'])
+
+
+# 发送消息类
+class ClientSendMsgThread(threading.Thread):
+
+    def __init__(self, cs):
+        super(ClientSendMsgThread, self).__init__()
+        self.__cs = cs
+
+    def run(self):
+        self.send_msg()
+
+    # 根据不同的输入格式来进行不同的聊天方式
+    def send_msg(self):
+        while True:
+            js = None
+            msg = input()
+            if msg.strip():
+                if not is_login:
+                    js = json.dumps({
+                        'type': 'login',
+                        'msg': msg
+                    })
+                elif msg[0] == "@":
+                    data = msg.split(' ')
+                    if not data:
+                        print("请重新输入")
+                        break
+                    nickname = data[0]
+                    nickname = nickname.strip("@")
+                    if len(data) == 1:
+                        data.append(" ")
+                    js = json.dumps({
+                        'type': 'sendto',
+                        'nickname': nickname,
+                        'msg': data[1]
+                    })
+                elif msg == "/help":
+                    js = json.dumps({
+                        'type': 'help',
+                        'msg': None
+                    })
+                elif msg == "/checkol":
+                    js = json.dumps({
+                        'type': 'ls',
+                        'msg': None
+                    })
+                elif msg == "/i":
+                    js = json.dumps({
+                        'type': 'ignore',
+                        'msg': None
+                    })
+                elif msg == "/exit":  # 添加退出功能
+                    js = json.dumps({
+                        'type': 'exit',
+                        'msg': None
+                    })
+                    self.__cs.sendall(bytes(js, 'utf-8'))
+                    self.__cs.close()
+                    break
+                else:
+                    if msg[0] != '/':
+                        js = json.dumps({
+                            'type': 'broadcast',
+                            'msg': msg
+                        })
+                if js is not None:
+                    self.__cs.sendall(bytes(js, 'utf-8'))
+                    if msg == "/exit":  # 如果输入了退出命令，直接退出循环
+                        break
+
+
+def main():
+    buf = 1024
+    # 改变这个的地址,变成服务器的地址,那么只要部署到服务器上就可以全网使用了
+    address = (input("请输入服务器 IP:"), 12231)
+    cs = socket(AF_INET, SOCK_STREAM, 0)
+    cs.connect(address)
+    data = cs.recv(buf).decode("utf-8")
+    if data:
+        print(data)
+    receive_thread = ClientReceiveThread(cs)
+    receive_thread.start()
+    send_thread = ClientSendMsgThread(cs)
+    send_thread.start()
+
+
+main()