python websockets封装成类的形式例子

以下是一个简单的例子，展示如何将 Python 的 websockets 封装成一个类：

import asyncio
import websockets

class WebSocketClient:
    def __init__(self, URI):
        self.URI = URI
        self.websocket = None

    async def connect(self):
        self.websocket = await websockets.connect(self.URI)

    async def send(self, message):
        await self.websocket.send(message)

    async def receive(self):
        message = await self.websocket.recv()
        return message

    async def close(self):
        await self.websocket.close()

async def example():
    ws = WebSocketClient('wss://echo.websocket.org')
    await ws.connect()
    await ws.send('Hello, world!')
    response = await ws.receive()
    print(response)
    await ws.close()

asyncio.get_event_loop().run_until_complete(example())

在这个例子中，我们创建了一个 WebSocketClient 类，有连接、发送、接收和关闭方法，分别对应了与 WebSocket 服务器建立连接、发送消息、接收消息和关闭连接的操作。在 `example()` 函数中，我们创建了一个 WebSocketClient 实例，连接到了 `wss://echo.websocket.org` 这个 WebSocket 服务器，并向其发送了一条消息。然后我们等待从服务器收到的消息，并将其打印出来，最后关闭了连接。

请注意，在这个例子中采用了异步编程方式，因此我们使用了 Python 内置的 asyncio 模块。我们需要调用 `asyncio.get_event_loop().run_until_complete(example())` 来运行 `example()` 函数，从而启动整个异步事件循环。