🐛 修复 ledger 在月初查询失败的问题

This commit is contained in:
xtaodada 2022-10-01 00:26:16 +08:00
parent b83c611516
commit 59942ba8aa
Signed by: xtaodada
GPG Key ID: 4CBB3F4FA8C85659

View File

@ -22,8 +22,14 @@ from utils.helpers import get_genshin_client
from utils.log import logger from utils.log import logger
def get_now() -> datetime:
now = datetime.now()
return (now - timedelta(days=1)) if now.day == 1 and now.hour <= 4 else now
def check_ledger_month(context: CallbackContext) -> int: def check_ledger_month(context: CallbackContext) -> int:
month = datetime.now().month now_time = get_now()
month = now_time.month
args = get_all_args(context) args = get_all_args(context)
if len(args) >= 1: if len(args) >= 1:
month = args[0] month = args[0]
@ -36,8 +42,7 @@ def check_ledger_month(context: CallbackContext) -> int:
"": 6, "": 7, "": 8, "": 9, "": 10} "": 6, "": 7, "": 8, "": 9, "": 10}
month = sum(num_dict.get(i, 0) for i in str(month)) month = sum(num_dict.get(i, 0) for i in str(month))
# check right # check right
now_time = datetime.now() allow_month = [now_time.month]
allow_month = [datetime.now().month]
last_month = now_time.replace(day=1) - timedelta(days=1) last_month = now_time.replace(day=1) - timedelta(days=1)
allow_month.append(last_month.month) allow_month.append(last_month.month)
last_month = last_month.replace(day=1) - timedelta(days=1) last_month = last_month.replace(day=1) - timedelta(days=1)