tinkering with the actions

Author: jakubdotpy3d

.github/workflows/periodic_update.yml

@@ -45,7 +45,7 @@ jobs:
         uses: astral-sh/setup-uv@v5
 
       - name: Install the project
-        run: uv sync --group actions --all-extras --dev
+        run: uv pip install -r requirements.txt
 
       - name: Install Chrome
         uses: browser-actions/setup-chrome@v1

.github/workflows/scripts/take_screenshot.py

@@ -0,0 +1,90 @@
+import os
+from pathlib import Path
+from typing import Tuple
+from PIL import Image
+from selenium import webdriver
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support import expected_conditions as EC
+from selenium.webdriver.support.ui import WebDriverWait
+
+
+def setup_driver(cookie_value: str) -> webdriver.Chrome:
+    """Set up a headless Chrome WebDriver and authenticate with a session cookie.
+
+    Args:
+        cookie_value (str): The session cookie value for authentication.
+
+    Returns:
+        webdriver.Chrome: Configured Chrome WebDriver instance.
+    """
+    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': 'CW_COOKIE', 'value': cookie_value.lstrip('CW_COOKIE='), 'domain': 'codewars.com'}
+    )
+    return driver
+
+
+def crop_image(
+    input_path: str, output_path: str, crop_box: Tuple[int, int, int, int] = (0, 0, 640, 621)
+) -> None:
+    """Crop an image to the specified dimensions and save it to the output path.
+
+    Args:
+        input_path (str): Path to the input image file.
+        output_path (str): Path to save the cropped image.
+        crop_box (Tuple[int, int, int, int], optional):
+            The cropping box defined as (left, upper, right, lower).
+            Defaults to (0, 0, 640, 621).
+    """
+    with Image.open(input_path) as img:
+        cropped = img.crop(crop_box)
+        cropped.save(output_path)
+
+
+def take_screenshot(driver: webdriver.Chrome, url: str, selector: str, output_name: str) -> None:
+    """Capture a screenshot of a web element specified by a CSS selector and crop it.
+
+    Args:
+        driver (webdriver.Chrome): The WebDriver instance used to navigate and take screenshots.
+        url (str): The URL of the web page to capture.
+        selector (str): The CSS selector of the element to capture.
+        output_name (str): Path to save the screenshot.
+    """
+    driver.get(url)
+    element = WebDriverWait(driver, 10).until(
+        EC.presence_of_element_located((By.CSS_SELECTOR, selector))
+    )
+    element.screenshot(output_name)
+    crop_image(output_name, output_name)
+
+
+def main() -> None:
+    """Main entry point for the script.
+
+    Sets up the environment, captures a screenshot, and ensures proper cleanup of resources.
+    """
+    os.makedirs('screenshots', exist_ok=True)
+    cookie = os.getenv('CW_COOKIE')
+    if not cookie:
+        raise ValueError('COOKIE environment variable is not set.')
+
+    driver = setup_driver(cookie)
+
+    try:
+        take_screenshot(
+            driver,
+            f'https://codewars.com',
+            'body > main > pre',
+            'screenshots/aoc-screenshot.png',
+        )
+    finally:
+        driver.quit()
+
+
+if __name__ == '__main__':
+    main()

requirements.txt

@@ -2,4 +2,6 @@ python-dotenv~=1.0.0
 requests-html
 lxml[html_clean]
 requests
-tqdm
+tqdm
+pillow>=11.0.0
+selenium>=4.27.1