feat: unify all system time records to Beijing Time (UTC+8)

backend/app/api.py

@@ -34,6 +34,7 @@ from .database_user import (
     register_user,
     delete_user as db_delete_user,
     sync_user_db_after_update,
+    get_beijing_time,
 )
 from .limiter import limiter
 
@@ -111,7 +112,7 @@ def _get_token_from_header(authorization: Optional[str]) -> Optional[str]:
 
 def check_vip_status(user_id: int, db: Session) -> dict:
     membership = db.get(UserMembership, user_id)
-    now = datetime.utcnow()
+    now = get_beijing_time()
     if membership is None or membership.vip_expire_at is None:
         return {"is_vip": False, "vip_expire_at": None}
 
@@ -175,7 +176,7 @@ async def register(request: Request, payload: RegisterRequest, db: Session = Dep
     session = UserSession(
         token=token,
         user_id=user.id,
-        expire_at=datetime.utcnow() + timedelta(days=SESSION_EXPIRE_DAYS)
+        expire_at=get_beijing_time() + timedelta(days=SESSION_EXPIRE_DAYS)
     )
     db.add(session)
     db.commit()
@@ -195,7 +196,7 @@ async def login(request: Request, payload: LoginRequest, db: Session = Depends(g
     session = UserSession(
         token=token,
         user_id=user.id,
-        expire_at=datetime.utcnow() + timedelta(days=SESSION_EXPIRE_DAYS)
+        expire_at=get_beijing_time() + timedelta(days=SESSION_EXPIRE_DAYS)
     )
     db.add(session)
     db.commit()

backend/app/core.py

@@ -11,6 +11,7 @@ from typing import Optional, List, Dict, Any
 from pydantic import BaseModel
 
 from .database import get_db
+from .database_user import get_beijing_time
 
 logger = logging.getLogger(__name__)
 
@@ -68,7 +69,7 @@ def get_eligible_stocks() -> List[Dict[str, Any]]:
     db = get_db()
     logger.info("Querying eligible stocks from database (first time)...")
     
-    cutoff_date = datetime.now() - timedelta(days=MIN_LISTING_YEARS * 365)
+    cutoff_date = get_beijing_time() - timedelta(days=MIN_LISTING_YEARS * 365)
     
 
     result = db.conn.execute("""

backend/app/database_user.py

@@ -17,7 +17,7 @@ from dotenv import load_dotenv
 load_dotenv()
 
 from dataclasses import dataclass
-from datetime import datetime, timedelta
+from datetime import datetime, timedelta, timezone
 from functools import wraps
 from pathlib import Path
 from typing import Optional, Callable, Any
@@ -51,13 +51,18 @@ class Base(DeclarativeBase):
     pass
 
 
+def get_beijing_time() -> datetime:
+    """获取北京时间 (UTC+8)"""
+    return datetime.now(timezone(timedelta(hours=8))).replace(tzinfo=None)
+
+
 class User(Base):
     __tablename__ = "users"
 
     id: Mapped[int] = mapped_column(Integer, primary_key=True, autoincrement=True)
     username: Mapped[str] = mapped_column(String(64), unique=True, nullable=False, index=True)
     password_hash: Mapped[str] = mapped_column(String(128), nullable=False)
-    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=get_beijing_time)
 
     @property
     def is_admin(self) -> bool:
@@ -70,7 +75,7 @@ class UserMembership(Base):
 
     user_id: Mapped[int] = mapped_column(Integer, ForeignKey("users.id"), primary_key=True)
     vip_expire_at: Mapped[Optional[datetime]] = mapped_column(DateTime, nullable=True)
-    updated_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
+    updated_at: Mapped[datetime] = mapped_column(DateTime, default=get_beijing_time, onupdate=get_beijing_time)
 
 
 class PaymentOrder(Base):
@@ -83,7 +88,7 @@ class PaymentOrder(Base):
     pay_type: Mapped[int] = mapped_column(Integer, default=1)  # 1: 支付宝, 2: 微信
     vip_duration_months: Mapped[int] = mapped_column(Integer, default=1)  # 购买月数
     status: Mapped[str] = mapped_column(String(32), default="pending", index=True)  # pending, paid, expired
-    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=get_beijing_time)
     paid_at: Mapped[Optional[datetime]] = mapped_column(DateTime, nullable=True)
     raw_payload: Mapped[Optional[str]] = mapped_column(Text, nullable=True)
 
@@ -94,7 +99,7 @@ class UserSession(Base):
     token: Mapped[str] = mapped_column(String(128), primary_key=True)
     user_id: Mapped[int] = mapped_column(Integer, ForeignKey("users.id"), nullable=False, index=True)
     expire_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
-    created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow)
+    created_at: Mapped[datetime] = mapped_column(DateTime, default=get_beijing_time)
 
 
 class DailyUsage(Base):
@@ -274,7 +279,7 @@ def create_session_token() -> str:
 def create_payment_order(db: Session, user_id: int, amount: float, pay_type: int = 1, months: int = 1) -> str:
     """创建待支付订单"""
     # 生成订单号: YYYYMMDDHHMMSS + 6位随机
-    order_id = datetime.utcnow().strftime("%Y%m%d%H%M%S") + secrets.token_hex(3)
+    order_id = get_beijing_time().strftime("%Y%m%d%H%M%S") + secrets.token_hex(3)
     
     order = PaymentOrder(
         order_id=order_id,
@@ -293,7 +298,7 @@ def create_payment_order(db: Session, user_id: int, amount: float, pay_type: int
 @sync_user_db_after_update
 def extend_vip_membership(db: Session, user_id: int, days: int = 30) -> datetime:
     membership = db.get(UserMembership, user_id)
-    now = datetime.utcnow()
+    now = get_beijing_time()
 
     if membership is None:
         membership = UserMembership(user_id=user_id, vip_expire_at=now + timedelta(days=days))
@@ -316,7 +321,7 @@ def update_order_status(db: Session, order_id: str, status: str) -> bool:
     
     order.status = status
     if status == "paid" and not order.paid_at:
-        order.paid_at = datetime.utcnow()
+        order.paid_at = get_beijing_time()
     elif status == "pending":
         order.paid_at = None
         
@@ -328,7 +333,7 @@ def get_user_by_token(db: Session, token: str) -> Optional[User]:
     session_row = db.get(UserSession, token)
     if not session_row:
         return None
-    if session_row.expire_at < datetime.utcnow():
+    if session_row.expire_at < get_beijing_time():
         db.delete(session_row)
         db.commit()
         return None
@@ -339,8 +344,8 @@ FREE_DAILY_LIMIT = int(os.getenv("FREE_DAILY_LIMIT", "3"))
 
 
 def get_daily_usage(db: Session, user_id: int) -> int:
-    """获取今日已使用次数（UTC 日期）"""
-    today = datetime.utcnow().date()
+    """获取今日已使用次数（北京时间日期）"""
+    today = get_beijing_time().date()
     row = db.get(DailyUsage, {"user_id": user_id, "use_date": today})
     return row.count if row else 0
 
@@ -348,7 +353,7 @@ def get_daily_usage(db: Session, user_id: int) -> int:
 @sync_user_db_after_update
 def increment_daily_usage(db: Session, user_id: int) -> int:
     """今日使用次数 +1，返回更新后的次数"""
-    today = datetime.utcnow().date()
+    today = get_beijing_time().date()
     row = db.get(DailyUsage, {"user_id": user_id, "use_date": today})
     if row is None:
         row = DailyUsage(user_id=user_id, use_date=today, count=1)

backend/scripts/sync_data.py

@@ -21,6 +21,7 @@ from huggingface_hub import hf_hub_download, upload_file
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 from app.database import get_db
+from app.database_user import get_beijing_time
 
 logging.basicConfig(
     level=logging.INFO,
@@ -92,7 +93,7 @@ def get_target_daily(code: str, start_date: str, market: str) -> Optional[pd.Dat
     max_retries = 3  # 增加重试次数
     for attempt in range(max_retries):
         try:
-            end_date = datetime.now().strftime('%Y%m%d')
+            end_date = get_beijing_time().strftime('%Y%m%d')
             fetch_start = start_date.replace('-', '')
             df = None
             if market == 'INDEX':
@@ -158,7 +159,7 @@ def get_last_trading_day() -> str:
         logger.warning(f"Failed to get last trading day from index data: {e}")
 
     # 回退：按工作日估算
-    d = datetime.now()
+    d = get_beijing_time()
     while d.weekday() >= 5:  # 5=周六, 6=周日
         d -= timedelta(days=1)
     return d.strftime('%Y-%m-%d')
@@ -217,7 +218,7 @@ def sync_stock_daily(targets: List[Dict[str, str]], last_trade_day: str) -> int:
             if latest_map[code] >= last_trade_day: continue
             start_dt = (pd.to_datetime(latest_map[code]) + timedelta(days=1)).strftime('%Y-%m-%d')
         else:
-            start_dt = (datetime.now() - timedelta(days=YEARS_OF_DATA * 365)).strftime('%Y-%m-%d')
+            start_dt = (get_beijing_time() - timedelta(days=YEARS_OF_DATA * 365)).strftime('%Y-%m-%d')
         t['start_dt'] = start_dt
         pending.append(t)