【Selenium 屏幕截图】

文章目录

前言
[一、Selenium 中的截图是什么？](#一、Selenium 中的截图是什么？)
二、实现整个页面截图
- [1. 基本写法](#1. 基本写法)
- [2. 说明](#2. 说明)
三、按照日期时间格式保存截图
- [1. 生成日期时间字符串](#1. 生成日期时间字符串)
- [2. 将时间戳作为文件名](#2. 将时间戳作为文件名)
四、只截取页面中的某个区域
- [1. 元素截图写法](#1. 元素截图写法)
- [2. 元素截图的特点](#2. 元素截图的特点)

前言

Selenium 提供了截图相关的接口，既可以截取整个浏览器窗口，也可以对页面中的某个元素进行截图。本文将重点介绍 Selenium 中屏幕截图的使用方法，并演示如何按照"年月日时分秒"的时间格式自动保存图片文件，方便后续管理和追踪。

一、Selenium 中的截图是什么？

Selenium 的截图功能，本质上是将当前浏览器窗口或页面元素的视觉内容保存为图片文件。它常见的使用场景包括：

自动化测试失败时保存错误现场
页面回归测试时记录对比截图
UI 调试时快速查看页面渲染效果
爬虫程序中保存关键页面内容
定时任务中留存运行结果

在 Python Selenium 中，截图主要有两种常见方式：

整个页面截图：保存当前浏览器窗口的图片
元素截图：只保存某个指定元素的图片

其中，整个页面截图最常用，元素截图则更适合局部定位和问题分析。

二、实现整个页面截图

Selenium 提供了 save_screenshot() 方法，用于把当前浏览器窗口保存成图片。

1. 基本写法

python 复制代码

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

# 创建浏览器配置
options = Options()
options.add_argument("--start-maximized")

driver = webdriver.Chrome(options=options)

# 打开网页
driver.get("https://www.example.com")

# 保存截图
driver.save_screenshot("page.png")

# 关闭浏览器
driver.quit()

2. 说明

save_screenshot("page.png") 会把当前页面截图保存为 page.png。

这个方法返回一个布尔值：

True：截图保存成功
False：截图保存失败

也可以直接使用 get_screenshot_as_file()，它的作用和 save_screenshot() 很接近。

python 复制代码

success = driver.get_screenshot_as_file("page.png")
print(success)

三、按照日期时间格式保存截图

在实际项目中，如果截图文件一直叫 page.png，那么新的截图会覆盖旧文件，不方便排查历史问题。因此，更常见的做法是使用日期时间作为文件名。

例如保存成：

20260412_153012.png
20260412_153045.png

这样既能避免覆盖，又能清晰记录截图生成时间。

1. 生成日期时间字符串

Python 的 datetime 模块非常适合生成时间戳。

python 复制代码

from datetime import datetime

now_str = datetime.now().strftime("%Y%m%d_%H%M%S")
print(now_str)

输出示例：

python 复制代码

20260412_153012

格式说明：

%Y：四位年份
%m：两位月份
%d：两位日期
%H：24小时制小时
%M：分钟
%S：秒

2. 将时间戳作为文件名

python 复制代码

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from datetime import datetime
import os

options = Options()
options.add_argument("--start-maximized")

driver = webdriver.Chrome(options=options)
driver.get("https://www.example.com")

# 创建保存目录
save_dir = "screenshots"
os.makedirs(save_dir, exist_ok=True)

# 生成时间戳文件名
filename = datetime.now().strftime("%Y%m%d_%H%M%S") + ".png"
filepath = os.path.join(save_dir, filename)

# 保存截图
driver.save_screenshot(filepath)

print(f"截图已保存：{filepath}")

driver.quit()

这段代码做了三件事：

创建截图目录 screenshots
使用当前时间生成文件名
将截图保存到指定目录中

四、只截取页面中的某个区域

除了整个页面截图，Selenium 还支持对单个元素截图。这在定位页面局部问题时特别有用，比如只想保存某个按钮、表单、提示框或错误信息。

1. 元素截图写法

python 复制代码

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options

options = Options()
options.add_argument("--start-maximized")

driver = webdriver.Chrome(options=options)
driver.get("https://www.example.com")

element = driver.find_element(By.TAG_NAME, "h1")
element.screenshot("title.png")

driver.quit()

2. 元素截图的特点

只保存目标元素区域
更适合局部问题定位
文件大小通常比整页截图更小
可用于 UI 自动化中的控件核验