viraaj.py

from selenium import webdriver
from selenium.webdriver.common.by import By

import constants
import elements


def get_driver():
    return webdriver.Chrome()


def take_screenshot(driver, css_selector, filename):
    element = driver.find_element(by=By.CSS_SELECTOR, value=css_selector)
    element.screenshot(filename)
    # driver.executeScript("arguments[0].scrollIntoView();", element)
    # File screenshot = ele.getScreenshotAs(OutputType.FILE)

def accept_if_cookies(driver):
    accept_btn = driver.find_element(by=By.CSS_SELECTOR, value=elements.COOKIES_ACCEPT)
    accept_btn.click()


def goto_contact_us():
    driver = get_driver()
    driver.get(constants.CONTACT_US)

    # https://www.selenium.dev/documentation/webdriver/elements/locators/
    # accept_if_cookies(driver)
    take_screenshot(
        driver,
        css_selector=elements.CONTACT_US_FORM,
        filename='contact_us_form.png'
    )
    driver.quit()


if __name__ == '__main__':
    goto_contact_us()