Gift-Code-Web/models/honkai.py

import re
from pytz import timezone
from datetime import datetime
from typing import List

from httpx import get
from bs4 import BeautifulSoup, Tag
from .code import Code, Reward


url = "https://honkai.gg/codes/"
reward_map = {
    "Stellar Jade": "星琼",
    "Credit": "信用点",
    "Credits": "信用点",
    "Traveler's Guide": "漫游指南",
    "Traveler’s Guide": "漫游指南",
    "Refined Aether": "提纯以太",
    "Adventure Log": "冒险记录",
    "Dust of Alacrity": "疾速粉尘",
    "Condensed Aether": "凝缩以太",
    "Cosmic Fried Rice": "大宇宙炒饭",
    "Travel Encounters": "旅情见闻",
    "Energy Drink": "能量饮料",
    "Startaro Bubble": "星芋啵啵",
    "Lost Gold Fragments": "遗失碎金",
}


def parse_reward(reward: List[str]) -> Reward:
    try:
        name = reward_map.get(reward[0])
        if not name:
            # 判断是否为中文
            if not re.search("[\u4e00-\u9fa5]", reward[0]):
                print("Unknown reward: ", reward[0])
            name = reward[0]
        return Reward(
            name=name,
            cnt=int(reward[1]),
        )
    except Exception as e:
        print("Bad reward data: ", reward)
        raise e


def parse_code(tr: Tag) -> Code:
    tds = tr.find_all("td")
    code = tds[0].text.strip()
    try:
        data = str(tds[2]).split("<br/>")
        expire = data[1]
    except (IndexError, TypeError):
        _, expire = datetime(1970, 1, 1, 1, 0, 0, 0), datetime(2099, 12, 31, 23, 59, 59, 999999)
    if isinstance(expire, str):
        try:
            expire = expire.split(": ")[1].replace("</td>", "")
            if "Unknown" in expire:
                expire = datetime(2099, 12, 31, 23, 59, 59, 999999)
            elif "," in expire:
                expire = datetime.strptime(expire, "%B %d, %Y")
            else:
                expire = datetime.strptime(expire, "%B %d")
                expire = expire.replace(year=datetime.now().year)
        except IndexError:
            expire = datetime(2099, 12, 31, 23, 59, 59, 999999)
        expire = timezone("Asia/Shanghai").localize(expire)
        expire = int(expire.timestamp() * 1000)
    rewards = []
    for reward in str(tds[1]).split("<br/>"):
        reward_soup = BeautifulSoup(reward, "lxml")
        reward_text = " ".join(reward_soup.text.strip().split())
        reward_div = []
        if " x " in reward_text:
            reward_div = reward_text.split(" x ")
        elif " x" in reward_text:
            reward_div = reward_text.split(" x")
        if len(reward_div) < 2:
            print("Bad td data: ", tds[1])
            continue
        parsed_reward = parse_reward(reward_div)
        if parsed_reward:
            rewards.append(parsed_reward)
    if not rewards:
        for reward in tds[1].find_all("a"):
            reward_a = reward.text.strip().split(" x ")
            if len(reward_a) < 2:
                print("Bad a data: ", tds[1])
                continue
            parsed_reward = parse_reward(reward_a)
            if parsed_reward:
                rewards.append(parsed_reward)
    return Code(code=code, reward=rewards, expire=expire)


def get_code():
    html = get(url).text
    soup = BeautifulSoup(html, "lxml")
    tables = soup.find_all("table")
    codes = []
    for table in tables:
        trs = table.find_all("tr")[1:]
        for tr in trs:
            codes.append(parse_code(tr))
    codes.sort(key=lambda x: x.expire, reverse=True)
    return codes