Upload database.py

src/core/database.py

@@ -935,23 +935,32 @@ class Database:
 
     async def get_generation_config(self) -> Optional[GenerationConfig]:
         """Get generation configuration"""
-        async with aiosqlite.connect(self.db_path) as db:
-            db.row_factory = aiosqlite.Row
-            cursor = await db.execute("SELECT * FROM generation_config WHERE id = 1")
-            row = await cursor.fetchone()
-            if row:
-                return GenerationConfig(**dict(row))
-            return None
+        try:
+            async with aiosqlite.connect(self.db_path) as db:
+                db.row_factory = aiosqlite.Row
+                cursor = await db.execute("SELECT * FROM generation_config WHERE id = 1")
+                row = await cursor.fetchone()
+                if row:
+                    return GenerationConfig(**dict(row))
+                return None
+        except Exception as e:
+            print(f"Error getting generation config: {e}")
+            # 返回默认配置，避免异常传播
+            return GenerationConfig(image_timeout=300, video_timeout=1500)
 
     async def update_generation_config(self, image_timeout: int, video_timeout: int):
         """Update generation configuration"""
-        async with aiosqlite.connect(self.db_path) as db:
-            await db.execute("""
-                UPDATE generation_config
-                SET image_timeout = ?, video_timeout = ?, updated_at = CURRENT_TIMESTAMP
-                WHERE id = 1
-            """, (image_timeout, video_timeout))
-            await db.commit()
+        try:
+            async with aiosqlite.connect(self.db_path) as db:
+                await db.execute("""
+                    UPDATE generation_config
+                    SET image_timeout = ?, video_timeout = ?, updated_at = CURRENT_TIMESTAMP
+                    WHERE id = 1
+                """, (image_timeout, video_timeout))
+                await db.commit()
+        except Exception as e:
+            print(f"Error updating generation config: {e}")
+            raise
 
     # Request log operations
     async def add_request_log(self, log: RequestLog):
@@ -1046,39 +1055,43 @@ class Database:
         - Generation config (image_timeout, video_timeout)
         - Proxy config will be handled by ProxyManager
         """
-        from .config import config
-
-        # Reload admin config
-        admin_config = await self.get_admin_config()
-        if admin_config:
-            config.set_admin_username_from_db(admin_config.username)
-            config.set_admin_password_from_db(admin_config.password)
-            config.api_key = admin_config.api_key
-
-        # Reload cache config
-        cache_config = await self.get_cache_config()
-        if cache_config:
-            config.set_cache_enabled(cache_config.cache_enabled)
-            config.set_cache_timeout(cache_config.cache_timeout)
-            config.set_cache_base_url(cache_config.cache_base_url or "")
-
-        # Reload generation config
-        generation_config = await self.get_generation_config()
-        if generation_config:
-            config.set_image_timeout(generation_config.image_timeout)
-            config.set_video_timeout(generation_config.video_timeout)
-
-        # Reload debug config
-        debug_config = await self.get_debug_config()
-        if debug_config:
-            config.set_debug_enabled(debug_config.enabled)
-
-        # Reload captcha config
-        captcha_config = await self.get_captcha_config()
-        if captcha_config:
-            config.set_captcha_method(captcha_config.captcha_method)
-            config.set_yescaptcha_api_key(captcha_config.yescaptcha_api_key)
-            config.set_yescaptcha_base_url(captcha_config.yescaptcha_base_url)
+        try:
+            from .config import config
+
+            # Reload admin config
+            admin_config = await self.get_admin_config()
+            if admin_config:
+                config.set_admin_username_from_db(admin_config.username)
+                config.set_admin_password_from_db(admin_config.password)
+                config.api_key = admin_config.api_key
+
+            # Reload cache config
+            cache_config = await self.get_cache_config()
+            if cache_config:
+                config.set_cache_enabled(cache_config.cache_enabled)
+                config.set_cache_timeout(cache_config.cache_timeout)
+                config.set_cache_base_url(cache_config.cache_base_url or "")
+
+            # Reload generation config
+            generation_config = await self.get_generation_config()
+            if generation_config:
+                config.set_image_timeout(generation_config.image_timeout)
+                config.set_video_timeout(generation_config.video_timeout)
+
+            # Reload debug config
+            debug_config = await self.get_debug_config()
+            if debug_config:
+                config.set_debug_enabled(debug_config.enabled)
+
+            # Reload captcha config
+            captcha_config = await self.get_captcha_config()
+            if captcha_config:
+                config.set_captcha_method(captcha_config.captcha_method)
+                config.set_yescaptcha_api_key(captcha_config.yescaptcha_api_key)
+                config.set_yescaptcha_base_url(captcha_config.yescaptcha_base_url)
+        except Exception as e:
+            print(f"Error reloading config to memory: {e}")
+            # 不要让异常传播，避免返回 500 错误
 
     # Cache config operations
     async def get_cache_config(self) -> CacheConfig: