selenium截屏代码

hys_guff2024-05-02 8:31

复制代码

六、截屏
   应用场景：失败截图，让错误看的更直观
   方法： 
      driver.get_screenshot_as_file(imgepath)
   参数：
      imagepath：为图片要保存的目录地址及文件名称
         如： 当前目录 ./test.png
             上一级目录 ../test.png
   扩展：
      1. 多条用例执行失败，会产生多张图片，可以采用时间戳的形式，进去区分。
      操作： 
         driver.get_screenshot_as_file("../image/%s.png"%(time.strftime("%Y_%m_%d %H_%M_%S")))
         strftime:将时间转为字符串函数
   注意：
      %Y_%m_%d %H_%M_%S：代表，年 月 日 时 分 秒

python 复制代码

# 导包
import time
from time import strftime
from selenium import webdriver
from time import sleep
# 获取浏览器驱动对象
from selenium.webdriver.common.by import By

driver = webdriver.Firefox()
# 最大化浏览器
driver.maximize_window()
# 隐式等待
driver.implicitly_wait(30)
# 打开url
url = r"C:\Users\黄永生\Desktop\软件测试\P2C电子市场\web自动化_day01_课件+笔记+资料+代码\02_其他资料\注册A.html"
driver.get(url)


"""
    目标： 截屏
    方法：
        driver.get_screenshot_as_file()
    需求：
        1. 输入用户名
        2. 截图 当前目录下 admin.png
"""
# 输入 admin
driver.find_element(By.CSS_SELECTOR,"#userA").send_keys("admin")
# 调用截图方法
# driver.get_screenshot_as_file("./admin.png")
# 存放指定目录
# driver.get_screenshot_as_file("../scripts/admin02.png")
# driver.get_screenshot_as_file("../image/admin.png")

# 动态获取文件名称 使用时间戳
# driver.get_screenshot_as_file("../image/%s.png"%(time.strftime("%Y_%m_%d %H_%M_%S")))
# driver.get_screenshot_as_file("../image/%s.jpg"%(time.strftime("%Y_%m_%d %H_%M_%S")))
driver.get_screenshot_as_file("../image/%s.jpg"%(strftime("%Y_%m_%d %H_%M_%S")))



# 暂停 2
sleep(2)
# 关闭驱动对象
driver.quit()