Flask-SocketIO，一个高效的 Python WebSocket 库！

最新推荐文章于 2025-04-15 19:26:12 发布

原创最新推荐文章于 2025-04-15 19:26:12 发布

· 657 阅读

14 ·

版权

文章标签：

#python #flask #websocket

面试同时被 3 个专栏收录

53 篇文章

订阅专栏

学习路线

53 篇文章

订阅专栏

阿里巴巴

28 篇文章

订阅专栏

大家好，我是“一行”。今天想带大家了解一个非常强大的工具：Flask-SocketIO。在现代的Web应用中，我们经常需要实时交互功能，比如即时聊天、实时通知等。Flask-SocketIO正是一个帮助我们轻松实现WebSocket实时通讯的库。它简单好用、功能强大，非常适合初学者上手。如果你已经熟悉Flask，那今天的内容将会让你的应用更上一层楼！

什么是 Flask-SocketIO？

Flask-SocketIO是一个基于Flask框架的扩展库，专门用于实现实时通讯。传统的HTTP协议是“请求-响应”模式，服务器只能在客户端请求后进行响应，而WebSocket可以让服务器主动推送数据给客户端，实现双向实时通信。

小贴士： Flask-SocketIO不仅支持WebSocket，还支持长轮询等多种通信方式，它会自动选择最佳方式，不用我们手动配置。

安装 Flask-SocketIO

首先，我们需要在项目中安装 Flask-SocketIO。在命令行中运行以下命令：

pip?install?flask-socketio

安装好后，就可以开始创建一个小项目，来体验实时通信的乐趣了！

创建第一个 Flask-SocketIO 应用

基础设置

我们从一个简单的应用开始，让服务器接收到消息后，能立即将消息广播给所有客户端。

from?flask?import?Flask,?render_template
from?flask_socketio?import?SocketIO,?send

app?=?Flask(__name__)
app.config['SECRET_KEY']?=?'mysecret'
socketio?=?SocketIO(app)

@socketio.on('message')
def?handle_message(msg):
????print('Received?message:?'?+?msg)
????send(msg,?broadcast=True)

if?__name__?==?'__main__':
????socketio.run(app)

在这个简单的代码示例中，我们：

创建了 Flask 应用：app = Flask(__name__)
初始化了 SocketIO 实例：socketio = SocketIO(app)
处理消息事件：当收到消息时，将消息广播给所有客户端。

代码解释：

@socketio.on('message')：监听事件类型message，当客户端发送消息时触发。
send(msg, broadcast=True)：发送消息给所有连接的客户端。

创建前端页面

我们还需要一个简单的前端页面来发送消息。我们使用HTML和JavaScript进行简单实现。

文件名：templates/chat.html

<!DOCTYPE?html>
<html?lang="en">
<head>
????<meta?charset="UTF-8">
????<title>Flask-SocketIO?Chat</title>
????<script?src="https://cdn.socket.io/4.0.0/socket.io.min.js"></script>
</head>
<body>
????<h1>聊天室</h1>
????<input?type="text"?id="messageInput"?placeholder="输入消息...">
????<button?onclick="sendMessage()">发送</button>
????<ul?id="messages"></ul>

????<script>
????????var?socket?=?io.connect('http://'?+?document.domain?+?':'?+?location.port);

????????socket.on('message',?function(msg)?{
????????????var?li?=?document.createElement("li");
????????????li.appendChild(document.createTextNode(msg));
????????????document.getElementById("messages").appendChild(li);
????????});

????????function?sendMessage()?{
????????????var?msg?=?document.getElementById("messageInput").value;
????????????socket.send(msg);
????????????document.getElementById("messageInput").value?=?'';
????????}
????</script>
</body>
</html>

在这个前端页面中，我们实现了：

连接服务器的WebSocket（通过Socket.IO客户端）。
监听message事件，并将收到的消息显示在页面上。
定义一个sendMessage函数，将输入框中的消息发送到服务器。

小贴士： io.connect()会自动连接到当前网页所在的服务器，无需指定完整URL，这在部署应用时非常方便。

运行应用

确保前端文件和后端文件在同一个项目目录下，然后在命令行运行以下命令启动应用：

python?app.py

访问http://127.0.0.1:5000，你应该会看到一个简单的聊天室界面。打开多个浏览器窗口，输入消息，观察消息会实时广播到其他窗口。

Flask-SocketIO 的高级功能

1. 命名空间（Namespace）

在大型应用中，我们可能需要多个命名空间来分割不同的功能。例如，一个命名空间用于聊天，另一个用于通知。我们可以使用命名空间来实现这种划分：

from?flask_socketio?import?emit

@socketio.on('message',?namespace='/chat')
def?handle_chat_message(msg):
????emit('message',?msg,?broadcast=True,?namespace='/chat')

@socketio.on('notification',?namespace='/notifications')
def?handle_notification(msg):
????emit('notification',?msg,?broadcast=True,?namespace='/notifications')

通过命名空间，消息只会在指定的空间内广播，互不干扰。

2. 房间（Rooms）

房间是命名空间的进一步细分。通过房间，我们可以让消息只发送给特定的用户组。例如在多人游戏或群聊中，只有在同一房间内的用户才能收到彼此的消息：

@socketio.on('join')
def?on_join(data):
????username?=?data['username']
????room?=?data['room']
????join_room(room)
????send(username?+?'?has?entered?the?room.',?to=room)

@socketio.on('leave')
def?on_leave(data):
????username?=?data['username']
????room?=?data['room']
????leave_room(room)
????send(username?+?'?has?left?the?room.',?to=room)

3. 自定义事件

除了message，我们可以定义更多自定义事件，帮助实现更细化的实时功能。例如，可以定义一个notification事件专门用于发送通知。

@socketio.on('notification')
def?handle_notification(data):
????print('Notification:',?data)
????emit('notification',?data,?broadcast=True)

代码解释：

emit('notification', data)：发送通知事件。
broadcast=True：将消息广播到所有客户端。

小贴士： 尽量使用自定义事件命名，保持事件名简洁且语义清晰。

异常处理

异常处理在WebSocket中也非常重要，特别是在消息通信较频繁时。Flask-SocketIO可以结合Flask的标准异常处理机制，我们可以捕获异常并返回错误消息。

@socketio.on_error()????????#?全局错误处理
def?error_handler(e):
????print('Error:',?e)

@socketio.on_error_default??#?捕获所有未定义命名空间的错误
def?default_error_handler(e):
????print('Unhandled?Error:',?e)