websockets，一个不可思议的 Python 库！

更多资料获取

个人网站：ipengtao.com

---

大家好，今天为大家分享一个不可思议的 Python 库 - websockets。

Github地址：https://github.com/python-websockets/websockets

WebSocket是一种在现代Web开发中变得越来越重要的协议。它允许客户端和服务器之间建立持久的双向通信，使得实时应用程序（如在线聊天、实时游戏和数据监控）变得更加容易实现。本文将深入探讨如何使用Python的websockets库构建WebSocket应用程序，并提供详细的示例代码。

什么是WebSocket？

WebSocket是一种协议，它允许浏览器和服务器之间建立持久的双向通信通道。与传统的HTTP请求-响应模型不同，WebSocket允许服务器主动向客户端发送数据，而不需要客户端发起请求。这种实时通信的特性使得WebSocket在实时应用程序中非常有用。

WebSocket与HTTP的区别

WebSocket与HTTP的主要区别在于持久性和双向通信。HTTP通常是一次性的请求-响应模型，而WebSocket连接保持打开，允许服务器和客户端随时发送数据。以下是一个简单的比较：

• HTTP：客户端发送请求，服务器响应，然后连接关闭。
• WebSocket：客户端和服务器建立持久连接，可以随时相互发送数据。

安装和设置

安装websockets库

首先，确保已经安装了websockets库。可以使用以下命令进行安装：

pip install websockets

创建虚拟环境

为了隔离WebSocket项目的依赖，建议使用Python虚拟环境。以下是创建和激活虚拟环境的示例：

python -m venv websocket-env
source websocket-env/bin/activate  # 在Windows上使用`activate.bat`

创建WebSocket服务器

下面是一个简单的WebSocket服务器示例，它监听客户端连接并处理WebSocket消息。

import asyncio
import websockets

async def server(websocket, path):
    async for message in websocket:
        await websocket.send("Received: " + message)

start_server = websockets.serve(server, "localhost", 8765)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

这个示例创建了一个名为server的异步函数，它接受来自客户端的消息并将其回传。WebSocket服务器通过websockets.serve启动，监听在本地主机的8765端口上。

创建WebSocket客户端

下面是一个简单的WebSocket客户端示例，它连接到服务器并发送和接收消息。

import asyncio
import websockets

async def client():
    async with websockets.connect("ws://localhost:8765") as websocket:
        await websocket.send("Hello, Server!")
        response = await websocket.recv()
        print("Received:", response)

asyncio.get_event_loop().run_until_complete(client())

这个示例使用websockets.connect连接到WebSocket服务器，并发送一条消息，然后等待服务器的响应。

实时应用示例：实时聊天应用

可以创建一个简单的实时聊天应用示例，展示WebSocket在实际应用中的用途。

服务器端代码

import asyncio
import websockets

connected_clients = set()

async def handle_client(websocket, path):
    connected_clients.add(websocket)
    try:
        async for message in websocket:
            # 广播消息给所有客户端
            for client in connected_clients:
                await client.send(message)
    except Exception:
        pass
    finally:
        # 客户端断开连接时移除
        connected_clients.remove(websocket)

start_server = websockets.serve(handle_client, "localhost", 8765)
asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

客户端代码

import asyncio
import websockets

async def main():
    async with websockets.connect("ws://localhost:8765") as websocket:
        while True:
            message = input("Enter a message: ")
            await websocket.send(message)
            response = await websocket.recv()
            print("Received:", response)

asyncio.run(main())

这个示例创建了一个简单的聊天室，允许多个客户端实时聊天。

WebSocket安全性

WebSocket通信可以通过使用TLS/SSL来加密以确保安全性。以下是如何通过WSS（WebSocket Secure）来实现WebSocket安全性的示例代码：

import asyncio
import websockets
import ssl

ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
ssl_context.load_cert_chain("server.crt", "server.key")

async def server(websocket, path):
    async for message in websocket:
        await websocket.send("Received: " + message)

start_server = websockets.serve(server, "localhost", 8765, ssl=ssl_context)

asyncio.get_event_loop().run_until_complete(start_server)
asyncio.get_event_loop().run_forever()

这个示例使用TLS证书来加密WebSocket通信。

WebSocket扩展功能

WebSocket支持一些扩展功能，例如子协议、自定义消息类型和心跳机制。下面是一个示例代码，演示如何使用子协议来区分不同类型的WebSocket连接：

import asyncio
import websockets

async def server(websocket, path):
    if "chat" in websocket.request_headers.get("Sec-WebSocket-Protocol", ""):
        # 处理聊天连接
        async for message in websocket:
            await websocket.send("Received: " + message)
    elif "sensor" in websocket.request_headers.get("Sec-WebSocket-Protocol", ""):
        # 处理传感器数据连接
        async for message in websocket:
            process_sensor_data(message)

start_server = websockets.serve(server, "localhost", 8765)

asyncio.get_event_loop().run_until_complete(start

_server)
asyncio.get_event_loop().run_forever()

总结

当构建WebSocket应用程序时，Python提供了强大的工具和库，其中最知名的是websockets库。WebSocket协议允许实现实时双向通信，使其在多种应用场景中都有广泛的用途。

在本文中，深入研究了WebSocket的核心概念，与传统HTTP通信的区别，以及如何使用Python的websockets库创建WebSocket服务器和客户端。提供了详细的示例代码，演示了WebSocket的基本用法，包括实时聊天应用、实时游戏、数据可视化等。

此外，还讨论了WebSocket的安全性，如使用TLS/SSL来加密通信以保护数据的安全性。介绍了WebSocket的扩展功能，如子协议、自定义消息类型和心跳机制，以满足特定应用需求。还探讨了WebSocket的实际用途，包括实时聊天、协作工具、数据可视化和实时游戏等示例，以及WebSocket库的选择、生态系统、性能优化、标准和协议、最佳实践、安全性和未来展望等主题。

WebSocket是一种强大的通信协议，为实时应用程序提供了强大的工具。通过深入学习和实践，可以充分利用WebSocket的潜力，构建出各种实时应用程序，满足不同的需求。

---

Python学习路线

更多资料获取

个人网站：ipengtao.com

如果还想要领取更多更丰富的资料，可以点击文章下方名片，回复【优质资料】，即可获取 全方位学习资料包。

点击文章下方链接卡片，回复【优质资料】，可直接领取资料大礼包。