diff --git a/app.py b/app.py index a8cca29..d519d83 100644 --- a/app.py +++ b/app.py @@ -93,11 +93,29 @@ def index(): """首页""" user = get_current_user() if user: + # 检查日期并重置计数(使用上海时区) + from datetime import timezone, timedelta + shanghai_tz = timezone(timedelta(hours=8)) + today = datetime.now(shanghai_tz).date() + if user.last_translate_date != today: + user.daily_count = 0 + user.last_translate_date = today + db.session.commit() + limits = USER_LIMITS.get(user.user_type, USER_LIMITS['free']) daily_remaining = limits['daily_translations'] - user.daily_count if limits['daily_translations'] > 0 else '无限' max_pages = limits['max_pages'] if limits['max_pages'] > 0 else '无限' else: guest = get_or_create_guest() + # 检查日期并重置访客计数 + from datetime import timezone, timedelta + shanghai_tz = timezone(timedelta(hours=8)) + today = datetime.now(shanghai_tz).date() + if guest.last_translate_date != today: + guest.daily_count = 0 + guest.last_translate_date = today + db.session.commit() + limits = USER_LIMITS['guest'] daily_remaining = limits['daily_translations'] - guest.daily_count max_pages = limits['max_pages']