testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -1,107 +1,50 @@
-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.support.wait import WebDriverWait
-
 from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
+from PIL import Image
+import os
 
-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 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'
-    )
-
-    print('Chrome options configured, initializing driver...')
 
-    try:
-        service = Service()
-        driver = webdriver.Chrome(service=service, options=chrome_options)
-        driver.set_page_load_timeout(30)
+def setup_driver(cookie_value):
+    options = Options()
+    options.add_argument('--headless=new')
+    options.add_argument('--window-size=1920,1080')
+    driver = webdriver.Chrome(options=options)
+    driver.get('https://adventofcode.com')
+    driver.add_cookie({'name': 'session', 'value': cookie_value, 'domain': 'adventofcode.com'})
+    return driver
 
-        # Set up cookie before any navigation
-        print('Setting up cookie...')
-        driver.get("https://adventofcode.com")
-        driver.add_cookie({
-            'name': 'session',
-            'value': cookie_value.lstrip('session='),
-            'domain': 'adventofcode.com',
-        })
 
-        print('Chrome driver initialized successfully')
-        return driver
-    except Exception as e:
-        print(f'Failed to initialize Chrome driver: {e!s}')
-        raise
+def crop_image(input_path, output_path, crop_box=(0, 0, 640, 621)):
+    with Image.open(input_path) as img:
+        cropped = img.crop(crop_box)
+        cropped.save(output_path)
 
 
 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:
-        print(f'Navigating to target URL: {url}')
-        driver.get(url)
-        time.sleep(3)  # Give the page time to load
-
-        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}')
-        print(f'Current URL: {driver.current_url}')
-        print('Page source:')
-        print(driver.page_source[:1000])  # Print first 1000 characters of source
-        raise
+    driver.get(url)
+    element = WebDriverWait(driver, 10).until(
+        EC.presence_of_element_located((By.CSS_SELECTOR, selector))
+    )
+    temp_output = output_name  # Directly save the temp screenshot as the final output
+    element.screenshot(temp_output)
+    crop_image(temp_output, output_name)
 
 
 def main():
-    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')
-
+    driver = setup_driver(cookie)
     try:
-        driver = setup_driver(cookie)
         take_screenshot(
-            driver=driver,
-            url='https://adventofcode.com/2024',
-            selector='body > main > pre',
-            output_name='aoc-content',
+            driver,
+            'https://adventofcode.com/2024',
+            'body > main > pre',
+            'aoc-screenshot.png',
         )
-
-    except Exception as e:
-        print(f'Fatal error in main: {e!s}')
-        raise
     finally:
-        if driver:
-            print('Cleaning up driver...')
-            driver.quit()
+        driver.quit()
 
 
 if __name__ == '__main__':

.github/scripts/update_readme.py

@@ -35,24 +35,15 @@ jobs:
           COOKIE: ${{ secrets.COOKIE }}
         run: |
           uv run .github/scripts/take_screenshot.py
-          tree
 
       - name: Upload screenshot artifact
         uses: actions/upload-artifact@v4
         with:
           name: webpage-screenshots
           path: screenshots/
-          if-no-files-found: error  # This will help us know if screenshots weren't created
-
-      - name: Update README
-        run: |
-          uv run .github/scripts/update_readme.py
-#
-#      - name: Commit and Push Changes
-#        run: |
-#          git config user.name "GitHub Actions"
-#          git config user.email "actions@github.com"
-#          git add README.md screenshot.png
-#          git commit -m "Update screenshot in README"
-#          git push
 
+      - name: Commit and Push Changes
+        uses: stefanzweifel/git-auto-commit-action@v4
+        with:
+          commit_message: "Update screenshot in README"
+          file_pattern: screenshots/aoc-screenshot.png

.github/workflows/update_readme.yml

@@ -3,6 +3,8 @@ advent of code 2024
 
 https://adventofcode.com/2024
 
+![Advent of Code Screenshot](screenshots/aoc-screenshot.png)
+
 ### about
 
 https://adventofcode.com/2024/about