File size: 802 Bytes
d7078ff
f6bab1b
d7078ff
 
 
 
 
 
 
 
f6bab1b
 
d7078ff
 
 
 
 
f6bab1b
d7078ff
 
 
 
 
 
 
a633179
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from selenium.webdriver.remote.webdriver import WebDriver
from selenium.common.exceptions import TimeoutException


def wait_for_element(
    driver: WebDriver, by_selector: str, selector_value: str, timeout: int = 2
):
    from selenium.webdriver.support.wait import WebDriverWait
    from selenium.webdriver.support import expected_conditions as EC

    element = None

    try:
        element = WebDriverWait(driver, timeout).until(
            EC.presence_of_element_located((by_selector, selector_value))
        )

    except TimeoutException:
        element = None
    finally:
        return element


def find_and_click(driver: WebDriver, by_selector: str, selector_value: str):
    element = driver.find_element(by=by_selector, value=selector_value)
    if element:
        element.click()