如果没有提供文件名回退到标准输出

Question

我有一个脚本接受作为参数的文件名，而不是打开它并写入一些东西。

我使用with声明：

with open(file_name, 'w') as out_file: 
    ... 
    out_file.write(...) 

现在，如果我想如果没有提供file_name写sys.stdout？

我是否需要将一个函数中的所有动作都包装起来，并在之前放置一个条件？

if file_name is None: 
    do_everything(sys.stdout) 
else: 
    with open(file_name, 'w') as out_file: 
     do_everything(out_file) 

Answer 1

from contextlib import contextmanager 
@contextmanager 
def file_or_stdout(file_name): 
    if file_name is None: 
     yield sys.stdout 
    else: 
     with open(file_name, 'w') as out_file: 
      yield out_file 

然后，你可以做

with file_or_stdout(file_name) as wfile: 
    do_stuff_writing_to(wfile) 

Answer 2

if file_name is None: 
    fd = sys.stdout 
else: 
    fd = open(file_name, 'w') 

# write to fd 

if fd != sys.stdout: 
    fd.close(); 

Answer 3

你可以编写自己的上下文管理。如果没有其他人，我会在以后发布示例代码

Answer 4

使用with ... as构造对于自动关闭文件非常有用。这意味着，使用它与sys.stdout，我猜你知道，将崩溃您的程序，因为它会尝试关闭系统标准输出！

这意味着像with open(name, 'w') if name else sys.stdout as:不会工作。

这让我说没有任何简单的好方法来更好地编写代码片段......但是可能有更好的方法来思考如何构建这样的代码！

要澄清的主要问题是当file_name存在时需要打开（并且更重要的是关闭）file_name的文件处理程序。

就我个人而言，我会简单地删除with .. as并照顾打开文件 - 更重要的是关闭它！ - 别的地方。里程可能会有所不同，这取决于您的软件的工作方式。

这意味着你可以简单地做：

out_file = open(file_name, 'w') if file_name else sys.stdout 

，并与整个程序out_file工作。

当您关闭，记得要检查它是否是一个文件或不:)

而且你有没有想过干脆使用logging模块？这很容易让你添加不同的处理程序，打印到文件，打印到stdout ...

Answer 5

你是如何处理的命令行参数？如果您使用​​，则可以使用add_argument的参数type和default来处理此问题。例如，你可以试试下面的：

import sys 
import argparse 

def main(argv=None): 

    if argv is None: 
     argv=sys.argv[1:] 

    parser = argparse.ArgumentParser() 

    parser.add_argument('infile', nargs='?', 
     type=argparse.FileType('w'), 
     default=sys.stdin) 

    args = parser.parse_args(argv) 

    print args.infile 

    return 0 

if __name__=="__main__": 
    sys.exit(main(sys.argv[1:])) 

如果文件名存在作为参数的脚本argparse会自动打开和关闭该文件，并args.infile将是一个句柄到这个文件。否则args.infile将只是sys.stdin。