From 801dd1e29bccf4c837b3d353ba8c5bd316fed81f Mon Sep 17 00:00:00 2001
From: coder <coder@tphai.com>
Date: Fri, 10 Apr 2026 18:29:37 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=A6=96=E9=A1=B5=E5=8A=A0=E8=BD=BD?=
 =?UTF-8?q?=E6=97=B6=E6=A3=80=E6=9F=A5=E6=97=A5=E6=9C=9F=E5=B9=B6=E9=87=8D?=
 =?UTF-8?q?=E7=BD=AEdaily=5Fcount?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 app.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

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']