使用Python实现基于WebSocket的实时通信的函数

WebSocket是一种在Web浏览器和服务器之间进行全双工通信的协议。它使得浏览器和服务器可以通过一个长连接进行实时的双向通信，而无需使用传统的HTTP请求和响应。在Python中，我们可以使用websockets库来实现基于WebSocket的实时通信。

首先，我们需要安装websockets库。可以通过运行以下命令来安装：

pip install websockets

接下来，我们可以编写一个简单的函数来实现WebSocket的实时通信。下面是一个基本的示例：

import asyncio
import websockets

# 实时通信的处理逻辑
async def handle_message(websocket, path):
    async for message in websocket:
        print("收到消息：", message)

        # 发送回复消息
        reply_message = "已收到消息：" + message
        await websocket.send(reply_message)

# 启动WebSocket的服务器
def start_server():
    start_server = websockets.serve(handle_message, "localhost", 8000)

    # 启动事件循环
    asyncio.get_event_loop().run_until_complete(start_server)
    asyncio.get_event_loop().run_forever()

# 调用启动函数
start_server()

在以上示例中，我们定义了一个名为handle_message的协程函数，用于处理实时通信的逻辑。每当收到客户端发送的消息时，它会将消息打印出来，并回复一个消息。然后，我们使用websockets.serve函数启动了一个WebSocket服务器，并指定了handle_message函数作为处理逻辑。最后，我们使用asyncio库中的事件循环来运行服务器。

下面是一个客户端使用WebSocket与服务器进行实时通信的示例代码：

import asyncio
import websockets

# 连接服务器并发送消息
async def send_message():
    async with websockets.connect("ws://localhost:8000") as websocket:
        while True:
            message = input("请输入要发送的消息：")
            await websocket.send(message)

            # 接收服务器的回复消息
            reply = await websocket.recv()
            print("收到回复：", reply)

# 调用发送消息的函数
asyncio.get_event_loop().run_until_complete(send_message())

在以上示例中，我们首先使用websockets.connect函数连接到服务器。然后，通过一个循环不断等待用户输入消息，并使用websocket.send发送消息到服务器。接下来，我们使用websocket.recv接收服务器的回复消息，并将其打印出来。

可以通过运行上述客户端的代码来测试WebSocket的实时通信功能。在运行客户端时，它会要求你输入要发送的消息。每次输入完消息后，它都会将输入的消息发送到服务器，并接收服务器的回复消息。

以上就是使用Python实现基于WebSocket的实时通信的函数及其使用例子。你可以使用这种方法，实现Web应用中的实时通信功能，例如实时聊天、实时数据更新等。