Update nb2.py

nb2.py

@@ -131,7 +131,7 @@ class MailGw:
 
         logger.error("未能获取到验证码")
         return None
-def get_cookie(driver):
+    def get_cookie(driver):
     try:
         cookie_script = """
         var cookieString = document.cookie;
@@ -313,9 +313,18 @@ class CookieGenerator:
             chrome_options.add_argument('--headless')
             chrome_options.add_argument('--no-sandbox')
             chrome_options.add_argument('--disable-dev-shm-usage')
+            chrome_options.add_argument('--disable-gpu')
+            chrome_options.add_argument('--disable-software-rasterizer')
+            chrome_options.add_argument('--disable-extensions')
+            chrome_options.add_argument('--remote-debugging-port=9222')
+            chrome_options.add_argument('--window-size=1920,1080')
             chrome_options.binary_location = os.getenv('CHROME_BIN', '/usr/bin/chromium')
             
-            service = Service(os.getenv('CHROMEDRIVER_PATH', '/usr/bin/chromedriver'))
+            service = Service(
+                os.getenv('CHROMEDRIVER_PATH', '/usr/bin/chromedriver'),
+                log_path='/tmp/chromedriver.log'
+            )
+            
             self.driver = webdriver.Chrome(service=service, options=chrome_options)
             wait = WebDriverWait(self.driver, 20)
 
@@ -329,6 +338,11 @@ class CookieGenerator:
 
         except Exception as e:
             logger.error(f"Cookie生成错误: {str(e)}")
+            if hasattr(e, 'msg'):
+                logger.error(f"错误消息: {e.msg}")
+            if os.path.exists('/tmp/chromedriver.log'):
+                with open('/tmp/chromedriver.log', 'r') as f:
+                    logger.error(f"ChromeDriver日志: {f.read()}")
         finally:
             self.running = False
             if self.driver:
@@ -371,4 +385,4 @@ async def get_cookies():
         return {"error": str(e)}
 
 if __name__ == "__main__":
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)