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

    def test_jieping(self):
        self.driver.find_element_by_id('kw').send_keys('selenium')
        self.driver.find_element_by_id('su').click()
        sleep(2)
        #现在截屏保存在当前路径
        #self.driver.save_screenshot('baidu.png')

        #截屏的名称拼接上日期
        #st=strftime("%Y-%d-%H-%M-%S",localtime(time()))
        #fime_time=st+'.png'
        #self.driver.save_screenshot(fime_time)

        #截屏保存到具体目录下
        st = strftime("%Y-%d-%H-%M-%S", localtime(time()))
        fime_time=st+'.png'
        path=os.path.abspath('screnshopt')
        file_path=path+'/'+fime_time
        self.driver.get_screenshot_as_file(file_path)

if __name__ == '__main__':

    case=TestCase()
    case.test_jieping()