Pythonでプロセス名でPIDを取得する方法はありますか?
PID USER PR NI VIRT RES SHR S %CPU %MEM TIME+ COMMAND
3110 meysam 20 0 971m 286m 63m S 14.0 7.9 14:24.50 chrome
たとえば、3110
by chrome
。
subprocess.check_output を使用して、pidof
を使用して名前でプロセスのpidを取得できます。
_from subprocess import check_output
def get_pid(name):
return check_output(["pidof",name])
In [5]: get_pid("Java")
Out[5]: '23366\n'
_
check_output(["pidof",name])
は、コマンドを_"pidof process_name"
_として実行します。戻りコードがゼロ以外の場合、CalledProcessErrorが発生します。
複数のエントリを処理し、intにキャストするには:
_from subprocess import check_output
def get_pid(name):
return map(int,check_output(["pidof",name]).split())
_
[21]で:get_pid( "chrome")
_Out[21]:
[27698, 27678, 27665, 27649, 27540, 27530, 27517, 14884, 14719, 13849, 13708, 7713, 7310, 7291, 7217, 7208, 7204, 7189, 7180, 7175, 7166, 7151, 7138, 7127, 7117, 7114, 7107, 7095, 7091, 7087, 7083, 7073, 7065, 7056, 7048, 7028, 7011, 6997]
_
または、_-s
_フラグを渡して単一のpidを取得します。
_def get_pid(name):
return int(check_output(["pidof","-s",name]))
In [25]: get_pid("chrome")
Out[25]: 27698
_
Posix(Linux、BSDなど、マウントする必要があるのは/ procディレクトリのみです)の場合、/ procのosファイルを操作する方が簡単です。その純粋なpythonで、外部でシェルプログラムを呼び出す必要はありません。
python 2 and 3(唯一の違い(2to3)は例外ツリーです。したがって、 "except、これは嫌いですが互換性を維持するために維持されます。また、カスタム例外を作成することもできました。)
#!/usr/bin/env python
import os
import sys
for dirname in os.listdir('/proc'):
if dirname == 'curproc':
continue
try:
with open('/proc/{}/cmdline'.format(dirname), mode='rb') as fd:
content = fd.read().decode().split('\x00')
except Exception:
continue
for i in sys.argv[1:]:
if i in content[0]:
print('{0:<12} : {1}'.format(dirname, ' '.join(content)))
サンプル出力(pgrepのように機能します):
phoemur ~/python $ ./pgrep.py bash
1487 : -bash
1779 : /bin/bash
pgrep
を使用することもできます。prgep
で一致するパターンを指定することもできます
import subprocess
child = subprocess.Popen(['pgrep','program_name'], stdout=subprocess.PIPE, Shell=True)
result = child.communicate()[0]
このようなpsでawk
を使用することもできます
ps aux | awk '/name/{print $2}'
優れた@Hackaholicの answer に基づく完全な例:
def get_process_id(name):
"""Return process ids found by (partial) name or regex.
>>> get_process_id('kthreadd')
[2]
>>> get_process_id('watchdog')
[10, 11, 16, 21, 26, 31, 36, 41, 46, 51, 56, 61] # ymmv
>>> get_process_id('non-existent process')
[]
"""
child = subprocess.Popen(['pgrep', '-f', name], stdout=subprocess.PIPE, Shell=False)
response = child.communicate()[0]
return [int(pid) for pid in response.split()]
Padraicの答えを改善するには:when check_output
はゼロ以外のコードを返し、CalledProcessErrorを発生させます。これは、プロセスが存在しないか実行されていないときに発生します。
この例外をキャッチするために私がすることは:
#!/usr/bin/python
from subprocess import check_output, CalledProcessError
def getPIDs(process):
try:
pidlist = map(int, check_output(["pidof", process]).split())
except CalledProcessError:
pidlist = []
print 'list of PIDs = ' + ', '.join(str(e) for e in pidlist)
if __name__ == '__main__':
getPIDs("chrome")
出力:
$ python pidproc.py
list of PIDS = 31840, 31841, 41942
OSがUnixベースの場合、次のコードを使用します。
import os
def check_process(name):
output = []
cmd = "ps -aef | grep -i '%s' | grep -v 'grep' | awk '{ print $2 }' > /tmp/out"
os.system(cmd % name)
with open('/tmp/out', 'r') as f:
line = f.readline()
while line:
output.append(line.strip())
line = f.readline()
if line.strip():
output.append(line.strip())
return output
次に、それを呼び出して、すべてのPIDを取得するプロセス名を渡します。
>>> check_process('firefox')
['499', '621', '623', '630', '11733']
Python 3.5、 subprocess.run() が subprocess.check_output() よりも推奨されているため、
>>> int(subprocess.run(["pidof", "-s", "your_process"], stdout=subprocess.PIPE).stdout)
また、Python 3.7なので、capture_output=true
パラメータは、stdoutとstderrをキャプチャします。
>>> int(subprocess.run(["pidof", "-s", "your process"], capture_output=True).stdout)