Hem "open (…)" hem de "sys.stdout" ile nasıl başa çıkılır?

Genellikle ya dosyaya ya da dosya belirtilmemişse stdout'a veri çıkışı yapmam gerekir. Aşağıdaki pasajı kullanıyorum:

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

Yeniden yazmak ve her iki hedefi de aynı şekilde ele almak istiyorum.

İdeal durumda şöyle olur:

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

ancak bu iyi çalışmayacak çünkü withbloktan çıkarken sys.stdout kapatılacak ve ben bunu istemiyorum. Ben de istemiyorum

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

çünkü orijinal stdout'u geri yüklemeyi hatırlamam gerekiyor.

İlişkili:

• Stdout Python'daki bir dosyaya yönlendirilsin mi?
• İstisnaları İşleme - C ++ ile karşılaştırıldığında Python'da istisnaları ele alma hakkında ilginç makale

Düzenle

Sarmalayabileceğimi target, ayrı işlev tanımlayabileceğimi veya bağlam yöneticisi kullanabileceğimi biliyorum . 5 satırdan fazlasını gerektirmeyen basit, zarif, deyimsel bir çözüm arıyorum

Burada kutunun dışında düşünmek, bir gelenek open() yönteme ?

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()

Bunu şu şekilde kullanın:

# 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)

Mevcut kodunuza bağlı kalın. Çok basit ve tam olarak söyleyebilirsin ne yaptığını sadece ona bakarak .

Başka bir yol da satır içi olabilir if:

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

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

Ancak bu, sahip olduğunuzdan çok daha kısa değil ve tartışmasız daha kötü görünüyor.

Ayrıca sys.stdoutkapatılamaz da yapabilirsiniz , ancak bu pek de Pythonic görünmüyor:

sys.stdout.close = lambda: None

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

EAFP yapabildiğinizde neden LBYL?

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

Kıvrımlı bir şekilde çalıştırmanız gerektiğinde with/ asbloğu tek tip olarak kullanmak için neden yeniden yazasınız? Daha fazla satır ekleyecek ve performansı düşüreceksiniz.

Başka bir olası çözüm: bağlam yöneticisi çıkış yönteminden kaçınmaya çalışmayın, sadece stdout'u kopyalayın.

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

Wolph'un cevabında bir gelişme

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

Bu, ikili IO'ya izin verir ve openeğer filenamegerçekten bir dosya adı ise , nihai harici argümanları iletir .

Ayrıca basit bir sarmalayıcı işlevini tercih ederim, eğer modu (ve dolayısıyla stdin'e karşı stdout'u) görmezden gelirseniz oldukça basit olabilir, örneğin:

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

Tamam, tek astarlı savaşlara giriyorsak, işte:

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

Bağlam yalnızca tek bir yerde yazıldığı sürece Jacob'ın orijinal örneğini seviyorum. Dosyayı birçok yazma işlemi için yeniden açmanız sorun olur. Sanırım kararı komut dosyasının en üstüne bir kez verirdim ve sistemin çıkışta dosyayı kapatmasına izin verirdim:

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

Daha düzenli olduğunu düşünüyorsanız kendi çıkış işleyicinizi dahil edebilirsiniz.

import atexit

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

atexit(cleanup_output)

Gerçekten daha "zarif" bir şeyde, yani tek satırlık bir şeyde ısrar etmeniz gerekiyorsa:

>>> 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.txtgörünür ve metni içerir foo.

Sys.stdout için yeni bir fd açmaya ne dersiniz? Bu şekilde kapatırken herhangi bir sorun yaşamazsınız:

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)

Bazı durumlarda hafif gelişme.

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 veya üstünü kullanıyorsanız sadece iki ekstra satır: ekstra importiçin bir satır ve stack.enter_context.

Vücuttan sys.stdoutsonra kapalı olması sorun değilse with, aşağıdaki gibi kalıpları da kullanabilirsiniz:

# Use stdout when target is "-"
with open(target, "w") if target != "-" else sys.stdout as f:
    f.write("hello world")

# Use stdout when target is falsy (None, empty string, ...)
with open(target, "w") if target else sys.stdout as f:
    f.write("hello world")

veya daha genel olarak:

with target if isinstance(target, io.IOBase) else open(target, "w") as f:
    f.write("hello world")

Aşağıdaki çözüm bir güzellik değil, çok uzun zaman öncesine ait; hemen önce ...

handler = open(path, mode = 'a') if path else sys.stdout
try:
    print('stuff', file = handler)
    ... # other stuff or more writes/prints, etc.
except Exception as e:
    if not (path is None): handler.close()
    raise e
handler.close()