`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を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)

複雑な方法で機能させる必要があるのに、with/ asブロックを均一に使用するように書き換えるのはなぜですか？行を追加してパフォーマンスを低下させます。

別の考えられる解決策：コンテキストマネージャーの終了メソッドを回避しようとせず、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と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)

コンテキストが1か所にのみ記述されている限り、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以降を使用している場合は、2行追加します。1行は追加用import、もう1行は用stack.enter_contextです。