Leer desde Archivo, o STDIN

El módulo fileinput puede hacer lo que quiera, asumiendo que los argumentos que no son opciones están en args entonces:

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

Si args está vacío, entonces fileinput.input() leerá desde stdin; de lo contrario, lee de cada archivo a su vez, de manera similar a while(<>) de Perl.

Me gusta el modismo general de usar un administrador de contexto, pero la solución (demasiado) trivial termina cerrándose sys.stdin cuando estás fuera de la instrucción with, lo que quiero evitar.

Tomando prestado de esta respuesta , aquí hay 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 podrías lograr algo similar con os.dup() pero el código que inventé para hacer eso resultó ser más complejo y más mágico, mientras que el anterior es algo torpe pero muy sencillo.

Para hacer uso de la instrucción with de python, se puede usar 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
    # ......

Prefiero usar " - " como un indicador que 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

No respuesta directa sino relacionada.

Normalmente cuando escribes un script en python puedes usar el paquete argparse. SI este es el caso se puede utilizar:

parser = argparse.ArgumentParser()
parser.add_argument('infile', nargs='?', type=argparse.FileType('r'), default=sys.stdin)

'?'. Si es posible, se consumirá un argumento desde la línea de comandos, y producido como un solo artículo. Si no hay ningún argumento de línea de comandos presente, se producirá el valor predeterminado.

Y aquí establecemos por defecto sys.stdin;

Así que si hay un archivo lo leerá , y si no tomará el input from stdin "Nota: que estamos usando el argumento posicional en el ejemplo anterior"

Para más visita: https://docs.python.org/2/library/argparse.html#nargs

Algo 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()