使用硒检查框的有效方法

Question

我在 python 中结合 selenium 编写了一个脚本来解开连接到三个按钮中的两个按钮的两个框。然后在这两个框中有几个字段可以选中或取消选中选项。

使用以下脚本，我可以检查所需的框是否连接到前两个按钮，但问题是我执行整个操作的方式非常不整洁。我怎样才能有效地做到这一点？

我尝试过的脚本：

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

driver = webdriver.Chrome()
driver.get("http://urbantoronto.ca/database/")
wait = WebDriverWait(driver, 10)

wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#filter_status_arrow"))).click()
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#check_pre_construction"))).click()
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#check_complete"))).click()

wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#filter_project_type_arrow"))).click()
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#check_condo"))).click()
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#check_co_op"))).click()
wait.until(EC.presence_of_element_located((By.CSS_SELECTOR, "#check_house"))).click()

driver.quit()

默认the buttons look like this。

当我选择所需的字段时，他们look like this。

我怎样才能有效地做到这一点？

Answer 1

仍然不确定您所说的“高效方式”是什么意思。

尝试以下解决方案，如果需要一些改进，请告诉我：

driver.get("http://urbantoronto.ca/database/")

# Use @id of filter as key, @ids of check-box as values
filters = {"filter_status": ["pre_construction", "under_construction", "complete"],
           "filter_project_type": ["condo", "hotel", "institutional"]}

for filter_item in filters:
    driver.find_element_by_id("%s_arrow" % filter_item).click()
    for checkbox in filters[filter_item]:
        driver.find_element_by_id(filter_item).find_element_by_id("check_%s" % checkbox).click()