Python 3.4.3 subprocess.Popen uzyskać wyjście polecenia bez rurociągów?

Question

Próbuję przypisać wyjście polecenia do zmiennej bez polecenia myślenie, że jest podłączony. Powodem tego jest to, że dane polecenie podaje niesformatowany tekst jako wynik, jeśli jest podłączony, ale daje tekst sformatowany w kolorze, jeśli jest uruchamiany z terminala. Muszę uzyskać ten sformatowany tekst w kolorze.

Do tej pory próbowałem kilku rzeczy. Próbowałem POPEN tak:

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

To pozwoli mi wydrukować wyjście, ale to daje mi wyjście niesformatowany że dostanę gdy polecenie jest rurami. To ma sens, ponieważ robię to tutaj w kodzie Pythona. Ale jestem ciekawy, czy istnieje sposób, aby przypisać wyjście tego polecenia bezpośrednio do zmiennej, bez polecenia uruchamiającego swoją wersję potokową.

Próbowałem również następującą wersję, która opiera się na check_output Zamiast:

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

I znowu uzyskać taki sam efekt niesformatowany że powraca dowodzenia, gdy polecenie jest przetłaczany.

Czy istnieje sposób na uzyskanie sformatowanego wyjścia, na wyjściu, które polecenie zwykle wydaje z terminala, gdy nie jest podłączone?

Answer 1

Korzystanie pexpect:

2.py:

import sys 

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

podproces:

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 

Oto ona z 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

Wiele programów automatycznie wyłącza kody drukowania w kolorze, gdy wykryją, że nie są podłączone bezpośrednio do terminala. Wiele programów będzie miało flagę, dzięki czemu możesz wymusić wydruk kolorowy. Możesz dodać tę flagę do swojego wywołania procesu. Na przykład:

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 

Ostrzegam jednak. Rzeczywisty wydruk kolorów jest wykonywany przez terminal. Terminal interpretuje kody espace specjalnego znaku i odpowiednio zmienia kolor tekstu i kolor tła. Bez terminala do interpretowania kodów kolorów zobaczysz tylko tekst w kolorze czarnym z tymi kodami ucieczkowymi przeplatanymi w całym tekście.

Answer 3

Tak, można użyć pty module.

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

Z 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 

Uwaga od docs:

Ponieważ obsługa pseudo-terminal jest wysoce zależny od platformy, tam jest kod to zrobić tylko dla systemu Linux. (Kod Linux ma pracować na innych platformach, ale nie został jeszcze przetestowany.)

Mniej niż piękny sposób czytania całe wyjście do końca za jednym zamachem:

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: ładniejszy sposób na uzyskanie zawartości naraz dzięki @Antti Haapala:

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

Edit: ludzie mają prawo zwrócić uwagę, że będzie to impas, jeśli proces generuje dużą moc, więc @Antti Haapala użytkownika odpowiedź jest lepsza.

Answer 4

Działający przykład polyglot (identycznie dla Pythonie 2 Python 3), przy 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>')