testing actions

Author: jakubdotpy3d

.github/scripts/take_screenshot.py

@@ -1,63 +1,58 @@
+from selenium import webdriver
+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
 import os
 
-cookie = os.environ['COOKIE']
 
-print(f'cookie: {cookie:10}')
+def setup_driver():
+    chrome_options = webdriver.ChromeOptions()
+    chrome_options.add_argument('--headless')
+    chrome_options.add_argument('--no-sandbox')
+    chrome_options.add_argument('--disable-dev-shm-usage')
+
+    service = Service()
+    driver = webdriver.Chrome(service=service, options=chrome_options)
+    return driver
+
+
+def take_element_screenshot(driver, url, selector, output_name):
+    # Create screenshots directory if it doesn't exist
+    os.makedirs('screenshots', exist_ok=True)
+
+    try:
+        driver.get(url)
+
+        # Wait for element to be present and visible
+        element = WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located((By.CSS_SELECTOR, selector))
+        )
+
+        # Ensure element is visible in viewport
+        driver.execute_script("arguments[0].scrollIntoView();", element)
+
+        # Take screenshot of specific element
+        element.screenshot(f'screenshots/{output_name}.png')
+        print(f"Screenshot saved as screenshots/{output_name}.png")
+
+    except Exception as e:
+        print(f"Error taking screenshot: {str(e)}")
+
+
+def main():
+    driver = setup_driver()
+    try:
+        # Example usage - replace with your target webpage and element selector
+        take_element_screenshot(
+            driver=driver,
+            url='https://github.com',
+            selector='.header',  # Example: capturing GitHub header
+            output_name='github-header'
+        )
+    finally:
+        driver.quit()
 
-from selenium import webdriver
-from selenium.webdriver.chrome.service import Service
-from selenium.webdriver.chrome.options import Options
-from PIL import Image
-from io import BytesIO
-import os
 
-# Configure Chrome options
-options = Options()
-options.headless = True  # Run in headless mode (no GUI)
-options.add_argument("--disable-gpu")
-options.add_argument("--no-sandbox")
-options.add_argument("--disable-dev-shm-usage")
-
-# Path to ChromeDriver (ensure this is set up in your environment)
-CHROMEDRIVER_PATH = os.getenv("CHROMEDRIVER_PATH", "/usr/local/bin/chromedriver")
-
-# Initialize WebDriver
-service = Service(executable_path=CHROMEDRIVER_PATH)
-driver = webdriver.Chrome(service=service, options=options)
-
-try:
-    # Navigate to the target URL
-    TARGET_URL = os.getenv("TARGET_URL", "http://stackoverflow.com/")
-    driver.get(TARGET_URL)
-
-    # Locate the element by its ID
-    element = driver.find_element("id", "hlogo")  # Replace 'hlogo' with the actual element ID
-
-    # Get the element's location and size
-    location = element.location
-    size = element.size
-
-    # Take a screenshot of the entire page
-    png = driver.get_screenshot_as_png()
-
-    # Open the image using PIL
-    im = Image.open(BytesIO(png))
-
-    # Define crop points
-    left = location['x']
-    top = location['y']
-    right = location['x'] + size['width']
-    bottom = location['y'] + size['height']
-
-    # Crop the image to the element and save it
-    im = im.crop((left, top, right, bottom))
-    screenshot_path = "screenshot.png"
-    im.save(screenshot_path)
-    print(f"Screenshot saved to {screenshot_path}")
-
-except Exception as e:
-    print(f"An error occurred: {e}")
-finally:
-    # Quit the WebDriver
-    driver.quit()
-    print("Driver closed")
+if __name__ == "__main__":
+    main()

.github/workflows/update_readme.yml

@@ -17,7 +17,12 @@ jobs:
       - name: Install the project
         run: uv sync --group actions --all-extras --dev
 
-      - name: setup chrome driver
+      - name: Install Chrome
+        uses: browser-actions/setup-chrome@v1
+        with:
+          chrome-version: stable
+
+      - name: Install Chrome WebDriver
         uses: nanasess/setup-chromedriver@v2
 
       - name: Take Screenshot
@@ -26,6 +31,12 @@ jobs:
         run: |
           uv run .github/scripts/take_screenshot.py
 
+      - name: Upload screenshot artifact
+        uses: actions/upload-artifact@v4
+        with:
+          name: webpage-screenshots
+          path: screenshots/
+
       - name: Update README
         run: |
           uv run .github/scripts/update_readme.py
@@ -37,3 +48,4 @@ jobs:
 #          git add README.md screenshot.png
 #          git commit -m "Update screenshot in README"
 #          git push
+