playwright_automation.py

from typing import Callable
from playwright.sync_api import Page, sync_playwright

CONSTANT_LOCATOR_TYPES = {
    "text": "get_by_text",
    "label": "get_by_label",
    "placeholder": "get_by_placeholder",
    "alt text": "get_by_alt_text",
    "title": "get_by_title"
}

def playwright_goto(page: Page, url: str) -> None:
    page.goto(url)

def playwright_click(page: Page, type: str, text: str, input: str) -> None:
    if type:
        locator_type = _get_locator_type(type)
        element = getattr(page, locator_type)(text)
    else:
        element = page.locator(text)
    element.click()

def playwright_iframe_click(page: Page, type: str, text: str, input: str) -> None:
    iframe = page.frame_locator(input)
    if type:
        locator_type = _get_locator_type(type)
        element = getattr(iframe, locator_type)(text)
    else:
        element = iframe.locator(text)
    element.click()

def playwright_click_download(page: Page, type: str, text: str, input: str) -> str:
    with page.expect_download() as download_info:
        # Perform the action that initiates download
        if type:
            locator_type = _get_locator_type(type)
            element = getattr(page, locator_type)(text)
        else:
            element = page.locator(text)
        element.click()
    # Save the downloaded file
    download = download_info.value
    download.save_as("static/downloads/" + download.suggested_filename)
    return f"static/downloads/{download.suggested_filename}"

def playwright_fill_input(page: Page, type: str, text: str, input: str) -> None:
    if type:
        locator_type = _get_locator_type(type)
        element = getattr(page, locator_type)(text)
    else:
        element = page.locator(text)
    element.fill(input)

def playwright_tick_checkbox(page: Page, type: str, text: str, input: str) -> None:
    if type:
        locator_type = _get_locator_type(type)
        element = getattr(page, locator_type)(text)
    else:
        element = page.locator(text)
    element.check()

def playwright_select_dropdown(page: Page, type: str, text: str, input: str) -> None:
    if type:
        locator_type = _get_locator_type(type)
        element = getattr(page, locator_type)(text)
    else:
        element = page.locator(text)
    element.select_option(input)

def _get_locator_type(type) -> str:
    return CONSTANT_LOCATOR_TYPES[type]