当前位置: 首页>编程语言>正文

pytest添加运行日志在报告文件中 pytest logging

python自带日志管理模块logging,使用时可进行模块化配置,详细可参考博文Python日志采集(详细)。

但logging配置起来比较繁琐,且在多进行多线程等场景下使用时,如果不经过特殊处理,则容易出现日志丢失或记录错乱的情况。

python中有一个用起来非常简便的第三方日志管理模块--loguru,不仅可以避免logging的繁琐配置,而且可以很简单地避免在logging中多进程多线程记录日志时出现的问题,甚至还可以自定义控制台输出的日志颜色。

接下来我们来学习怎么使用loguru模块进行日志管理。

安装

第三方模块,需要先安装,安装命令如下:

pip install loguru

简单示例

简单使用示例如下:

from loguru import logger

# 日志写入文件则需使用add()方法,"../log/test.log"即日志文件路径,可自定义。# 写入文件的日志里有中文的话,需要加上encoding="utf-8",否则会显示乱码。
logger.add("../log/test.log", encoding="utf-8", rotation="500MB",
           enqueue=True, retention="7 days")

# 打印不同类型的日志
logger.debug("这是一段debug级别日志")
logger.info("这是一段info级别日志")
logger.warning("这是一段warning级别日志")
logger.critical("这是一段critical级别日志")

上面的示例代码运行之后,会在控制台打印如下日志:




pytest添加运行日志在报告文件中 pytest logging,pytest添加运行日志在报告文件中 pytest logging_pytest,第1张


同时也会在 "../log/test.log" 路径中写入日志,如下:


pytest添加运行日志在报告文件中 pytest logging,pytest添加运行日志在报告文件中 pytest logging_接口测试_02,第2张


由以上结果可以看出:

  • loguru中直接提供了一个日志记录对象logger,拿来即用;
  • loguru默认的输出格式为:时间 | 级别 | 模块名 | 行号 | 日志信息;
  • 不同级别的日志在控制台显示颜色不一样。

当然,在实际项目中我们还需要日志管理器有更丰富的配置,loguru中的add()方法便具有强大的配置功能,接下来介绍add()方法提供的常用配置功能。

add()常用参数说明

loguru模块源码中add()方法的参数如下:


pytest添加运行日志在报告文件中 pytest logging,pytest添加运行日志在报告文件中 pytest logging_接口测试_03,第3张


sink(官方文档简单翻译如下):

  • 可以传入类file对象,如sys.stderr或者open('test.log', 'w')。
  • 可以传入文件路径str或pathlib,如示例代码中的写入日志文件路径。路径可以使用一些附加参数对其进行参数化,如test_{time}.log,{time}即时间参数,创建文件时文件名称中会加入时间。
  • 可以传入像lambda这样的可调用函数的简单函数,如lambda msg:print(msg)。这允许完全由用户偏好和需求定义日志记录过程。
  • 还可以是使用async def语句定义的异步协程函数。该函数返回的协程对象将使用loop.create_task()添加到事件循环中。在使用complete()结束循环之前,应该等待这些任务。
  • 也支持传入logging模块的Handler,如FileHandler、StreamHandler等,Loguru记录会自动转换为日志模块预期的结构。

level:发送到sink的日志消息的最低日志级别,即输出的最低日志级别。

format:定义日志的输出格式。

filter:过滤日志。

colorize:终端日志输出的颜色。

serializer:bool值,输出日志时是否先格式化成JSON数据格式。

backtrace:bool值,是否进行异常跟踪(即backtrace信息记录)。

diagnose:bool值,异常跟踪是否应显示变量值以简化调试。在生产中应将其设置为False,以避免泄漏敏感数据。

enqueue:bool值,日志消息输出之前是否先通过多进程安全队列,多进程多线程运行写入日志时需用到,避免日志记录丢失或混乱。

catch:bool值,是否自动捕获接收器处理日志消息时发生的错误。如果为True,则在sys上显示异常消息。

继续看loguru模块源码,如下:


pytest添加运行日志在报告文件中 pytest logging,pytest添加运行日志在报告文件中 pytest logging_自动化_04,第4张


即我们需要将日志写入指定的文件时,可以使用截图中新增的参数。常用的参数说明如下:

rotation:指定日志文件记录条件,日志输出太多需要隔一段时间写入新文件时适用。

# rotation指定日志文件最大为500MB,超过则新建文件记录日志
logger.add("../log/test.log", rotation="500MB")

# 指定每天12:00新建日志记录文件:rotation="12:00"# 指定每隔10天新建日志记录文件:rotation="10 days"# 指定每隔一个星期新建日志记录文件:rotation="1 week"# 指定每隔两个月新建日志记录文件:rotation="2 months"

retention:指定日志保留时长,会清除超过指定时长的数据。

# rotation指定日志保留时长,如"1 week", "3 days", "2 months"
logger.add("../log/test.log", retention="10 days")

compression:配置文件压缩格式。

# rotation指定日志保留时长,如"gz", "bz2", "xz", "lzma", "tar", "tar.gz", "tar.bz2", "tar.xz", "zip"
logger.add("../log/test.log", compression="zip")

encoding:指定写入文件时的编码格式,这在示例代码中已说明。

使用

接下来,我们使用常用的参数来组合定义日志记录方式。

from loguru import logger
import sys

# format建议直接使用默认的格式
logger.add("../log/test_{time}.log", level="INFO", encoding="utf-8", enqueue=True, rotation="500MB", retention="1 week")

# 当然也可以自定义format,如下示例# logger.add("../log/test_{time}.log", format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {name} | {line} | {message}", encoding="utf-8", enqueue=True, rotation="500MB", retention="1 week")# 打印不同类型的日志
logger.debug("这是一段debug级别日志")
logger.info("这是一段info级别日志")
logger.warning("这是一段warning级别日志")
logger.critical("这是一段critical级别日志")

参数说明:

  • sink="../log/test_{time}.log",即指定日志文件路径,且加入了{time},创建.log文件时名称会加上时间。
  • level="INFO",写入文件时只会写入INFO及以上级别的日志 (需要大写),即INFO、WARNING、CRITICAL。
  • encoding="utf-8",日志写入文件时指定编码格式为"utf-8",否则有中文的话会显示乱码。
  • enqueue=True,支持多线程多进程执行时按照队列写入。
  • rotation="500MB",日志文件最大为500MB,超过则新建。
  • retention="1 week",日志保留一周,过后则清除。
  • format没有指定即使用默认格式,建议直接使用默认的格式,见代码注释。

执行代码后,结果如下:


pytest添加运行日志在报告文件中 pytest logging,pytest添加运行日志在报告文件中 pytest logging_接口测试_05,第5张


从结果中可以看出来,输出内容中模块部分显示的是如__main__:<module>:20的形式,而并不是我们想要看到的对应的模块名称,这是因为直接执运行了当前文件的文件,如果是其他模块调用执行便会显示对应的模块名称。



https://www.xamrdz.com/lan/59k1939577.html

相关文章: