将stdout重定向到Python中的记录器

前端之家收集整理的这篇文章主要介绍了将stdout重定向到Python中的记录器前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我可以将stdout的所有输出重定向到我使用标准日志记录模块设置的记录器吗?
(我有os.system调用,其输出我也想看到或掩盖打印语句)

最佳答案
您可以使用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

猜你在找的Python相关文章