Comment répliquer le comportement de tee en Python lors de l'utilisation de subprocess?

je vois que c'est un vieux post, mais juste au cas où quelqu'un est toujours à la recherche d'un moyen de faire ceci:

proc = subprocess.Popen(["ping", "localhost"], 
                        stdout=subprocess.PIPE, 
                        stderr=subprocess.PIPE)

with open("logfile.txt", "w") as log_file:
  while proc.poll() is None:
     line = proc.stderr.readline()
     if line:
        print "err: " + line.strip()
        log_file.write(line)
     line = proc.stdout.readline()
     if line:
        print "out: " + line.strip()
        log_file.write(line)

ceci est un port simple de tee à Python.

import sys
sinks = sys.argv[1:]
sinks = [open(sink, "w") for sink in sinks]
sinks.append(sys.stderr)
while True:
  input = sys.stdin.read(1024)
  if input:
    for sink in sinks:
      sink.write(input)
  else:
    break

J'utilise Linux en ce moment, mais cela devrait fonctionner sur la plupart des plateformes.

Maintenant, pour le subprocess partie, je ne sais pas comment vous voulez "fil de fer" le sous-processus stdin,stdout et stderr pour votre stdin,stdout,stderr et le fichier de lavabos, mais je sais que vous pouvez faire ceci:

import subprocess
callee = subprocess.Popen( ["python", "-i"],
                           stdin = subprocess.PIPE,
                           stdout = subprocess.PIPE,
                           stderr = subprocess.PIPE
                         )

Maintenant vous pouvez accéder callee.stdin, callee.stdout et callee.stderr comme les fichiers normaux, permettant à la "solution" ci-dessus de fonctionner. Si vous voulez obtenir le callee.returncode, vous aurez besoin de faire un appel supplémentaire à callee.poll().

soyez prudent avec l'écriture de callee.stdin: si le processus s'est terminé quand vous faites cela, une erreur peut être monté (sur Linux, j'obtiens IOError: [Errno 32] Broken pipe).

si vous ne voulez pas interagir avec le processus, vous pouvez utiliser le module subprocess très bien.

Exemple:

tester.py

import os
import sys

for file in os.listdir('.'):
    print file

sys.stderr.write("Oh noes, a shrubbery!")
sys.stderr.flush()
sys.stderr.close()

testing.py

import subprocess

p = subprocess.Popen(['python', 'tester.py'], stdout=subprocess.PIPE,
                     stdin=subprocess.PIPE, stderr=subprocess.PIPE)

stdout, stderr = p.communicate()
print stdout, stderr

Dans votre situation, vous pouvez simplement écrire stdout/stderr dans un fichier en premier. Vous pouvez envoyer des arguments à votre processus avec communicate aussi bien, bien que je n'ai pas été en mesure de comprendre comment interagir continuellement avec le sous-processus.

il y a des problèmes/bogues subtils en python liés au sous-processus.TUYAU: http://bugs.python.org/issue1652

apparemment ceci a été corrigé dans python3+, mais pas dans python 2.7 et plus ancien. Pour cela, vous devez utiliser: code.google.com/p/python-subprocess32/

essaye ceci :

import sys

class tee-function :

    def __init__(self, _var1, _var2) :

        self.var1 = _var1
        self.var2 = _var2

    def __del__(self) :

        if self.var1 != sys.stdout and self.var1 != sys.stderr :
            self.var1.close()
        if self.var2 != sys.stdout and self.var2 != sys.stderr :
            self.var2.close()

    def write(self, text) :

        self.var1.write(text)
        self.var2.write(text)

    def flush(self) :

        self.var1.flush()
        self.var2.flush()

stderrsav = sys.stderr

out = open(log, "w")

sys.stderr = tee-function(stderrsav, out)

j'ai écrit une chose qui enveloppe les commandes shell en Python.

avantages:

1. C'util capture stdout/stderr toujours
2. C'util fournit une option pour echo stdout/stderr vers stdout/stderr pour le processus
3. lorsque l'écho stdout/stderr Le out/err il n'y a pas de délai

inconvénient:

• fonctionne seulement sur bash / unix

source: https://gist.github.com/AndrewHoos/9f03c74988469b517a7a