#coding:utf-8

 class Screen(object):

     '''

         封装的截图类，webdriver自带的get_screenshot_as_file()

         在使用过程中，注意driver参数的传递

     '''

     def __init__(self, driver):

         '''

             写一个构造函数，有一个参数driver

         '''

         self.driver = driver

     def save_screen(self):

         '''

             截图并保存在根目录下的Screenshots文件夹下,并传输参数caseName

         '''

         self.log = TestLog().getlog()

         img_time = time.strftime("%Y_%m_%d_%H_%M_%S_")

         img_path = "D:\\python\\workspace\\pythontest\\screenshot\\"

         img_name = img_path + img_time + 'screen.png'

         self.img_info = "截图路径：" + img_name

         try:

             self.driver.get_screenshot_as_file(self.img_info)

             self.log.info(self.img_info)

         except Exception as e:

             self.log.info("截图异常->", e)

             format(e)