如何在python中查找运行特定命令的进程数

Question

的输出

ps uaxw | egrep 'kms' | grep -v 'grep'

产量：

user1  8148  0.0  0.0 128988  3916 pts/8    S+   18:34   0:00 kms 
user2 11782  0.7  0.3 653568 56564 pts/14   Sl+  20:29   0:01 kms

显然有两个进程在运行该程序。我想将此数字（此处为 2）存储为变量。关于如何在 python 中执行此操作的任何建议？

我尝试了以下方法：

procs = subprocess.check_output("ps uaxw | egrep 'kmns' |grep -v 'grep'",shell=True)

但我得到以下信息（我认为当作业当前未运行时，因此运行作业的进程数为零）：

Traceback（最近一次调用最后一次）：文件“”，第 1 行，in 文件“/usr/lib64/python2.7/subprocess.py”，第 573 行，在 检查输出 raise CalledProcessError(retcode, cmd, output=output) subprocess.CalledProcessError: Command 'ps uaxw | egrep 'kmns' |grep -v 'grep'' 返回非零退出状态 1

我该如何解决这个问题？

顺便说一句，这是我编写的用于检测我的系统是否繁忙的函数（这意味着如果 cpu 的数量 > 安装的总数量，并且负载 avg > 0.9 per cpu）：

def busy():
    import subprocess
    output = subprocess.check_output("uptime", shell=False)
    words = output.split()
    sys.stderr.write("%s\n"%(output)) 
    procs = subprocess.check_output("ps uaxw | egrep '(kmns)' | grep -v 'grep'", shell=True)
    kmns_wrds = procs.split("\n")
    wrds=words[9]
    ldavg=float(wrds.strip(','))+0.8
    sys.stderr.write("%s %s\n"%(ldavg,len(kmns_wrds)))
    return max(ldavg, len(kmns_wrds)) > ncpus

上面的调用者：

def wait_til_free(myseconds):
    while busy():
        import time
        import sys
        time.sleep(myseconds)
        """ sys.stderr.write("Waiting %s seconds\n"%(myseconds)) """

这基本上告诉系统在所有 cpu 被占用时等待。

有什么建议吗？

非常感谢！

Answer 1

如果您打算使用一个大的 shell 命令来完成这一切，只需将 -c 参数添加到 grep，这样它就会为您提供行数而不是实际行数：

$ ps uaxw |grep python |grep -v grep
abarnert         1028   0.0  0.3  2529488  55252 s000  S+    9:46PM   0:02.80 /Library/Frameworks/Python.framework/Versions/3.4/Resources/Python.app/Contents/MacOS/Python /Library/Frameworks/Python.framework/Versions/3.4/bin/ipython3
abarnert         9639   0.0  0.1  2512928  19228 s002  T     3:06PM   0:00.40 /System/Library/Frameworks/Python.framework/Versions/2.7/Resources/Python.app/Contents/MacOS/Python /usr/local/bin/ipython2
$
$ ps uaxw |grep python |grep -c -v grep
2

当然，您可以通过在末尾添加 | wc -l 或通过计算 Python 中的行数来使这更复杂，但为什么呢？

---

另外，为什么还要涉及外壳？您可以像运行grep 一样轻松地在Python 中进行搜索——然后您就不会遇到不小心创建了grep 进程的问题，ps 将重复匹配您的搜索然后需要grep -v 退出：

procs = subprocess.check_output(['ps', 'uaxw']).splitlines()
kms_procs = [proc for proc in procs if 'kms' in proc]
count = len(kms_procs)

---

或者，更简单地说，不要让ps 给你一大堆你不想要的信息，然后想办法忽略它，只要问你想要的信息：

procs = subprocess.check_output(['ps', '-a', '-c', '-ocomm=']).splitlines()
count = procs.count('kms')

---

或者，更简单的是，安装psutil，甚至不尝试运行子进程并解析它们的输出：

count = sum(1 for proc in psutil.process_iter() if proc.name() == 'kms')

Answer 2

如果你想模拟管道，你可以使用 Popen：

p1 = Popen(["ps", "uaxw"], stdout=PIPE)
p2 = Popen(["grep", 'kms'], stdout=PIPE, stdin=p1.stdout)
p1.stdout.close()

out,_ = p2.communicate()
print(len(out.splitlines()))

或者使用pgrep（如果有的话）：

count  = check_output(["pgrep", "-c", "kms"])

您可能会从两者获得不同的输出，因为 pgrep 仅获取可执行文件的名称，但 ps -aux 与 ps -a 也是如此。