finddiv.py

#! /usr/bin/env python

"""finddiv - a grep-like tool that looks for division operators.

Usage: finddiv [-l] file_or_directory ...

For directory arguments, all files in the directory whose name ends in
.py are processed, and subdirectories are processed recursively.

This actually tokenizes the files to avoid false hits in comments or
strings literals.

By default, this prints all lines containing a / or /= operator, in
grep -n style. With the -l option specified, it prints the filename
of files that contain at least one / or /= operator.
"""

importos
importsys
importgetopt
importtokenize

defmain():
try:
opts, args=getopt.getopt(sys.argv[1:], "lh")
exceptgetopt.error, msg:
usage(msg)
return2
ifnotargs:
usage("at least one file argument is required")
return2
listnames=0
foro, ainopts:
ifo=="-h":
print__doc__
return
ifo=="-l":
listnames=1
exit=None
forfilenameinargs:
x=process(filename, listnames)
exit=exitorx
returnexit

defusage(msg):
sys.stderr.write("%s: %s\n"% (sys.argv[0], msg))
sys.stderr.write("Usage: %s [-l] file ...\n"%sys.argv[0])
sys.stderr.write("Try `%s -h' for more information.\n"%sys.argv[0])

defprocess(filename, listnames):
ifos.path.isdir(filename):
returnprocessdir(filename, listnames)
try:
fp=open(filename)
exceptIOError, msg:
sys.stderr.write("Can't open: %s\n"%msg)
return1
g=tokenize.generate_tokens(fp.readline)
lastrow=None
fortype, token, (row, col), end, lineing:
iftokenin ("/", "/="):
iflistnames:
printfilename
break
ifrow!=lastrow:
lastrow=row
print"%s:%d:%s"% (filename, row, line),
fp.close()

defprocessdir(dir, listnames):
try:
names=os.listdir(dir)
exceptos.error, msg:
sys.stderr.write("Can't list directory: %s\n"%dir)
return1
files= []
fornameinnames:
fn=os.path.join(dir, name)
ifos.path.normcase(fn).endswith(".py") oros.path.isdir(fn):
files.append(fn)
files.sort(lambdaa, b: cmp(os.path.normcase(a), os.path.normcase(b)))
exit=None
forfninfiles:
x=process(fn, listnames)
exit=exitorx
returnexit

if__name__=="__main__":
sys.exit(main())