用python的websocket库封装一个websocket服务端的class

以下是一个简单的Python WebSocket服务器类的封装：

import asyncio
import websockets

class WebSocketServer:
    def __init__(self, host, port):
        self.host = host
        self.port = port
        self.server = None
        self.clients = set()

    async def handler(self, websocket, path):
        self.clients.add(websocket)
        try:
            async for message in websocket:
                await self.broadcast(message)
        finally:
            self.clients.remove(websocket)

    async def broadcast(self, message):
        if self.clients:
            await asyncio.wait([client.send(message) for client in self.clients])

    async def start(self):
        self.server = await websockets.serve(self.handler, self.host, self.port)
        await self.server.wait_closed()

    async def stop(self):
        if self.server:
            self.server.close()
            await self.server.wait_closed()

    def run(self):
        asyncio.run(self.start())

if __name__ == '__main__':
    server = WebSocketServer('localhost', 8765)
    server.run()

这个类使用了Python的asyncio库和websockets库。它包含一个构造函数，一个处理程序，一个广播函数和三个方法：启动、停止和运行。

启动方法使用asyncio库中的run函数来运行一个协程，该协程启动WebSocket服务器并等待它关闭。停止方法关闭WebSocket服务器。运行方法使用asyncio库中的run函数来运行启动方法。

此类通过handler函数处理WebSocket连接。当客户端连接到服务器时，它将添加到客户端集合中。当客户端发送消息时，它将被广播到所有连接的客户端。当客户端断开连接时，它将从客户端集合中移除。

注意：这个类只是一个简单的例子，不适用于生产环境。在实际应用中，您需要添加更多的错误处理和安全性检查。