selenium 屏幕截图按照日期时间格式保存图片

from selenium import webdriver
from time import sleep, strftime, localtime, time
import os

class Testcase(object):
    def __init__(self):
        self.driver = webdriver.Chrome()
        self.driver.get("http://www.baidu.com")
        self.driver.maximize_window()

    def test01(self):
        self.driver.find_element_by_id("kw").send_keys("selenium")
        self.driver.find_element_by_id("su").click()

        sleep(2)
        # 加入时间戳
        st = strftime("%Y-%m-%d-%H-%M-%s", localtime(time()))

        file_name = st + ".png"
        # 屏幕截屏
        self.driver.save_screenshot(file_name)
        path = os.getcwd()
        file_path = path + "./" + file_name
        # 给一个文件路径
        self.driver.get_screenshot_as_file(file_path)


if __name__ == '__main__':
    case = Testcase()
    case.test01()
    case.driver.quit()