Procházet zdrojové kódy

bugfix: 月底 日期+1 报错 修复

jihuaqiang před 1 měsícem
rodič
revize
1252b0ee74
1 změnil soubory, kde provedl 2 přidání a 2 odebrání
  1. 2 2
      examples/content_finder/server.py

+ 2 - 2
examples/content_finder/server.py

@@ -16,7 +16,7 @@ import logging
 import os
 import sys
 import uuid
-from datetime import datetime
+from datetime import datetime, timedelta
 from decimal import Decimal, ROUND_HALF_UP
 from pathlib import Path
 from typing import Optional
@@ -278,7 +278,7 @@ async def scheduled_tick():
     if day_limit_coast is not None:
         now_local = datetime.now(SCHEDULER_TZ)
         day_start = now_local.replace(hour=0, minute=0, second=0, microsecond=0)
-        day_end = day_start.replace(day=day_start.day + 1)
+        day_end = day_start + timedelta(days=1)
 
         used_today = get_total_token_coast_between(day_start, day_end)
         if used_today >= day_limit_coast: