| |
| import aiohttp |
| from bs4 import BeautifulSoup |
| from typing import List, Dict |
|
|
| GENSHIN_API = ( |
| "https://genshin-impact.fandom.com/api.php" |
| "?action=parse" |
| "&page=Promotional_Code" |
| "&prop=text" |
| "&format=json" |
| ) |
|
|
| class GenshinCodeLogic: |
| def __init__(self): |
| self.session: aiohttp.ClientSession | None = None |
|
|
| async def start(self): |
| if not self.session: |
| self.session = aiohttp.ClientSession( |
| headers={"User-Agent": "Mozilla/5.0 (GenshinCodeBot)"} |
| ) |
|
|
| async def stop(self): |
| if self.session: |
| await self.session.close() |
| self.session = None |
|
|
| async def fetch_codes(self) -> Dict: |
| async with self.session.get( |
| GENSHIN_API, |
| timeout=aiohttp.ClientTimeout(total=20) |
| ) as resp: |
| if resp.status != 200: |
| return {"count": 0, "cards": []} |
|
|
| data = await resp.json() |
|
|
| html = data.get("parse", {}).get("text", {}).get("*") |
| if not html: |
| return {"count": 0, "cards": []} |
|
|
| soup = BeautifulSoup(html, "html.parser") |
|
|
| cards: List[Dict] = [] |
|
|
| for row in soup.select("table.wikitable tbody tr"): |
| tds = row.find_all("td") |
| if len(tds) < 3: |
| continue |
|
|
| code_els = tds[0].find_all("code") |
| if not code_els: |
| continue |
|
|
| codes = [c.get_text(strip=True) for c in code_els] |
|
|
| server = tds[1].get_text(" ", strip=True) |
|
|
| rewards = [] |
| for item in tds[2].select(".item-text"): |
| text = item.get_text(" ", strip=True) |
| if text: |
| rewards.append(text) |
|
|
| validity = ( |
| tds[3].get_text(" ", strip=True) |
| if len(tds) >= 4 else None |
| ) |
|
|
| cards.append({ |
| "codes": codes, |
| "server": server, |
| "rewards": rewards, |
| "validity": validity |
| }) |
|
|
| return { |
| "count": sum(len(card["codes"]) for card in cards), |
| "cards": cards |
| } |
