使用上下文管理器将脚本输出重定向到文件？

Question

我正在编写一个脚本，其中有一个选项可以在命令行上传递，无论脚本是否应将其结果打印到标准输出或预定义的结果文件。其代码概要如下所示。我现在已经阅读了一些有关 Python 上下文管理器的内容，但不太确定在这种特定情况下是否以及如何使用上下文管理器。因此，我正在寻求建议，

1. 在这个问题中使用上下文管理器是否有意义，
2. 以及如何实施它。

所以，没有上下文管理器的代码是：

option_file = True # would come from OptionParser in real code

if option_file:
    out = open("resultsfile", "w")
else:
    out = sys.stdout

# do some computations
out.write("Results of script")
# more computations and calls to out.write

if option_file:
    out.close()

Answer 1

上下文管理器可以与 with 语句一起使用。它的明确设计目的是：

• 执行一些设置，
• 给您一个对象，然后
• 再次执行一些拆卸（即使您引发异常）。

例如，open 可以用作上下文管理器。在下面的代码中，

with open(...) as f:
    # do stuff

无论 stuff 是什么，文件都将始终关闭。 （嗯，通常。除了一些愚蠢的情况，比如电源被关闭或进程被终止。）

当你处于这种情况时，你应该使用上下文管理器。在我看来，它不像你那样，所以我认为没有理由使用上下文管理器。

---

还有一种使用上下文管理器编写代码的替代方法（没有更好或更坏，只是不同）。如果您想暂时重定向 stdout - 但确保完成后恢复它 - 那么您就处于上述情况。这是一个示例：

@contextlib.contextmanager
def redirect_stdout(stream):
    import sys
    sys.stdout = stream
    yield
    sys.stdout = sys.__stdout__

然后您可以编写类似的代码

with open(...) as f:
    with redirect_stdout(f):
        # do stuff

，并且对 stuff 中的 stdout 的任何写入都将转至 f 。

---

编辑：您是正确的，没有办法有条件地拥有上下文管理器：要么您在其中，要么不在。你总是可以编写自己的，但可能什么也不做：

@contextlib.contextmanager
def maybe_open(path, do_nothing=True):
    if do_nothing:
        f = None
        yield sys.stdout
    else:
        f = open(path)
        yield f

    if f:
        f.close()

这几乎肯定是矫枉过正。