更优雅地在 Python 中写日志

原来我们在 Python 中写日志,使用的是 Python 自带的 logging 模块,要实现既在终端输出,又能写文件,并且日志文件还能 rotate ,代码需要十多行:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
import logging
from logging.handlers import RotatingFileHandler

os.makedirs('Logs', exists_ok=True)
logger = logging.getLogger('Robot')
handler = logging.StreamHandler()
handler.setLevel(logging.DEBUG)
file_handler = RotatingFileHandler(os.path.join('Logs', 'robot.log'),
maxBytes=5 * 1024 * 1024,
backupCount=10,
encoding='utf-8')
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
handler.setFormatter(formatter)
file_handler.setFormatter(formatter)
logger.addHandler(handler)
logger.addHandler(file_handler)
logger.setLevel(logging.DEBUG)

logger.info('终于可以写日志了...')

那有没有什么简单好用,for human的写日志方案呢?

当然有,那就是使用loguru这个模块。

它有多好用呢?我们用一段代码来演示:

1
2
3
4
5
6
from loguru import logger
logger.add('logs/z_project.log',
level='DEBUG',
format='{time:YYYY-MM-DD 
:mm:ss} - {level} - {file} - {line} - {message}',
rotation="10 MB")
logger.info('可以写日志了')

不需要提前创建文件夹。不需要设置奇奇怪怪的handler

甚至你可以直接把它当做装饰器使用,当一个函数报错的时候,自动记录日志:

1
2
3
4
5
6
from loguru import logger


@logger.catch
def test():
'a' + 1

函数报错时,自动记录报错信息到日志中。

他记录的报错信息有多准确呢?我们来看这个图:

每个变量的值都给你标出来了。

使用 loguru 也只需要定义一次格式。只需要在入口文件定义好格式,在同一个项目中的其他文件中,直接

1
2
from loguru import logger
logger.info('可以用了')