如何通过Python的Selenium自动移除网页弹窗后再打印PDF？

以下是几个无需手动指定CSS、适配多数网站的实用方法，用Selenium实现：

1. 优先处理标准化Cookie弹窗 #

多数网站的Cookie同意按钮会包含固定文本，比如"Accept All"、"同意全部"、"Accept Cookies"、"接受所有Cookie"等。可以通过文本匹配尝试定位并点击：

from selenium import webdriver
from selenium.webdriver.common.by import By
import time

driver = webdriver.Chrome()
driver.get("目标网页URL")

# 尝试点击Cookie同意按钮
try:
    # 匹配多种常见文本的按钮
    accept_buttons = driver.find_elements(By.XPATH, """//button[contains(text(), 'Accept All') or contains(text(), '同意全部') or contains(text(), '接受所有Cookie') or contains(text(), 'Accept Cookies')]""")
    for btn in accept_buttons:
        if btn.is_displayed():
            btn.click()
            time.sleep(1)
            break
except Exception:
    pass  # 没有找到就跳过

2. 通用弹窗移除（基于元素特征） #

弹窗通常有position: fixed、z-index值极高的特征，可通过JavaScript遍历页面元素，识别并隐藏/移除这类元素：

# 执行JS移除固定定位的弹窗元素
remove_popups_js = """
// 获取所有fixed/absolute定位的元素
const fixedElements = document.querySelectorAll('*[style*="position: fixed"], *[style*="position:absolute"]');
fixedElements.forEach(el => {
    // 筛选高度/宽度占比大、z-index高的元素（疑似弹窗）
    const rect = el.getBoundingClientRect();
    const isLarge = rect.width > window.innerWidth * 0.5 || rect.height > window.innerHeight * 0.5;
    const zIndex = parseInt(window.getComputedStyle(el).zIndex) || 0;
    if (isLarge || zIndex > 999) {
        el.style.display = 'none';
        // 或者直接移除：el.remove();
    }
});
"""
driver.execute_script(remove_popups_js)
time.sleep(1)

3. 处理订阅类弹窗 #

订阅弹窗常包含"Close"、"×"、"关闭"等按钮，同样可以通过文本或图标定位：

# 尝试关闭订阅弹窗
try:
    # 查找关闭按钮（文本或类名匹配）
    close_buttons = driver.find_elements(By.XPATH, """//button[contains(text(), 'Close') or contains(text(), '关闭') or @aria-label='Close' or contains(@class, 'close')]""")
    for btn in close_buttons:
        if btn.is_displayed() and btn.is_enabled():
            btn.click()
            time.sleep(1)
            break
    # 匹配叉号图标
    close_icons = driver.find_elements(By.XPATH, """//*[text()='×' or text()='✕']""")
    for icon in close_icons:
        if icon.is_displayed():
            icon.click()
            time.sleep(1)
            break
except Exception:
    pass

4. 执行打印PDF操作 #

处理完弹窗后，调用Selenium的打印功能：

import base64

# 打印为PDF（Chrome为例）
pdf_options = {
    'landscape': False,
    'displayHeaderFooter': False,
    'printBackground': True,
    'preferCSSPageSize': True
}
pdf_data = driver.execute_cdp_cmd("Page.printToPDF", pdf_options)

# 保存PDF文件
with open("output.pdf", "wb") as f:
    f.write(base64.b64decode(pdf_data['data']))

driver.quit()

注意事项 #

• 以上方法覆盖了绝大多数常见弹窗，若遇到特殊网站仍可能失效，但无需手动写每个网站的CSS定位
• 可以用WebDriverWait替代time.sleep()让代码更健壮，比如等待弹窗消失：

from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

WebDriverWait(driver, 5).until(EC.invisibility_of_element_located((By.XPATH, "//*[contains(@class, 'popup')]")))

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