Đọc từ Tệp, hoặc STDIN

Question

Tôi đã viết một tiện ích dòng lệnh sử dụng getopt để phân tích các đối số được đưa ra trên dòng lệnh. Tôi cũng muốn có một tên tập tin là một đối số tùy chọn, chẳng hạn như trong các tiện ích khác như grep, cắt vv Vì vậy, tôi muốn nó phải có việc sử dụng sau

tool -d character -f integer [filename] 

Làm thế nào tôi có thể thực hiện như sau ?

• nếu tên tệp được cung cấp, hãy đọc từ tệp.
• nếu tên tệp không được cung cấp, hãy đọc từ STDIN.

Answer 1

Trong thuật ngữ đơn giản:

import sys 
# parse command line 
if file_name_given: 
    inf = open(file_name_given) 
else: 
    inf = sys.stdin 

Tại thời điểm này, bạn sẽ sử dụng inf để đọc từ tập tin. Tùy thuộc vào việc một tên tệp đã được đưa ra, điều này sẽ đọc từ tệp đã cho hay từ stdin.

Khi bạn cần để đóng tập tin, bạn có thể làm điều này:

if inf is not sys.stdin: 
    inf.close() 

Tuy nhiên, trong nhiều trường hợp nó sẽ là vô hại để đóng sys.stdin nếu bạn đang thực hiện với nó.

Answer 2

Module fileinput có thể làm những gì bạn muốn - giả định các đối số không lựa chọn là ở args thì:

import fileinput 
for line in fileinput.input(args): 
    print line 

Nếu args trống sau đó fileinput.input() sẽ đọc từ stdin; nếu không nó sẽ đọc từ mỗi tập tin lần lượt, theo cách tương tự như của Perl while(<>).

Answer 3

Cái gì như:

if input_from_file: 
    f = open(file_name, "rt") 
else: 
    f = sys.stdin 
inL = f.readline() 
while inL: 
    print inL.rstrip() 
    inL = f.readline() 

Answer 4

Để tận dụng tuyên bố python của with, người ta có thể sử dụng đoạn mã sau:

import sys 
with open(sys.argv[1], 'r') if len(sys.argv) > 1 else sys.stdin as f: 
    # read data using f 
    # ...... 

Answer 5

Tôi thích sử dụng "-" như một chỉ báo rằng bạn nên đọc từ stdin, nó rõ ràng hơn:

import sys 
with open(sys.argv[1], 'r') if sys.argv[1] is not "-" else sys.stdin as f: 
    pass # do something here 

Answer 6

Tôi thích thành ngữ chung của việc sử dụng ngữ cảnh manag er, nhưng các giải pháp tầm thường (quá) kết thúc lên đóng sys.stdin khi bạn ra khỏi tuyên bố with, mà tôi muốn tránh.

Vay từ this answer, đây là một cách giải quyết:

import sys 
import contextlib 

@contextlib.contextmanager 
def _smart_open(filename, mode='Ur'): 
    if filename == '-': 
     if mode is None or mode == '' or 'r' in mode: 
      fh = sys.stdin 
     else: 
      fh = sys.stdout 
    else: 
     fh = open(filename, mode) 
    try: 
     yield fh 
    finally: 
     if filename is not '-': 
      fh.close() 

if __name__ == '__main__': 
    args = sys.argv[1:] 
    if args == []: 
     args = ['-'] 
    for filearg in args: 
     with _smart_open(filearg) as handle: 
      do_stuff(handle) 

Tôi cho rằng bạn có thể đạt được something similar with os.dup() nhưng mã tôi nấu chín lên để làm điều đó hóa ra là phức tạp hơn và kỳ diệu hơn, trong khi ở trên là hơi clunky nhưng rất đơn giản.