home *** CD-ROM | disk | FTP | other *** search
/ PC World 2005 June / PCWorld_2005-06_cd.bin / software / vyzkuste / firewally / firewally.exe / framework-2.3.exe / compileall.py < prev    next >
Text File  |  2003-12-30  |  5KB  |  156 lines
  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.                     print err.msg
  70.                     success = 0
  71.                 except IOError, e:
  72.                     print "Sorry", e
  73.                     success = 0
  74.                 else:
  75.                     if ok == 0:
  76.                         success = 0
  77.         elif maxlevels > 0 and \
  78.              name != os.curdir and name != os.pardir and \
  79.              os.path.isdir(fullname) and \
  80.              not os.path.islink(fullname):
  81.             if not compile_dir(fullname, maxlevels - 1, dfile, force, rx, quiet):
  82.                 success = 0
  83.     return success
  84.  
  85. def compile_path(skip_curdir=1, maxlevels=0, force=0, quiet=0):
  86.     """Byte-compile all module on sys.path.
  87.  
  88.     Arguments (all optional):
  89.  
  90.     skip_curdir: if true, skip current directory (default true)
  91.     maxlevels:   max recursion level (default 0)
  92.     force: as for compile_dir() (default 0)
  93.     quiet: as for compile_dir() (default 0)
  94.  
  95.     """
  96.     success = 1
  97.     for dir in sys.path:
  98.         if (not dir or dir == os.curdir) and skip_curdir:
  99.             print 'Skipping current directory'
  100.         else:
  101.             success = success and compile_dir(dir, maxlevels, None,
  102.                                               force, quiet=quiet)
  103.     return success
  104.  
  105. def main():
  106.     """Script main program."""
  107.     import getopt
  108.     try:
  109.         opts, args = getopt.getopt(sys.argv[1:], 'lfqd:x:')
  110.     except getopt.error, msg:
  111.         print msg
  112.         print "usage: python compileall.py [-l] [-f] [-q] [-d destdir] " \
  113.               "[-x regexp] [directory ...]"
  114.         print "-l: don't recurse down"
  115.         print "-f: force rebuild even if timestamps are up-to-date"
  116.         print "-q: quiet operation"
  117.         print "-d destdir: purported directory name for error messages"
  118.         print "   if no directory arguments, -l sys.path is assumed"
  119.         print "-x regexp: skip files matching the regular expression regexp"
  120.         print "   the regexp is search for in the full path of the file"
  121.         sys.exit(2)
  122.     maxlevels = 10
  123.     ddir = None
  124.     force = 0
  125.     quiet = 0
  126.     rx = None
  127.     for o, a in opts:
  128.         if o == '-l': maxlevels = 0
  129.         if o == '-d': ddir = a
  130.         if o == '-f': force = 1
  131.         if o == '-q': quiet = 1
  132.         if o == '-x':
  133.             import re
  134.             rx = re.compile(a)
  135.     if ddir:
  136.         if len(args) != 1:
  137.             print "-d destdir require exactly one directory argument"
  138.             sys.exit(2)
  139.     success = 1
  140.     try:
  141.         if args:
  142.             for dir in args:
  143.                 if not compile_dir(dir, maxlevels, ddir,
  144.                                    force, rx, quiet):
  145.                     success = 0
  146.         else:
  147.             success = compile_path()
  148.     except KeyboardInterrupt:
  149.         print "\n[interrupt]"
  150.         success = 0
  151.     return success
  152.  
  153. if __name__ == '__main__':
  154.     exit_status = not main()
  155.     sys.exit(exit_status)
  156.