如何在没有 html/js 的情况下捕获 fastapi websocket 消息？答案

【问题标题】：How can I capture fastapi websocket messages without html/js?如何在没有 html/js 的情况下捕获 fastapi websocket 消息？
【发布时间】：2021-05-23 04:32:42
【问题描述】：

Fastapi 文档包含一个 websocket example，它通过 html/javascript 接收数据。将脚本保存为main.py 并运行uvicorn main:app --reload，示例按预期工作：

from fastapi import FastAPI, WebSocket
from fastapi.responses import HTMLResponse

app = FastAPI()

html = """
<!DOCTYPE html>
<html>
    <head>
        <title>Chat</title>
    </head>
    <body>
        <h1>WebSocket Chat</h1>
        <form action="" onsubmit="sendMessage(event)">
            <input type="text" id="messageText" autocomplete="off"/>
            <button>Send</button>
        </form>
        <ul id='messages'>
        </ul>
        <script>
            var ws = new WebSocket("ws://localhost:8000/ws");
            ws.onmessage = function(event) {
                var messages = document.getElementById('messages')
                var message = document.createElement('li')
                var content = document.createTextNode(event.data)
                message.appendChild(content)
                messages.appendChild(message)
            };
            function sendMessage(event) {
                var input = document.getElementById("messageText")
                ws.send(input.value)
                input.value = ''
                event.preventDefault()
            }
        </script>
    </body>
</html>
"""


@app.get("/")
async def get():
    return HTMLResponse(html)


@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    while True:
        data = await websocket.receive_text()
        await websocket.send_text(f"Message text was: {data}")

如何修改此示例以在不使用任何 html/js 的情况下将 websocket 消息写入文件？我想用 python 直接访问传入的数据（文本/json），但我无法直接捕获它。任何额外的信息/清晰度表示赞赏。

【问题讨论】：

你试过什么？您介意在您的问题中进一步详细说明吗？此外，您是否需要将服务器或客户端收到的消息写入文件？我不清楚
我想将客户端收到的消息写入文件。
Javascript 在浏览器上运行，不能直接访问本地文件。如果可以保留界面（UI），则获取消息然后对其进行破解以模拟文件的下载。如果适合您的情况，请参阅stackoverflow.com/questions/30872790/…
如何使用 python 脚本连接和记录消息，如 websockets.readthedocs.io/en/stable/index.html ？
是的，但这个问题是关于使用 fastapi 的 websocket 支持

标签： python web websocket fastapi

【解决方案1】：

所以，可能在 cmets 中我没有很好地解释自己。

据我了解，您希望通过 python 连接到您的 web 服务器，并使用 websocket 并记录服务器发送到客户端上的 python 脚本的消息。简单来说，你需要通过 python 来模仿 html/js 部分。

TL;DR

服务器已经存在，您只需连接到它。

这是代码 sn-p，您必须将其复制并粘贴到不同的文件中，并在网络服务器已经运行时运行。请注意，不需要更改网络服务器，如果不是 while True 循环中的两行。这些可能会消失，您可以使用 await websocket.send_text("text") 之类的方式更改它们

import asyncio
import websockets


async def hello():
    uri = "ws://localhost:8000/ws"
    async with websockets.connect(uri) as websocket:
        await websocket.send("Hello world!")
        res = await websocket.recv()
        print(res)

asyncio.get_event_loop().run_until_complete(hello())

我不确定您需要将哪些消息写入文件，但上面的代码 sn-p 正在运行，并且是您需要的基础。

【讨论】：