如何通过管道使用“subprocess”命令

问题描述：

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

解决方案 1：

要将管道与subprocess模块一起使用，您可以通过shell=True但请注意安全注意事项。不鼓励使用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输出。

解决方案 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)

通信方法返回标准输出和标准错误的元组。

解决方案 3：

使用inputsubprocess.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.decode('utf-8').strip())

解决方案 4：

请参阅有关使用子进程设置管道的文档：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]

解决方案 5：

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

解决方案 6：

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

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

解决方案 7：

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

解决方案 8：

我认为启动一个 shell 只是为了享受流水线的乐趣并不那么优雅。

以下代码使用本subprocess机管道支持并且确实有效。

您可以轻松修改它以向管道添加两个以上的进程。

#!/usr/bin/env python3

import subprocess


def ps_grep(pattern):
    # First command-line
    ps_command = ["ps", "-A"]

    # Second command-line
    grep_command = ["grep", pattern]

    # Launch first process
    ps_process = subprocess.Popen(ps_command, stdout=subprocess.PIPE)

    # Launch second process and connect it to the first one
    grep_process = subprocess.Popen(
        grep_command, stdin=ps_process.stdout, stdout=subprocess.PIPE
    )

    # Let stream flow between them
    output, _ = grep_process.communicate()

    return output.decode()


if __name__ == "__main__":
    print(ps_grep("python"))

解决方案 9：

我正在回答一个老问题，因为没有人提到shlex。在Unix中，您可以使用管道安全地构造命令shlex.join()，如以下示例所示。

import shlex
import subprocess

print(subprocess.check_output(
    shlex.join(['ps', '-ef']) + ' | ' +
    shlex.join(['grep', 'ps -ef']), # The arguments will be quoted.
    shell=True
).decode('utf-8'))