| 1 | #! /usr/bin/env python
|
|---|
| 2 |
|
|---|
| 3 | """finddiv - a grep-like tool that looks for division operators.
|
|---|
| 4 |
|
|---|
| 5 | Usage: finddiv [-l] file_or_directory ...
|
|---|
| 6 |
|
|---|
| 7 | For directory arguments, all files in the directory whose name ends in
|
|---|
| 8 | .py are processed, and subdirectories are processed recursively.
|
|---|
| 9 |
|
|---|
| 10 | This actually tokenizes the files to avoid false hits in comments or
|
|---|
| 11 | strings literals.
|
|---|
| 12 |
|
|---|
| 13 | By default, this prints all lines containing a / or /= operator, in
|
|---|
| 14 | grep -n style. With the -l option specified, it prints the filename
|
|---|
| 15 | of files that contain at least one / or /= operator.
|
|---|
| 16 | """
|
|---|
| 17 |
|
|---|
| 18 | import os
|
|---|
| 19 | import sys
|
|---|
| 20 | import getopt
|
|---|
| 21 | import tokenize
|
|---|
| 22 |
|
|---|
| 23 | def main():
|
|---|
| 24 | try:
|
|---|
| 25 | opts, args = getopt.getopt(sys.argv[1:], "lh")
|
|---|
| 26 | except getopt.error, msg:
|
|---|
| 27 | usage(msg)
|
|---|
| 28 | return 2
|
|---|
| 29 | if not args:
|
|---|
| 30 | usage("at least one file argument is required")
|
|---|
| 31 | return 2
|
|---|
| 32 | listnames = 0
|
|---|
| 33 | for o, a in opts:
|
|---|
| 34 | if o == "-h":
|
|---|
| 35 | print __doc__
|
|---|
| 36 | return
|
|---|
| 37 | if o == "-l":
|
|---|
| 38 | listnames = 1
|
|---|
| 39 | exit = None
|
|---|
| 40 | for filename in args:
|
|---|
| 41 | x = process(filename, listnames)
|
|---|
| 42 | exit = exit or x
|
|---|
| 43 | return exit
|
|---|
| 44 |
|
|---|
| 45 | def usage(msg):
|
|---|
| 46 | sys.stderr.write("%s: %s\n" % (sys.argv[0], msg))
|
|---|
| 47 | sys.stderr.write("Usage: %s [-l] file ...\n" % sys.argv[0])
|
|---|
| 48 | sys.stderr.write("Try `%s -h' for more information.\n" % sys.argv[0])
|
|---|
| 49 |
|
|---|
| 50 | def process(filename, listnames):
|
|---|
| 51 | if os.path.isdir(filename):
|
|---|
| 52 | return processdir(filename, listnames)
|
|---|
| 53 | try:
|
|---|
| 54 | fp = open(filename)
|
|---|
| 55 | except IOError, msg:
|
|---|
| 56 | sys.stderr.write("Can't open: %s\n" % msg)
|
|---|
| 57 | return 1
|
|---|
| 58 | g = tokenize.generate_tokens(fp.readline)
|
|---|
| 59 | lastrow = None
|
|---|
| 60 | for type, token, (row, col), end, line in g:
|
|---|
| 61 | if token in ("/", "/="):
|
|---|
| 62 | if listnames:
|
|---|
| 63 | print filename
|
|---|
| 64 | break
|
|---|
| 65 | if row != lastrow:
|
|---|
| 66 | lastrow = row
|
|---|
| 67 | print "%s:%d:%s" % (filename, row, line),
|
|---|
| 68 | fp.close()
|
|---|
| 69 |
|
|---|
| 70 | def processdir(dir, listnames):
|
|---|
| 71 | try:
|
|---|
| 72 | names = os.listdir(dir)
|
|---|
| 73 | except os.error, msg:
|
|---|
| 74 | sys.stderr.write("Can't list directory: %s\n" % dir)
|
|---|
| 75 | return 1
|
|---|
| 76 | files = []
|
|---|
| 77 | for name in names:
|
|---|
| 78 | fn = os.path.join(dir, name)
|
|---|
| 79 | if os.path.normcase(fn).endswith(".py") or os.path.isdir(fn):
|
|---|
| 80 | files.append(fn)
|
|---|
| 81 | files.sort(lambda a, b: cmp(os.path.normcase(a), os.path.normcase(b)))
|
|---|
| 82 | exit = None
|
|---|
| 83 | for fn in files:
|
|---|
| 84 | x = process(fn, listnames)
|
|---|
| 85 | exit = exit or x
|
|---|
| 86 | return exit
|
|---|
| 87 |
|
|---|
| 88 | if __name__ == "__main__":
|
|---|
| 89 | sys.exit(main())
|
|---|
