Update app/gscode_logic.py

app/gscode_logic.py

@@ -18,9 +18,7 @@ class GenshinCodeLogic:
     async def start(self):
         if not self.session:
             self.session = aiohttp.ClientSession(
-                headers={
-                    "User-Agent": "Mozilla/5.0 (GenshinCodeBot)"
-                }
+                headers={"User-Agent": "Mozilla/5.0 (GenshinCodeBot)"}
             )
 
     async def stop(self):
@@ -29,52 +27,55 @@ class GenshinCodeLogic:
             self.session = None
 
     async def fetch_codes(self) -> Dict:
-        async with self.session.get(GENSHIN_API, timeout=aiohttp.ClientTimeout(total=20)) as resp:
+        async with self.session.get(
+            GENSHIN_API,
+            timeout=aiohttp.ClientTimeout(total=20)
+        ) as resp:
             if resp.status != 200:
-                return {"count": 0, "codes": []}
+                return {"count": 0, "cards": []}
 
             data = await resp.json()
 
         html = data.get("parse", {}).get("text", {}).get("*")
         if not html:
-            return {"count": 0, "codes": []}
+            return {"count": 0, "cards": []}
 
         soup = BeautifulSoup(html, "html.parser")
 
-        results: List[Dict] = []
-        
+        cards: List[Dict] = []
+
         for row in soup.select("table.wikitable tbody tr"):
             tds = row.find_all("td")
             if len(tds) < 3:
                 continue
 
-            code_el = tds[0].find("code")
-            if not code_el:
+            code_els = tds[0].find_all("code")
+            if not code_els:
                 continue
 
-            code = code_el.get_text(strip=True)
+            codes = [c.get_text(strip=True) for c in code_els]
 
-            # server (China / Global / etc)
             server = tds[1].get_text(" ", strip=True)
 
-            # rewards
             rewards = []
             for item in tds[2].select(".item-text"):
                 text = item.get_text(" ", strip=True)
                 if text:
                     rewards.append(text)
 
-            # validity (nếu có)
-            validity = tds[3].get_text(" ", strip=True) if len(tds) >= 4 else None
+            validity = (
+                tds[3].get_text(" ", strip=True)
+                if len(tds) >= 4 else None
+            )
 
-            results.append({
-                "code": code,
+            cards.append({
+                "codes": codes,
                 "server": server,
                 "rewards": rewards,
                 "validity": validity
             })
 
         return {
-            "count": len(results),
-            "codes": results
+            "count": sum(len(card["codes"]) for card in cards),
+            "cards": cards
         }