Leer desde archivo, o STDIN

Question

He escrito una utilidad de línea de comandos que usa getopt para analizar argumentos dados en la línea de comando. También me gustaría tener un nombre de archivo sea un argumento opcional, como lo es en otras utilidades como grep, corte, etc. Por lo tanto, me gustaría tener el siguiente uso

tool -d character -f integer [filename] 

¿Cómo puedo aplicar la siguiente ?

• si se da un nombre de archivo, lea del archivo.
• si no se proporciona un nombre de archivo, lea de STDIN.

Answer 1

En los términos más simples:

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

En este punto se usaría inf a leer el archivo. Dependiendo de si se dio un nombre de archivo, esto se leería desde el archivo dado o desde stdin.

Cuando tenga que cerrar el archivo, se puede hacer esto:

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

Sin embargo, en la mayoría de los casos quedará inofensivo para cerrar sys.stdin si está terminado con él.

Answer 2

El módulo fileinput puede hacer lo que quiera - suponiendo que los argumentos que no sean opciones están en args a continuación:

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

Si args es vacía, entonces fileinput.input() leerá de la entrada estándar; de lo contrario, se lee de cada archivo uno por uno, de manera similar al while(<>) de Perl.

Answer 3

Algo así como:

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

para hacer uso de la declaración del pitón with, se puede utilizar el siguiente código:

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

Answer 5

Yo prefiero usar "-" como un indicador de que usted debe leer de stdin, es más explícito:

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

Me gusta el lenguaje general de usar un contexto de gestión er, pero la solución (demasiado) trivial termina cerrando sys.stdin cuando está fuera de la declaración with, que quiero evitar.

préstamos de this answer, aquí es una solución:

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) 

supongo que se podría lograr something similar with os.dup() pero el código cociné hasta hacer eso resultó ser más complejo y más mágico, mientras que el anterior es un tanto torpe pero muy directo.