python 重定向控制台stdout与stderr到loguru

python 重定向stdout与stderr到loguru

安装

pip install loguru

重定向文件：log_init.py

import sys
from loguru import logger

DEBUG = 0
INFO = 1
ERROR = 2


class LoggerWriter:
    def __init__(self, level):
        # 日志级别
        self.level = level

    def write(self, message):
        # 只有message非空时才记录
        if message.rstrip() != "":
            if self.level == DEBUG:
                logger.debug(message)
            elif self.level == ERROR:
                logger.error(message)
            else:
                logger.info(message)

    def flush(self):
        # 这个方法是为了满足文件对象的接口，但是什么都不需要做
        pass


# 重定向标准输出和错误输出
def init_log():
    # 只定义一次add就可以了
    logger.add('log/log_{time:YYYY-MM-DD}.log', rotation='00:00')
    # 重定向
    sys.stdout = LoggerWriter(DEBUG)
    sys.stderr = LoggerWriter(ERROR)
    logger.info("--------------------------- start -----------------------")


# 引入时，自动调用初始化log
init_log()

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

main.py引用
建议如果自己写的代码直接使用 from loguru import logger

import log_init  # print 重定向到日志，放第一行
1