Python 3.4.3 subprocess.Popen Ausgabe des Befehls ohne Piping erhalten?

Question

Ich versuche, die Ausgabe eines Befehls zu einer Variablen ohne den Befehl zu übergeben, der denkt, dass es piped ist. Der Grund dafür ist, dass der fragliche Befehl unformatierten Text als Ausgabe gibt, wenn er ausgegeben wird, aber es gibt farbig formatierten Text, wenn er vom Terminal aus ausgeführt wird. Ich brauche diesen farbig formatierten Text.

Bis jetzt habe ich ein paar Dinge ausprobiert. Ich habe versucht, Popen wie so:

output = subprocess.Popen(command, stdout=subprocess.PIPE) 
output = output.communicate()[0] 
output = output.decode() 
print(output) 

Dies läßt mich die Ausgabe drucken, aber es gibt mir die unformatierte Ausgabe, die ich bekomme, wenn der Befehl geleitet wird. Das macht Sinn, da ich es hier im Python-Code übergebe. Aber ich bin neugierig, ob es eine Möglichkeit gibt, die Ausgabe dieses Befehls direkt einer Variablen zuzuordnen, ohne dass der Befehl die verrohrte Version von sich selbst ausführt.

Ich habe auch versucht, die folgende Version, die statt auf check_output beruht:

output = subprocess.check_output(command) 
output = output.decode() 
print(output) 

Und wieder ich die gleiche unformatierte Ausgabe zu erhalten, die den Befehl, wenn der Befehl geleitet wird.

Gibt es eine Möglichkeit, die formatierte Ausgabe zu erhalten, die Ausgabe, die der Befehl normalerweise vom Terminal geben würde, wenn es nicht verrohrt wird?

Answer 1

pexpect Verwendung:

2.py:

import sys 

if sys.stdout.isatty(): 
    print('hello') 
else: 
    print('goodbye') 

subprocess:

import subprocess 

p = subprocess.Popen(
    ['python3.4', '2.py'], 
    stdout=subprocess.PIPE 
) 

print(p.stdout.read()) 

--output:-- 
goodbye 

pexpect:

import pexpect 

child = pexpect.spawn('python3.4 2.py') 

child.expect(pexpect.EOF) 
print(child.before) #Print all the output before the expectation. 

--output:-- 
hello 

Hier ist es mit grep --colour=auto :

import subprocess 

p = subprocess.Popen(
    ['grep', '--colour=auto', 'hello', 'data.txt'], 
    stdout=subprocess.PIPE 
) 

print(p.stdout.read()) 

import pexpect 

child = pexpect.spawn('grep --colour=auto hello data.txt') 
child.expect(pexpect.EOF) 
print(child.before) 

--output:-- 
b'hello world\n' 
b'\x1b[01;31mhello\x1b[00m world\r\n' 

Answer 2

Viele Programme schalten Farbdruckcodes automatisch aus, wenn sie erkennen, dass sie nicht direkt an ein Terminal angeschlossen sind. Viele Programme haben ein Flag, so dass Sie die Farbausgabe erzwingen können. Sie könnten dieses Flag zu Ihrem Prozessaufruf hinzufügen. Zum Beispiel:

grep "search term" inputfile.txt 
# prints colour to the terminal in most OSes 

grep "search term" inputfile.txt | less 
# output goes to less rather than terminal, so colour is turned off 

grep "search term" inputfile.txt --color | less 
# forces colour output even when not connected to terminal 

Seien Sie gewarnt, obwohl. Die eigentliche Farbausgabe erfolgt über das Terminal. Das Terminal interpretiert Sonderzeichen-Espace-Codes und ändert die Textfarbe und die Hintergrundfarbe entsprechend. Ohne das Terminal, um die Farbcodes zu interpretieren, sehen Sie nur den Text in schwarz mit diesen durchgestrichenen Escape-Codes.

Answer 3

Ja, Sie können die pty module verwenden.

>>> import subprocess 
>>> p = subprocess.Popen(["ls", "--color=auto"], stdout=subprocess.PIPE) 
>>> p.communicate()[0] 
# Output does not appear in colour 

Mit pty:

import subprocess 
import pty 
import os 

master, slave = pty.openpty() 
p = subprocess.Popen(["ls", "--color=auto"], stdout=slave) 
p.communicate() 
print(os.read(master, 100)) # Print 100 bytes 
# Prints with colour formatting info 

Hinweis aus der Dokumentation:

Da Pseudo-Terminal Handling sehr plattformabhängig ist, gibt -Code ist es nur für Linux zu tun. (Der Linux-Code sollte auf anderen Plattformen arbeiten, aber noch nicht getestet.)

A weniger als schöne Art und Weise, die gesamte Ausgabe in einem Rutsch zu Ende zu lesen:

def num_bytes_readable(fd): 
    import array 
    import fcntl 
    import termios 
    buf = array.array('i', [0]) 
    if fcntl.ioctl(fd, termios.FIONREAD, buf, 1) == -1: 
     raise Exception("We really should have had data") 
    return buf[0] 

print(os.read(master, num_bytes_readable(master))) 

Edit: schönere Art und Weise, den Inhalt sofort dank @Antti Haapala bekommen:

os.close(slave) 
f = os.fdopen(master) 
print(f.read()) 

Edit: Menschen sind zu Recht darauf hin, dass dies eine Sackgasse, wenn der Prozess eine große Leistung erzeugt, so @Antti Haapala des Antwort ist besser.

Answer 4

Ein Arbeits polyglot Beispiel (funktioniert auf das gleiche für Python 2 und Python 3) mit pty.

import subprocess 
import pty 
import os 
import sys 

master, slave = pty.openpty() 
# direct stderr also to the pty! 
process = subprocess.Popen(
    ['ls', '-al', '--color=auto'], 
    stdout=slave, 
    stderr=subprocess.STDOUT 
) 

# close the slave descriptor! otherwise we will 
# hang forever waiting for input 
os.close(slave) 

def reader(fd): 
    try: 
     while True: 
      buffer = os.read(fd, 1024) 
      if not buffer: 
       return 

      yield buffer 

    # Unfortunately with a pty, an 
    # IOError will be thrown at EOF 
    # On Python 2, OSError will be thrown instead. 
    except (IOError, OSError) as e: 
     pass 

# read chunks (yields bytes) 
for i in reader(master): 
    # and write them to stdout file descriptor 
    os.write(1, b'<chunk>' + i + b'</chunk>')