Update main.py

main.py

@@ -16,6 +16,7 @@ import os
 import re
 import threading
 from DrissionPage import ChromiumPage, ChromiumOptions
+from DrissionPage.common import Settings
 import logging
 from dotenv import load_dotenv
 
@@ -29,6 +30,10 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+# 全局配置 DrissionPage 设置
+Settings.set_browser_connect_timeout(60)  # 增加连接超时时间
+Settings.set_language('en')  # 设置错误信息为英文，方便调试
+
 # 修改全局数据存储
 global_data = {
     "cookie": None,
@@ -59,10 +64,40 @@ def get_cookie():
     try:
         logger.info("Starting cookie retrieval process...")
         
-        # 使用全局配置创建页面
-        # DrissionPage 会自动使用 /usr/local/lib/python3.10/site-packages/DrissionPage/_configs/configs.ini 中的配置
-        logger.info("Creating ChromiumPage instance with global config...")
-        page = ChromiumPage()
+        # 创建配置对象
+        options = ChromiumOptions()
+        
+        # 设置浏览器路径
+        chrome_path = os.getenv('CHROME_PATH', '/usr/bin/google-chrome-stable')
+        logger.info(f"Using Chrome path: {chrome_path}")
+        options.set_browser_path(chrome_path)
+        
+        # 设置用户数据目录
+        user_data_dir = os.getenv('CHROME_USER_DATA_DIR', '/tmp/chrome-data')
+        logger.info(f"Using user data directory: {user_data_dir}")
+        options.set_user_data_dir(user_data_dir)  # 使用正确的方法设置用户数据目录
+        
+        # 设置无头模式和其他参数
+        options.set_headless(True)  # 使用 DrissionPage 提供的方法设置无头模式
+        options.set_argument('--no-sandbox')
+        options.set_argument('--disable-dev-shm-usage')
+        options.set_argument('--disable-gpu')
+        options.set_argument('--disable-software-rasterizer')
+        options.set_argument('--disable-extensions')
+        options.set_argument('--disable-setuid-sandbox')
+        options.set_argument('--no-first-run')
+        options.set_argument('--no-zygote')
+        options.set_argument('--single-process')
+        options.set_argument('--window-size=1920,1080')
+        options.set_argument('--user-agent=Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36')
+        
+        # 启用详细日志
+        options.set_argument('--enable-logging')
+        options.set_argument('--v=1')
+        
+        # 使用配置对象创建页面
+        logger.info("Creating ChromiumPage instance...")
+        page = ChromiumPage(options)
         
         logger.info("Navigating to target website...")
         page.get("https://chat.akash.network/")