【进阶】Python使用selenium 访问网页完成登录操作页面异常时——最全重试机制汇总

文章目录

• • • 思路一：使用添加异常重试机制 [常用]
    • 思路二：使用装饰器实现重试机制 [常用]
    • 思路三：使用更复杂的异常处理和恢复机制
    • 思路四：结合日志记录与重试
    • 思路五：使用递归实现重试机制
    • 思路六：结合外部配置文件管理重试参数
    • 思路七：使用自定义异常类来处理特定错误
    • 思路八：使用外部库如 `tenacity` 处理重试

思路一：使用添加异常重试机制 [常用]

from selenium import webdriver from selenium.webdriver.common.by import By from selenium.webdriver.common.keys import Keys from selenium.common.exceptions import NoSuchElementException, WebDriverException, TimeoutException from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC import time  def login(driver, username, password):     try:         # 访问登录页面         driver.get("https://example.com/login")          # 等待页面加载完成         WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "username")))          # 输入用户名         username_field = driver.find_element(By.ID, "username")         username_field.send_keys(username)          # 输入密码         password_field = driver.find_element(By.ID, "password")         password_field.send_keys(password)          # 点击登录按钮         login_button = driver.find_element(By.ID, "loginButton")         login_button.click()          # 等待页面加载完成并检查是否登录成功         WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "logoutButton")))          print("登录成功")         return True      except NoSuchElementException as e:         print(f"登录失败，未找到元素: {      e}")     except WebDriverException as e:         print(f"登录失败，Web驱动异常: {      e}")     except TimeoutException as e:         print(f"登录失败，操作超时: {      e}")      return False  def main():     # 设置WebDriver     driver = webdriver.Chrome()  # 根据实际使用的浏览器选择对应的驱动      username = "your_username"     password = "your_password"      max_retries = 3     for attempt in range(max_retries):         print(f"尝试登录次数: {      attempt + 1}")         if login(driver, username, password):             break         else:             print("登录失败，重试中...")             time.sleep(3)  # 等待一段时间后重试      driver.quit()  if __name__ == "__main__":     main() 

代码解释

1. 导入库: 引入必要的Selenium库、WebDriver等待机制、预期条件以及异常处理模块。
2. login函数: 实现具体的登录操作，加入等待页面加载的逻辑。
   • 使用WebDriverWait和expected_conditions来等待特定元素的出现，避免因页面加载缓慢导致的异常。
   • 捕获NoSuchElementException、WebDriverException和TimeoutException异常，并返回False表示登录失败。
3. 重试机制: 在main函数中实现重试逻辑，最多尝试登录三次，每次失败后等待3秒再重试。
4. WebDriver设置: 根据实际使用的浏览器选择对应的WebDriver（如Chrome、Firefox等）。

异常处理

• NoSuchElementException: 当无法找到指定的元素时抛出。
• WebDriverException: 当Web驱动程序出现问题时抛出。
• TimeoutException: 当等待操作超时时抛出。

通过这种方式，可以更全面地处理登录过程中可能遇到的异常，并实现自动重试机制，提高登录操作的成功率和稳定性。

思路二：使用装饰器实现重试机制 [常用]

使用装饰器可以使代码更加简洁和模块化：

from selenium import webdriver from selenium.webdriver.common.by import By from selenium.common.exceptions import NoSuchElementException, WebDriverException, TimeoutException from selenium.webdriver.support.ui import WebDriverWait from selenium.webdriver.support import expected_conditions as EC import time import functools  def retry_on_exception(max_retries, exceptions, delay=3):     def decorator(func):         @functools.wraps(func)         def wrapper(*args, **kwargs):             for attempt in range(max_retries):                 try:                     return func(*args, **kwargs)                 except exceptions as e:                     print(f"Exception: {      e}. Retrying {      attempt + 1}/{      max_retries}...")                     time.sleep(delay)             print("Max retries reached. Exiting.")             return