From 682b2e3c8ba66eb06b5a07ec9fa2b914fbebb35e Mon Sep 17 00:00:00 2001
From: likai1995 <45533391+likai1995@users.noreply.github.com>
Date: Wed, 24 Jun 2020 20:36:48 +0800
Subject: [PATCH] add python_udp.py
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

基于socket和udp协议实现简单的聊天室系统（多人群聊）
功能介绍：
1.进入聊天室时，判断用户名是否重复
2.某个成员进入聊天室后给其他所有所有成员发送消息
3.任何一个人发消息，其他人都会收到
4.某人退出聊天室，其他成员也会收到通知
5.服务器记录当前聊天室的人数
---
 python_udp.py | 100 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 python_udp.py

diff --git a/python_udp.py b/python_udp.py
new file mode 100644
index 0000000..636e397
--- /dev/null
+++ b/python_udp.py
@@ -0,0 +1,100 @@
+import socket
+import threading
+
+"""服务器"""
+def main():
+    #建立字节流套接字
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    #设置端口复用
+    s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
+    addr = ('127.0.0.1', 9999)
+    #套接字绑定地址，端口号
+    s.bind(addr)
+    print("----------快乐聊天室已开启----------")
+    #地址：用户名
+    user = {}
+    while True:
+        try:
+            data, addr = s.recvfrom(1024)
+
+            #新用户
+            if not addr in user:
+                #判断是否重名
+                if data.decode('utf-8') in user.values():
+                    s.sendto('用户名已存在!'.encode(), addr)
+                    continue
+                else:
+                    s.sendto('OK'.encode(), addr)
+                #给其他用户发送欢迎信息
+                for address in user:
+                    s.sendto(data + ' 进入聊天室...'.encode(), address)
+                user[addr] = data.decode('utf-8')
+                print('聊天室当前人数：%s' % len(user))
+                continue
+
+            #退出聊天室
+            if 'EXIT' in data.decode('utf-8'):
+                name = user[addr]
+                user.pop(addr)
+                for address in user:
+                    s.sendto((name + ' 离开了聊天室...').encode(), address)
+                print('聊天室当前人数：%s' % len(user))
+            else:
+                print('"%s" from %s:%s' %
+                      (data.decode('utf-8'), addr[0], addr[1]))
+                for address in user:
+                    if address != addr:
+                        s.sendto(data, address)
+
+        except ConnectionResetError:
+            raise Exception('Someone left unexcept')
+
+if __name__ == '__main__':
+    main()
+
+
+"""客户端"""
+def recv(sock, addr):
+    while True:
+        data = sock.recv(1024)
+        print(data.decode('utf-8'))
+
+
+def send(sock, name, addr):
+    while True:
+        string = input()
+        message = name + ' : ' + string
+        data = message.encode('utf-8')
+        sock.sendto(data, addr)
+        if string == 'EXIT':
+            break
+
+def main():
+    #创建udp套接字
+    s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+    server = ('127.0.0.1', 9999)
+    print("-----欢迎来到聊天室,退出聊天室请输入'EXIT'-----")
+    name = input('请输入你的名称:')
+    #判断名字是否重复
+    while True:
+        s.sendto(name.encode('utf-8'), server)
+        data = s.recv(1024)
+        if data.decode('utf-8') == '用户名已存在!':
+            print('用户名已存在，请重新输入!')
+            name = input('请输入你的名称:')
+            continue
+        elif data.decode('utf-8') == 'OK':
+            break
+    print('-----------------%s------------------' % name)
+    #建立两个子进程，分别用于接收数据和发送数据，其中接受线程是守护线程
+    pr = threading.Thread(target=recv, args=(s, server), daemon=True)
+    ps = threading.Thread(target=send, args=(s, name, server))
+    #子进程运行
+    pr.start()
+    ps.start()
+    #阻塞主进程，等待发送线程完成
+    ps.join()
+    s.close()
+
+if __name__ == '__main__':
+    main()