Update main.py

main.py

@@ -32,7 +32,6 @@ def load_cookies():
     
     converted_cookies = []
     for cookie in raw_cookies:
-        # 构建Playwright兼容的cookie格式
         new_cookie = {
             "name": cookie.get("name"),
             "value": cookie.get("value"),
@@ -40,13 +39,11 @@ def load_cookies():
             "path": cookie.get("path", "/"),
         }
         
-        # 处理过期时间：expirationDate -> expires (秒转为秒，Playwright接受Unix时间戳)
         if "expirationDate" in cookie:
             new_cookie["expires"] = cookie["expirationDate"]
         elif "expires" in cookie:
             new_cookie["expires"] = cookie["expires"]
         
-        # 处理sameSite：转换为Playwright格式
         same_site = cookie.get("sameSite", "").lower()
         if same_site in ["strict", "lax", "none"]:
             new_cookie["sameSite"] = same_site.capitalize()
@@ -55,13 +52,11 @@ def load_cookies():
         elif same_site == "unspecified" or same_site == "no_restriction":
             new_cookie["sameSite"] = "None"
         else:
-            new_cookie["sameSite"] = "Lax"  # 默认值
+            new_cookie["sameSite"] = "Lax"
         
-        # 处理secure和httpOnly
         new_cookie["secure"] = cookie.get("secure", False)
         new_cookie["httpOnly"] = cookie.get("httpOnly", False)
         
-        # sameSite=None时必须secure=True
         if new_cookie["sameSite"] == "None":
             new_cookie["secure"] = True
         
@@ -79,6 +74,36 @@ async def activate_task():
         except Exception as e:
             print(f"[Activate] Error: {e}")
 
+async def auto_click_task():
+    """自动点击 'Got it' 按钮"""
+    global page
+    if not page:
+        return
+    
+    try:
+        # 查找包含 "Got it" 文本的按钮
+        selectors = [
+            'button:has-text("Got it")',
+            'a:has-text("Got it")',
+            '[role="button"]:has-text("Got it")',
+            'input[type="button"][value="Got it"]',
+            'input[type="submit"][value="Got it"]',
+        ]
+        
+        for selector in selectors:
+            try:
+                element = page.locator(selector).first
+                if await element.is_visible(timeout=500):
+                    await element.click()
+                    print(f'[AutoClick] Clicked "Got it" button')
+                    return
+            except:
+                continue
+                
+    except Exception as e:
+        # 静默处理，避免日志刷屏
+        pass
+
 async def init_browser():
     """初始化浏览器"""
     global browser, page, context, config
@@ -93,13 +118,10 @@ async def init_browser():
         user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36"
     )
     
-    # 按域名分组导入cookies
     if cookies:
-        # 过滤有效的cookies
         valid_cookies = []
         for cookie in cookies:
             try:
-                # 确保必要字段存在
                 if cookie.get("name") and cookie.get("value") and cookie.get("domain"):
                     valid_cookies.append(cookie)
             except Exception as e:
@@ -111,7 +133,6 @@ async def init_browser():
                 print(f"[Browser] Loaded {len(valid_cookies)} cookies")
             except Exception as e:
                 print(f"[Browser] Error loading cookies: {e}")
-                # 尝试逐个添加
                 success_count = 0
                 for cookie in valid_cookies:
                     try:
@@ -123,7 +144,6 @@ async def init_browser():
     
     page = await context.new_page()
     
-    # 打开目标页面
     target_link = config.get("targetLink", "https://www.example.com")
     try:
         await page.goto(target_link, wait_until="networkidle", timeout=60000)
@@ -140,19 +160,22 @@ async def close_browser():
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     """应用生命周期管理"""
-    # 启动时
     await init_browser()
     
-    # 启动定时任务
+    # 定时激活任务
     if config and config.get("activateLink"):
         interval = config.get("activateInterval", 60)
         scheduler.add_job(activate_task, "interval", seconds=interval)
-        scheduler.start()
-        print(f"[Scheduler] Started with interval: {interval}s")
+        print(f"[Scheduler] Activate task started with interval: {interval}s")
+    
+    # 自动点击 "Got it" 按钮任务
+    scheduler.add_job(auto_click_task, "interval", seconds=5)
+    print(f"[Scheduler] Auto-click task started with interval: 5s")
+    
+    scheduler.start()
     
     yield
     
-    # 关闭时
     scheduler.shutdown(wait=False)
     await close_browser()