Selenium Python自定义显式等待助手方法/函数

Question

首先感谢您的反馈

因此：

我将SeleniumWebDriver与Python一起使用（主要用于自动测试）

我有一个名为：Automation.py的Python自动化文件——例如

1-我想打开Google.com（用于解释）

2-我想等待，直到使用自定义功能显示搜索字段

目前我可以做到这一点：

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.Firefox()
driver.get("https://www.google.com/")
element = WebDriverWait(driver, 10).until(
       EC.presence_of_element_located((By.NAME, "q"))

我想创建另一个Python文件，从中可以导入自定义方法/函数来重构“wait”代码

比如：

waitfor(e, locator, 10)

其中：

e=元素

定位器=定位器（XPATH、名称、ID等）

10=等待时间

谢谢，很抱歉打扰您

Answer 1

您可以按如下方式对其进行自定义

def waitfor(locator, timeout):
    element = WebDriverWait(driver, timeout).until(EC.presence_of_element_located(locator)

您可以从另一个文件调用它，如

locator = (By.NAME, "q")
waitfor(locator,10)