stdin Extra in Python con argparse

Question

ho trovato la sintassi molto utile

parser.add_argument('-i', '--input-file', type=argparse.FileType('r'), default='-') 

per specificare un file di input o utilizzando standard input-entrambi i quali voglio nel mio programma. Tuttavia, il file di input non è sempre richiesto. Se io non sto usando -i o il reindirizzamento di ingresso con uno dei

$ someprog | my_python_prog 
$ my_python_prog < inputfile 

io non voglio che il mio programma Python attendere per l'input. Voglio che passi e utilizzi i valori predefiniti.

Answer 1

La libreria standard documentation for argparse suggerisce questa soluzione per consentire i file di ingresso opzionale/uscita:

>>> parser = argparse.ArgumentParser() 
>>> parser.add_argument('infile', nargs='?', type=argparse.FileType('r'), 
...      default=sys.stdin) 
>>> parser.add_argument('outfile', nargs='?', type=argparse.FileType('w'), 
...      default=sys.stdout) 
>>> parser.parse_args(['input.txt', 'output.txt']) 
Namespace(infile=<_io.TextIOWrapper name='input.txt' encoding='UTF-8'>, 
      outfile=<_io.TextIOWrapper name='output.txt' encoding='UTF-8'>) 
>>> parser.parse_args([]) 
Namespace(infile=<_io.TextIOWrapper name='<stdin>' encoding='UTF-8'>, 
      outfile=<_io.TextIOWrapper name='<stdout>' encoding='UTF-8'>) 

Answer 2

Usa isatty-detect se il programma è in una sessione interattiva o la lettura da un file:

if not sys.stdin.isatty(): # Not an interactive device. 
    # ... read from stdin 

Tuttavia, per motivi di coerenza e di riproducibilità, in considerazione dopo la norma e la lettura da stdin se il nome del file è -. Potresti considerare di lasciare che il modulo fileinput lo gestisca.