输出重定向实现日志记录

以你之姓@ 2022-10-06 05:54 206阅读 0赞

本文的主要内容是：介绍使用aop的方式，对输出重定向，然后记录日志。

python中的日志记录，有其专门的log模块，目前还没有看，但是通过输出重定向，自己实现一个日志记录功能还是比较简单的。

python重定向的方法如下：

import sys
    
    def myfun():
        print('hello world')
    
    savedStdout = sys.stdout  # 保存标准输出流
    with open('output.log', 'w+') as file:
        sys.stdout = file  # 标准输出重定向至文件
        myfun()  # 调用相关的需要记录日志的函数
    
    sys.stdout = savedStdout  # 恢复标准输出流
    print('This message is for screen!')

上述代码中，代码的灵活性比较差，因为，如果我不想记录日志时，只能删除上面输出重定向的代码，而希望达到的状态时，我写的关于记录日志的代码，不对原来的调用，代码等进行改动，像是在函数执行的流程中横向地插入了一段逻辑，这时候就需要把AOP（面向切面编程的思想）应用过来，关于AOP的内容，可以查看这篇博客[轻松让你理解装饰器与面向切面编程（AOP)][AOP]

import sys
    
    def myfun():
        print('hello world')
    
    def write_log(func):
        # 定义一个内嵌的包装函数，给传入的函数加上输出重定向功能的包装，进而记录日志
        def wrapper():
            savedStdout = sys.stdout  # 保存标准输出流
            with open('out.log', 'w+') as file:
                sys.stdout = file  # 标准输出重定向至文件
                func()
            sys.stdout = savedStdout  # 恢复标准输出流
    
        # 将包装后的函数返回
        return wrapper
    
    
    myfun = write_log(myfun)
    myfun()

再利用python的语法糖精简一下：

import sys
    
    
    def write_log(func):
        # 定义一个内嵌的包装函数，给传入的函数加上输出重定向功能的包装，进而记录日志
        def wrapper():
            savedStdout = sys.stdout  # 保存标准输出流
            with open('out.log', 'w+') as file:
                sys.stdout = file  # 标准输出重定向至文件
                func()
            sys.stdout = savedStdout  # 恢复标准输出流
    
        # 将包装后的函数返回
        return wrapper
     
    @write_log    
    def myfun():
        print('hello world')
    
    myfun()  # 直接调用就可以了

此时，一个结合AOP以及输出重定向的日志记录功能就实现了。程序中所有print的内容就会实时的记录到日志文件中了。

**参考：**  
关于输出重定向的内容可参考下面这篇博客，大佬写的很详细：  
https://www.cnblogs.com/clover-toeic/p/5491073.html

[AOP]: https://blog.csdn.net/qq_38048756/article/details/117377952