clean.py

   1 #!/usr/bin/python
   2 #
   3 # File:   clean.py
   4 #
   5 # Author: Sergey Satskiy
   6 #
   7 # Date:   Apr 13, 2010
   8 #
   9 # $Id$
  10 #
  11
  12 """ Deletes the compiled python files """
  13
  14 import sys, os.path
  15 from optparse  import OptionParser
  16
  17
  18 ext = [ '.pyc', '.out', '.pyo', '.o' ]
  19
  20
  21 def mainClean():
  22     """ Counter main function """
  23
  24     parser = OptionParser(
  25     """
  26     %prog  <dir name>
  27     Deletes the compiled python files
  28     """ )
  29
  30     options, args = parser.parse_args()
  31
  32     if len( args ) != 1:
  33         raise Exception( "One argument expected" )
  34
  35
  36     if not os.path.exists( args[0] ) or \
  37        not os.path.isdir( args[0] ):
  38         raise Exception( "A directory name is expected" )
  39
  40     files = processDir( args[0] )
  41
  42     print "Project directory: " + args[0]
  43     print "Deleted files:    " + str( files )
  44
  45     return 0
  46
  47
  48 def processDir( path ):
  49     " Recursively deletes files in the given directory "
  50
  51     files = 0
  52
  53     for item in os.listdir( path ):
  54         if os.path.isdir( path + '/' + item ):
  55             files += processDir( path + '/' + item )
  56             continue
  57         for ex in ext:
  58             if item.endswith( ex ):
  59                 files += 1
  60                 os.remove( path + '/' + item )
  61             continue
  62     return files
  63
  64
  65 # The script execution entry point
  66 if __name__ == "__main__":
  67     returnCode = 1
  68     try:
  69         returnCode = mainClean()
  70     except Exception, exception:
  71         print >> sys.stderr, str( exception )
  72     sys.exit( returnCode )
  73