Selenium Firefox：页面加载完成后执行URL条件判断代码的需求

要实现类似C# WebBrowser控件中DocumentCompleted事件的效果——每次页面（包括刷新、跳转后）完全加载完成时，根据当前URL执行不同操作，可以按以下步骤实现：

1. 配置Firefox驱动等待完全加载 #

Firefox默认的页面加载策略是normal，会等待所有资源（图片、脚本、样式表等）加载完成后才判定页面加载完成。显式声明该策略更稳妥：

from selenium import webdriver
from selenium.webdriver.firefox.options import Options

options = Options()
options.page_load_strategy = 'normal'  # 明确设置为完全加载模式
driver = webdriver.Firefox(options=options)

2. 监听页面加载完成事件（兼容刷新/跳转） #

通过注入JavaScript监听window.load事件，确保每次页面刷新或导航后都能触发后续逻辑。同时配合Selenium的显式等待，避免过早执行操作：

from selenium.webdriver.support.ui import WebDriverWait

# 注入全局加载状态监听脚本
driver.execute_script("""
window.pageLoadedFlag = false;
window.addEventListener('load', () => {
    window.pageLoadedFlag = true;
});
""")

def wait_for_full_load():
    # 等待加载完成标志为true，同时确保URL稳定（避免跳转过程中误判）
    WebDriverWait(driver, 30).until(
        lambda d: d.execute_script("return window.pageLoadedFlag")
    )
    # 重置标志，为下一次加载做准备
    driver.execute_script("window.pageLoadedFlag = false")

3. 根据URL执行分支操作 #

每次页面加载完成后，获取当前URL并执行对应逻辑：

def run_url_specific_logic():
    current_url = driver.current_url
    if "target-url-1.com" in current_url:
        # 此处替换为URL1对应的操作
        print("执行URL1的业务逻辑")
        # 示例：driver.find_element(By.CSS_SELECTOR, ".submit-btn").click()
    else:
        # 其他URL的处理逻辑
        print("执行通用/其他URL的逻辑")
        # 示例：driver.execute_script("console.log('页面加载完成')")

# 测试场景：
# 1. 首次导航
driver.get("https://target-url-1.com")
wait_for_full_load()
run_url_specific_logic()

# 2. 刷新页面
driver.refresh()
wait_for_full_load()
run_url_specific_logic()

# 3. 跳转到其他页面
driver.get("https://another-site.com")
wait_for_full_load()
run_url_specific_logic()

核心注意事项 #

• 若需更精准的等待（比如特定元素加载完成），可在wait_for_full_load中添加额外的等待条件（如EC.presence_of_element_located）。
• 避免使用implicitly_wait替代显式等待，显式等待更可控，能确保仅在页面完全加载后执行逻辑。

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