refactor(douyu): runner 按功能域拆分 Mixin + 精英手册兑换对齐真机流程 (HAR 审核通过)

拆分: douyu_runner.py (-3186行) → core/bind/manual/gold/donate/goods/xpd 七个
Mixin + 入口聚合类, douyu_batch_registry 供 routers 重导出

精英手册兑换对齐 HAR 抓包 (activity_client +379 行):
- csrf 复用 Cookie 值不再每次 generateCsrf (HAR 实测全程零 csrf 请求),
  仅服务端报 csrf 错误时 force_refresh 重试一次
- pay Referer 补 roomId 对齐浏览器
- 新增手册链路接口: elite_user_info/storedetail/pre_exchange check+confirm/
  subscribe/batch_limit/赠品兑换/自动转换
- resolve_exchange_plan: 浏览器兑换按钮状态机路由 (normal/subscribe/
  pre_exchange/wait/blocked)
- 兑换执行: 人类节奏抖动 → 锁单(火爆重试1次) → 支付分级退避
  (频控 30/60/90s 4次上限, 普错 2/5/10/20s 5次上限, 300s 锁单期 15s 余量,
  可中断睡眠)

测试: 新增 17 项 (csrf 复用语义/状态机路由/Referer), 全量 94 通过;
未夹带代理改动
This commit is contained in:
yml2213
2026-08-30 11:05:18 +08:00
parent 6e17a575dd
commit 39b202bc3f
11 changed files with 3959 additions and 3196 deletions
+163
View File
@@ -0,0 +1,163 @@
"""精英手册兑换:csrf 复用、新链路接口与浏览器状态机路由的测试。"""
import unittest
from unittest.mock import Mock, call
from core.douyu.activity_client import DouyuActivityClient, DouyuActivityError
class CsrfReuseTests(unittest.TestCase):
def test_reuses_cookie_token_without_any_http_call(self):
client = DouyuActivityClient("acf_uid=100; cvl_csrf_token=existing-token")
client._request_json = Mock()
self.assertEqual(client.csrf_token(), "existing-token")
client._request_json.assert_not_called()
def test_generates_csrf_when_cookie_missing(self):
client = DouyuActivityClient("acf_uid=100")
def fake_generate(*_args, **_kwargs):
# 模拟 generateCsrf 响应把新 token 写回 Cookie
client._set_cookie_value("cvl_csrf_token", "new-token")
return {"error": 0}
client._request_json = Mock(side_effect=fake_generate)
self.assertEqual(client.csrf_token(), "new-token")
client._request_json.assert_called_once()
def test_force_refresh_rotates_token(self):
client = DouyuActivityClient("acf_uid=100; cvl_csrf_token=old-token")
client._request_json = Mock(return_value={"error": 0})
client._set_cookie_value("cvl_csrf_token", "rotated-token")
self.assertEqual(client.csrf_token(force_refresh=True), "rotated-token")
def test_refreshes_csrf_once_on_token_error(self):
client = DouyuActivityClient("acf_uid=100")
client.csrf_token = Mock(side_effect=["bad-token", "good-token"])
client._request_json = Mock(side_effect=[
{"error": 1, "msg": "csrfToken 校验失败"},
{"error": 0, "data": {"orderId": "9", "expireSeconds": "300"}},
])
result = client.create_exchange_order(manual_id="m", rid="r", commodity_id="c")
self.assertEqual(result["order_id"], "9")
self.assertEqual(client.csrf_token.call_args_list, [call(), call(force_refresh=True)])
def test_no_refresh_on_regular_error(self):
client = DouyuActivityClient("acf_uid=100")
client.csrf_token = Mock(return_value="t")
client._request_json = Mock(return_value={"error": 1, "msg": "需要开通精英手册,才可以兑换本道具哦"})
with self.assertRaisesRegex(DouyuActivityError, "需要开通精英手册"):
client.create_exchange_order(manual_id="m", rid="r", commodity_id="c")
client.csrf_token.assert_called_once_with()
def test_pay_referer_includes_room_id_like_browser(self):
client = DouyuActivityClient("acf_uid=100")
client.csrf_token = Mock(return_value="t")
client._request_json = Mock(return_value={"error": 0, "data": {"exchangeId": "11098933"}})
client.pay_exchange_order(manual_id="m", order_id="5874", rid="9263298")
headers = client._request_json.call_args.kwargs["headers"]
self.assertEqual(
headers["Referer"],
"https://www.douyu.com/pages/live-peace-handbook/web/shop?ditchname=pass0&roomId=9263298",
)
class ExchangePlanTests(unittest.TestCase):
"""对齐浏览器兑换按钮状态机($e / Ve / Ge)的路由判定。"""
@staticmethod
def detail(**overrides):
base = {
"status": 0,
"storeStatus": 1,
"openStatus": 1,
"roomLimited": 0,
"isLimitRoom": 0,
"eliteLimited": 0,
"subscribeType": 0,
"exchangeInfo": {"userStatus": 0},
"preExchange": {"isExists": 0, "expiringTime": "0"},
"batchExchange": 0,
}
base.update(overrides)
return base
def test_normal_item_uses_classic_flow(self):
plan = DouyuActivityClient.resolve_exchange_plan(self.detail())
self.assertEqual(plan["action"], "classic")
self.assertEqual(plan["max_num"], 1)
def test_pre_exchange_order_still_classic(self):
plan = DouyuActivityClient.resolve_exchange_plan(self.detail(preExchange={"isExists": 1}))
self.assertEqual(plan["action"], "classic")
def test_reached_exchange_limit_blocked(self):
plan = DouyuActivityClient.resolve_exchange_plan(self.detail(status=1))
self.assertEqual(plan["action"], "blocked")
self.assertIn("上限", plan["text"])
def test_no_stock_blocked(self):
plan = DouyuActivityClient.resolve_exchange_plan(self.detail(storeStatus=0))
self.assertEqual(plan["action"], "blocked")
self.assertIn("无库存", plan["text"])
def test_room_limited_and_not_in_room_blocked(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(roomLimited=1, isLimitRoom=0)
)
self.assertEqual(plan["action"], "blocked")
self.assertIn("限定房间", plan["text"])
def test_not_open_subscribe_item_subscribes(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(openStatus=0, subscribeType=1)
)
self.assertEqual(plan["action"], "subscribe")
def test_not_open_pre_exchange_item_pre_exchanges(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(openStatus=0, subscribeType=2)
)
self.assertEqual(plan["action"], "pre_exchange")
def test_not_open_plain_item_blocked(self):
plan = DouyuActivityClient.resolve_exchange_plan(self.detail(openStatus=0))
self.assertEqual(plan["action"], "blocked")
self.assertIn("未开启", plan["text"])
def test_elite_limited_requires_manual(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(eliteLimited=1), manual_type=0
)
self.assertEqual(plan["action"], "blocked")
self.assertIn("精英手册", plan["text"])
# 已开通手册(manualType=1)不受精英专享限制
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(eliteLimited=1), manual_type=1
)
self.assertEqual(plan["action"], "classic")
def test_pre_exchanged_waits_for_open(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(subscribeType=2, exchangeInfo={"userStatus": 1})
)
self.assertEqual(plan["action"], "wait")
def test_batch_exchange_clamps_num(self):
plan = DouyuActivityClient.resolve_exchange_plan(
self.detail(batchExchange=1), batch_num_limit="5"
)
self.assertEqual(plan["max_num"], 5)
self.assertTrue(plan["batch_exchange"])
if __name__ == "__main__":
unittest.main()
+3 -3
View File
@@ -54,7 +54,7 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
Base.metadata.drop_all(self.engine)
self.engine.dispose()
@patch("web.backend.services.douyu_runner.FishFinRechargeClient")
@patch("web.backend.services.douyu_runner_gold.FishFinRechargeClient")
def test_supplier_channel_creates_order_using_nickname_and_finishes_on_success(self, client_class):
supplier = Mock()
supplier.create_order.return_value = {
@@ -99,7 +99,7 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
self.assertNotIn("pay_url", self.task.result)
self.assertNotIn("sign", self.task.result["supplier_order"])
@patch("web.backend.services.douyu_runner.FishFinRechargeClient")
@patch("web.backend.services.douyu_runner_gold.FishFinRechargeClient")
def test_supplier_channel_rejects_account_without_nickname(self, client_class):
self.account.nickname = ""
self.session.commit()
@@ -157,7 +157,7 @@ class DouyuGoldRechargeChannelTests(unittest.TestCase):
self.assertIn("10001", audit.detail)
self.assertNotIn("password", audit.detail)
@patch("web.backend.services.douyu_runner.FishFinRechargeClient")
@patch("web.backend.services.douyu_runner_gold.FishFinRechargeClient")
def test_supplier_channel_marks_non_200_create_response_as_failed(self, client_class):
supplier = Mock()
supplier.create_order.return_value = {"code": 400, "msg": "商品已下架"}