DemoUI/public/models/screenshot.py

18 lines
486 B
Python

#!/usr/bin/env python
# _*_ coding:utf-8 _*_
__author__ = 'YinJia'
import os,sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
from config import setting
def insert_img(driver,file_name):
"""
截图
:param driver: 启动浏览器
:param file_name: 截图文件名
:return: 返回指定路径的截图文件
"""
file_path = setting.TEST_REPORT + "/screenshot/" + file_name
return driver.get_screenshot_as_file(file_path)