将sys.stdout重定向到Python日志

Question

现在我们有很多Python脚本，想把它们合并起来，并修复一些重复的部分。我们正在做的一件事是确保所有的sys.stdout和sys.stderr的输出都能进入Python的日志模块。

我们主要想打印出以下内容：

[<ERROR LEVEL>] | <TIME> | <WHERE> | <MSG>

现在，几乎所有的Python错误信息在sys.stdout和sys.stderr中的格式都是[LEVEL] - MSG，这些信息都是通过sys.stdout和sys.stderr输出的。我可以在我的sys.stdout和sys.stderr的包装器中解析这些信息，然后根据解析的内容调用相应的日志级别。

简单来说，我们有一个叫做foo的包，还有一个叫做log的子包。在__init__.py文件中，我们定义了以下内容：

def initLogging(default_level = logging.INFO, stdout_wrapper = None, \
                stderr_wrapper = None):
    """
        Initialize the default logging sub system
    """
    root_logger = logging.getLogger('')
    strm_out = logging.StreamHandler(sys.__stdout__)
    strm_out.setFormatter(logging.Formatter(DEFAULT_LOG_TIME_FORMAT, \
                                            DEFAULT_LOG_TIME_FORMAT))
    root_logger.setLevel(default_level)
    root_logger.addHandler(strm_out)

    console_logger = logging.getLogger(LOGGER_CONSOLE)
    strm_out = logging.StreamHandler(sys.__stdout__)
    #strm_out.setFormatter(logging.Formatter(DEFAULT_LOG_MSG_FORMAT, \
    #                                        DEFAULT_LOG_TIME_FORMAT))
    console_logger.setLevel(logging.INFO)
    console_logger.addHandler(strm_out)

    if stdout_wrapper:
        sys.stdout = stdout_wrapper
    if stderr_wrapper:
        sys.stderr = stderr_wrapper


def cleanMsg(msg, is_stderr = False):
    logy = logging.getLogger('MSG')
    msg = msg.rstrip('\n').lstrip('\n')
    p_level = r'^(\s+)?\[(?P<LEVEL>\w+)\](\s+)?(?P<MSG>.*)$'
    m = re.match(p_level, msg)
    if m:
        msg = m.group('MSG')
        if m.group('LEVEL') in ('WARNING'):
            logy.warning(msg)
            return
        elif m.group('LEVEL') in ('ERROR'):
            logy.error(msg)
            return
    if is_stderr:
        logy.error(msg)
    else:
        logy.info(msg)

class StdOutWrapper:
    """
        Call wrapper for stdout
    """
    def write(self, s):
        cleanMsg(s, False)

class StdErrWrapper:
    """
        Call wrapper for stderr
    """
    def write(self, s):
        cleanMsg(s, True)

然后我们会在我们的某个脚本中调用这个，比如：

import foo.log
foo.log.initLogging(20, foo.log.StdOutWrapper(), foo.log.StdErrWrapper())
sys.stdout.write('[ERROR] Foobar blew')

这会被转换成一个错误日志信息，像这样：

[ERROR] | 20090610 083215 | __init__.py | Foobar Blew

现在问题是，当我们这样做时，错误信息被记录的模块变成了__init__（对应于foo.log.__init__.py文件），这就违背了我们的初衷。

我尝试对stderr/stdout对象进行深拷贝和浅拷贝，但这没有任何效果，信息仍然显示发生错误的模块是__init__.py。我该怎么做才能避免这种情况呢？

错误处理深拷贝输出解析日志模块日志重定向 sys.stdout sys.stderr 模块记录

将sys.stdout重定向到Python日志

2 个回答

撰写回答