psutil

1
#!/usr/bin/env python3
2

3
# Copyright (c) 2009, Giampaolo Rodola', karthikrev. All rights reserved.
4
# Use of this source code is governed by a BSD-style license that can be
5
# found in the LICENSE file.
6

7

8
"""A clone of 'pidof' cmdline utility.
9

10
$ pidof python
11
1140 1138 1136 1134 1133 1129 1127 1125 1121 1120 1119
12
"""
13

14
from __future__ import print_function
15

16
import sys
17

18
import psutil
19

20

21
def pidof(pgname):
22
    pids = []
23
    for proc in psutil.process_iter(['name', 'cmdline']):
24
        # search for matches in the process name and cmdline
25
        if (
26
            proc.info['name'] == pgname
27
            or proc.info['cmdline']
28
            and proc.info['cmdline'][0] == pgname
29
        ):
30
            pids.append(str(proc.pid))
31
    return pids
32

33

34
def main():
35
    if len(sys.argv) != 2:
36
        sys.exit('usage: %s pgname' % __file__)
37
    else:
38
        pgname = sys.argv[1]
39
    pids = pidof(pgname)
40
    if pids:
41
        print(" ".join(pids))
42

43

44
if __name__ == '__main__':
45
    main()
46