SeleniumBase项目中实现全页面截图的两种方法

在自动化测试和网页抓取过程中，我们经常需要获取整个网页的完整截图，包括那些不在当前视窗内的部分。SeleniumBase作为基于Selenium的增强框架，提供了多种实现这一需求的方法。本文将详细介绍两种有效的全页面截图技术方案。

方法一：使用CDP模式截图

Chrome DevTools Protocol(CDP)模式是SeleniumBase提供的一个强大功能，可以直接与浏览器底层交互。在这种模式下，我们可以轻松获取整个页面的截图：

from seleniumbase import SB

with SB(uc=True, test=True) as sb:
    url = "https://www.example.com/"
    sb.activate_cdp_mode(url)  # 激活CDP模式
    sb.sleep(1)  # 等待页面加载
    sb.save_screenshot("full_page.png", selector="body")

这种方法的优势在于：

1. 直接获取整个DOM结构的渲染结果
2. 不受浏览器窗口大小的限制
3. 截图包含所有滚动区域的内容

方法二：调整窗口尺寸后截图

当我们需要在普通模式或UC模式下工作，特别是需要处理某些安全验证时，可以采用调整浏览器窗口尺寸的方法：

from seleniumbase import SB

with SB(uc=True, headless=True) as sb:
    # 设置足够大的窗口尺寸以容纳整个页面
    sb.set_window_size(1200, 5000)  
    sb.uc_open_with_reconnect("example.com", 3)  # 带重连功能的打开方式
    sb.save_screenshot("full_page.png", selector="body")

这种方法的关键点：

1. 必须先设置足够大的窗口尺寸
2. 适合需要特殊验证的场景
3. 高度值需要根据页面实际内容调整

实际应用中的选择建议

在需要处理某些安全验证的场景下，推荐使用方法二。虽然需要预估页面高度，但能确保验证流程正常通过。而在不需要特殊验证的场合，CDP模式更为简洁高效。

对于高度动态的页面，可以结合两种方法：先通过UC模式完成验证，然后切换到CDP模式获取完整截图。不过需要注意会话状态的保持问题。

无论采用哪种方法，selector="body"参数都是关键，它告诉SeleniumBase捕获整个body元素的内容，而不仅仅是当前视窗部分。掌握这些技巧后，开发者可以灵活应对各种网页截图需求。