mirror of
https://github.com/PaiGramTeam/Gift-Code-Web.git
synced 2024-11-16 04:45:25 +00:00
95 lines
3.0 KiB
Python
95 lines
3.0 KiB
Python
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": "旅情见闻",
|
||
}
|
||
|
||
|
||
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:
|
||
_, expire = str(tds[2]).split("<br/>")
|
||
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 expire == "Unknown":
|
||
expire = datetime(2099, 12, 31, 23, 59, 59, 999999)
|
||
else:
|
||
expire = datetime.strptime(expire, "%B %d, %Y")
|
||
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 = BeautifulSoup(reward, "lxml")
|
||
reward_div = " ".join(reward.text.strip().split()).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
|