testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -4,7 +4,10 @@
 from selenium import webdriver
 from selenium.webdriver.chrome.options import Options
 from selenium.webdriver.chrome.service import Service
+from selenium.webdriver.support.wait import WebDriverWait
 
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support import expected_conditions as EC
 
 def setup_driver(cookie_value):
     print('Setting up Chrome driver...')
@@ -32,11 +35,10 @@ def setup_driver(cookie_value):
 
         # Set up cookie before any navigation
         print('Setting up cookie...')
-
-        driver.execute_cdp_cmd('Network.setExtraHTTPHeaders', {
-            'headers': {
-                'Cookie': cookie_value,
-            }
+        driver.add_cookie({
+            'name': 'session',
+            'value': cookie_value.lstrip('session='),
+            'domain': '.adventofcode.com',
         })
 
         print('Chrome driver initialized successfully')
@@ -55,20 +57,16 @@ def take_screenshot(driver, url, selector, output_name):
         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')
-
-        # 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('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')
 
     except Exception as e:
         print(f'Error during screenshot process: {e!s}')