วิธีจัดการทั้ง "ด้วย 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)

แต่จะไม่ได้ผลเพราะ sys.stdout ถูกปิดเมื่อออกจากwithบล็อกและฉันไม่ต้องการสิ่งนั้น ฉันไม่ต้องการ

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

เพราะฉันจะต้องจำไว้ว่าต้องกู้คืนมาตรฐานเดิม

ที่เกี่ยวข้อง:

• เปลี่ยนเส้นทาง stdout ไปยังไฟล์ใน Python?
• การจัดการข้อยกเว้น - บทความที่น่าสนใจเกี่ยวกับการจัดการข้อยกเว้นใน Python เมื่อเทียบกับ C ++

แก้ไข

ฉันรู้ว่าฉันสามารถ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ปิดได้ แต่ดูเหมือนว่าจะไม่เหมือน Pythonic:

sys.stdout.close = lambda: None

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

ทำไมต้อง LBYL เมื่อคุณสามารถ EAFP?

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

ทำไมต้องเขียนซ้ำเพื่อใช้with/ asblock อย่างสม่ำเสมอในเมื่อคุณต้องทำให้มันทำงานในลักษณะที่ซับซ้อน คุณจะได้เพิ่มมากขึ้นเส้นและลดประสิทธิภาพ

อีกวิธีหนึ่งที่เป็นไปได้: อย่าพยายามหลีกเลี่ยงวิธีการออกจากตัวจัดการบริบทเพียงทำซ้ำ stdout

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

การปรับปรุงคำตอบของ Wolph

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เป็นชื่อไฟล์

ฉันจะเลือกใช้ฟังก์ชัน wrapper แบบธรรมดาซึ่งอาจค่อนข้างง่ายหากคุณสามารถเพิกเฉยต่อโหมด (และส่งผลให้ 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)

ฉันชอบตัวอย่างดั้งเดิมของเจคอบตราบใดที่มีการเขียนบริบทในที่เดียวเท่านั้น มันจะเป็นปัญหาถ้าคุณเปิดไฟล์ขึ้นมาใหม่เพื่อเขียนหลาย ๆ ฉันคิดว่าฉันจะตัดสินใจเพียงครั้งเดียวที่ด้านบนของสคริปต์และปล่อยให้ระบบปิดไฟล์เมื่อออก:

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.txtfooปรากฏขึ้นและมีข้อความ

แล้วการเปิด fd ใหม่สำหรับ sys.stdout ล่ะ ด้วยวิธีนี้คุณจะไม่มีปัญหาในการปิด:

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.

หากsys.stdoutปิดตามwithร่างกายได้ดีคุณสามารถใช้รูปแบบเช่นนี้:

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

หรือมากกว่านั้นโดยทั่วไป:

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