which.py 1.59 KB
Newer Older
1
#! /usr/bin/env python
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 32 33 34 35
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)
            except os.error:
                continue
            if not S_ISREG(st[ST_MODE]):
                msg(filename + ': not a disk file')
Tim Peters's avatar
Tim Peters committed
36
            else:
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60
                mode = S_IMODE(st[ST_MODE])
                if mode & 0111:
                    if not ident:
                        print filename
                        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()