Aus Datei lesen, oder STDIN

Question

Ich habe ein Befehlszeilenprogramm geschrieben, das getopt zum Parsen von Argumenten verwendet, die in der Befehlszeile angegeben werden. Ich möchte auch einen Dateinamen haben ein optionales Argument sein, wie es in anderen Programmen wie grep, schneiden usw. Also, ich würde es wie die folgende Nutzungs

tool -d character -f integer [filename] 

haben Wie kann ich implementieren die folgenden ?

• Wenn ein Dateiname angegeben wird, lesen Sie aus der Datei.
• Wenn ein Dateiname nicht angegeben ist, lesen Sie von STDIN.

Answer 1

In einfachen Worten ausgedrückt:

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

An dieser Stelle würden Sie inf verwenden aus der Datei zu lesen. Je nachdem, ob ein Dateiname angegeben wurde, würde dies von der angegebenen Datei oder von stdin gelesen werden.

Wenn Sie die Datei schließen, dann können Sie dies tun:

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

Doch in den meisten Fällen wird es harmlos sein sys.stdin zu schließen, wenn Sie mit ihm fertig sind.

Answer 2

Das fileinput Modul kann das tun, was Sie wollen - in args die Nicht-Option Argumente unter der Annahme, sind dann:

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

Wenn args leer ist, dann wird fileinput.input() von stdin lesen; Ansonsten liest es aus jeder Datei der Reihe nach ähnlich wie Perls while(<>).

Answer 3

Etwas wie:

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

Um die Verwendung der Python with Aussage zu machen, kann man den folgenden Code verwenden:

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

Answer 5

ich lieber verwenden "-" als Indikator, die Sie lesen sollten von stdin, es ist expliziter:

Answer 6

Ich mag das allgemeine Idiom der Verwendung eines Kontextmanag ähm, aber die (zu) triviale Lösung endet mit dem Schließen sys.stdin, wenn Sie aus der with Anweisung herauskommen, die ich vermeiden möchte.

von this answer Leihen, hier eine Abhilfe:

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) 

Ich nehme an, Sie something similar with os.dup() erreichen könnte, aber der Code ich gekocht bis zu, dass komplexere entpuppte und magischer, während die oben etwas ist klobig, aber sehr geradlinig.