Upload admin.py

src/api/admin.py

@@ -519,8 +519,13 @@ async def update_proxy_config_alias(
     token: str = Depends(verify_admin_token)
 ):
     """Update proxy configuration (alias for frontend compatibility)"""
-    await proxy_manager.update_proxy_config(request.proxy_enabled, request.proxy_url)
-    return {"success": True, "message": "代理配置更新成功"}
+    try:
+        await proxy_manager.update_proxy_config(request.proxy_enabled, request.proxy_url)
+        return {"success": True, "message": "代理配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating proxy config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.post("/api/config/proxy")
@@ -529,8 +534,13 @@ async def update_proxy_config(
     token: str = Depends(verify_admin_token)
 ):
     """Update proxy configuration"""
-    await proxy_manager.update_proxy_config(request.proxy_enabled, request.proxy_url)
-    return {"success": True, "message": "代理配置更新成功"}
+    try:
+        await proxy_manager.update_proxy_config(request.proxy_enabled, request.proxy_url)
+        return {"success": True, "message": "代理配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating proxy config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.get("/api/config/generation")
@@ -682,10 +692,18 @@ async def update_admin_config(
     token: str = Depends(verify_admin_token)
 ):
     """Update admin configuration (error_ban_threshold)"""
-    # Update error_ban_threshold in database
-    await db.update_admin_config(error_ban_threshold=request.error_ban_threshold)
+    try:
+        # Update error_ban_threshold in database
+        await db.update_admin_config(error_ban_threshold=request.error_ban_threshold)
 
-    return {"success": True, "message": "配置更新成功"}
+        # 🔥 Hot reload: sync database config to memory
+        await db.reload_config_to_memory()
+
+        return {"success": True, "message": "配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating admin config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.post("/api/admin/password")
@@ -741,12 +759,17 @@ async def update_generation_timeout(
     token: str = Depends(verify_admin_token)
 ):
     """Update generation timeout configuration"""
-    await db.update_generation_config(request.image_timeout, request.video_timeout)
+    try:
+        await db.update_generation_config(request.image_timeout, request.video_timeout)
 
-    # 🔥 Hot reload: sync database config to memory
-    await db.reload_config_to_memory()
+        # 🔥 Hot reload: sync database config to memory
+        await db.reload_config_to_memory()
 
-    return {"success": True, "message": "生成配置更新成功"}
+        return {"success": True, "message": "生成配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating generation timeout: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 # ========== AT Auto Refresh Config ==========
@@ -815,16 +838,21 @@ async def update_cache_config_full(
     token: str = Depends(verify_admin_token)
 ):
     """Update complete cache configuration"""
-    enabled = request.get("enabled")
-    timeout = request.get("timeout")
-    base_url = request.get("base_url")
+    try:
+        enabled = request.get("enabled")
+        timeout = request.get("timeout")
+        base_url = request.get("base_url")
 
-    await db.update_cache_config(enabled=enabled, timeout=timeout, base_url=base_url)
+        await db.update_cache_config(enabled=enabled, timeout=timeout, base_url=base_url)
 
-    # 🔥 Hot reload: sync database config to memory
-    await db.reload_config_to_memory()
+        # 🔥 Hot reload: sync database config to memory
+        await db.reload_config_to_memory()
 
-    return {"success": True, "message": "缓存配置更新成功"}
+        return {"success": True, "message": "缓存配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating cache config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.post("/api/cache/base-url")
@@ -833,13 +861,18 @@ async def update_cache_base_url(
     token: str = Depends(verify_admin_token)
 ):
     """Update cache base URL"""
-    base_url = request.get("base_url", "")
-    await db.update_cache_config(base_url=base_url)
+    try:
+        base_url = request.get("base_url", "")
+        await db.update_cache_config(base_url=base_url)
 
-    # 🔥 Hot reload: sync database config to memory
-    await db.reload_config_to_memory()
+        # 🔥 Hot reload: sync database config to memory
+        await db.reload_config_to_memory()
 
-    return {"success": True, "message": "缓存Base URL更新成功"}
+        return {"success": True, "message": "缓存Base URL更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating cache base URL: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.post("/api/captcha/config")
@@ -848,32 +881,37 @@ async def update_captcha_config(
     token: str = Depends(verify_admin_token)
 ):
     """Update captcha configuration"""
-    from ..services.browser_captcha import validate_browser_proxy_url
-
-    captcha_method = request.get("captcha_method")
-    yescaptcha_api_key = request.get("yescaptcha_api_key")
-    yescaptcha_base_url = request.get("yescaptcha_base_url")
-    browser_proxy_enabled = request.get("browser_proxy_enabled", False)
-    browser_proxy_url = request.get("browser_proxy_url", "")
-
-    # 验证浏览器代理URL格式
-    if browser_proxy_enabled and browser_proxy_url:
-        is_valid, error_msg = validate_browser_proxy_url(browser_proxy_url)
-        if not is_valid:
-            return {"success": False, "message": error_msg}
-
-    await db.update_captcha_config(
-        captcha_method=captcha_method,
-        yescaptcha_api_key=yescaptcha_api_key,
-        yescaptcha_base_url=yescaptcha_base_url,
-        browser_proxy_enabled=browser_proxy_enabled,
-        browser_proxy_url=browser_proxy_url if browser_proxy_enabled else None
-    )
+    try:
+        from ..services.browser_captcha import validate_browser_proxy_url
+
+        captcha_method = request.get("captcha_method")
+        yescaptcha_api_key = request.get("yescaptcha_api_key")
+        yescaptcha_base_url = request.get("yescaptcha_base_url")
+        browser_proxy_enabled = request.get("browser_proxy_enabled", False)
+        browser_proxy_url = request.get("browser_proxy_url", "")
+
+        # 验证浏览器代理URL格式
+        if browser_proxy_enabled and browser_proxy_url:
+            is_valid, error_msg = validate_browser_proxy_url(browser_proxy_url)
+            if not is_valid:
+                return {"success": False, "message": error_msg}
+
+        await db.update_captcha_config(
+            captcha_method=captcha_method,
+            yescaptcha_api_key=yescaptcha_api_key,
+            yescaptcha_base_url=yescaptcha_base_url,
+            browser_proxy_enabled=browser_proxy_enabled,
+            browser_proxy_url=browser_proxy_url if browser_proxy_enabled else None
+        )
 
-    # 🔥 Hot reload: sync database config to memory
-    await db.reload_config_to_memory()
+        # 🔥 Hot reload: sync database config to memory
+        await db.reload_config_to_memory()
 
-    return {"success": True, "message": "验证码配置更新成功"}
+        return {"success": True, "message": "验证码配置更新成功"}
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating captcha config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.get("/api/captcha/config")
@@ -931,24 +969,29 @@ async def update_plugin_config(
     token: str = Depends(verify_admin_token)
 ):
     """Update plugin configuration"""
-    connection_token = request.get("connection_token", "")
-    auto_enable_on_update = request.get("auto_enable_on_update", True)  # 默认开启
+    try:
+        connection_token = request.get("connection_token", "")
+        auto_enable_on_update = request.get("auto_enable_on_update", True)  # 默认开启
 
-    # Generate random token if empty
-    if not connection_token:
-        connection_token = secrets.token_urlsafe(32)
+        # Generate random token if empty
+        if not connection_token:
+            connection_token = secrets.token_urlsafe(32)
 
-    await db.update_plugin_config(
-        connection_token=connection_token,
-        auto_enable_on_update=auto_enable_on_update
-    )
+        await db.update_plugin_config(
+            connection_token=connection_token,
+            auto_enable_on_update=auto_enable_on_update
+        )
 
-    return {
-        "success": True,
-        "message": "插件配置更新成功",
-        "connection_token": connection_token,
-        "auto_enable_on_update": auto_enable_on_update
-    }
+        return {
+            "success": True,
+            "message": "插件配置更新成功",
+            "connection_token": connection_token,
+            "auto_enable_on_update": auto_enable_on_update
+        }
+    except Exception as e:
+        # 捕获所有异常，确保返回有效的 JSON 响应
+        print(f"Error updating plugin config: {e}")
+        return {"success": False, "message": f"保存失败: {str(e)}"}
 
 
 @router.post("/api/plugin/update-token")