Как получить идентификатор процесса программы в Unix или Linux с помощью Python?



Я пишу некоторые сценарии мониторинга на Python, и я пытаюсь найти самый чистый способ получить идентификатор процесса любой случайной запущенной программы, учитывая имя этой программы



что-то вроде



ps -ef | grep MyProgram


Я мог бы разобрать вывод этого, однако я думал, что может быть лучший способ в python

336   9  

9 ответов:

попробовать pgrep. Его выходной формат намного проще и поэтому легче анализировать.

Если вы не ограничиваете себя стандартной библиотекой, мне нравится psutil для этого.

например, чтобы найти все процессы "python":

>>> import psutil
>>> [p.info for p in psutil.process_iter(attrs=['pid', 'name']) if 'python' in p.info['name']]
[{'name': 'python3', 'pid': 21947},
 {'name': 'python', 'pid': 23835}]

также: Python: как получить PID по имени процесса?

адаптация к предыдущим опубликованным ответам.

def getpid(process_name):
    import os
    return [item.split()[1] for item in os.popen('tasklist').read().splitlines()[4:] if process_name in item.split()]

getpid('cmd.exe')
['6560', '3244', '9024', '4828']

Для Windows

способ получить все pids программ на вашем компьютере без загрузки каких-либо модулей:

import os

pids = []
a = os.popen("tasklist").readlines()
for x in a:
      try:
         pids.append(int(x[29:34]))
      except:
           pass
for each in pids:
         print(each)

Если вы просто хотели одну программу или все программы с тем же именем, и вы хотели, чтобы убить процесс или что-то:

import os, sys, win32api

tasklistrl = os.popen("tasklist").readlines()
tasklistr = os.popen("tasklist").read()

print(tasklistr)

def kill(process):
     process_exists_forsure = False
     gotpid = False
     for examine in tasklistrl:
            if process == examine[0:len(process)]:
                process_exists_forsure = True
     if process_exists_forsure:
         print("That process exists.")
     else:
        print("That process does not exist.")
        raw_input()
        sys.exit()
     for getpid in tasklistrl:
         if process == getpid[0:len(process)]:
                pid = int(getpid[29:34])
                gotpid = True
                try:
                  handle = win32api.OpenProcess(1, False, pid)
                  win32api.TerminateProcess(handle, 0)
                  win32api.CloseHandle(handle)
                  print("Successfully killed process %s on pid %d." % (getpid[0:len(prompt)], pid))
                except win32api.error as err:
                  print(err)
                  raw_input()
                  sys.exit()
    if not gotpid:
       print("Could not get process pid.")
       raw_input()
       sys.exit()

   raw_input()
   sys.exit()

prompt = raw_input("Which process would you like to kill? ")
kill(prompt)

Это была просто вставка моей программы убийства процесса, я мог бы сделать ее намного лучше, но все в порядке.

С psutil:

(может быть установлен с [sudo] pip install psutil)

import psutil

# Get current process pid
current_process_pid = psutil.Process().pid
print(current_process_pid)  # e.g 12971

# Get pids by program name
program_name = 'chrome'
process_pids = [process.pid for process in psutil.process_iter() if process.name == program_name]
print(process_pids)  # e.g [1059, 2343, ..., ..., 9645]

для posix (Linux, BSD и др... нужно только смонтировать каталог /proc) проще работать с файлами ОС в /proc

работает на python 2 и 3 ( единственное отличие-это дерево исключений, поэтому "кроме исключения", который мне не нравится, но сохраняется для поддержания совместимости. Также можно было создать пользовательское исключение.)

#!/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]:
            # dirname is also the number of PID
            print('{0:<12} : {1}'.format(dirname, ' '.join(content)))

образец вывода (он работает как pgrep):

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

это упрощенный вариант ответа Фернандо. Это для Linux и Python 2 или 3. Никакая внешняя библиотека не требуется, и никакой внешний процесс не выполняется.

import glob

def get_command_pid(command):
    for path in glob.glob('/proc/*/comm'):
        if open(path).read().rstrip() == command:
            return path.split('/')[2]

будет возвращен только первый найденный процесс сопоставления, который хорошо работает для некоторых целей. Чтобы получить PIDs нескольких процессов сопоставления, вы можете просто заменить return С yield, а затем получить список с pids = list(get_command_pid(command)).

кроме того, как один выражение:

для одного процесса:

next(path.split('/')[2] for path in glob.glob('/proc/*/comm') if open(path).read().rstrip() == command)

для нескольких процессов:

[path.split('/')[2] for path in glob.glob('/proc/*/comm') if open(path).read().rstrip() == command]

задача может быть решена с помощью следующего фрагмента кода, [0:28] являющегося интервалом, где удерживается имя, в то время как [29:34] содержит фактический pid.

import os

program_pid = 0
program_name = "notepad.exe"

task_manager_lines = os.popen("tasklist").readlines()
for line in task_manager_lines:
    try:
        if str(line[0:28]) == program_name + (28 - len(program_name) * ' ': #so it includes the whitespaces
            program_pid = int(line[29:34])
            break
    except:
        pass

print(program_pid)

Comments

    Ничего не найдено.