testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -12,17 +12,10 @@ def setup_driver():
     print("Setting up Chrome driver...")
     chrome_options = Options()
 
-    # Required arguments for running in GitHub Actions
-    chrome_options.add_argument('--headless=new')  # Updated headless argument
+    chrome_options.add_argument('--headless=new')
     chrome_options.add_argument('--no-sandbox')
     chrome_options.add_argument('--disable-dev-shm-usage')
-    chrome_options.add_argument('--disable-gpu')
     chrome_options.add_argument('--window-size=1920,1080')
-    chrome_options.add_argument('--disable-extensions')
-
-    # Additional debugging options
-    chrome_options.add_argument('--verbose')
-    chrome_options.add_argument('--log-level=0')
 
     print("Chrome options configured, initializing driver...")
 
@@ -36,64 +29,57 @@ def setup_driver():
         raise
 
 
-def take_element_screenshot(driver, url, selector, output_name):
+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:
-        print("Navigating to page...")
-        driver.get(url)
+        # First navigate to the domain to set cookie
+        driver.get("https://adventofcode.com")
 
-        # Add a small delay to ensure page loads
-        time.sleep(5)
-        print("Page loaded, waiting for element...")
+        # Set the session cookie
+        driver.add_cookie({
+            'name'  : 'session',
+            'value' : cookie_value,
+            'domain': '.adventofcode.com'
+        })
+
+        print("Cookie set, navigating to target page...")
+        driver.get(url)
+        time.sleep(3)  # Allow page to load
 
-        # Wait for element to be present and visible
-        element = WebDriverWait(driver, 20).until(
+        print("Waiting for element...")
+        element = WebDriverWait(driver, 10).until(
             EC.presence_of_element_located((By.CSS_SELECTOR, selector))
         )
 
-        print("Element found, preparing for screenshot...")
-
-        # Ensure element is visible in viewport
-        driver.execute_script("arguments[0].scrollIntoView();", element)
-        time.sleep(2)  # Give time for any animations to complete
-
-        # Take full page screenshot first for debugging
-        driver.save_screenshot('screenshots/full_page.png')
-        print("Full page screenshot saved for debugging")
-
-        # Take screenshot of specific element
+        print("Element found, taking screenshot...")
         element.screenshot(f'screenshots/{output_name}.png')
-        print(f"Element 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)}")
         print(f"Current URL: {driver.current_url}")
-        print(f"Page source length: {len(driver.page_source)}")
         raise
 
 
 def main():
     print("Starting screenshot process...")
     driver = None
-    try:
-        driver = setup_driver()
 
-        # Test with a simple, reliable page first
-        take_element_screenshot(
-            driver=driver,
-            url='https://example.com',  # Using example.com as a test
-            selector='h1',  # Example.com has a simple h1 element
-            output_name='test-screenshot'
-        )
+    # Get cookie from environment variable
+    cookie = os.getenv('COOKIE')
+    if not cookie:
+        raise ValueError("COOKIE environment variable not set")
 
-        # If the above works, try your actual target
-        take_element_screenshot(
+    try:
+        driver = setup_driver()
+        take_screenshot(
             driver=driver,
-            url='https://github.com',
-            selector='.header',
-            output_name='github-header'
+            url='https://adventofcode.com/2024',
+            cookie_value=cookie,
+            selector='body > main > pre',
+            output_name='aoc-content'
         )
 
     except Exception as e: