Come leggere da stdin o da un file se nessun dato viene reindirizzato in Python?

Question

Ho uno script CLI e voglio che legga i dati da un file. Dovrebbe essere in grado di leggere in due modi:

• cat data.txt | ./my_script.py
• ./my_script.py data.txt

po '-a come grep, per esempio.

quello che so:

• sys.argv e optparse Permettetemi di leggere qualsiasi args e opzioni facilmente.
• sys.stdin Permettetemi di leggere i dati di sottofondo
• fileinput rendere il processo completamente automatico

Sfortunatamente:

• utilizzando fileinput utilizza standard input e gli eventuali argomenti come input. Quindi non posso usare opzioni che non sono nomi di file mentre cerca di aprirli.
• sys.stdin.readlines() funziona bene, ma se non lo faccio tubo di tutti i dati, si blocca fino a quando io entro Ctrl + D
• non so come implementare "se nulla in stdin, lette da un file in args "perché stdin è sempre True in un contesto booleano.

Mi piacerebbe un modo portatile per farlo, se possibile.

Answer 1

processo tuoi argomenti non nomi di file tuttavia vuoi, così si finisce con una serie di argomenti che non siano opzioni, quindi passare questo array come parametro per fileinput.input():

import fileinput 
for line in fileinput.input(remaining_args): 
    process(line) 

Answer 2

Non esiste un modo affidabile per rilevare se sys.stdin è collegato a qualcosa, né è opportuno farlo (ad esempio, l'utente desidera incollare i dati in) Rileva la presenza di un nome di file come argomento e usa stdin se non viene trovato alcun nome.

Answer 3

per UNIX/Linux è possibile rilevare se i dati vengono convogliata in, cercando in os.isatty(0)

$ date | python -c "import os;print os.isatty(0)" 
False 
$ python -c "import os;print os.isatty(0)" 
True 

io non sono sicuro che ci sia un equivalente per Windows.

modificare Ok, ho provato con python2.6 su Windows XP

C:\Python26>echo "hello" | python.exe -c "import os;print os.isatty(0)" 
False 

C:\Python26> python.exe -c "import os;print os.isatty(0)" 
True 

Quindi forse non tutto senza speranza per le finestre

Answer 4

Argparse permette che questo essere fatto in un abbastanza facile modo, e dovresti davvero usarlo invece di optparse a meno che tu non abbia problemi di compatibilità.

Il codice dovrebbe andare qualcosa come questo:

import argparse 
parser = argparse.ArgumentParser() 
parser.add_argument('--input', type = argparse.FileType('r'), default = '-') 

Ora avete un parser in grado di analizzare i tuoi argomenti della riga di comando, utilizzare un file se vede uno, o utilizzare lo standard input se non lo fa.

Answer 5

È possibile utilizzare questa funzione per rilevare se l'input proviene da una pipeline o meno.

sys.stdin.isatty() 

Restituisce false se l'input proviene dalla pipeline o vero altrimenti.

Answer 6

Sono un noob, quindi questa potrebbe non essere una buona risposta, ma sto cercando di fare la stessa cosa (consenti uno o più file sulla riga di comando, altrimenti STDIN di default).

Il combo finale ho messo insieme:

parser = argparse.ArgumentParser() 
parser.add_argument("infiles", nargs="*") 
args = parser.parse_args() 

for line in fileinput.input(args.infiles): 
    process(line) 

Questo mi sembra l'unico modo per ottenere tutto il comportamento desiderato in un unico pacchetto elegante, senza la necessità di args nome. Proprio come i comandi Unix sono utilizzati come tali:

cat file1 file2 
wc -l < file1 

Non

:

cat --file file1 --file file2 

Apprezzerei un feedback/conferma da Pythonisti idiomatiche veterani per assicurarsi che ho la migliore risposta. Non ho visto questa soluzione completa menzionata altrove, solo frammenti.