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
|
import locale
import os
import subprocess
try:
import psutil
except Exception:
psutil = None
def normpath(location):
if isinstance(location, (list, tuple)):
return os.pathsep.join(normpath(l) for l in location)
return os.path.normcase(os.path.realpath(os.path.expanduser(location.strip(';:'))))
def location_in_path(location, path):
return normpath(location) in (normpath(p) for p in path.split(os.pathsep) if p != '')
def in_current_path(location):
return location_in_path(location, os.environ.get('PATH', ''))
def ensure_parent_dir_exists(path):
parent_dir = os.path.dirname(os.path.abspath(path))
if not os.path.isdir(parent_dir):
os.makedirs(parent_dir)
def get_flat_output(command, sep=os.pathsep, **kwargs):
process = subprocess.Popen(command, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE, **kwargs)
output = process.communicate()[0].decode(locale.getpreferredencoding(False)).strip()
# We do this because the output may contain new lines.
lines = [line.strip() for line in output.splitlines()]
return sep.join(line for line in lines if line)
def get_parent_process_name():
# We want this to never throw an exception
try:
if psutil:
try:
pid = os.getpid()
process = psutil.Process(pid)
ppid = process.ppid()
pprocess = psutil.Process(ppid)
return pprocess.name()
except Exception:
pass
ppid = os.getppid()
process_name = subprocess.check_output(['ps', '-o', 'args=', str(ppid)]).decode('utf-8')
return process_name.strip().lstrip("-")
except Exception:
pass
return ''
|
