162 lines
5.9 KiB
Python
162 lines
5.9 KiB
Python
"""虎牙精英宝典的同步 WSS 会话封装。
|
|
|
|
任务执行器运行在线程中，使用本类把异步 WSS 生命周期限制在一个账号任务内，
|
|
避免在每次 RPC 时重复建连，也不把 asyncio 对象跨线程传递。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import asyncio
|
|
import urllib.parse
|
|
from collections.abc import Callable
|
|
from typing import Literal
|
|
|
|
from .http_client import HuyaHttpClient, generate_http_baseinfo
|
|
from .wss_client import ACTIVITY_WS_HOST, SHOP_BASEINFO, SHOP_WS_HOST, HuyaWssClient
|
|
|
|
|
|
class HuyaEliteWssSession:
|
|
"""一个账号、一个通道、一个可复用的同步 WSS 会话。"""
|
|
|
|
def __init__(
|
|
self,
|
|
uid: int,
|
|
cookie: str,
|
|
kind: Literal["activity", "shop"],
|
|
logger: Callable[[str], None] | None = None,
|
|
host: str | None = None,
|
|
):
|
|
self.uid = int(uid or 0)
|
|
self.cookie = cookie or ""
|
|
self.guid = HuyaHttpClient._resolve_cookie_guid(self.cookie)
|
|
self.kind = kind
|
|
self.logger = logger or (lambda _message: None)
|
|
self.http = HuyaHttpClient(logger=self.logger)
|
|
self.loop = asyncio.new_event_loop()
|
|
self.client: HuyaWssClient | None = None
|
|
self._closed = False
|
|
|
|
if kind == "activity":
|
|
baseinfo = urllib.parse.unquote(
|
|
generate_http_baseinfo(self.uid, self.guid, "")
|
|
)
|
|
connect_host = host or ACTIVITY_WS_HOST
|
|
origin = "https://zt.huya.com"
|
|
else:
|
|
baseinfo = SHOP_BASEINFO
|
|
connect_host = host or SHOP_WS_HOST
|
|
origin = "https://m-shop.yaoguo.com"
|
|
|
|
self.client = HuyaWssClient(baseinfo=baseinfo, logger=self.logger)
|
|
try:
|
|
self.loop.run_until_complete(
|
|
self.client.connect(host=connect_host, origin=origin, cookie=self.cookie)
|
|
)
|
|
initialized = self.loop.run_until_complete(
|
|
self.client.initialize_activity(self.uid, self.guid, self.cookie)
|
|
if kind == "activity"
|
|
else self.client.initialize(self.uid, "", self.cookie)
|
|
)
|
|
if not initialized:
|
|
raise RuntimeError(f"虎牙 {kind} WSS 初始化失败")
|
|
except Exception:
|
|
self.close()
|
|
raise
|
|
|
|
def _run(self, coroutine):
|
|
if self._closed or self.client is None:
|
|
raise RuntimeError("虎牙 WSS 会话已关闭")
|
|
return self.loop.run_until_complete(coroutine)
|
|
|
|
def close(self):
|
|
if self._closed:
|
|
return
|
|
self._closed = True
|
|
if self.client is not None and self.client.ws is not None:
|
|
try:
|
|
self.loop.run_until_complete(self.client.disconnect())
|
|
except Exception: # noqa: BLE001 - cleanup must not mask task result
|
|
self.logger("[WSS] 会话清理异常")
|
|
self.loop.close()
|
|
|
|
def __enter__(self):
|
|
return self
|
|
|
|
def __exit__(self, _exc_type, _exc, _tb):
|
|
self.close()
|
|
|
|
def get_act_info(self, act_id: int):
|
|
return self._run(self.client.get_act_info(act_id))
|
|
|
|
def get_act_task_detail(self, uid: int, cookie: str, act_id: int):
|
|
return self._run(self.client.get_act_task_detail(uid, cookie, act_id))
|
|
|
|
def get_act_user_task_detail(self, uid: int, cookie: str, act_id: int):
|
|
return self._run(self.client.get_act_user_task_detail(uid, cookie, act_id))
|
|
|
|
def get_user_score(self, uid: int, cookie: str, sid: int):
|
|
return self._run(self.client.get_user_score(uid, cookie, sid))
|
|
|
|
def query_user_score(self, uid: int, cookie: str, sid: int):
|
|
return self.get_user_score(uid, cookie, sid)
|
|
|
|
def get_act_prize_list(self, uid: int, cookie: str, sid: int):
|
|
return self._run(self.client.get_act_prize_list(uid, cookie, sid))
|
|
|
|
def get_act_prize_detail(self, uid: int, cookie: str, sid: int, pid: int):
|
|
return self._run(self.client.get_act_prize_detail(uid, cookie, sid, pid))
|
|
|
|
def score_exchange_prize(self, uid: int, cookie: str, sid: int, pid: int):
|
|
return self._run(self.client.score_exchange_prize(uid, cookie, sid, pid))
|
|
|
|
def get_user_prize_records(self, uid: int, cookie: str, sid: int):
|
|
return self._run(self.client.get_user_prize_records(uid, cookie, sid))
|
|
|
|
def check_user_bind_game_account(self, **kwargs):
|
|
return self._run(self.client.check_user_bind_game_account(**kwargs))
|
|
|
|
def confirm_bind_act_account(self, **kwargs):
|
|
return self._run(self.client.confirm_bind_act_account(**kwargs))
|
|
|
|
def get_live_link_param(self, **kwargs):
|
|
return self._run(self.client.get_live_link_param(**kwargs))
|
|
|
|
def get_user_profile_batch(self, **kwargs):
|
|
return self._run(self.client.get_user_profile_batch(**kwargs))
|
|
|
|
def build_bind_urls(self, *args, **kwargs):
|
|
return self.http.build_bind_urls(*args, **kwargs)
|
|
|
|
def get_livelink_mini_qrcode(self, *args, **kwargs):
|
|
return self.http.get_livelink_mini_qrcode(*args, **kwargs)
|
|
|
|
def get_livelink_qrcode_status(self, *args, **kwargs):
|
|
return self.http.get_livelink_qrcode_status(*args, **kwargs)
|
|
|
|
def get_goods_info(self, **kwargs):
|
|
return self._run(self.client.get_goods_info(**kwargs))
|
|
|
|
def list_pay_channels(self, **kwargs):
|
|
return self._run(self.client.list_pay_channels(**kwargs))
|
|
|
|
def check_hy_protocol(self, **kwargs):
|
|
return self._run(self.client.check_hy_protocol(**kwargs))
|
|
|
|
def check_user_buy_auth(self, **kwargs):
|
|
return self._run(self.client.check_user_buy_auth(**kwargs))
|
|
|
|
def create_order(self, **kwargs):
|
|
return self._run(self.client.create_order(**kwargs))
|
|
|
|
def pay_order_submit(self, **kwargs):
|
|
return self._run(self.client.pay_order_submit(**kwargs))
|
|
|
|
def order_detail(self, **kwargs):
|
|
return self._run(self.client.order_detail(**kwargs))
|
|
|
|
def query_user_order_list(self, **kwargs):
|
|
return self._run(self.client.query_user_order_list(**kwargs))
|
|
|
|
|
|
__all__ = ["HuyaEliteWssSession"]
|