feat: 触发429自动禁用和解禁

src/core/database.py

@@ -193,6 +193,8 @@ class Database:
                     ("video_enabled", "BOOLEAN DEFAULT 1"),
                     ("image_concurrency", "INTEGER DEFAULT -1"),
                     ("video_concurrency", "INTEGER DEFAULT -1"),
+                    ("ban_reason", "TEXT"),  # 禁用原因
+                    ("banned_at", "TIMESTAMP"),  # 禁用时间
                 ]
 
                 for col_name, col_type in columns_to_add:
@@ -262,7 +264,9 @@ class Database:
                     image_enabled BOOLEAN DEFAULT 1,
                     video_enabled BOOLEAN DEFAULT 1,
                     image_concurrency INTEGER DEFAULT -1,
-                    video_concurrency INTEGER DEFAULT -1
+                    video_concurrency INTEGER DEFAULT -1,
+                    ban_reason TEXT,
+                    banned_at TIMESTAMP
                 )
             """)
 

src/core/models.py

@@ -38,6 +38,10 @@ class Token(BaseModel):
     image_concurrency: int = -1  # -1表示无限制
     video_concurrency: int = -1  # -1表示无限制
 
+    # 429禁用相关
+    ban_reason: Optional[str] = None  # 禁用原因: "429_rate_limit" 或 None
+    banned_at: Optional[datetime] = None  # 禁用时间
+
 
 class Project(BaseModel):
     """Project model for VideoFX"""

src/main.py

@@ -73,10 +73,24 @@ async def lifespan(app: FastAPI):
     # Start file cache cleanup task
     await generation_handler.file_cache.start_cleanup_task()
 
+    # Start 429 auto-unban task
+    import asyncio
+    async def auto_unban_task():
+        """定时任务：每小时检查并解禁429被禁用的token"""
+        while True:
+            try:
+                await asyncio.sleep(3600)  # 每小时执行一次
+                await token_manager.auto_unban_429_tokens()
+            except Exception as e:
+                print(f"❌ Auto-unban task error: {e}")
+
+    auto_unban_task_handle = asyncio.create_task(auto_unban_task())
+
     print(f"✓ Database initialized")
     print(f"✓ Total tokens: {len(tokens)}")
     print(f"✓ Cache: {'Enabled' if config.cache_enabled else 'Disabled'} (timeout: {config.cache_timeout}s)")
     print(f"✓ File cache cleanup task started")
+    print(f"✓ 429 auto-unban task started (runs every hour)")
     print(f"✓ Server running on http://{config.server_host}:{config.server_port}")
     print("=" * 60)
 
@@ -86,7 +100,14 @@ async def lifespan(app: FastAPI):
     print("Flow2API Shutting down...")
     # Stop file cache cleanup task
     await generation_handler.file_cache.stop_cleanup_task()
+    # Stop auto-unban task
+    auto_unban_task_handle.cancel()
+    try:
+        await auto_unban_task_handle
+    except asyncio.CancelledError:
+        pass
     print("✓ File cache cleanup task stopped")
+    print("✓ 429 auto-unban task stopped")
 
 
 # Initialize components

src/services/generation_handler.py

@@ -358,7 +358,12 @@ class GenerationHandler:
             if stream:
                 yield self._create_stream_chunk(f"❌ {error_msg}\n")
             if token:
-                await self.token_manager.record_error(token.id)
+                # 检测429错误，立即禁用token
+                if "429" in str(e) or "HTTP Error 429" in str(e):
+                    debug_logger.log_warning(f"[429_BAN] Token {token.id} 遇到429错误，立即禁用")
+                    await self.token_manager.ban_token_for_429(token.id)
+                else:
+                    await self.token_manager.record_error(token.id)
             yield self._create_error_response(error_msg)
 
             # 记录失败日志

src/services/token_manager.py

@@ -179,7 +179,10 @@ class TokenManager:
         image_concurrency: Optional[int] = None,
         video_concurrency: Optional[int] = None
     ):
-        """Update token (支持修改project_id和project_name)"""
+        """Update token (支持修改project_id和project_name)
+
+        当用户编辑保存token时，如果token未过期，自动清空429禁用状态
+        """
         update_fields = {}
 
         if st is not None:
@@ -203,6 +206,25 @@ class TokenManager:
         if video_concurrency is not None:
             update_fields["video_concurrency"] = video_concurrency
 
+        # 检查token是否因429被禁用，如果是且未过期，则清空429状态
+        token = await self.db.get_token(token_id)
+        if token and token.ban_reason == "429_rate_limit":
+            # 检查token是否过期
+            is_expired = False
+            if token.at_expires:
+                now = datetime.now(timezone.utc)
+                if token.at_expires.tzinfo is None:
+                    at_expires_aware = token.at_expires.replace(tzinfo=timezone.utc)
+                else:
+                    at_expires_aware = token.at_expires
+                is_expired = at_expires_aware <= now
+
+            # 如果未过期，清空429禁用状态
+            if not is_expired:
+                debug_logger.log_info(f"[UPDATE_TOKEN] Token {token_id} 编辑保存，清空429禁用状态")
+                update_fields["ban_reason"] = None
+                update_fields["banned_at"] = None
+
         if update_fields:
             await self.db.update_token(token_id, **update_fields)
 
@@ -377,6 +399,79 @@ class TokenManager:
         """
         await self.db.reset_error_count(token_id)
 
+    async def ban_token_for_429(self, token_id: int):
+        """因429错误立即禁用token
+
+        Args:
+            token_id: Token ID
+        """
+        debug_logger.log_warning(f"[429_BAN] 禁用Token {token_id} (原因: 429 Rate Limit)")
+        await self.db.update_token(
+            token_id,
+            is_active=False,
+            ban_reason="429_rate_limit",
+            banned_at=datetime.now(timezone.utc)
+        )
+
+    async def auto_unban_429_tokens(self):
+        """自动解禁因429被禁用的token
+
+        规则:
+        - 距离禁用时间12小时后自动解禁
+        - 仅解禁未过期的token
+        - 仅解禁因429被禁用的token
+        """
+        all_tokens = await self.db.get_all_tokens()
+        now = datetime.now(timezone.utc)
+
+        for token in all_tokens:
+            # 跳过非429禁用的token
+            if token.ban_reason != "429_rate_limit":
+                continue
+
+            # 跳过未禁用的token
+            if token.is_active:
+                continue
+
+            # 跳过没有禁用时间的token
+            if not token.banned_at:
+                continue
+
+            # 检查token是否已过期
+            if token.at_expires:
+                # 确保时区一致
+                if token.at_expires.tzinfo is None:
+                    at_expires_aware = token.at_expires.replace(tzinfo=timezone.utc)
+                else:
+                    at_expires_aware = token.at_expires
+
+                # 如果已过期，跳过
+                if at_expires_aware <= now:
+                    debug_logger.log_info(f"[AUTO_UNBAN] Token {token.id} 已过期，跳过解禁")
+                    continue
+
+            # 确保banned_at时区一致
+            if token.banned_at.tzinfo is None:
+                banned_at_aware = token.banned_at.replace(tzinfo=timezone.utc)
+            else:
+                banned_at_aware = token.banned_at
+
+            # 检查是否已过12小时
+            time_since_ban = now - banned_at_aware
+            if time_since_ban.total_seconds() >= 12 * 3600:  # 12小时
+                debug_logger.log_info(
+                    f"[AUTO_UNBAN] 解禁Token {token.id} (禁用时间: {banned_at_aware}, "
+                    f"已过 {time_since_ban.total_seconds() / 3600:.1f} 小时)"
+                )
+                await self.db.update_token(
+                    token.id,
+                    is_active=True,
+                    ban_reason=None,
+                    banned_at=None
+                )
+                # 重置错误计数
+                await self.db.reset_error_count(token.id)
+
     # ========== 余额刷新 ==========
 
     async def refresh_credits(self, token_id: int) -> int: