testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -1,153 +1,119 @@
 import os
 import time
+
 from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
 from selenium.webdriver.common.by import By
-from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
-from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.support.ui import WebDriverWait
 
 
-def setup_driver():
-    print("Setting up Chrome driver...")
+def setup_driver(cookie_value):
+    print('Setting up Chrome driver...')
     chrome_options = Options()
 
+    # Basic Chrome options
     chrome_options.add_argument('--headless=new')
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
     chrome_options.add_argument('--window-size=1920,1080')
 
-    # Add user agent to avoid detection
+    # Add headers
+    chrome_options.add_argument('--accept-encoding=gzip, deflate, br, zstd')
+    chrome_options.add_argument('--accept-language=en,cs;q=0.9')
     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')
+        '--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...")
+    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])
+        # Set up cookie before any navigation
+        print('Setting up cookie...')
+        driver.get('https://adventofcode.com/robots.txt')
+        driver.delete_all_cookies()
 
-    # 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)}")
+        driver.add_cookie(
+            {
+                'name': 'session',
+                'value': cookie_value,
+            }
+        )
 
-    # Try to find main element
-    main_elements = driver.find_elements(By.TAG_NAME, "main")
-    print(f"Number of <main> elements: {len(main_elements)}")
+        # Verify cookie
+        cookies = driver.get_cookies()
+        print(f"Cookies after setting: {[f"{c['name']}={c['domain']}" for c in cookies]}")
 
-    # Try to find pre elements
-    pre_elements = driver.find_elements(By.TAG_NAME, "pre")
-    print(f"Number of <pre> elements: {len(pre_elements)}")
+        print('Chrome driver initialized successfully')
+        return driver
+    except Exception as e:
+        print(f'Failed to initialize Chrome driver: {e!s}')
+        raise
 
 
-def take_screenshot(driver, url, cookie_value, selector, output_name):
-    print(f"Attempting to take screenshot of {selector} at {url}")
+def take_screenshot(driver, url, selector, output_name):
+    print(f'Attempting to take screenshot of {selector} at {url}')
     os.makedirs('screenshots', exist_ok=True)
 
     try:
-        # 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'
-        })
-
-        # 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}")
+        print(f'Navigating to target URL: {url}')
         driver.get(url)
+        time.sleep(3)  # Give the page time 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...")
-        # 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)
+        print('Saved full page screenshot for debugging')
+
+        print('Waiting for element...')
+        element = WebDriverWait(driver, 15).until(
+            EC.presence_of_element_located((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")
+        print(f'Screenshot saved as screenshots/{output_name}.png')
 
     except Exception as e:
-        print(f"Error during screenshot process: {str(e)}")
-        check_page_state(driver)  # Get debug info even on failure
+        print(f'Error during screenshot process: {e!s}')
+        print(f'Current URL: {driver.current_url}')
+        print('Page source:')
+        print(driver.page_source[:1000])  # Print first 1000 characters of source
         raise
 
 
 def main():
-    print("Starting screenshot process...")
+    print('Starting screenshot process...')
     driver = None
 
     # Get cookie from environment variable
     cookie = os.getenv('COOKIE')
     if not cookie:
-        raise ValueError("COOKIE environment variable not set")
+        raise ValueError('COOKIE environment variable not set')
 
     try:
-        driver = setup_driver()
+        driver = setup_driver(cookie)
         take_screenshot(
             driver=driver,
             url='https://adventofcode.com/2024',
-            cookie_value=cookie,
             selector='body > main > pre',
-            output_name='aoc-content'
+            output_name='aoc-content',
         )
 
     except Exception as e:
-        print(f"Fatal error in main: {str(e)}")
+        print(f'Fatal error in main: {e!s}')
         raise
     finally:
         if driver:
-            print("Cleaning up driver...")
+            print('Cleaning up driver...')
             driver.quit()
 
 
-if __name__ == "__main__":
-    main()
+if __name__ == '__main__':
+    main()