如何在异步websockets中使用代理并修复收不到消息问题

你的代码无法收到消息是几个显性问题+版本兼容问题共同导致的，逐一修复即可：

• 版本兼容问题：目前主流版本的websockets库（v10及以上）已经移除了内置的http_proxy_host/http_proxy_port/proxy_type/http_proxy_auth代理参数，你写的这些参数不会被识别，连接根本没有走代理，自然无法和目标wss服务建立有效连接。
• 参数类型错误：就算使用支持内置代理参数的旧版本websockets，http_proxy_port要求传入整数类型，你传的字符串"1111"会触发参数校验错误。
• 缺少异常捕获：当前代码没有任何错误打印逻辑，连接握手失败、代理不通、服务端拒绝连接的时候会直接进入下一轮循环重连，你看不到任何报错信息，无法定位问题。

正确实现步骤 #

1. 安装所需依赖

pip install websockets websockets-proxy

websockets-proxy是专门适配新版websockets的代理转发组件，支持HTTP/HTTPS/SOCKS5类型代理，也支持代理认证。

2. 修复后的可运行代码

import asyncio
import websockets
from websockets_proxy import Proxy, proxy_connect

# 替换为你实际的参数值
USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/96.0.4664.110 Safari/537.36"
PROXY_ADDR = "1.1.1.1"
PROXY_PORT = 1111  # 注意端口是整数类型
PROXY_LOGIN = "your_proxy_account"
PROXY_PASSWORD = "your_proxy_password"
WS_TARGET_URL = "wss://your_actual_ws_endpoint"

async def messages_check():
    # 初始化代理对象
    proxy = Proxy.from_url(f"http://{PROXY_LOGIN}:{PROXY_PASSWORD}@{PROXY_ADDR}:{PROXY_PORT}")
    while True:
        try:
            async with proxy_connect(
                WS_TARGET_URL,
                proxy=proxy,
                extra_headers=[("user-agent", USER_AGENT)]
            ) as ws_conn:
                print("Websocket连接建立成功，等待接收消息...")
                async for msg in ws_conn:
                    print(f"收到消息: {msg}")
        except Exception as e:
            print(f"连接异常，5秒后尝试重连，错误信息: {str(e)}")
            await asyncio.sleep(5)

if __name__ == "__main__":
    asyncio.run(messages_check())

额外排查点 #

如果替换代码后还是收不到消息，逐一检查以下项：

• 确认你使用的HTTP代理支持CONNECT方法转发Websocket流量，部分普通HTTP代理只支持HTTP/HTTPS网页流量转发，不支持wss协议，可以先用Postman配置同个代理测试目标wss地址连通性。
• 如果目标wss服务有Origin校验，需要在extra_headers里加上服务端认可的Origin请求头，否则握手阶段会被服务端直接拒绝，不会推送任何消息。
• 如果代理是SOCKS5类型，只需要把初始化代理的URL前缀从http://改成socks5://即可，不需要修改其他逻辑。

内容的提问来源于stack exchange，提问作者Parzival