Selenium：正确使用ExpectedConditions与page_is_loaded函数

1. ExpectedConditions 的工作机制：不需要额外加 while 循环 #

WebDriverWait 本身就是带轮询的阻塞等待工具，你完全不需要自己套 while 循环加 sleep。

它的工作逻辑是：

• 初始化时指定超时时间（你这里设的是60秒）
• 默认每隔0.5秒就检查一次 ExpectedCondition 传入的条件（轮询间隔可以通过 poll_frequency 参数修改）
• 一旦条件满足（比如你这里的加载spinner变得不可见），立刻返回结果，程序继续执行后续代码
• 如果超时还没满足条件，直接抛出 TimeoutException

你现在的 page_is_loaded 函数里，wait.until(EC.invisibility_of_element_located(...)) 已经在做阻塞等待了，外面再套 while 循环完全是画蛇添足——不仅会重复等待，还可能因为 sleep(1) 导致不必要的延迟（比如spinner在0.3秒就消失了，你的循环还要等1秒才会再检查）。

直接调用 page_is_loaded(driver) 就够了，不用加循环。

2. 优化代码复用的几种方案 #

方案一：封装成通用等待函数，抽离可变参数 #

把加载提示的定位器（比如spinner的XPATH）作为参数传入，同一个函数可以复用在需要等待不同加载提示的场景：

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

def wait_for_loading_complete(driver, locator, timeout=60):
    """等待加载提示元素消失"""
    WebDriverWait(driver, timeout).until(EC.invisibility_of_element_located(locator))
    return True

调用时直接传定位器：

# 定义全局的加载spinner定位器
LOADING_SPINNER = (By.XPATH, 'path/to/spinner')

# 任何需要等待加载完成的地方调用
wait_for_loading_complete(driver, LOADING_SPINNER)

方案二：基于页面对象模式封装 #

如果你的项目用页面对象管理页面元素，把等待逻辑放到基页类里，所有页面类继承这个基类，实现全局复用：

class BasePage:
    def __init__(self, driver):
        self.driver = driver
        self.loading_spinner_locator = (By.XPATH, 'path/to/spinner')
    
    def wait_for_page_load(self, timeout=60):
        WebDriverWait(self.driver, timeout).until(EC.invisibility_of_element_located(self.loading_spinner_locator))

# 具体页面类继承基类
class HomePage(BasePage):
    def __init__(self, driver):
        super().__init__(driver)
        # 其他页面元素定义
    
    def click_some_button(self):
        # 点击按钮后自动等待加载完成
        self.driver.find_element(By.ID, 'button').click()
        self.wait_for_page_load()

这样每个页面类都能直接调用 wait_for_page_load()，不用重复写等待逻辑。

方案三：自定义 ExpectedCondition #

如果这个等待逻辑是全局通用的，可以自定义一个EC，直接用在 WebDriverWait 里：

class wait_for_loading_complete:
    def __init__(self, locator):
        self.locator = locator
    
    def __call__(self, driver):
        try:
            element = driver.find_element(*self.locator)
            return not element.is_displayed()
        except:
            # 元素不存在也视为加载完成
            return True

使用时直接传入自定义条件：

WebDriverWait(driver, 60).until(wait_for_loading_complete((By.XPATH, 'path/to/spinner')))

内容的提问来源于stack exchange，提问作者michaelbane