UI 自动化中截图必不可少，截图可以展现测试过程 ，也可以检查错误情况。甚至在某种情况下还可以通过截图对比来判断程序的正确性。
自动化测试过程中 ，需要截图有以下几个用途：
- 记录测试过程
- 脚本发生异常或者失败情况时截图
- 部分页面需要截图对比断言
- 获取验证码图片用于验证码识别
- 其他需要记录的情况
Selenium 中提供的截图有两种截图方式。一种是截取当前浏览器窗口，第二种是截取可见元素。
截取当前窗口
截取当前浏览器窗口的截图方法由 WebDriver 提供：
- 截图保存为文件
driver.save_screenshot(".\screenshot.png")
driver.get_screenshot_as_file(".\screenshot.png")
上面两个方法的结果一样，通过参数指定保存的文件位置，可以是相对路径也可以是绝对路径。最后截取图片并保存。
注意，必须保存为 png 格式的图片。
- 截图保存为二进制数据
png = driver.get_screenshot_as_png()
该方法返回二进制数据，保存于内存中，可以写入图片。
- 截图保存为 base64 位字符串
img = get_screenshot_as_base64()
该方法是将截图后的图像信息保存为 base64 位格式的字符串。可以直接写入 HTML 页面中。
以下例子，将图像直接写入 HTML 页面中。可以用于将图像放入测试报告。
from selenium import webdriver
driver.get("http://www.baidu.com")
img = driver.get_screenshot_as_base64()
string = """
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>图像演示</title>
</head>
<body>
<div id="img">
<img src="https://img-blog.csdnimg.cn/2022010613461584562.png"> <!-- <= 替换到这里-->
</div>
</body>
</html>
""" % img
with open('t1.html', 'w', encoding='utf-8') as f:
f.write(string)
注意：
窗口截取时，只会截取当前能看到的窗口，如果看不到的部分是不会截图的。
如果需要将未显示的元素显示到当前窗口再截取，请使用指定元素的location_once_scrolled_into_view 将元素滚动到窗口可见范围。具体参考《元素对象操作》。
截取页面元素
元素对象 WebElement 也具有上面提到的几种截图方式，只是上面针对的是整个浏览器窗口，而元素对象的截图针对的是具体的某个元素，值会截取到元素。
from selenium import webdriver
driver.get("http://www.baidu.com")
driver.find_element_by_id('su').screenshot('百度.png')
截取的图像如下：
只包含元素的截图。比如要做验证码处理的时候，可以通过此方式获取验证码图片。
比如 [showdoc] 网站的验证码：
driver.find_element_by_class_name('v_code_img').screenshot('验证码.png')
获取到的图片结果如下：
然后尝试去识别吧，😁
同样，元素对象也支持另外两种截图，返回二进制或者 base64 格式。
driver.find_element_by_id('su').screenshot_as_png() # => 保存为二进制格式
driver.find_element_by_id('su').screenshot_as_base64() # => 保存为 base64
base64： Base64 是一种用64个字符来表示任意二进制数据的方法。
用记事本打开exe、jpg、pdf这些文件时，我们都会看到一大堆乱码，因为二进制文件包含很多无法显示和打印的字符，所以，如果要让记事本这样的文本处理软件能处理二进制数据，就需要一个二进制到字符串的转换方法。Base64是一种最常见的二进制编码方法。
– 参考自廖雪峰网站《base64》
965
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
