Python+Selenium 自动化框架篇之4 - 封装日志类和浏览器引擎类到框架内_python

这一篇，我们介绍，如何封装自己的日志类和浏览器引擎类。

相关步骤：

打开PyCharm，创建如下格式的项目层级结构，文件名我保持一样；

相关文件代码如下

配置文件config.ini相关设计如下：

#  this is config file, only store browser type and server URL

[browserType]
#browserName = Firefox
browserName = Chrome
#browserName = IE

[testServer]
URL = https://www.baidu.com

浏览器引擎类browser_engine.py相关脚本代码如下：

# -*- coding:utf-8 -*-
import configparser
import os.path
from selenium import webdriver
from framework.logger import Logger
from webdriver_manager.chrome import ChromeDriverManager
from webdriver_manager.firefox import GeckoDriverManager
from webdriver_manager.microsoft import IEDriverManager

logger = Logger(logger="BrowserEngine").getlog()

class BrowserEngine(object):

    def __init__(self, driver):
        self.driver = driver

    # 从 config.ini 文件中读取浏览器类型，返回驱动
    def open_browser(self, driver):
        config = configparser.ConfigParser()
        # file_path = os.path.dirname(os.getcwd()) + '/config/config.ini'
        file_path = os.path.dirname(os.path.abspath('.')) + '/config/config.ini'
        # print(file_path)
        config.read(file_path,encoding='UTF-8'), # 如果代码有中文注释，用这个，不然报解码错误

        browser = config.get("browserType", "browserName")
        logger.info("选择的 %s 浏览器." % browser)
        url = config.get("testServer", "URL")
        logger.info("测试服务器地址 url 是: %s" % url)

        # 判断config.ini 文件的浏览器类型
        if browser == "Firefox":
            driver = webdriver.Firefox(executable_path=GeckoDriverManager().install())
            logger.info("启动 Firefox 浏览器.")
        elif browser == "Chrome":
            driver = webdriver.Chrome(ChromeDriverManager().install())
            logger.info("启动 Chrome 浏览器.")
        elif browser == "IE":
            driver = webdriver.Ie(IEDriverManager().install())
            logger.info("启动 IE 浏览器.")

        driver.get(url)
        logger.info("打开 url: %s" % url)
        driver.maximize_window()
        logger.info("最大化当前窗口.")
        driver.implicitly_wait(10)
        logger.info("设置隐式等待 10 秒.")
        return driver

    def quit_browser(self):
        logger.info("关闭并退出浏览器.")
        self.driver.quit()

日志类logger.py相关脚本代码如下：

# _*_ coding: utf-8 _*_
import logging
import os.path
import time


class Logger(object):

    def __init__(self, logger):
        '''
            指定保存日志的文件路径，日志级别，以及调用文件
            将日志存入到指定的文件中
        '''

        # 创建一个logger
        self.logger = logging.getLogger(logger)
        self.logger.setLevel(logging.DEBUG)

        # 创建一个handler，用于写入日志文件
        rq = time.strftime('%Y年%m月%d日%H时%M分%S秒', time.localtime(time.time()))
        # log_path = os.path.dirname(os.getcwd()) + '/Logs/'  # 项目根目录下/Logs 保存日志
        log_path = os.path.dirname(os.path.abspath('.')) + '/logs/'
        # 如果case组织结构式 /testsuit/featuremodel/xxx.py ， 那么得到的相对路径的父路径就是项目根目录
        log_name = log_path + rq + '.log'
        fh = logging.FileHandler(log_name,mode='a',encoding='utf-8')
        fh.setLevel(logging.INFO)

        # 再创建一个handler，用于输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.INFO)

        # 定义handler的输出格式
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        fh.setFormatter(formatter)
        ch.setFormatter(formatter)

        # 给logger添加handler
        self.logger.addHandler(fh)
        self.logger.addHandler(ch)

    def getlog(self):
        return self.logger

页面对象pageobject这里暂不用，下一篇介绍

测试脚本test_baidu_search.py相关脚本如下：

# coding=utf-8
import time
import unittest
from framework.browser_engine import BrowserEngine
from selenium.webdriver.common.by import By


class BaiduSearch(unittest.TestCase):

    def setUp(self):
        """
        测试固件的setUp()的代码，主要是测试的前提准备工作
        :return:
        """
        browse = BrowserEngine(self)
        self.driver = browse.open_browser(self)

    def tearDown(self):
        """
        测试结束后的 *** 作，这里基本上都是关闭浏览器
        :return:
        """
        self.driver.quit()

    def test_baidu_search(self):
        """
        这里一定要test开头，把测试逻辑代码封装到一个test开头的方法里。
        :return:
        """
        self.driver.find_element(By.ID, 'kw').send_keys('selenium')
        time.sleep(1)
        try:
            assert 'selenium' in self.driver.title
            print('测试通过.')
        except Exception as e:
            print('测试失败.', format(e))


if __name__ == '__main__':
    unittest.main()