38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
#!/usr/bin/env python3
|
|
"""Prepare the dropdown argument form screenshot."""
|
|
|
|
import time
|
|
|
|
from selenium.webdriver.common.by import By
|
|
from selenium.webdriver.support.ui import WebDriverWait
|
|
|
|
|
|
def _wait_for_body_attr(driver, attr, timeout=15):
|
|
WebDriverWait(driver, timeout).until(
|
|
lambda d: bool(d.find_element(By.TAG_NAME, "body").get_attribute(attr))
|
|
)
|
|
|
|
|
|
def _open_choice_list(driver, input_id):
|
|
combobox_input = driver.find_element(By.ID, input_id)
|
|
combobox_input.click()
|
|
WebDriverWait(driver, 15).until(
|
|
lambda d: len(
|
|
d.find_elements(
|
|
By.CSS_SELECTOR,
|
|
f"#{input_id}-listbox li, #{input_id} + input + ul li",
|
|
)
|
|
)
|
|
>= 2
|
|
or len(d.find_elements(By.CSS_SELECTOR, ".choice-combobox-list li")) >= 2
|
|
)
|
|
|
|
|
|
def run(driver):
|
|
_wait_for_body_attr(driver, "loaded-dashboard")
|
|
|
|
driver.find_element(By.CSS_SELECTOR, '[title="Print a message"]').click()
|
|
_wait_for_body_attr(driver, "loaded-argument-form")
|
|
_open_choice_list(driver, "message")
|
|
time.sleep(0.2)
|