从python中的请求会话创建websocket连接

Question

我需要通过 websocket 连接从网站接收数据。只有在登录网站后才能访问 websocket。在 python 请求中使用会话我可以将我的登录信息发布到登录页面并验证我的详细信息。Websocket-client 将用于创建到网站的 websocket 连接，但是创建的 websocket 连接不会通过登录会话。

通过请求登录后如何创建 websocket 连接？

我在这里发现了一个类似的问题，但尚未得到解答 Python - Websockets 无法验证我的会话

到目前为止，这是我的代码（稍微简化了一点）。

import requests
from bs4 import BeautifulSoup
import base64
import random
import websocket

headers = {
    'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/51.0.2704.103 Safari/537.36'}

payload = {
    'username': 'username',
    'password': 'password'}

session = requests.Session()

r = session.get("https://www.example.com/login", headers=headers)

soup = BeautifulSoup(r.content, 'html.parser')
payload['token'] = soup.find('input', attrs={'name': 'token'})['value']

r = session.post("https://www.example.com/login", data=payload, headers=headers)
r = session.get("https://www.example.com/", headers=headers)

headers['Sec-WebSocket-Key'] = str(base64.b64encode(bytes([random.randint(0, 255) for _ in range(16)])), 'ascii')
headers['Sec-WebSocket-Version'] = '13'
headers['Upgrade'] = 'websocket'

ws = websocket.WebSocketApp('wss://www.example.com/streaming/',
                            header=headers,
                            #session=session????)
ws.run_forever()

Answer 1

经过大量搜索，我最终找到了问题的根源。我在通过 Chrome 浏览器访问网站时使用 Chrome Dev Tools 查看网络数据包，但是 Chrome Dev Tools 不显示 websocket 请求的 cookie。Firefox Developer Tools 确实显示了 websocket 请求的 cookie，而我只在使用 Firefox 时才发现发送了哪些 cookie。解决方案是提取Requests Session的cookies，并将这些cookies传递给WebSocketApp。这样 websocket 服务器就可以验证您是否已登录。

cookies = session.cookies.get_dict()
ws = websocket.WebSocketApp('wss://www.example.com/streaming/',
                        header=headers,
                        cookie="; ".join(["%s=%s" %(i, j) for i, j in cookies.items()]))