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

时间：2024-04-04 15:03:16

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

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

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

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

在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 这样能把完整的报错信息都打印出来

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

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

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

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

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

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

相关文章