selenium中截屏以及按照时间格式保存到相应文件夹

#   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('https://www.baidu.com')
        self.driver.maximize_window()

    #   初步截屏
    def test_screen(self):
        self.driver.find_element_by_id('kw').send_keys('中国移动云视讯')
        self.driver.find_element_by_id('su').click()
        #   生成截图
        self.driver.save_screenshot('baidu.png')
    #   将截屏文件名称改为当前本地时间时分秒
    def test_filetime_screen(self):
        self.driver.find_element_by_id('kw').send_keys('中国移动云视讯')
        self.driver.find_element_by_id('su').click()
        #   生成截图
        st = strftime("%Y-%m-%d-%H-%M_%S",localtime(time()))
        file_time = st +'.png'
        sleep(2)
        self.driver.save_screenshot