testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -17,49 +17,107 @@ def setup_driver():
     chrome_options.add_argument('--disable-dev-shm-usage')
     chrome_options.add_argument('--window-size=1920,1080')
 
+    # Add user agent to avoid detection
+    chrome_options.add_argument(
+        '--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')
+
     print("Chrome options configured, initializing driver...")
 
     try:
         service = Service()
         driver = webdriver.Chrome(service=service, options=chrome_options)
+        driver.set_page_load_timeout(30)
         print("Chrome driver initialized successfully")
         return driver
     except Exception as e:
         print(f"Failed to initialize Chrome driver: {str(e)}")
         raise
 
 
+def verify_cookie(cookie_value):
+    if not cookie_value:
+        raise ValueError("Cookie value is empty")
+    print(f"Cookie length: {len(cookie_value)} characters")
+    return cookie_value
+
+
+def check_page_state(driver):
+    """Debug page state"""
+    print("\nPage debugging information:")
+    print(f"Current URL: {driver.current_url}")
+    print(f"Page title: {driver.title}")
+    print("\nFirst 500 characters of page source:")
+    print(driver.page_source[:500])
+
+    # Try to find any elements to verify page is loading
+    all_elements = driver.find_elements(By.TAG_NAME, "*")
+    print(f"\nTotal elements found on page: {len(all_elements)}")
+
+    # Try to find main element
+    main_elements = driver.find_elements(By.TAG_NAME, "main")
+    print(f"Number of <main> elements: {len(main_elements)}")
+
+    # Try to find pre elements
+    pre_elements = driver.find_elements(By.TAG_NAME, "pre")
+    print(f"Number of <pre> elements: {len(pre_elements)}")
+
+
 def take_screenshot(driver, url, cookie_value, selector, output_name):
     print(f"Attempting to take screenshot of {selector} at {url}")
     os.makedirs('screenshots', exist_ok=True)
 
     try:
-        # First navigate to the domain to set cookie
+        # Verify cookie before using
+        cookie_value = verify_cookie(cookie_value)
+
+        # First navigate to the domain root
+        print("Navigating to domain root...")
         driver.get("https://adventofcode.com")
 
         # Set the session cookie
+        print("Setting cookie...")
         driver.add_cookie({
             'name'  : 'session',
             'value' : cookie_value,
             'domain': '.adventofcode.com'
         })
 
-        print("Cookie set, navigating to target page...")
+        # Get cookies for debugging
+        cookies = driver.get_cookies()
+        print(f"Current cookies: {[cookie['name'] for cookie in cookies]}")
+
+        print(f"Navigating to target URL: {url}")
         driver.get(url)
-        time.sleep(3)  # Allow page to load
+
+        # Take full page screenshot for debugging
+        driver.save_screenshot(f'screenshots/{output_name}_full.png')
+        print("Saved full page screenshot for debugging")
+
+        # Check page state
+        check_page_state(driver)
 
         print("Waiting for element...")
-        element = WebDriverWait(driver, 10).until(
-            EC.presence_of_element_located((By.CSS_SELECTOR, selector))
-        )
+        # Try different methods to find the element
+        try:
+            # First try with explicit wait
+            element = WebDriverWait(driver, 15).until(
+                EC.presence_of_element_located((By.CSS_SELECTOR, selector))
+            )
+        except Exception as e:
+            print(f"Failed with explicit wait: {str(e)}")
+            print("Trying direct find_element...")
+            # Try direct find_element
+            element = driver.find_element(By.CSS_SELECTOR, selector)
 
         print("Element found, taking screenshot...")
+        driver.execute_script("arguments[0].scrollIntoView();", element)
+        time.sleep(2)
         element.screenshot(f'screenshots/{output_name}.png')
         print(f"Screenshot saved as screenshots/{output_name}.png")
 
     except Exception as e:
         print(f"Error during screenshot process: {str(e)}")
-        print(f"Current URL: {driver.current_url}")
+        check_page_state(driver)  # Get debug info even on failure
         raise