python3+selenium框架设计03-封装日志类

首先我们先来实现日志的功能，日志可以使用python3自带logging模块，不会的可以百度一下相关文章，也可以看我另外一篇文章Python3学习笔记24-logging模块

在封装日志类前，我们需要想一下，我们需要实现怎样的日志功能，需要把日志放到项目下指定路径，文件名为日期.log，需要把所有日志输出到一个文件中，需要把报错的日志单独输出到另一个文件，info和error需要不同的日志格式

那先在PyCharm里新建一个项目，在这个项目下新建一个Logs文件夹，和getcwd.py文件。然后在Logs文件下建一个Alll_Logs文件夹,Error_Logs文件夹和log.py文件。项目结构如下图

​

 

在getcwd.py中输入如下代码

import os
def get_cwd():
    path = os.path.dirname(os.path.abspath(__file__))
    #当前文件的绝对路径
    return path

这样子是通过getcwd.py文件的绝对路径，来获得项目所在文件夹地址。其他一些获取路径的方法，会获取当前执行文件的路径，会导致路径错误

然后在log.py文件封装日志功能，输入以下代码

#-*- coding: UTF-8 -*-
import logging
import time
import os
import getcwd
def get_log(logger_name):

        #创建一个logger
        logger= logging.getLogger(logger_name)
        logger.setLevel(logging.INFO)

        #设置日志存放路径，日志文件名
        #获取本地时间，转换为设置的格式
        rq = time.strftime('%Y%m%d%H%M',time.localtime(time.time()))
        #设置所有日志和错误日志的存放路径
        path = getcwd.get_cwd()
        #通过getcwd.py文件的绝对路径来拼接日志存放路径
        all_log_path = os.path.join(path,'Logs/All_Logs/')
        error_log_path = os.path.join(path,'Logs/Error_Logs/')
        #设置日志文件名
        all_log_name = all_log_path + rq +'.log'
        error_log_name = error_log_path + rq +'.log'

        #创建handler
        #创建一个handler写入所有日志
        fh = logging.FileHandler(all_log_name)
        fh.setLevel(logging.INFO)
        #创建一个handler写入错误日志
        eh = logging.FileHandler(error_log_name)
        eh.setLevel(logging.ERROR)
        #创建一个handler输出到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.INFO)

        #定义日志输出格式
        #以时间-日志器名称-日志级别-日志内容的形式展示
        all_log_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        #以时间-日志器名称-日志级别-文件名-函数行号-错误内容
        error_log_formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(module)s  - %(lineno)s - %(message)s')
        #将定义好的输出形式添加到handler
        fh.setFormatter(all_log_formatter)
        ch.setFormatter(all_log_formatter)
        eh.setFormatter(error_log_formatter)


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

log1 = get_log("selenium")

最后一行代码是直接在日志类创建了一个实例，括号里面的selenium可以填项目名称表示哪个项目，在调用日志类的时候，直接import这个log1，不用在使用的时候再去初始化，一来避免重复去初始化，二来可以避免偶尔info日志重复打印的问题

接下来测试一下日志类，随便新建一个.py文件，输入以下代码

from Logs.log import log1
try:
    log1.info("开始测试")
    r = 10/0
    log1.info("resuit:",r)
except ZeroDivisionError as e :
    log1.error("报错信息：",exc_info = 1)
log1.info('end')

打印报错信息请使用exc_info=1 这样能把完整的报错信息都打印出来

​

这是在控制台日志输出的截图

​

 

这是All_Logs文件夹下的日志输出截图

​

 

这是Error_Logs文件夹下日志输出截图，可以看到两个日志文件输出正确，输出的格式也是对的