python - 将标准输出重定向到 Python 中的记录器

Question

我可以将所有输出从 stdout 重定向到我使用标准 logging 模块设置的记录器吗？ (我有 os.system 调用，我也想查看其输出或偶尔的打印语句)

Answer 1

您可以利用 this post 中的建议，总结如下:

import logging

class LoggerWriter:
    def __init__(self, logger, level):
        self.logger = logger
        self.level = level

    def write(self, message):
        if message != '\n':
            self.logger.log(self.level, message)

def main():
    logging.basicConfig(level=logging.DEBUG)
    logger = logging.getLogger("demo")
    info_fp = LoggerWriter(logger, logging.INFO)
    debug_fp = LoggerWriter(logger, logging.DEBUG)
    print >> info_fp, "An INFO message"
    print >> debug_fp, "A DEBUG message"

if __name__ == "__main__":
    main()

运行时，脚本打印:

INFO:demo:An INFO message
DEBUG:demo:An DEBUG message