登录
首页 >  文章 >  python教程

Python网页版WebSocket通信实现方法

时间:2026-04-07 13:35:14 197浏览 收藏

本文详细介绍了如何使用FastAPI与原生JavaScript快速搭建一个稳定、可运行的Python网页版WebSocket实时通信系统,涵盖后端服务定义、连接管理、消息广播、前端连接与交互、异常处理、心跳保活及生产部署注意事项,通过一个简洁的聊天示例,让开发者轻松掌握双向实时通信的核心实现逻辑与关键实践要点。

Python网页版如何实现WebSocket通信_Python网页版WebSocket实时通信开发指南

要在Python网页版中实现WebSocket通信,核心是使用支持异步处理的后端框架配合前端JavaScript建立长连接。WebSocket能实现客户端与服务器之间的双向实时通信,非常适合聊天应用、实时数据推送等场景。下面从后端搭建、前端连接到完整示例,一步步说明如何开发。

选择合适的Python后端框架

Python中常用的WebSocket支持框架有以下几种:

  • FastAPI + Uvicorn:基于ASGI的现代框架,原生支持WebSocket,开发效率高
  • Sanic:异步Web框架,性能优秀,语法类似Flask
  • Tornado:老牌异步框架,自带WebSocket模块,适合独立部署

推荐使用FastAPI,因其文档清晰、集成简单,且可直接通过@app.websocket装饰器定义接口。

使用FastAPI搭建WebSocket服务端

安装依赖:

pip install fastapi uvicorn websockets

编写后端代码(main.py):

from fastapi import FastAPI, WebSocket
from fastapi.responses import HTMLResponse
<p>app = FastAPI()</p><h1>存储所有活动连接</h1><p>active_connections = []</p><p>@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
await websocket.accept()
active_connections.append(websocket)
try:
while True:
data = await websocket.receive_text()</p><h1>广播消息给所有连接</h1><pre class="brush:php;toolbar:false"><code>        for connection in active_connections:
            if connection != websocket:
                await connection.send_text(f"收到: {data}")
except:
    active_connections.remove(websocket)</code>

启动服务:

uvicorn main:app --reload

前端页面通过JavaScript连接WebSocket

创建一个简单的HTML页面,使用浏览器原生WebSocket API连接后端:

<!DOCTYPE html>
<html>
<body>
  &lt;input type=&quot;text&quot; id=&quot;message&quot; placeholder=&quot;输入消息&quot;/&gt;
  <button onclick="sendMessage()">发送</button>
  <div id="output"></div>
<p><script>
const ws = new WebSocket("ws://127.0.0.1:8000/ws");</p><pre class="brush:php;toolbar:false"><code>ws.onmessage = function(event) {
  const output = document.getElementById("output");
  const msg = document.createElement("p");
  msg.textContent = event.data;
  output.appendChild(msg);
};

function sendMessage() {
  const input = document.getElementById("message");
  ws.send(input.value);
  input.value = "";
}</code>

将该页面返回给用户,可在FastAPI中添加路由:

@app.get("/", response_class=HTMLResponse)
def get_page():
    with open("index.html", "r") as f:
        return f.read()

处理连接管理与异常情况

实际开发中需注意连接状态管理:

  • try...finally中确保断开时清理连接列表
  • 设置心跳机制防止连接超时关闭
  • 对敏感数据做校验和过滤
  • 生产环境使用Nginx反向代理时,确保配置了正确的Upgrade头

例如增强异常处理:

finally:
    if websocket in active_connections:
        active_connections.remove(websocket)

基本上就这些。只要后端用FastAPI这类支持异步的框架开启WebSocket路径,前端用new WebSocket(url)连接并监听消息,就能实现双向实时通信。不复杂但容易忽略的是连接维护和错误恢复逻辑,上线前务必测试断网重连等情况。

以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于文章的相关知识,也可关注golang学习网公众号。

资料下载
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>