Làm cách nào để lấy PID bằng tên quy trình trong Python?

Question

Có cách nào tôi có thể lấy PID bằng tên quy trình trong Python không?

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 

Ví dụ: tôi cần nhận 3110 bởi chrome.

Answer 1

Bạn có thể lấy pid của tiến trình theo tên sử dụng pidof qua subprocess.check_output:

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]) sẽ chạy lệnh như "pidof process_name", Nếu mã trở lại là khác không nó sẽ gây nên CalledProcessError.

Để xử lý nhiều mục và dàn diễn viên để ints:

from subprocess import check_output 
def get_pid(name): 
    return map(int,check_output(["pidof",name]).split()) 

Trong [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] 

Hoặc pas các -s cờ để có được một pid duy nhất:

def get_pid(name): 
    return int(check_output(["pidof","-s",name])) 

In [25]: get_pid("chrome") 
Out[25]: 27698 

Answer 2

bạn cũng có thể sử dụng pgrep, trong prgep bạn cũng có thể cung cấp cho mô hình cho trận đấu

import subprocess 
child = subprocess.Popen(['pgrep','program_name'], stdout=subprocess.PIPE, shell=True) 
result = child.communicate()[0] 

bạn cũng có thể sử dụng awk với ps như thế này

ps aux | awk '/name/{print $2}' 

Answer 3

Đối posix (Linux, BSD, vv ... chỉ cần/thư mục proc được gắn kết) dễ dàng hơn khi làm việc với các tệp os trong/proc. Trăn tinh khiết của nó, không cần phải gọi các chương trình shell bên ngoài.

Hoạt động trên python 2 và 3 (Sự khác biệt duy (2to3) là cây ngoại lệ, do đó "trừ ngoại lệ", mà tôi không thích nhưng vẫn giữ để duy trì khả năng tương thích. Ngoài ra có thể đã tạo ngoại lệ tùy chỉnh.)

#!/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))) 

Sample Output (nó hoạt động như pgrep):

phoemur ~/python $ ./pgrep.py bash 
1487   : -bash 
1779   : /bin/bash 

Answer 4

Để cải thiện câu trả lời của Padraic: khi check_output trả về một tổ chức phi zero mã, nó sẽ gây nên CalledProcessError. Điều này xảy ra khi quá trình không tồn tại hoặc không chạy.

Những gì tôi sẽ làm gì để bắt ngoại lệ này là:

#!/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") 

Sản lượng:

$ python pidproc.py 
list of PIDS = 31840, 31841, 41942 

Answer 5

Hoàn dụ dựa trên tuyệt vời @answer Hackaholic của:

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()]