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