Tools/scripts/finddiv.py

   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())