Upload main.py

src/main.py

@@ -75,14 +75,56 @@ async def lifespan(app: FastAPI):
     # Initialize browser captcha service if needed
     browser_service = None
     if captcha_config.captcha_method == "personal":
-        from .services.browser_captcha_personal import BrowserCaptchaService
-        browser_service = await BrowserCaptchaService.get_instance(db)
-        await browser_service.open_login_window()
-        print("✓ Browser captcha service initialized (webui mode)")
+        try:
+            from .services.browser_captcha_personal import PLAYWRIGHT_AVAILABLE
+            if PLAYWRIGHT_AVAILABLE:
+                from .services.browser_captcha_personal import BrowserCaptchaService
+                browser_service = await BrowserCaptchaService.get_instance(db)
+                await browser_service.open_login_window()
+                print("✓ Browser captcha service initialized (webui mode)")
+            else:
+                print("⚠️ Playwright not available. Please use YesCaptcha service instead.")
+                # 自动切换到 yescaptcha 方法
+                await db.update_captcha_config(
+                    captcha_method="yescaptcha",
+                    yescaptcha_api_key=captcha_config.yescaptcha_api_key,
+                    yescaptcha_base_url=captcha_config.yescaptcha_base_url
+                )
+                print("✓ Captcha method automatically switched to yescaptcha")
+        except ImportError:
+            print("⚠️ Playwright not available. Please use YesCaptcha service instead.")
+            # 自动切换到 yescaptcha 方法
+            await db.update_captcha_config(
+                captcha_method="yescaptcha",
+                yescaptcha_api_key=captcha_config.yescaptcha_api_key,
+                yescaptcha_base_url=captcha_config.yescaptcha_base_url
+            )
+            print("✓ Captcha method automatically switched to yescaptcha")
     elif captcha_config.captcha_method == "browser":
-        from .services.browser_captcha import BrowserCaptchaService
-        browser_service = await BrowserCaptchaService.get_instance(db)
-        print("✓ Browser captcha service initialized (headless mode)")
+        try:
+            from .services.browser_captcha import PLAYWRIGHT_AVAILABLE
+            if PLAYWRIGHT_AVAILABLE:
+                from .services.browser_captcha import BrowserCaptchaService
+                browser_service = await BrowserCaptchaService.get_instance(db)
+                print("✓ Browser captcha service initialized (headless mode)")
+            else:
+                print("⚠️ Playwright not available. Please use YesCaptcha service instead.")
+                # 自动切换到 yescaptcha 方法
+                await db.update_captcha_config(
+                    captcha_method="yescaptcha",
+                    yescaptcha_api_key=captcha_config.yescaptcha_api_key,
+                    yescaptcha_base_url=captcha_config.yescaptcha_base_url
+                )
+                print("✓ Captcha method automatically switched to yescaptcha")
+        except ImportError:
+            print("⚠️ Playwright not available. Please use YesCaptcha service instead.")
+            # 自动切换到 yescaptcha 方法
+            await db.update_captcha_config(
+                captcha_method="yescaptcha",
+                yescaptcha_api_key=captcha_config.yescaptcha_api_key,
+                yescaptcha_base_url=captcha_config.yescaptcha_base_url
+            )
+            print("✓ Captcha method automatically switched to yescaptcha")
 
     # Initialize concurrency manager
     tokens = await token_manager.get_all_tokens()