Quelle est la meilleure façon de savoir si un programme Python a quelque chose à lire à partir de stdin?

Question

Je veux un programme pour faire une chose si elle est exécutée comme ceci:

cat something | my_program.py 

et faire autre chose si courir comme ce

my_program.py 

Mais si je lis de stdin, il attendra l'utilisateur entrée, donc je veux voir s'il y a quelque chose à lire avant d'essayer de lire à partir de stdin.

Answer 1

Si vous voulez détecter si quelqu'un est des données de tuyauterie dans votre programme, ou l'exécuter, vous pouvez utiliser de manière interactive isatty pour voir si stdin est un terminal:

$ python -c 'import sys; print sys.stdin.isatty()' 
True 
$ echo | python -c 'import sys; print sys.stdin.isatty()' 
False 

Answer 2

Je ne connais pas les commandes Python du haut de ma tête, mais vous devriez être capable de faire quelque chose avec poll ou de sélectionner des données prêtes à lire sur une entrée standard.

Cela peut être spécifique à un système d'exploitation Unix et différent sur Windows Python.

Answer 3

Les mauvaises nouvelles. Du point de vue de la ligne de commande Unix, ces deux invocations de votre programme sont identiques.

Unix ne peut pas les distinguer facilement. Ce que vous demandez n'est pas vraiment raisonnable, et vous devez penser à une autre façon d'utiliser votre programme.

Dans le cas où ce n'est pas dans un pipeline, que doit-il lire s'il ne lit pas stdin?

Est-il censé lancer une interface graphique? Si c'est le cas, vous pouvez avoir une option "-i" (--interactive) pour indiquer que vous voulez une interface graphique, pas de lecture de stdin.

Vous pouvez parfois distinguer les tubes de la console car le périphérique de console est "/ dev/tty", mais ce n'est pas portable.

Answer 4

Vous voulez le module de sélection (man select sur unix) Il vous permettra de tester s'il y a quelque chose de lisible sur stdin. Notez que select ne fonctionnera pas sur Window avec des objets fichier. Mais à partir de votre pipe chargée question que je suppose que vous êtes sur un os unix :)

http://docs.python.org/library/select.html

root::2832 jobs:0 [~] # cat stdin_test.py 
#!/usr/bin/env python 
import sys 
import select 

r, w, x = select.select([sys.stdin], [], [], 0) 
if r: 
    print "READABLES:", r 
else: 
    print "no pipe" 

root::2832 jobs:0 [~] # ./stdin_test.py 
no pipe 

root::2832 jobs:0 [~] # echo "foo" | ./stdin_test.py 
READABLES: [<open file '<stdin>', mode 'r' at 0xb7d79020>]