Lib/compileall.py

   1 """Module/script to "compile" all .py files to .pyc (or .pyo) file.
   2
   3 When called as a script with arguments, this compiles the directories
   4 given as arguments recursively; the -l option prevents it from
   5 recursing into directories.
   6
   7 Without arguments, if compiles all modules on sys.path, without
   8 recursing into subdirectories.  (Even though it should do so for
   9 packages -- for now, you'll have to deal with packages separately.)
  10
  11 See module py_compile for details of the actual byte-compilation.
  12
  13 """
  14
  15 import os
  16 import stat
  17 import sys
  18 import py_compile
  19
  20 __all__ = ["compile_dir","compile_path"]
  21
  22 def compile_dir(dir, maxlevels=10, ddir=None, force=0):
  23     """Byte-compile all modules in the given directory tree.
  24
  25     Arguments (only dir is required):
  26
  27     dir:       the directory to byte-compile
  28     maxlevels: maximum recursion level (default 10)
  29     ddir:      if given, purported directory name (this is the
  30                directory name that will show up in error messages)
  31     force:     if 1, force compilation, even if timestamps are up-to-date
  32
  33     """
  34     print 'Listing', dir, '...'
  35     try:
  36         names = os.listdir(dir)
  37     except os.error:
  38         print "Can't list", dir
  39         names = []
  40     names.sort()
  41     success = 1
  42     for name in names:
  43         fullname = os.path.join(dir, name)
  44         if ddir:
  45             dfile = os.path.join(ddir, name)
  46         else:
  47             dfile = None
  48         if os.path.isfile(fullname):
  49             head, tail = name[:-3], name[-3:]
  50             if tail == '.py':
  51                 cfile = fullname + (__debug__ and 'c' or 'o')
  52                 ftime = os.stat(fullname)[stat.ST_MTIME]
  53                 try: ctime = os.stat(cfile)[stat.ST_MTIME]
  54                 except os.error: ctime = 0
  55                 if (ctime > ftime) and not force: continue
  56                 print 'Compiling', fullname, '...'
  57                 try:
  58                     py_compile.compile(fullname, None, dfile)
  59                 except KeyboardInterrupt:
  60                     raise KeyboardInterrupt
  61                 except:
  62                     if type(sys.exc_type) == type(''):
  63                         exc_type_name = sys.exc_type
  64                     else: exc_type_name = sys.exc_type.__name__
  65                     print 'Sorry:', exc_type_name + ':',
  66                     print sys.exc_value
  67                     success = 0
  68         elif maxlevels > 0 and \
  69              name != os.curdir and name != os.pardir and \
  70              os.path.isdir(fullname) and \
  71              not os.path.islink(fullname):
  72             compile_dir(fullname, maxlevels - 1, dfile, force)
  73     return success
  74
  75 def compile_path(skip_curdir=1, maxlevels=0, force=0):
  76     """Byte-compile all module on sys.path.
  77
  78     Arguments (all optional):
  79
  80     skip_curdir: if true, skip current directory (default true)
  81     maxlevels:   max recursion level (default 0)
  82     force: as for compile_dir() (default 0)
  83
  84     """
  85     success = 1
  86     for dir in sys.path:
  87         if (not dir or dir == os.curdir) and skip_curdir:
  88             print 'Skipping current directory'
  89         else:
  90             success = success and compile_dir(dir, maxlevels, None, force)
  91     return success
  92
  93 def main():
  94     """Script main program."""
  95     import getopt
  96     try:
  97         opts, args = getopt.getopt(sys.argv[1:], 'lfd:')
  98     except getopt.error, msg:
  99         print msg
 100         print "usage: compileall [-l] [-f] [-d destdir] [directory ...]"
 101         print "-l: don't recurse down"
 102         print "-f: force rebuild even if timestamps are up-to-date"
 103         print "-d destdir: purported directory name for error messages"
 104         print "if no directory arguments, -l sys.path is assumed"
 105         sys.exit(2)
 106     maxlevels = 10
 107     ddir = None
 108     force = 0
 109     for o, a in opts:
 110         if o == '-l': maxlevels = 0
 111         if o == '-d': ddir = a
 112         if o == '-f': force = 1
 113     if ddir:
 114         if len(args) != 1:
 115             print "-d destdir require exactly one directory argument"
 116             sys.exit(2)
 117     success = 1
 118     try:
 119         if args:
 120             for dir in args:
 121                 success = success and compile_dir(dir, maxlevels, ddir, force)
 122         else:
 123             success = compile_path()
 124     except KeyboardInterrupt:
 125         print "\n[interrupt]"
 126         success = 0
 127     return success
 128
 129 if __name__ == '__main__':
 130     sys.exit(not main())