python使用websocket简单组建聊天室

server端

 ###websocket_server###
 import socket
 import threading

 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 sock.bind(('localhost', 900))

 sock.listen(5)
 print('Server', socket.gethostbyname('localhost'), 'listening ...')

 mydict = dict()
 mylist = list()

 # 把whatToSay传给除了exceptNum的所有人
 def tellOthers(exceptNum, whatToSay):
     for c in mylist:
         if c.fileno() != exceptNum:
             try:
                 c.send(whatToSay.encode())
             except:
                 pass

 def subThreadIn(myconnection, connNumber):
     nickname = myconnection.recv(1024).decode()
     mydict[myconnection.fileno()] = nickname
     mylist.append(myconnection)
     print('connection', connNumber, ' has nickname :', nickname)
     tellOthers(connNumber, '【系统提示：' + mydict[connNumber] + ' 进入聊天室】')
     while True:
         try:
             recvedMsg = myconnection.recv(1024).decode()
             if recvedMsg:
                 print(mydict[connNumber], ':', recvedMsg)
                 tellOthers(connNumber, mydict[connNumber] + ' :' + recvedMsg)

         except (OSError, ConnectionResetError):
             try:
                 mylist.remove(myconnection)
             except:
                 pass
             print(mydict[connNumber], 'exit, ', len(mylist), ' person left')
             tellOthers(connNumber, '【系统提示：' + mydict[connNumber] + ' 离开聊天室】')
             myconnection.close()
             return

 while True:
     connection, addr = sock.accept()
     print('Accept a new connection', connection.getsockname(), connection.fileno())
     try:
         # connection.settimeout(5)
         buf = connection.recv(1024).decode()
         if buf == '':
             connection.send(b'welcome to server!')

             # 为当前连接开辟一个新的线程
             mythread = threading.Thread(target=subThreadIn, args=(connection, connection.fileno()))
             mythread.setDaemon(True)
             mythread.start()

         else:
             connection.send(b'please go out!')
             connection.close()
     except:
         pass

client端

 #####socket_client####
 import socket
 import time
 import threading

 sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

 sock.connect(('localhost', 900))
 sock.send(b'')
 print(sock.recv(1024).decode())
 nickName = input('input your nickname: ')
 sock.send(nickName.encode())

 def sendThreadFunc():
     while True:
         try:
             myword = input()
             sock.send(myword.encode())
             # print(sock.recv(1024).decode())
         except ConnectionAbortedError:
             print('Server closed this connection!')
         except ConnectionResetError:
             print('Server is closed!')

 def recvThreadFunc():
     while True:
         try:
             otherword = sock.recv(1024)
             if otherword:
                 print(otherword.decode())
             else:
                 pass
         except ConnectionAbortedError:
             print('Server closed this connection!')

         except ConnectionResetError:
             print('Server is closed!')

 th1 = threading.Thread(target=sendThreadFunc)
 th2 = threading.Thread(target=recvThreadFunc)
 threads = [th1, th2]

 for t in threads:
     t.setDaemon(True)
     t.start()
 t.join()

测试结果