如何很好地处理带有open（…）和sys.stdout的两者？

通常，我需要将数据输出到文件，或者，如果未指定文件，则将数据输出到stdout。我使用以下代码段：

if target:
    with open(target, 'w') as h:
        h.write(content)
else:
    sys.stdout.write(content)

我想重写它并统一处理两个目标。

在理想情况下，它将是：

with open(target, 'w') as h:
    h.write(content)

但这不能很好地工作，因为在离开with块时sys.stdout是关闭的，我不希望这样。我都不想要

stdout = open(target, 'w')
...

因为我需要记住恢复原始的stdout。

有关：

• 将stdout重定向到Python中的文件？
• 处理异常 -有关与C ++相比在Python中处理异常的有趣文章

编辑

我知道我可以包装target，定义单独的功能或使用上下文管理器。我正在寻找一种简单，优雅，惯用的解决方案配件，它不需要多于5条线

只是想在这里跳出框框，自定义open()方法怎么样？

import sys
import contextlib

@contextlib.contextmanager
def smart_open(filename=None):
    if filename and filename != '-':
        fh = open(filename, 'w')
    else:
        fh = sys.stdout

    try:
        yield fh
    finally:
        if fh is not sys.stdout:
            fh.close()

像这样使用它：

# For Python 2 you need this line
from __future__ import print_function

# writes to some_file
with smart_open('some_file') as fh:
    print('some output', file=fh)

# writes to stdout
with smart_open() as fh:
    print('some output', file=fh)

# writes to stdout
with smart_open('-') as fh:
    print('some output', file=fh)

坚持使用您当前的代码。这很简单，你可以告诉究竟是什么它是由它一眼正是这样做。

另一种方法是使用内联if：

handle = open(target, 'w') if target else sys.stdout
handle.write(content)

if handle is not sys.stdout:
    handle.close()

但这并不比您所拥有的要短得多，而且看起来可以说更糟。

您也可以将其sys.stdout设为无法关闭，但这似乎不太像Python风格：

sys.stdout.close = lambda: None

with (open(target, 'w') if target else sys.stdout) as handle:
    handle.write(content)

当您可以使用EAFP时为什么选择LBYL？

try:
    with open(target, 'w') as h:
        h.write(content)
except TypeError:
    sys.stdout.write(content)

为什么要重写它以使用 withas当必须使它以复杂的方式工作时，统一 / 块？您将添加更多行并降低性能。

另一个可能的解决方案：不要尝试避免使用上下文管理器退出方法，而要重复使用stdout。

with (os.fdopen(os.dup(sys.stdout.fileno()), 'w')
      if target == '-'
      else open(target, 'w')) as f:
      f.write("Foo")

沃尔夫答案的改进

import sys
import contextlib

@contextlib.contextmanager
def smart_open(filename: str, mode: str = 'r', *args, **kwargs):
    '''Open files and i/o streams transparently.'''
    if filename == '-':
        if 'r' in mode:
            stream = sys.stdin
        else:
            stream = sys.stdout
        if 'b' in mode:
            fh = stream.buffer  # type: IO
        else:
            fh = stream
        close = False
    else:
        fh = open(filename, mode, *args, **kwargs)
        close = True

    try:
        yield fh
    finally:
        if close:
            try:
                fh.close()
            except AttributeError:
                pass

这允许二进制IO并将最终无关的参数传递给openif filename确实是文件名。

我还将使用一个简单的包装函数，如果您可以忽略该模式（因此可以忽略stdin vs. stdout），则它可以非常简单，例如：

from contextlib import contextmanager
import sys

@contextmanager
def open_or_stdout(filename):
    if filename != '-':
        with open(filename, 'w') as f:
            yield f
    else:
        yield sys.stdout

好吧，如果我们要进行单线战争，那就是：

(target and open(target, 'w') or sys.stdout).write(content)

只要上下文仅在一个地方编写，我就喜欢Jacob的原始示例。如果最终要多次打开该文件，那将是一个问题。我想我只在脚本顶部做出一次决定，然后让系统在退出时关闭文件：

output = target and open(target, 'w') or sys.stdout
...
output.write('thing one\n')
...
output.write('thing two\n')

如果您认为它更整洁，则可以包括自己的退出处理程序

import atexit

def cleanup_output():
    global output
    if output is not sys.stdout:
        output.close()

atexit(cleanup_output)

如果您真的必须坚持一些更“优雅”的东西，即单线：

>>> import sys
>>> target = "foo.txt"
>>> content = "foo"
>>> (lambda target, content: (lambda target, content: filter(lambda h: not h.write(content), (target,))[0].close())(open(target, 'w'), content) if target else sys.stdout.write(content))(target, content)

foo.txt出现并包含文本foo。

如何为sys.stdout打开新的fd？这样一来，关闭它就不会有任何问题：

if not target:
    target = "/dev/stdout"
with open(target, 'w') as f:
    f.write(content)

if (out != sys.stdout):
    with open(out, 'wb') as f:
        f.write(data)
else:
    out.write(data)

在某些情况下略有改善。

import contextlib
import sys

with contextlib.ExitStack() as stack:
    h = stack.enter_context(open(target, 'w')) if target else sys.stdout
    h.write(content)

如果您使用的是Python 3.3或更高版本，则只需增加两行：一行用于额外的import行，另一行用于stack.enter_context。