在 contextlib.redirect_stdout() 中， print() 函数会导致 AttributeError: 'str' object has no attribute 'write'

Question

我尝试按照this post 的建议将我的python 代码的特定部分的整个控制台输出重定向到一个文本文件。

根据那篇文章，它应该与 contextlib.redirect_stdout() 一起使用，但是当第一个 print() 命令出现在我的自定义函数 custom_function_with_internal_print_calls() 中时，它会抛出 AttributeError : 'str' 对象没有属性 'write'.

dummy_file = "dummy_textfile.txt"  # file to which the stdout shall be redirected

with contextlib.redirect_stdout(dummy_file):
    # Direct print()-function call
    print("Informative string which shall go to the textfile instead of the console.")

    # Indirect print()-function calls (internally)
    custom_function_with_internal_print_calls(**kwargs)

编辑： 在我的特定情况下，我有一个自定义函数，而不是 with 环境中的 print() 函数，它具有对 Python 内置 print() 函数的内部调用。如果 print() 函数出现在另一个函数中，或者直接出现在顶层以重现错误，这无关紧要。

已经尝试过的选项： 其他解决方法似乎无法提供我正在寻找的解决方案，例如

redefining the print()-function 或

writing a decorator for python 2.7x（因为我使用的是 python 3.7x）， 或者也

redirecting the print()-function to a file object 通过print('Filename:', filename, file=f)。 在我的情况下，后者意味着将文件处理程序传递给我选择的封闭函数中的所有子函数，对于这种特殊情况，这需要太多额外的代码修改。

我希望能够只使用环绕我的函数的环境with contextlib.redirect_stdout(dummy_file):，以便将这个函数的每个控制台输出重定向到 dummy_file。

documentation of contextlib.redirect_stdout() 也对我没有帮助。 提前感谢您的建议。

Answer 1

我通过将 file-handler "f" 传递给函数 contextlib.redirect_stdout(f) 而不是文件路径 dummy_file 找到了解决方案，如建议in this post。

dummy_file = "dummy_textfile.txt"  # file to which the stdout shall be redirected

with open(dummy_file, 'w') as f:
    with contextlib.redirect_stdout(f):
        # Indirect print()-function calls (internally)
        custom_function_with_internal_print_calls(**kwargs)

这样，所有 print() 函数调用都会写入 dummy_file 并在之后恢复标准控制台输出。