python+selenium:利用WebDriverWait实现多条件等待

       WebDriverWait类结合其until或者not_until方法可以实现显性等待,即通过until或者not_until的method参数把外部callable的函数或者实例传给WebDriverWait,然后进行判断。该方法通常会结合expected_conditions模块使用,该模块提供了很多的判断页面元素各种状态的类以及方法,因此两者结合可以实现很多的需求。但是这里有一个问题就是,如何只是单纯的和expected_conditiond结合,只能一次实现一个给定条件的判断,如果我们有多个条件需要判断,则应该如何实现呢?

       其实很简单,只需要自己构造一个callable的类或者函数就行,我们可以把多个条件打包进行这个类的__call__函数中或者callable的函数中,然后如果满足条件,则返回True,不满足则返回False即可。如下代码所示,目的是:当网页出现了alert弹窗或者出现了“提示重新登录”的窗口,则关闭页面。这里通过构造了一个callable的类alert_or_relogin来实现这样的条件,然后再把这个类传给until()即可。这里需要注意的一点是,alert_or_relogin中对于两个条件的判断是依次分开判断的,先判断是否出现alert弹窗,如果没有再判断是否需要重新登录,这样做的原因是一旦出现alert弹窗,则页面就会被锁定,driver也就不能再获取页面其他的元素,所以此时如果再强制driver获取判断是否需要重新登录,则会报错。

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

#import time

class alert_or_relogin:
    def __call__(self,driver):
        '''用来结合webDriverWait判断是否出现alert或者需要重新登陆'''
        is_alert=bool(EC.alert_is_present()(driver))
        if is_alert:
            return True
        else:
            is_invisible=EC.invisibility_of_element_located((By.XPATH,'//*[@id="repeatcourseDialog"]/div/div[3]/a'))(driver)
            is_visible=not bool(is_invisible)
            if is_visible:
                return True
            else: return False

driver=webdriver.Chrome()
driver.get('url')
WebDriverWait(driver,timeout=60,poll_frequency=5).until(alert_or_relogin())
driver.close()

 

  • 7
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 3
    评论
在使用 PythonSelenium 库进行元素定位时,可以创建一个类来管理定位相关的操作。下面是一个示例代码: ```python 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 class ElementLocator: def __init__(self, driver): self.driver = driver def find_element_by_id(self, element_id): return self.driver.find_element(By.ID, element_id) def find_element_by_xpath(self, xpath): return self.driver.find_element(By.XPATH, xpath) def wait_for_element_visible(self, locator, timeout=10): return WebDriverWait(self.driver, timeout).until(EC.visibility_of_element_located(locator)) # 可以添加其他常用的元素定位方法,比如 find_element_by_css_selector、find_element_by_class_name 等等 # 示例用法 driver = webdriver.Chrome() element_locator = ElementLocator(driver) element = element_locator.find_element_by_id("element_id") element.click() visible_element = element_locator.wait_for_element_visible((By.XPATH, "//div[@class='my-element']")) ``` 在上面的示例代码中,我们创建了一个名为 `ElementLocator` 的类,它接收一个 Selenium WebDriver 对象作为参数,并提供了一些常用的元素定位方法,比如 `find_element_by_id` 和 `find_element_by_xpath`。此外,还添加了一个 `wait_for_element_visible` 方法,用于等待元素可见。 你可以根据自己的需要扩展 `ElementLocator` 类,添加其他常用的元素定位方法,以便于在测试或爬虫等场景中使用。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值