Python+Selenium+Unittest实现PO模式web自动化框架（2）

1.Common目录下的具体模块讲解。

2.basepage.py

basepage.py模块里面是封装的对元素的操作。例如：查找元素、点击元素、文本输入等等。

# --^_^-- coding:utf-8 --^_^--
# @Remark:webdriver的封装

from Common import logger
import logging
import time
import datetime
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from Common.dir_config import screenshot_dir
from selenium.webdriver.common.action_chains import ActionChains
from selenium.webdriver.support.select import Select
import win32gui
import win32con

class BasePage:
    '''
    包含了PageObjects当中，用到所有的selenium底层方法。
    还可以包含通用的一些元素操作，如alert,iframe,windows...
    还可以自己额外封装一些web相关的断言
    实现日志记录、实现失败截图
    '''
    def __init__(self,driver):
        self.driver = driver

    # 实现网页截图操作
    def save_web_screenshot(self,img_doc):
        # 页面_功能_时间.png
        now = time.strftime("%Y-%m-%d %H_%M_%S")
        filepath = "{}-{}.png".format(now,img_doc)
        try:
            self.driver.save_screenshot(screenshot_dir + "/" + filepath)
            logging.info("网页截图成功！图片存储在：{}!".format(screenshot_dir +"/" + filepath))
        except:
            logging.exception("网页截图失败！")
            raise

    # 等待元素可见
    def wait_eleVisible(self,loc,img_doc="",timeout=30,frequency=0.5):
        logging.info("等待元素{}可见！".format(loc))
        try:
            # 起始等待的时间datetime
            startTime = datetime.datetime.now()
            # 等待元素
            WebDriverWait(self.driver,timeout,frequency).until(EC.visibility_of_element_located)
            # 结束等待时间
            endTime = datetime.datetime.now()
            logging.info("等待元素成功！开始等待时间：{}，结束等待时间：{}，等待时长为：{}！".format(startTime, endTime, endTime - startTime))
        except:
            # 日志
            logging.exception("等待元素可见失败！")
            # 截图-哪一个页面哪一个操作导致的失败 + 当前时间
            self.save_web_screenshot(img_doc)
            raise

    # 查找一个元素
    def get_element(self, loc, img_doc=""):
        '''
        :param loc: 元素定位，以元组的形式。（定位类型、定位时间）
        :param img_doc:截图的说明。例如：登录页面_输入用户名
        :return:webElement对象。
        '''
        logging.info("查找{}中的元素{}".format(img_doc, loc))
        try:
            ele = self.driver.find_element(*loc)
            logging.info("查找{}元素成功！".format(ele))
            return ele
        except:
            # 日志
            logging.exception("查找元素失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 等待元素出现，点击元素
    def click_element(self, loc, img_doc,timeout=30, frequency=0.5):
        '''
        :param loc:元素定位，以元组的形式。（定位类型、定位时间）
        :param img_doc:截图的说明。例如：登录页面_输入用户名
        :param timeout:等待元素的超时上限
        :param frequency:等待元素可见时，轮询周期
        :return:
        '''
        # 等待元素可见
        self.wait_eleVisible(loc, img_doc, timeout, frequency)
        # 查找元素
        ele = self.get_element(loc, img_doc)
        # 点击操作
        logging.info("点击元素{}！".format(loc))
        try:
            ele.click()
            logging.info("点击元素{}成功！".format(ele))
        except:
            # 日志
            logging.exception("点击元素失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 文本输入
    def input_text(self, loc, img_doc, *args):
        '''
        等待元素可见，找到元素，并在元素中输入文本信息
        :param loc:元组形式的元素定位表达式
        :param img_doc:执行失败时，截图的文件命名
        :param args:输入操作中，可以输入多个值，也可以组合按键
        :return:
        '''
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找元素
        ele = self.get_element(loc, img_doc)
        # 清空输入框内容
        self.clear_data(loc,img_doc)
        # 文本输入
        logging.info("给元素{}输入文本内容：{}".format(loc, *args))
        try:
            ele.send_keys(*args)
            logging.info("文本输入成功！")
        except:
            # 日志
            logging.exception("文本输入失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 获取元素的属性值
    def get_element_attribute(self, loc, attr_name, img_doc):
        '''
        获取元素标签的内容
        :param loc:元素定位，以元组的形式。（定位类型、定位时间）
        :param attr_name:属性名称
        :param img_doc:执行失败时，截图的文件命名
        :return:
        '''
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找元素
        ele = self.get_element(loc, img_doc)
        # 获取属性
        try:
            attr_value = ele.get_attribute(attr_name)
            logging.info("获取元素{}的属性{}值为：{}！".format(loc, attr_name, attr_value))
            return attr_value
        except:
            # 日志
            logging.exception("获取元素属性失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 获取元素的文本值
    def get_element_text(self, loc, img_doc):
        '''
        :param loc:元素定位，以元组的形式。（定位类型、定位时间）
        :param img_doc:执行失败时，截图的文件命名
        :return:
        '''
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找元素
        ele = self.get_element(loc,img_doc)
        # 获取文本值
        try:
            text = ele.text
            logging.info("获取元素{}的文本值为：{}！".format(loc, text))
            return text
        except:
            # 日志
            logging.exception("获取元素文本值失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 鼠标悬浮操作
    def mouse_suspension(self,loc, img_doc):
        # 实例化ActionChains
        ac = ActionChains(self.driver)
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找元素
        ele = self.get_element(loc)
        # 鼠标悬浮
        try:
            ac.move_to_element(ele).perform()
            time.sleep(1)
            logging.info("鼠标悬浮成功！")
        except:
            # 日志
            logging.exception("鼠标悬浮失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 进入到iframe中
    def get_iframe(self,loc,img_doc):
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找元素
        ele = self.get_element(loc)
        # 进入iframe
        try:
            self.driver.switch_to.frame(ele)
            logging.info("进入iframe中成功！")
        except:
            # 日志
            logging.exception("进入到iframe中失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 通过文本选择下拉选择框
    def drop_down_select(self,loc,text,img_doc):
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找select元素
        ele = self.get_element(loc)
        # 下拉选择
        try:
            Select(ele.select_by_visible_text(text))
            logging.info("选择select成功！")
        except:
            # 日志
            logging.exception("下拉选择失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 选择非select的下拉选择框
    def drop_down_not_select(self,loc1,loc2,img_doc):
        # 先定位到下拉菜单
        # 等待元素出现
        self.wait_eleVisible(loc1, img_doc)
        # 查找下拉菜单元素
        self.click_element(loc1,"下拉菜单输入框")
        logging.info("查找下拉菜单输入框成功！")
        time.sleep(2)
        # 在对下拉菜单的选项进行选择
        try:
            # 选择下拉菜单选项
            # 等待元素出现
            self.wait_eleVisible(loc2, img_doc)
            self.click_element(loc2,"选择下拉菜单选项")
            logging.info("选择下拉菜单成功！")
        except:
            # 日志
            logging.exception("下拉选择失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 清空输入框中的内容
    def clear_data(self,loc,img_doc):
        # 等待元素出现
        self.wait_eleVisible(loc, img_doc)
        # 查找一个元素
        ele = self.get_element(loc)
        # 清空输入框
        try:
            ele.clear()
            logging.info("清空输入框成功！")
        except:
            # 日志
            logging.exception("清空输入框失败！")
            # 截图
            self.save_web_screenshot(img_doc)
            raise

    # 上传操作（前提 ：windows上传窗口已经出现。sleep1-2秒等待弹出的出现。）
    def upload(filePath, browser_type="chrome"):
        if browser_type == "chrome":
            title = "打开"
        else:
            title = ""

        # 找元素
        # 一级窗口"#32770","打开"
        dialog = win32gui.FindWindow("#32770", title)
        ComboBoxEx32 = win32gui.FindWindowEx(dialog, 0, "ComboBoxEx32", None)  # 二级
        comboBox = win32gui.FindWindowEx(ComboBoxEx32, 0, "ComboBox", None)  # 三级
        # 编辑按钮
        edit = win32gui.FindWindowEx(comboBox, 0, 'Edit', None)  # 四级
        # 打开按钮
        button = win32gui.FindWindowEx(dialog, 0, 'Button', "打开(&O)")  # 四级

        # 往编辑当中，输入文件路径.
        win32gui.SendMessage(edit, win32con.WM_SETTEXT, None, filePath)  # 发送文件路径
        win32gui.SendMessage(dialog, win32con.WM_COMMAND, 1, button)  # 点击打开按钮

3.dir_config.py

dir_config.py模块里面封装的是项目当中需要用到的路径配置，该配置是相对项目来说的相对配置，这样项目无论放在哪里都能使用而不报错。

# --^_^-- coding:utf-8 --^_^--
# @Remark:路径配置

import os

# 框架项目顶层目录
base_dir = os.path.split(os.path.split(os.path.abspath(__file__))[0])[0]

# 日志目录
logs_dir = os.path.join(base_dir, "Outputs/logs")

# 截屏目录
screenshot_dir = os.path.join(base_dir, "Outputs/screenshots")

# 测试数据目录
testdatas_dir = os.path.join(base_dir, "TestDatas")

# 测试用例目录
testcases_dir = os.path.join(base_dir, "TestCases")

# 测试报告目录
htmlreport_dir = os.path.join(base_dir, "Outputs/reports")

4.HTMLTestRunnerNew.py

HTMLTestRunnerNew.py模块是用于生成HTML的测试报告模块。（具体的模块代码太长，不方便展示。有需要可私聊。）

5.logger.py

logger.py模块是自定义的log日志模块。

# --^_^-- coding:utf-8 --^_^--
# @Remark:日志模块

import logging
from logging.handlers import TimedRotatingFileHandler
import datetime
from Common import dir_config

fmt = "%(asctime)s  %(levelname)s %(filename)s %(funcName)s [ line:%(lineno)d ] %(message)s"
datefmt = "%Y-%m-%d %H:%M:%S"
# 获取当前时间
now_time = datetime.datetime.now().strftime('%Y-%m-%d')
# 把当前时间转换成str
now_time_str = str(now_time)

handler_1 = logging.StreamHandler()
handler_2 = TimedRotatingFileHandler(dir_config.logs_dir + "/" + now_time_str + "_Web.log", when='D', interval=10,backupCount=0, encoding='utf-8')

# 设置rootlogger 的输出内容形式，输出渠道
logging.basicConfig(format=fmt, datefmt=datefmt, level=logging.INFO, handlers=[handler_1, handler_2])