Selenium+Python-get_screenshot_as_file保存截图

一、保存截图

1.所谓有图有真相，在测试过程中遇到错误的时候最好截图保存下来，最后在生成测试报告的时候可以一并附上

2.常用的保存截图的方法get_screenshot_as_file()；方法中的参数为保存的路径

注意点：最好保存为png格式，保存为其他格式如：jpg，也不会报错，生成的截图也可以查看，但是运行时会有一个警告，当然警告也可忽略，不影响程序运行

二、代码

这里是登录为例，登录失败的时候截图保存下来

1.代码

# coding:utf-8
from selenium import webdriver
import time

# 初始设置
driver = webdriver.Chrome()
driver.get("http://192.168.1.9:8080/login.do?gotoURL=http%3A%2F%2F192.168.1.9%3A8080%2Fportal%2Fgeoindex.do")
driver.maximize_window()
time.sleep(1)

# 执行登录操作
driver.find_element_by_id("userAccount").clear()
driver.find_element_by_id("userAccount").send_keys("xxxxx")  # 输入登录用户名
driver.find_element_by_id("passwd").clear()
driver.find_element_by_id("passwd").send_keys("xxxx")  # 输入登录密码
driver.find_element_by_id("btnlogin").click()  # 点击登录按钮

# 获取登录失败的弹框提示，判断如果有提示，即为登录失败，截图保存；
res = driver.find_element_by_xpath("//div[@class='sTipsText']/span[2]").text
if res:
    driver.get_screenshot_as_file("C://Users/PC/PycharmProjects/ui_auto_test/img/登录失败.png")
    print("登录失败...")
else:
    print("登录成功...")

2.结果：在img目录下会保存一张名为“登录失败”的截图