33 lines
979 B
Python
33 lines
979 B
Python
|
from selenium import webdriver
|
||
|
|
from selenium.webdriver.common.by import By
|
||
|
|
|
||
|
|
|
||
|
|
def get_input_fields(
|
||
|
|
driver: webdriver.firefox.webdriver.WebDriver,
|
||
|
|
) -> tuple[list, list]:
|
||
|
|
extract_list: list[str] = ["header", "date", "header_link", "subheader"]
|
||
|
|
results: list = [None] * len(extract_list)
|
||
|
|
found_element_list = driver.find_elements(By.TAG_NAME, "input")
|
||
|
|
|
||
|
|
for i in found_element_list:
|
||
|
|
value = str(i.get_attribute("value"))
|
||
|
|
temp_str = i.get_dom_attribute("data-formengine-input-name")
|
||
|
|
if temp_str is not None:
|
||
|
|
data_lines = temp_str.split("[")
|
||
|
|
if len(data_lines) > 0:
|
||
|
|
data = data_lines[-1]
|
||
|
|
if len(data) > 1:
|
||
|
|
data = data[:-1]
|
||
|
|
else:
|
||
|
|
data = None
|
||
|
|
|
||
|
|
try:
|
||
|
|
idx = extract_list.index(data)
|
||
|
|
except ValueError:
|
||
|
|
idx = None
|
||
|
|
|
||
|
|
if idx is not None:
|
||
|
|
results[idx] = value
|
||
|
|
|
||
|
|
return extract_list, results
|