如何使用带有管道的`subprocess`命令

Question

我想将subprocess.check_output() 与ps -A | grep 'process_name' 一起使用。 我尝试了各种解决方案，但到目前为止没有任何效果。有人可以指导我怎么做吗？

Answer 1

要将管道与subprocess 模块一起使用，您必须传递shell=True。

但是，出于各种原因，这并不是真正可取的，尤其是安全性。相反，分别创建 ps 和 grep 进程，并将输出从一个管道传输到另一个，如下所示：

ps = subprocess.Popen(('ps', '-A'), stdout=subprocess.PIPE)
output = subprocess.check_output(('grep', 'process_name'), stdin=ps.stdout)
ps.wait()

但是，在您的特定情况下，简单的解决方案是调用 subprocess.check_output(('ps', '-A'))，然后在输出上调用 str.find。

Answer 2

或者你总是可以在子进程对象上使用通信方法。

cmd = "ps -A|grep 'process_name'"
ps = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
output = ps.communicate()[0]
print(output)

communicate 方法返回一个标准输出和标准错误的元组。

Answer 3

请参阅有关使用子流程设置管道的文档：http://docs.python.org/2/library/subprocess.html#replacing-shell-pipeline

我还没有测试过下面的代码示例，但它应该大致是你想要的：

query = "process_name"
ps_process = Popen(["ps", "-A"], stdout=PIPE)
grep_process = Popen(["grep", query], stdin=ps_process.stdout, stdout=PIPE)
ps_process.stdout.close()  # Allow ps_process to receive a SIGPIPE if grep_process exits.
output = grep_process.communicate()[0]

Answer 4

使用subprocess.run

import subprocess
    
ps = subprocess.run(['ps', '-A'], check=True, capture_output=True)
processNames = subprocess.run(['grep', 'process_name'],
                              input=ps.stdout, capture_output=True)
print(processNames.stdout)

Answer 5

你可以试试sh.py中的管道功能：

import sh
print sh.grep(sh.ps("-ax"), "process_name")

Answer 6

另外，请尝试使用'pgrep' 命令而不是'ps -A | grep 'process_name'

Answer 7

JKALAVIS 解决方案很好，但是我会添加一个改进以使用 shlex 而不是 SHELL=TRUE。下面我正在找出查询时间

#!/bin/python
import subprocess
import shlex

cmd = "dig @8.8.4.4 +notcp www.google.com|grep 'Query'"
ps = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
output = ps.communicate()[0]
print(output)

Answer 8

command = "ps -A | grep 'process_name'"
output = subprocess.check_output(["bash", "-c", command])

Answer 9

Python 3.5之后还可以使用：

    import subprocess

    f = open('test.txt', 'w')
    process = subprocess.run(['ls', '-la'], stdout=subprocess.PIPE, universal_newlines=True)
    f.write(process.stdout)
    f.close()

命令的执行是阻塞的，输出将在process.stdout中。