1
|
|
|
# -*- coding: utf-8 -*- |
2
|
|
|
"""Print where on the path an executable is located. |
3
|
|
|
""" |
4
|
|
|
from __future__ import print_function |
5
|
|
|
import sys |
6
|
|
|
import os |
7
|
|
|
from stat import ST_MODE, S_IXUSR, S_IXGRP, S_IXOTH |
8
|
|
|
|
9
|
|
|
|
10
|
|
|
def get_executable(name): |
11
|
|
|
for result in which(name): |
12
|
|
|
return result |
13
|
|
|
return None |
14
|
|
|
|
15
|
|
|
|
16
|
|
|
def get_path_directories(): |
17
|
|
|
pth = os.environ['PATH'] |
18
|
|
|
if sys.platform == 'win32' and os.environ.get("BASH"): |
19
|
|
|
# winbash has a bug.. |
20
|
|
|
if pth[1] == ';': # pragma: nocover |
21
|
|
|
pth = pth.replace(';', ':', 1) |
22
|
|
|
return pth.split(os.pathsep) |
23
|
|
|
|
24
|
|
|
|
25
|
|
|
def is_executable(fname): |
26
|
|
|
return os.stat(fname)[ST_MODE] & (S_IXUSR | S_IXGRP | S_IXOTH) |
27
|
|
|
|
28
|
|
|
|
29
|
|
|
def which(filename, interactive=False, verbose=False): |
30
|
|
|
exe = os.environ.get('PATHEXT', ['.cmd', '.bat', '.exe', '.com']) |
31
|
|
|
|
32
|
|
|
name, ext = os.path.splitext(filename) |
33
|
|
|
if ext and (ext in exe): # pragma: nocover |
34
|
|
|
exe = [] |
35
|
|
|
|
36
|
|
|
def match(filenames): |
37
|
|
|
res = set() |
38
|
|
|
for fname in filenames: |
39
|
|
|
if fname == filename: # pragma: nocover |
40
|
|
|
res.add(fname) |
41
|
|
|
continue |
42
|
|
|
|
43
|
|
|
fn_name, fn_ext = os.path.splitext(fname) |
44
|
|
|
if name == fn_name: |
45
|
|
|
for suffix in exe: |
46
|
|
|
if name + fn_ext == fname: |
47
|
|
|
res.add(fname) |
48
|
|
|
|
49
|
|
|
return sorted(res) |
50
|
|
|
|
51
|
|
|
returnset = set() |
52
|
|
|
found = False |
53
|
|
|
for pth in get_path_directories(): |
54
|
|
|
if not pth.strip(): # pragma: nocover |
55
|
|
|
continue |
56
|
|
|
|
57
|
|
|
if verbose: # pragma: nocover |
58
|
|
|
print('checking pth..') |
59
|
|
|
|
60
|
|
|
try: |
61
|
|
|
fnames = os.listdir(pth) |
62
|
|
|
except: # pragma: nocover |
63
|
|
|
continue |
64
|
|
|
|
65
|
|
|
matched = match(fnames) |
66
|
|
|
|
67
|
|
|
if matched: |
68
|
|
|
for m in matched: |
69
|
|
|
found_file = os.path.normcase(os.path.normpath(os.path.join(pth, m))) |
70
|
|
|
if found_file not in returnset: |
71
|
|
|
if is_executable(found_file): |
72
|
|
|
yield found_file |
73
|
|
|
returnset.add(found_file) |
74
|
|
|
found = True |
75
|
|
|
|
76
|
|
|
if not found and interactive: # pragma: nocover |
77
|
|
|
print("Couldn't find %r anywhere on the path.." % filename) |
78
|
|
|
sys.exit(1) |
79
|
|
|
|
80
|
|
|
|
81
|
|
|
if __name__ == "__main__": # pragma: nocover |
82
|
|
|
for _fname in which(sys.argv[1], interactive=True, verbose='-v' in sys.argv): |
83
|
|
|
print(_fname) |
84
|
|
|
sys.exit(0) |
85
|
|
|
|