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 sys
  17 import py_compile
  18
  19 __all__ = ["compile_dir","compile_path"]
  20
  21 def compile_dir(dir, maxlevels=10, ddir=None,
  22                 force=0, rx=None, quiet=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     quiet:     if 1, be quiet during compilation
  33
  34     """
  35     if not quiet:
  36         print 'Listing', dir, '...'
  37     try:
  38         names = os.listdir(dir)
  39     except os.error:
  40         print "Can't list", dir
  41         names = []
  42     names.sort()
  43     success = 1
  44     for name in names:
  45         fullname = os.path.join(dir, name)
  46         if ddir is not None:
  47             dfile = os.path.join(ddir, name)
  48         else:
  49             dfile = None
  50         if rx is not None:
  51             mo = rx.search(fullname)
  52             if mo:
  53                 continue
  54         if os.path.isfile(fullname):
  55             head, tail = name[:-3], name[-3:]
  56             if tail == '.py':
  57                 cfile = fullname + (__debug__ and 'c' or 'o')
  58                 ftime = os.stat(fullname).st_mtime
  59                 try: ctime = os.stat(cfile).st_mtime
  60                 except os.error: ctime = 0
  61                 if (ctime > ftime) and not force: continue
  62                 if not quiet:
  63                     print 'Compiling', fullname, '...'
  64                 try:
  65                     ok = py_compile.compile(fullname, None, dfile, True)
  66                 except KeyboardInterrupt:
  67                     raise KeyboardInterrupt
  68                 except py_compile.PyCompileError,err:
  69                     if quiet:
  70                         print 'Compiling', fullname, '...'
  71                     print err.msg
  72                     success = 0
  73                 except IOError, e:
  74                     print "Sorry", e
  75                     success = 0
  76                 else:
  77                     if ok == 0:
  78                         success = 0
  79         elif maxlevels > 0 and \
  80              name != os.curdir and name != os.pardir and \
  81              os.path.isdir(fullname) and \
  82              not os.path.islink(fullname):
  83             if not compile_dir(fullname, maxlevels - 1, dfile, force, rx, quiet):
  84                 success = 0
  85     return success
  86
  87 def compile_path(skip_curdir=1, maxlevels=0, force=0, quiet=0):
  88     """Byte-compile all module on sys.path.
  89
  90     Arguments (all optional):
  91
  92     skip_curdir: if true, skip current directory (default true)
  93     maxlevels:   max recursion level (default 0)
  94     force: as for compile_dir() (default 0)
  95     quiet: as for compile_dir() (default 0)
  96
  97     """
  98     success = 1
  99     for dir in sys.path:
 100         if (not dir or dir == os.curdir) and skip_curdir:
 101             print 'Skipping current directory'
 102         else:
 103             success = success and compile_dir(dir, maxlevels, None,
 104                                               force, quiet=quiet)
 105     return success
 106
 107 def main():
 108     """Script main program."""
 109     import getopt
 110     try:
 111         opts, args = getopt.getopt(sys.argv[1:], 'lfqd:x:')
 112     except getopt.error, msg:
 113         print msg
 114         print "usage: python compileall.py [-l] [-f] [-q] [-d destdir] " \
 115               "[-x regexp] [directory ...]"
 116         print "-l: don't recurse down"
 117         print "-f: force rebuild even if timestamps are up-to-date"
 118         print "-q: quiet operation"
 119         print "-d destdir: purported directory name for error messages"
 120         print "   if no directory arguments, -l sys.path is assumed"
 121         print "-x regexp: skip files matching the regular expression regexp"
 122         print "   the regexp is searched for in the full path of the file"
 123         sys.exit(2)
 124     maxlevels = 10
 125     ddir = None
 126     force = 0
 127     quiet = 0
 128     rx = None
 129     for o, a in opts:
 130         if o == '-l': maxlevels = 0
 131         if o == '-d': ddir = a
 132         if o == '-f': force = 1
 133         if o == '-q': quiet = 1
 134         if o == '-x':
 135             import re
 136             rx = re.compile(a)
 137     if ddir:
 138         if len(args) != 1:
 139             print "-d destdir require exactly one directory argument"
 140             sys.exit(2)
 141     success = 1
 142     try:
 143         if args:
 144             for dir in args:
 145                 if not compile_dir(dir, maxlevels, ddir,
 146                                    force, rx, quiet):
 147                     success = 0
 148         else:
 149             success = compile_path()
 150     except KeyboardInterrupt:
 151         print "\n[interrupt]"
 152         success = 0
 153     return success
 154
 155 if __name__ == '__main__':
 156     exit_status = int(not main())
 157     sys.exit(exit_status)