Method _setup_leetcode_cookie

src/benchmark/leetcode.py:236–280

Use Playwright to login and extract cookies to bypass Cloudflare 403

(self)

        return stdout.decode().strip()
    
    async def _setup_leetcode_cookie(self):
        """Use Playwright to login and extract cookies to bypass Cloudflare 403"""
        logger.info("| 🔍 Logging in to LeetCode via browser to fetch session cookies...")
        
        try:
            if self.leetcode_cookie:
                logger.info("| 🔍 LeetCode cookie already exists, skipping login")
                return
            
            # 1. Navigate to login page
            await self.page.goto("https://leetcode.com/accounts/login/", wait_until="networkidle")
            await asyncio.sleep(2)

            # 2. Check if already logged in (due to persistent context)
            if "login" not in self.page.url:
                logger.info("| 🔍 Already logged in to LeetCode.")
            else:
                logger.info("| 🔍 Entering LeetCode credentials...")
                # LeetCode uses specific IDs for login inputs
                await self.page.fill('input[name="login"]', os.getenv("LEETCODE_USERNAME"))
                await self.page.fill('input[name="password"]', os.getenv("LEETCODE_PASSWORD"))
                
                # Click sign in and wait for navigation
                await self.page.click('button[type="submit"], #signin_btn')
                
                # Wait for navigation back to home or dashboard
                try:
                    await self.page.wait_for_url("https://leetcode.com/", timeout=20000)
                    logger.info("| ✅ LeetCode web login successful.")
                except:
                    logger.warning("| ⚠️ Login navigation timed out, checking cookies anyway...")

            # 3. Extract cookies from context
            cookies = await self.context.cookies()
            cookie_str = "; ".join([f"{c['name']}={c['value']}" for c in cookies if c['domain'].endswith("leetcode.com")])
            
            if "LEETCODE_SESSION" in cookie_str:
                self.leetcode_cookie = cookie_str
                logger.info(f"| ✅ Successfully extracted LeetCode cookies (length: {len(cookie_str)})")
            else:
                logger.error("| ❌ Failed to find LEETCODE_SESSION in cookies.")
                
        except Exception as e:
            logger.error(f"| ❌ Error during LeetCode cookie setup: {str(e)}")
            raise e

    async def _login_and_navigate(self):
        """Login GitHub and enter specified Codespace"""