Python 3.4.3 subprocess.Popen 在没有管道的情况下获取命令输出？

Question

我正在尝试将命令的输出分配给变量，而命令不会认为它正在通过管道传输。这样做的原因是，有问题的命令在通过管道传输时会给出未格式化的文本作为输出，但如果是从终端运行，它会给出彩色格式的文本。我需要得到这个颜色格式的文本。

到目前为止，我已经尝试了一些事情。我已经像这样尝试过 Popen：

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

这将让我打印输出，但它给了我在命令通过管道时得到的未格式化输出。这是有道理的，因为我在 Python 代码中使用了管道。但是我很好奇是否有一种方法可以将此命令的输出直接分配给一个变量，而无需该命令运行其自身的管道版本。

我还尝试了以下依赖 check_output 的版本：

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

我再次得到与命令通过管道传输时命令返回的相同的未格式化输出。

有没有办法获得格式化的输出，当它没有被管道传输时，命令通常会从终端给出输出？

Answer 1

使用pexpect：

2.py：

import sys

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

子进程：

import subprocess

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

print(p.stdout.read())

--output:--
goodbye

期待：

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

这里是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

是的，您可以使用pty module。

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

与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

来自文档的注释：

由于伪终端处理高度依赖于平台，因此 是仅适用于 Linux 的代码。 （Linux代码应该可以工作 在其他平台上，但尚未经过测试。）

一口气读完整个输出的一种不太漂亮的方式：

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)))

编辑：感谢@Antti Haapala，更好地立即获取内容：

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

编辑：人们正确地指出，如果该过程产生大量输出，这将死锁，所以@Antti Haapala 的答案更好。

Answer 3

使用pty 的多语言工作示例（Python 2 和 Python 3 的工作方式相同）。

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>')

Answer 4

许多程序在检测到未直接连接到终端时会自动关闭彩色打印代码。许多程序都会有一个标志，因此您可以强制颜色输出。您可以将此标志添加到您的流程调用中。例如：

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 

请注意。实际的颜色输出由终端完成。终端解释特殊字符 espace 代码并相应地更改文本颜色和背景颜色。如果没有终端来解释颜色代码，您只会看到黑色文本，其中散布着这些转义代码。