python - 从python中的请求 session 创建websocket连接

Question

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

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

我在这里找到了一个类似的问题，但尚未得到解答
Python - Websockets Unable to validate my session

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

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()]))