python - 使用文件参数抑制python3中的打印输出

Question

我有几个要设置详细程度的函数。目前，我是这样处理的:

class Foo:
    def __init__(self, foo_stuff, verbose=True):
        self.print_file = None if verbose else open(os.devnull, 'w')

    def do_stuff(self):
        print('doing stuff', file=self.print_file)

这行得通，但我不喜欢我从不关闭文件 self.print_file.

为了简洁起见，我不希望将每个打印函数都包装在 with open(...) 中。我想知道是否有人可以建议另一种方法来做到这一点。对于这个应用程序，我认为 python 日志记录模块不起作用。

Answer 1

atexit 模块可能有帮助。在您的 Foo 类中编写一个清理函数并通过调用

注册它

atexit.register(function, args)

在您的程序退出之前调用。