实现虎牙刷新商品列表

core/huya/activity_structs.py

@@ -104,6 +104,191 @@ class GetUserScoreResp(TafStruct):
         }
 
 
+class GetActPrizeListReq(TafStruct):
+    """webActUI.getActPrizeList 请求。"""
+
+    def __init__(self):
+        self.userId = ActivityUserId()
+        self.sid: int = 0
+
+    def write_to(self, os: TafOutputStream):
+        os.write_struct(0, self.userId)
+        os.write_int32(1, self.sid)
+
+    def read_from(self, ins: TafInputStream):
+        self.userId = ins.read_struct(0, ActivityUserId) or self.userId
+        self.sid = ins.read_int32(1, default=self.sid)
+
+
+class ActPrizeItem(TafStruct):
+    """活动商品项，字段位置来自 getActPrizeList 实测响应。"""
+
+    def __init__(self):
+        self.prizeId: int = 0
+        self.prizeType: int = 0
+        self.name: str = ""
+        self.score: int = 0
+        self.icon: str = ""
+        self.stock: int = 0
+        self.remainText: str = ""
+        self.price: int = 0
+        self.categoryId: str = ""
+        self.totalRemain: int = 0
+        self.rawOrder: int = 0
+        self.startTime: int = 0
+        self.endTime: int = 0
+
+    def read_from(self, ins: TafInputStream):
+        self.prizeId = ins.read_int64(0, default=self.prizeId)
+        self.prizeType = ins.read_int32(1, default=self.prizeType)
+        self.name = ins.read_string(2, default=self.name)
+        self.score = ins.read_int64(3, default=self.score)
+        ins.read_int32(4, default=0)
+        self.icon = ins.read_string(5, default=self.icon)
+        self.stock = ins.read_int64(10, default=self.stock)
+        self.remainText = ins.read_string(12, default=self.remainText)
+        self.price = ins.read_int64(13, default=self.price)
+        self.categoryId = ins.read_string(14, default=self.categoryId)
+        self.totalRemain = ins.read_int64(17, default=self.totalRemain)
+        self.rawOrder = ins.read_int64(22, default=self.rawOrder)
+        self.startTime = ins.read_int64(25, default=self.startTime)
+        self.endTime = ins.read_int64(26, default=self.endTime)
+
+    def write_to(self, os: TafOutputStream):
+        os.write_int64(0, self.prizeId)
+        os.write_int32(1, self.prizeType)
+        os.write_string(2, self.name)
+        os.write_int64(3, self.score)
+        os.write_string(5, self.icon)
+        os.write_int64(10, self.stock)
+        os.write_string(12, self.remainText)
+        os.write_int64(13, self.price)
+        os.write_string(14, self.categoryId)
+        os.write_int64(17, self.totalRemain)
+        os.write_int64(22, self.rawOrder)
+        os.write_int64(25, self.startTime)
+        os.write_int64(26, self.endTime)
+
+    @staticmethod
+    def read_list_item(ins: TafInputStream, _tag: int):
+        item = ActPrizeItem()
+        item.read_from(ins)
+        while True:
+            _end_tag, dtype = ins.read_head()
+            if dtype == TafType.STRUCT_END:
+                break
+            ins.skip_field(dtype)
+        return item
+
+    @staticmethod
+    def _remain_percent_text(value: str) -> str:
+        text = str(value or "").strip()
+        if not text or text.endswith("%"):
+            return text
+        try:
+            float(text)
+        except ValueError:
+            return text
+        return f"{text}%"
+
+    def to_dict(self) -> dict:
+        return {
+            "product_id": str(self.prizeId),
+            "name": self.name,
+            "price": self.price,
+            "remain_text": self._remain_percent_text(self.remainText),
+            "remain_percent": self.remainText,
+            "stock": self.stock,
+            "score": self.score,
+            "icon": self.icon,
+            "prize_type": self.prizeType,
+            "category_id": self.categoryId,
+            "total_remain": self.totalRemain,
+            "raw_order": self.rawOrder,
+            "start_time": self.startTime,
+            "end_time": self.endTime,
+        }
+
+
+class ActPrizeTag(TafStruct):
+    """活动商品分类。"""
+
+    def __init__(self):
+        self.tagId: int = 0
+        self.name: str = ""
+        self.sort: int = 0
+
+    def read_from(self, ins: TafInputStream):
+        self.tagId = ins.read_int64(0, default=self.tagId)
+        self.name = ins.read_string(1, default=self.name)
+        self.sort = ins.read_int32(2, default=self.sort)
+
+    def write_to(self, os: TafOutputStream):
+        os.write_int64(0, self.tagId)
+        os.write_string(1, self.name)
+        os.write_int32(2, self.sort)
+
+    @staticmethod
+    def read_list_item(ins: TafInputStream, _tag: int):
+        item = ActPrizeTag()
+        item.read_from(ins)
+        while True:
+            _end_tag, dtype = ins.read_head()
+            if dtype == TafType.STRUCT_END:
+                break
+            ins.skip_field(dtype)
+        return item
+
+    def to_dict(self) -> dict:
+        return {
+            "category_id": str(self.tagId),
+            "category_name": self.name,
+            "sort": self.sort,
+        }
+
+
+class GetActPrizeListResp(TafStruct):
+    """webActUI.getActPrizeList 响应。"""
+
+    def __init__(self):
+        self.status: int = 0
+        self.msg: str = ""
+        self.prizes: list[ActPrizeItem] = []
+        self.tags: list[ActPrizeTag] = []
+
+    def read_from(self, ins: TafInputStream):
+        self.status = ins.read_int32(0, default=self.status)
+        self.msg = ins.read_string(1, default=self.msg)
+        self.prizes = ins.read_list(2, item_reader=ActPrizeItem.read_list_item)
+        self.tags = ins.read_list(4, item_reader=ActPrizeTag.read_list_item)
+
+    def write_to(self, os: TafOutputStream):
+        os.write_int32(0, self.status)
+        os.write_string(1, self.msg)
+        os.write_list(2, self.prizes)
+        os.write_list(4, self.tags)
+
+    def to_dict(self) -> dict:
+        tag_info = {
+            str(item.tagId): {"name": item.name, "sort": index, "raw_sort": item.sort}
+            for index, item in enumerate(self.tags)
+        }
+        goods = []
+        for item in self.prizes:
+            data = item.to_dict()
+            category = tag_info.get(data["category_id"], {})
+            data["category_name"] = category.get("name", "")
+            data["category_sort"] = category.get("sort", 0)
+            goods.append(data)
+        return {
+            "status": self.status,
+            "msg": self.msg,
+            "goods_count": len(self.prizes),
+            "categories": [item.to_dict() for item in self.tags],
+            "goods": goods,
+        }
+
+
 class GameAccount(TafStruct):
     """游戏账号信息。"""
 
@@ -532,6 +717,35 @@ def _self_check():
     assert parsed.status == 200
     assert parsed.available_score == 180
 
+    prize_rsp = GetActPrizeListResp()
+    prize = ActPrizeItem()
+    prize.prizeId = 12885
+    prize.name = "赤蝎幽灵大礼包"
+    prize.price = 3600
+    prize.remainText = "39"
+    prize.categoryId = "85"
+    tag = ActPrizeTag()
+    tag.tagId = 85
+    tag.name = "限量返场"
+    tag.sort = 2
+    prize_rsp.status = 200
+    prize_rsp.msg = "请求成功"
+    prize_rsp.prizes = [prize]
+    prize_rsp.tags = [tag]
+    os = TafOutputStream()
+    os.write_struct(0, prize_rsp)
+    ins = TafInputStream(os.get_bytes())
+    _, dtype = ins.read_head()
+    assert dtype == TafType.STRUCT_BEGIN
+    parsed_prize = GetActPrizeListResp()
+    parsed_prize.read_from(ins)
+    assert parsed_prize.status == 200
+    assert parsed_prize.prizes[0].name == "赤蝎幽灵大礼包"
+    assert parsed_prize.prizes[0].price == 3600
+    parsed_prize_data = parsed_prize.to_dict()
+    assert parsed_prize_data["goods"][0]["category_name"] == "限量返场"
+    assert parsed_prize_data["goods"][0]["remain_text"] == "39%"
+
     link_rsp = GetLiveLinkParamResp()
     link_rsp.status = 200
     link_rsp.livelinkParam = {

core/huya/http_client.py

@@ -270,6 +270,22 @@ class HuyaHttpClient:
             timeout=timeout,
         )
 
+    def get_act_prize_list(self, uid: int, cookie: str, sid: int, timeout: float = 15.0):
+        """查询活动可兑换商品列表。"""
+        from .activity_structs import GetActPrizeListReq, GetActPrizeListResp
+        req = GetActPrizeListReq()
+        req.userId = self._build_activity_user(uid, cookie)
+        req.sid = sid
+        return self.call_rpc(
+            "webActUI",
+            "getActPrizeList",
+            req,
+            GetActPrizeListResp,
+            uid=uid,
+            cookie=cookie,
+            timeout=timeout,
+        )
+
     def check_user_bind_game_account(
         self,
         uid: int,