如何获取外部命令与管道结合的输出

Question

我有这样的命令。

wmctrl -lp | awk '/gedit/ { print $1 }'

我想在Python脚本中获取它的输出，我试过这个代码

>>> import subprocess
>>> proc =  subprocess.Popen(["wmctrl -lp", "|","awk '/gedit/ {print $1}"], shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
>>> proc.stdout.readline()
'0x0160001b -1 6504   beer-laptop x-nautilus-desktop\n'
>>> proc.stdout.readline()
'0x0352f117  0 6963   beer-laptop How to get output from external command combine with Pipe - Stack Overflow - Chromium\n'
>>> proc.stdout.readline()
'0x01400003 -1 6503   beer-laptop Bottom Expanded Edge Panel\n'
>>> 

看起来我的代码有问题，只有 wmctrl -lp 被执行了，而 | awk '{print $1}' 被忽略了。
我期望的输出是 0x03800081

$ wmctrl -lp | awk '/gedit/ {print $1}'
0x03800081

请问有人能帮忙吗。

Answer 1

经过一些研究，我写了以下代码，它对我来说效果很好。这个代码基本上可以实时打印标准输出和错误输出。希望它能帮助到其他需要的人。

stdout_result = 1
stderr_result = 1


def stdout_thread(pipe):
    global stdout_result
    while True:
        out = pipe.stdout.read(1)
        stdout_result = pipe.poll()
        if out == '' and stdout_result is not None:
            break

        if out != '':
            sys.stdout.write(out)
            sys.stdout.flush()


def stderr_thread(pipe):
    global stderr_result
    while True:
        err = pipe.stderr.read(1)
        stderr_result = pipe.poll()
        if err == '' and stderr_result is not None:
            break

        if err != '':
            sys.stdout.write(err)
            sys.stdout.flush()


def exec_command(command, cwd=None):
    if cwd is not None:
        print '[' + ' '.join(command) + '] in ' + cwd
    else:
        print '[' + ' '.join(command) + ']'

    p = subprocess.Popen(
        command, stdout=subprocess.PIPE, stderr=subprocess.PIPE, cwd=cwd
    )

    out_thread = threading.Thread(name='stdout_thread', target=stdout_thread, args=(p,))
    err_thread = threading.Thread(name='stderr_thread', target=stderr_thread, args=(p,))

    err_thread.start()
    out_thread.start()

    out_thread.join()
    err_thread.join()

    return stdout_result + stderr_result

如果需要的话，我觉得把输出或错误信息收集到一个字符串里并返回是很简单的。

Answer 2

因为你给程序传递的是一个序列，所以它把管道当成了参数，就像你这样做：

wmctrl -lp "|"

这样一来，实际的管道操作就没了。

把它变成一个单一的字符串，确实应该能得到正确的结果：

>>> import subprocess as s
>>> proc = s.Popen("echo hello | grep e", shell=True, stdout=s.PIPE, stderr=s.PIPE)
>>> proc.stdout.readline()
'hello\n'
>>> proc.stdout.readline()
''

Answer 3

当你使用 shell=True 时，应该用一整行命令，而不是用一个数组。否则，你额外添加的参数会被当作命令行的参数来处理。根据subprocess 的文档：

在Unix系统中，如果 shell=True：如果 args 是一个字符串，那就表示要通过命令行执行的命令。如果 args 是一个序列（比如列表），那么第一个项目是命令字符串，后面的项目会被当作额外的命令行参数。

所以你的调用应该是：

subprocess.Popen("wmctrl -lp | sed /gedit/ '{print $1}'", shell=True, ...

我觉得你可能还有一个不匹配的单引号。