我想将subprocess.check_output()
与ps -A | grep 'process_name'
一起使用。
我尝试了各种解决方案,但到目前为止没有任有人可以指导我怎么做吗?
答案 0 :(得分:356)
要使用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
。
答案 1 :(得分:48)
或者您始终可以在子进程对象上使用communication方法。
cmd = "ps -A|grep 'process_name'"
ps = subprocess.Popen(cmd,shell=True,stdout=subprocess.PIPE,stderr=subprocess.STDOUT)
output = ps.communicate()[0]
print output
communication方法返回标准输出的元组和标准错误。
答案 2 :(得分:20)
请参阅有关使用子流程设置管道的文档: 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]
答案 3 :(得分:3)
另外,请尝试使用'pgrep'
命令而不是'ps -A | grep 'process_name'
答案 4 :(得分:3)
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
干杯,
答案 5 :(得分:2)
您可以在sh.py中尝试管道功能:</ p>
import sh
print sh.grep(sh.ps("-ax"), "process_name")
答案 6 :(得分:2)
import subprocess
ps = subprocess.run(['ps', '-A'], check=True, capture_output=True)
processNames = subprocess.run(['grep', 'process_name'], input=ps.stdout)
print(processNames.stdout)
答案 7 :(得分:1)
command = "ps -A | grep 'process_name'"
output = subprocess.check_output(["bash", "-c", command])
答案 8 :(得分:0)
在 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 中。