Mac/scripts/zappycfiles.py

   1 #!/usr/local/bin/python
   2 """Recursively zap all .pyc and .pyo files"""
   3 import os
   4 import sys
   5
   6 # set doit true to actually delete files
   7 # set doit false to just print what would be deleted
   8 doit = 1
   9
  10 def main():
  11     if not sys.argv[1:]:
  12         if os.name == 'mac':
  13             import EasyDialogs
  14             dir = EasyDialogs.AskFolder(message='Directory to zap pyc files in')
  15             if not dir:
  16                 sys.exit(0)
  17             zappyc(dir)
  18         else:
  19             print 'Usage: zappyc dir ...'
  20             sys.exit(1)
  21     for dir in sys.argv[1:]:
  22         zappyc(dir)
  23
  24 def zappyc(dir):
  25     os.path.walk(dir, walker, None)
  26
  27 def walker(dummy, top, names):
  28     for name in names:
  29         if name[-4:] in ('.pyc', '.pyo'):
  30             path = os.path.join(top, name)
  31             print 'Zapping', path
  32             if doit:
  33                 os.unlink(path)
  34
  35 if __name__ == '__main__':
  36     main()