简单学习Selenium的截图功能--python

之前写过java的截图功能，现在学习python，补充一个。

'''
自定义一个截图类；
初始化时需要传入一个webdriver；
方法screen_shot需要传入一个截图的名称，比如登录模块的截图，就传一个“登录”咯；
设置一个时间参数，这样就知道是什么时候截下来的图片；
最后就是  driver.save_screenshot(pic_path)了。
'''
import time

class ScreenShot(object):

	def __init__(self, driver):
		self.driver = driver

	def screen_shot(self, pic_name):
		current_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
		pic_path = 'E:\\workspace\\screenshot\\' + pic_name + '-' + current_time + '.png'
		self.driver.save_screenshot(pic_path)


screen_shot = ScreenShot(self.driver)
screen_shot.screen_shot("testscreenshotname")