diff --git a/add.py b/add.py index 3d8e5bf..9b8a142 100644 --- a/add.py +++ b/add.py @@ -1,4 +1,5 @@ from pathlib import Path +from pytz import timezone from sys import argv from datetime import datetime from typing import List @@ -16,6 +17,7 @@ def add(code: str, expire_str: str, rewards_str: List[str]) -> Code: else: expire = datetime.strptime(expire_str, "%Y-%m-%d") expire = expire.replace(hour=23, minute=59, second=59, microsecond=999999) + expire = timezone("Asia/Shanghai").localize(expire) rewards = [] for reward_str in rewards_str: reward_list = reward_str.split(":") diff --git a/data/custom.json b/data/custom.json index e3ddcdd..cc0a2e6 100644 --- a/data/custom.json +++ b/data/custom.json @@ -8,7 +8,7 @@ "cnt": 60 } ], - "expire": 1689724799999 + "expire": 1689695999999 }, { "code": "9A6BHRKX4XNL", diff --git a/models/honkai.py b/models/honkai.py index 32c64dc..c9ff17e 100644 --- a/models/honkai.py +++ b/models/honkai.py @@ -1,4 +1,5 @@ import re +from pytz import timezone from datetime import datetime from typing import List @@ -57,6 +58,7 @@ def parse_code(tr: Tag) -> Code: now = datetime.now() expire = datetime.strptime(f"{day} {month}", "%d %b") expire = expire.replace(year=now.year, hour=23, minute=59, second=59, microsecond=999999) + expire = timezone("Asia/Shanghai").localize(expire) expire = int(expire.timestamp() * 1000) rewards = [] for reward in tds[1].find_all("div", {"class": "flex"}): diff --git a/requirements.txt b/requirements.txt index fe60500..027342d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -2,3 +2,4 @@ pydantic httpx beautifulsoup4 lxml +pytz