如果沒有提供文件名回退到標準輸出

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。