selenium之截图

下面和大家分享一下，ui自动化中截图方法。

代码如下：

def  screen_shot(self):

current_time=datatime.fromtimestamp(int(time.time()))  #获取当前时间

self.driver.sava_screenshot(os.path.join(IMG_PATH,current_time+‘png‘))    #调用sava_screenshot方法，将报错截图存放在指定目录

获取根目录py文件：

ROOT_PATH=os.path.dirname(os.path.abspath(__file__))  #获取项目路径

LOG_PATH=os.path.join(ROOT_PATH,‘LOG‘)  #创建log目录

if not  os.path.exists(LOG_PATH):

        os.mkdir(LOG_PATH)

IMG_PATH=os.path.join(LOG_PATH,‘img‘)  #创建img文件夹

if  not os.path.exists(IMG_PATH):

            os.mkdir(IMG_PATH)