blob: d1f0d933688286685521b7cc0d0ce042dfa57f86 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
#!/usr/bin/env python
# This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this file,
# You can obtain one at http://mozilla.org/MPL/2.0/.
import os
import mozinfo
import shlex
import subprocess
import sys
# determine the platform-specific invocation of `ps`
if mozinfo.isMac:
psarg = '-Acj'
elif mozinfo.isLinux:
psarg = 'axwww'
else:
psarg = 'ax'
def ps(arg=psarg):
"""
python front-end to `ps`
http://en.wikipedia.org/wiki/Ps_%28Unix%29
returns a list of process dicts based on the `ps` header
"""
retval = []
process = subprocess.Popen(['ps', arg], stdout=subprocess.PIPE)
stdout, _ = process.communicate()
header = None
for line in stdout.splitlines():
line = line.strip()
if header is None:
# first line is the header
header = line.split()
continue
split = line.split(None, len(header)-1)
process_dict = dict(zip(header, split))
retval.append(process_dict)
return retval
def running_processes(name, psarg=psarg, defunct=True):
"""
returns a list of
{'PID': PID of process (int)
'command': command line of process (list)}
with the executable named `name`.
- defunct: whether to return defunct processes
"""
retval = []
for process in ps(psarg):
# Support for both BSD and UNIX syntax
# `ps aux` returns COMMAND, `ps -ef` returns CMD
try:
command = process['COMMAND']
except KeyError:
command = process['CMD']
command = shlex.split(command)
if command[-1] == '<defunct>':
command = command[:-1]
if not command or not defunct:
continue
if 'STAT' in process and not defunct:
if process['STAT'] == 'Z+':
continue
prog = command[0]
basename = os.path.basename(prog)
if basename == name:
retval.append((int(process['PID']), command))
return retval
def get_pids(name):
"""Get all the pids matching name"""
if mozinfo.isWin:
# use the windows-specific implementation
import wpk
return wpk.get_pids(name)
else:
return [pid for pid,_ in running_processes(name)]
if __name__ == '__main__':
pids = set()
for i in sys.argv[1:]:
pids.update(get_pids(i))
for i in sorted(pids):
print i
|
