网络聊天室2000

import asyncio, websockets, json, ssl, os
asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) #注册为Windows服务时需要用

class Chatroom:
    def __init__(self):
        self.USERS, self.history = {}, []
        
    def start(self):
        print('wss://www.shishuyang.cn:23457 websocket...')
        ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
        ssl_context.load_cert_chain('xxx.pem', 'xxx.key')
        asyncio.get_event_loop().run_until_complete(websockets.serve(self.run, '0.0.0.0', 23457, ssl=ssl_context))
        asyncio.get_event_loop().run_forever()

    async def recv_user_msg(self, ws):    
        while True: 
            recv_text = await ws.recv()
            print(self.USERS[ws], recv_text)
            message = json.dumps({'state':1, 'msg':self.USERS[ws]+'：'+recv_text})
            if len(self.history)>15: self.history.pop(0)
            self.history.append((self.USERS[ws], recv_text))
            await asyncio.gather(*[user.send(message) for user in self.USERS.keys()])

    async def register(self, ws):
        recv_text = await ws.recv()
        print('加入', recv_text, list(self.USERS.values()))
        self.USERS[ws] = recv_text    
        s = '\n'.join([i[0]+'：'+i[1] for i in self.history])
        await ws.send(json.dumps({'state':1, 'msg':s}))
        message = json.dumps({'state':1, 'msg':'【'+self.USERS[ws]+'】进入了聊天室！'})
        await asyncio.gather(*[user.send(message) for user in self.USERS.keys()])    

    async def unregister(self, ws):
        message = json.dumps({'state':1, 'msg':'【'+self.USERS[ws]+'】离开了聊天室！'})
        print('离开', self.USERS.pop(ws), list(self.USERS.values()))
        if self.USERS: await asyncio.gather(*[user.send(message) for user in self.USERS.keys()])

    async def run(self, websocket, path):
        await self.register(websocket)
        while True:
            try:
                await self.recv_user_msg(websocket)
            except websockets.ConnectionClosed:
                await self.unregister(websocket)
                break
            except websockets.InvalidState:
                print("InvalidState...") 
                break
            except Exception as e:
                print("Exception:", e) 
    
if __name__ == '__main__':
    Chatroom().start()