正文
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！
