【爬虫】Selenium打开新tab页截图并关闭

如果说你曾苦过我的甜
我愿活成你的愿
愿不枉啊愿勇往啊
这盛世每一天
山河无恙烟火寻常
可是你如愿的眺望
孩子们啊安睡梦乡
像你深爱的那样
🎵 王菲《如愿》

在自动化测试和网页抓取中，Selenium WebDriver 是一个强大的工具，能够模拟用户在浏览器中的各种操作。本文将围绕使用 Selenium 打开新窗口、捕获屏幕截图以及避免截图错误这一主题进行探讨，分享一些实用技巧和最佳实践。

使用 Selenium 打开新窗口

在自动化脚本中打开新窗口是一项常见的需求。这可以用于同时对多个网页进行操作或抓取，而无需在同一个窗口中反复加载不同的URL。以下是一个如何使用Selenium打开新窗口的例子：

from selenium.webdriver import Chrome, ChromeOptionsoptions = ChromeOptions()
options.add_argument("--headless")  # 在后台运行
browser = Chrome(options=options)
browser.execute_script("window.open('https://www.example.com', 'new window')")

在上述代码中，execute_script 方法用于执行JavaScript代码，其中 window.open 函数用于打开一个新窗口。通过指定URL和窗口名称（在这个例子中是 'new window'），我们可以轻松地在新窗口中打开一个网页。

切换窗口和捕获屏幕截图
打开新窗口后，我们可能需要在原始窗口和新窗口之间切换，以便执行不同的操作。Selenium 提供了 switch_to.window 方法来实现这一点。以下是如何切换到新打开的窗口并捕获屏幕截图的示例：

browser.switch_to.window(browser.window_handles[-1])  # 切换到新窗口
browser.get("https://www.example.com")
browser.save_screenshot("screenshot.png")  # 保存屏幕截图

在这段代码中，window_handles 属性返回当前会话中所有窗口的句柄列表。通过选择列表的最后一个元素（-1），我们可以切换到最近打开的窗口。

避免截图错误

在自动化测试或网页抓取过程中，捕获屏幕截图是一个重要的步骤，它可以用于记录测试结果或保存网页的状态。然而，在尝试捕获截图时可能会遇到各种问题，比如截图不完整或超时错误。以下是一些避免这些问题的策略：

设置合适的页面尺寸：在捕获截图之前，确保浏览器窗口的尺寸足够大，以容纳整个网页。可以使用 set_window_size 方法来设置窗口大小。
等待页面完全加载：在页面的所有元素加载完成之前尝试捕获截图可能会导致截图不完整。使用 implicitly_wait 或显式等待（Explicit Waits）可以确保页面完全加载。
异常处理：在执行捕获截图的操作时，使用 try-except 语句来捕获可能发生的异常，这样即使遇到错误，脚本也不会立即中断。