# Zap .pyc files
import os
import sys

doit = 1

def main():
	if os.name == 'mac':
		import macfs
		fss, ok = macfs.GetDirectory('Directory to zap pyc files in')
		if not ok:
			sys.exit(0)
		dir = fss.as_pathname()
		zappyc(dir)
	else:
		if not sys.argv[1:]:
			print 'Usage: zappyc dir ...'
			sys.exit(1)
		for dir in sys.argv[1:]:
			zappyc(dir)

def zappyc(dir):
	os.path.walk(dir, walker, None)
	
def walker(dummy, top, names):
	for name in names:
		if name[-4:] == '.pyc':
			path = os.path.join(top, name)
			print 'Zapping', path
			if doit:
				os.unlink(path)
				
if __name__ == '__main__':
	main()