# -*- coding:utf-8 -*-
# explain : 截图操作
import os, time
class Picture:
"""
截图操作
分为正常的操作截图truepicture
和异常的操作截图errorpicture
"""
def __init__(self, driver, path):
"""
:type driver: selenium.webdriver.remote.webdriver.WebDriver
:param driver WebDriver对象
:param path 截图存放位置
"""
self.__driver = driver
self.__path = path
def __picturelocation(self):
# path = Primary.readini.getvalue("picture", "picturelocation")
path = self.__path
if not os.path.exists(path): # 判断图片位置是否存在，不存在则先创建
os.makedirs(path)
picdatetime = time.strftime("%Y-%m-%d", time.localtime()) # 文件名为日期
# 正确图片目录
turepath = os.path.join(path, picdatetime + '/turepicture') # 正确图片位置
if not os.path.exists(turepath):
os.makedirs(turepath)
# 错误图片目录
errorpath = os.path.join(path, picdatetime + '/errorpicture') # 错误图片位置
if not os.path.exists(errorpath):
os.makedirs(errorpath)
return [turepath, errorpath]
def truepicture(self, picusername):
"""
正常操作的截图
文件的保存位置在配置文件的turepicture
:param picusername: 图片的文件名称
"""
turepath = self.__picturelocation()[0]
# 通过路径来拼接图片存放路径
self.__driver.get_screenshot_as_file(turepath + os.sep + picusername +
time.strftime("%Y-%m-%d%H%M", time.localtime()) + '.png')
def errorpicture(self, picusername):
"""
出现错误的截图操作
文件的保存位置在配置文件的errorpicture
:param picusername: 图片的文件名称
"""
# 通过路径来拼接图片存放路径
errorpath = self.__picturelocation()[1]
# 截图
self.__driver.get_screenshot_as_file(errorpath + os.sep + picusername +
time.strftime("%Y-%m-%d%H%M", time.localtime()) + '.png')
UI自动化截图操作简单样例
)
