如何抽象标准输入/标准输出和文件？

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)