如何抽象stdin/stdout和文件？

Question

我想寫一個Python腳本，它允許用戶在標準輸入/標準輸出和輸入/輸出文件之間切換。但我不知道如果我的實現是正確的：

#!/usr/bin/python3 

import fileinput 
import sys 

from contextlib import contextmanager 
from contextlib import redirect_stdout 


def foo(file): 
    return ''.join([line for line in file]) 


@contextmanager 
def run(input_file, output_file): 
    stdin = '-' if input_file is None else input_file 
    stdout = sys.stdout if output_file is None else open(output_file, 'w') 
    with fileinput.input(stdin) as stdin, stdout: 
     with redirect_stdout(stdout): 
      print(foo(stdin)) 

1. 是代碼管理資源是否正確？
2. 如何正確處理字符編碼？我想在utf-8中擁有一切。
3. 上述代碼有沒有改進的餘地？

Answer 1

我認爲你需要這樣做。你不需要使run()本身成爲@contextmanager-所有它需要做的就是使用一個（或更多，fileinput.input和redirect_stdout已經是）。我不知道你爲什麼使用fileinput.input，因爲似乎只涉及一個輸入文件。

from contextlib import redirect_stdout 
import fileinput 
import sys 

def foo(file): 
    return ''.join([line for line in file]) 

def run(input_file, output_file): 
    stdin = '-' if input_file is None else input_file 
    stdout = sys.stdout if output_file is None else open(output_file, 'w') 
    with fileinput.input(stdin) as stdin, redirect_stdout(stdout): 
     print(foo(stdin)) 

run('input.txt', 'output.txt') 
run('input.txt', None)