which.py 1.59 KB
Newer Older
1
#! /usr/bin/env python3
Guido van Rossum's avatar
Guido van Rossum committed
2 3 4

# Variant of "which".
# On stderr, near and total misses are reported.
5
# '-l<flags>' argument adds ls -l<flags> of each file found.
Guido van Rossum's avatar
Guido van Rossum committed
6

7 8 9
import sys
if sys.path[0] in (".", ""): del sys.path[0]

10
import sys, os
Guido van Rossum's avatar
Guido van Rossum committed
11 12 13
from stat import *

def msg(str):
Tim Peters's avatar
Tim Peters committed
14
    sys.stderr.write(str + '\n')
Guido van Rossum's avatar
Guido van Rossum committed
15

16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
def main():
    pathlist = os.environ['PATH'].split(os.pathsep)

    sts = 0
    longlist = ''

    if sys.argv[1:] and sys.argv[1][:2] == '-l':
        longlist = sys.argv[1]
        del sys.argv[1]

    for prog in sys.argv[1:]:
        ident = ()
        for dir in pathlist:
            filename = os.path.join(dir, prog)
            try:
                st = os.stat(filename)
32
            except OSError:
33 34 35
                continue
            if not S_ISREG(st[ST_MODE]):
                msg(filename + ': not a disk file')
Tim Peters's avatar
Tim Peters committed
36
            else:
37
                mode = S_IMODE(st[ST_MODE])
38
                if mode & 0o111:
39
                    if not ident:
40
                        print(filename)
41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
                        ident = st[:3]
                    else:
                        if st[:3] == ident:
                            s = 'same as: '
                        else:
                            s = 'also: '
                        msg(s + filename)
                else:
                    msg(filename + ': not executable')
            if longlist:
                sts = os.system('ls ' + longlist + ' ' + filename)
                if sts: msg('"ls -l" exit status: ' + repr(sts))
        if not ident:
            msg(prog + ': not found')
            sts = 1

    sys.exit(sts)

if __name__ == '__main__':
    main()