selenium 截图加上时间戳

思路:

1  新建screenshot文件夹,不存在则创建该目录

2  在screenshot文件夹下新建当日日期文件夹,比如20190110;不存在则创建该目录

3  截图保存到当日文件夹,且截图文件名加上时间戳。 避免用例执行多次,截图被覆盖。

 

 

import os
from datetime import datetime


def getscreenshot(driver, filename="页面截图"): 
    """带有时间戳的截图"""
    screenshot_dir = './screenshot'  # 当前目录下的screenshot文件夹;或设置其他目录
    if not os.path.exists(screenshot_dir):  # 不存在则创建该目录
        os.mkdir(screenshot_dir)

    nowdate = datetime.now().strftime('%Y%m%d')  # 当日日期
    screenshot_today_dir = os.path.join(screenshot_dir, nowdate)  # 当前日期文件夹
    if not os.path.exists(screenshot_today_dir):
        os.mkdir(screenshot_today_dir)  # 不存在则创建

    nowtime = datetime.now().strftime('%H%M%S%f')  # 时间戳
    filename = nowtime + filename + ".png"  # 拼接文件名 时间戳+文件名+.png
    filepath = os.path.join(screenshot_today_dir, filename)
driver.get_screenshot_as_file(filepath)
# 截图,文件名=filename+时间戳

 

 

调用:

from selenium import webdriver

driver = webdriver.Chrome()
getscreenshot(driver)
# getscreenshot(driver,'走势图异常')

 

 

运行后:

 

 

the end!

 

posted @ 2019-01-10 20:42  dinghanhua  阅读(1122)  评论(0编辑  收藏  举报