python - 使用 Selenium 的预期条件 [Python]

Question

我想加载我的驱动程序，直到我的 current_url 包含“某些内容”。我尝试过以下代码:

self.url = self.driver.current_url
try:
    element = WebDriverWait(self.driver, 20).until(EC.title_contains("XXX", "YYY", "ZZZ"))
except:
    print "\n IMPERFECT URL \n"
finally:
    self.driver.quit()

但是这种方法使用标题搜索..我想检查我当前的网址是否有可能的字符串集。我怎么做？另外我想检查同一网址中的三组字符串。有人可以帮忙吗。我是 Selenium 的新手。

Answer 1

我不太明白你想要执行什么确切的测试。无论如何，您可以通过可调用并测试您想要的任何内容。下面是一个工作代码示例，用于测试当前 URL 中是否存在 google、blah 或 foo:

from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait

driver = webdriver.Chrome("/home/ldd/src/selenium/chromedriver")

driver.get("http://google.com")

def condition(driver):
    look_for = ("google", "blah", "foo")
    url = driver.current_url
    for s in look_for:
        if url.find(s) != -1:
            return True

    return False

WebDriverWait(driver, 10).until(condition)

driver.quit()

(显然，Chrome驱动程序的路径要根据自己的情况进行调整。)

一旦condition的返回值为真值，等待就结束。否则，将引发TimeoutException。如果您从 ("google", ...) 中删除 "google"，您将收到 TimeoutException。