Python WebSocket 客户端|极客教程

Python WebSocket 客户端

在网络通信中，WebSocket 是一种在单个 TCP 连接上进行全双工通信的协议。通过WebSocket，客户端和服务器之间可以实现实时的双向通信，而不需要像传统的HTTP协议那样每次都需要发起新的请求。

在Python中，我们可以使用第三方库来快速构建WebSocket客户端。下面我们将详细介绍如何在Python中实现WebSocket客户端，并提供多个示例代码来演示其用法。

安装依赖库

首先，我们需要安装websockets库来帮助我们实现WebSocket客户端。你可以通过以下命令来安装：

pip install websockets

连接WebSocket服务器

接下来，我们将演示如何连接到WebSocket服务器，并向其发送消息。首先，创建一个Python脚本，命名为websocket_client.py，然后添加以下代码：

import asyncio
import websockets

async def send_message():
    async with websockets.connect('ws://127.0.0.1:8000') as websocket:
        message = "Hello, from WebSocket client!"
        await websocket.send(message)
        print(f"> Sent: {message}")

asyncio.get_event_loop().run_until_complete(send_message())

在上面的示例代码中，我们使用websockets.connect()函数来连接到WebSocket服务器的地址ws://127.0.0.1:8000，然后发送消息并打印发送的消息。

运行上面的代码，你将看到输出如下：

> Sent: Hello, from WebSocket client!

接收服务器消息

除了向WebSocket服务器发送消息，我们还可以接收来自服务器的消息。下面的示例代码将演示如何执行这一操作：

import asyncio
import websockets

async def receive_message():
    async with websockets.connect('ws://127.0.0.1:8000') as websocket:
        message = await websocket.recv()
        print(f"< Received: {message}")

asyncio.get_event_loop().run_until_complete(receive_message())

运行上面的代码，你将看到输出如下：

< Received: Hello, from WebSocket server!

实现简单的聊天室

最后，我们将结合上述的发送和接收消息的功能，实现一个简单的WebSocket聊天室。下面的示例代码展示了如何实现：

import asyncio
import websockets

async def send_message():
    async with websockets.connect('ws://127.0.0.1:8000') as websocket:
        while True:
            message = input("Enter message: ")
            await websocket.send(message)
            print(f"> Sent: {message}")

async def receive_message():
    async with websockets.connect('ws://127.0.0.1:8000') as websocket:
        while True:
            message = await websocket.recv()
            print(f"< Received: {message}")

asyncio.get_event_loop().run_until_complete(asyncio.gather(send_message(), receive_message()))

在上面的代码中，我们同时创建了两个异步任务，一个用于发送消息，一个用于接收消息。在该示例中，我们可以开启两个终端分别运行客户端程序，实现双方实时的聊天对话。