【Python自动化测试】：屏幕截图

截屏方式一：

• driver.save_screenshot('绝对路径\文件名.png')

截屏方式二：

• driver.get_screenshot_as_file('绝对路径\文件名.png')

截屏方式三：

• driver.get_screenshot_as_png()

截屏方式四：

• driver.get_screenshot_as_base64()

示例代码：截取必应浏览器首页并保存

# 导包
from selenium import webdriver
import base64

# 创建一个谷歌浏览器对象
driver = webdriver.Chrome()

# 打开必应首页
driver.get('https://cn.bing.com')

# 截屏方式一：driver.save_screenshot('绝对路径\文件名.png')
driver.save_screenshot(r'E:\软件测试学习\test\必应首页1.png')

# 截屏方式二：driver.get_screenshot_as_file('绝对路径\文件名.png')
driver.get_screenshot_as_file(r'E:\软件测试学习\test\必应首页2.png')

# 截屏方式三：driver.get_screenshot_as_png()
# 通过driver.get_screenshot_as_png()方法得到图片的二进制数据
datas = driver.get_screenshot_as_png()
# 通过文件读写的形式，把数据存储到本地
with open(r'E:\软件测试学习\test\必应首页3', 'wb') as file:
    file.write(datas)

# 截屏方式四：driver.get_screenshot_as_base64()
# 通过driver.get_screenshot_as_base64()方法得到图片的base64格式数据
base64_datas = driver.get_screenshot_as_base64()
# 对base64格式数据进行解码操作
decode_datas = base64.b64decode(base64_datas)
# 通过文件读写的形式，把数据存储到本地
with open(r'E:\软件测试学习\test\必应首页4', 'wb') as file2:
    file2.write(decode_datas)

# 退出浏览器
driver.quit()