Update app/gscode_logic.py

app/gscode_logic.py

@@ -1,40 +1,54 @@
 # gscode_logic.py
-import json
+import aiohttp
 from bs4 import BeautifulSoup
-from fastapi import HTTPException
 
-class GenshinCodeLogic:
-    async def fetch_codes(self, fetcher, url: str):
-        # 1. fetch raw
-        res = await fetcher.fetch(url)
-
-        if not res or "text" not in res:
-            raise HTTPException(500, "Fetch failed")
+GENSHIN_API = (
+    "https://genshin-impact.fandom.com/api.php"
+    "?action=parse"
+    "&page=Promotional_Code"
+    "&prop=text"
+    "&format=json"
+)
 
-        try:
-            data = json.loads(res["text"])
-        except Exception:
-            raise HTTPException(400, "Invalid JSON response")
+class GenshinCodeLogic:
+    def __init__(self):
+        self.session: aiohttp.ClientSession | None = None
+
+    async def start(self):
+        if not self.session:
+            self.session = aiohttp.ClientSession()
+
+    async def stop(self):
+        if self.session:
+            await self.session.close()
+            self.session = None
+
+    async def fetch_codes(self):
+        async with self.session.get(
+            GENSHIN_API,
+            headers={
+                "User-Agent": "Mozilla/5.0"
+            },
+            timeout=aiohttp.ClientTimeout(total=20)
+        ) as resp:
+            if resp.status != 200:
+                return {"count": 0, "codes": []}
+
+            data = await resp.json()
 
-        # 2. lấy HTML string
         html = data.get("parse", {}).get("text", {}).get("*")
         if not html:
             return {"count": 0, "codes": []}
 
-        # 3. parse HTML
         soup = BeautifulSoup(html, "html.parser")
 
-        # 4. lấy <code>
-        codes = []
-        for c in soup.find_all("code"):
-            val = c.get_text(strip=True)
-            if val and len(val) >= 6:
-                codes.append(val)
-
-        # 5. unique + sort
-        codes = sorted(set(codes))
+        codes = {
+            c.text.strip()
+            for c in soup.select("code")
+            if c.text and len(c.text.strip()) >= 6
+        }
 
         return {
             "count": len(codes),
-            "codes": codes
+            "codes": sorted(codes)
         }