diff options
Diffstat (limited to 'Lib/py_compile.py')
| -rw-r--r-- | Lib/py_compile.py | 83 |
1 files changed, 50 insertions, 33 deletions
diff --git a/Lib/py_compile.py b/Lib/py_compile.py index dc1cae98dd9..5adb70a2939 100644 --- a/Lib/py_compile.py +++ b/Lib/py_compile.py @@ -3,11 +3,13 @@ This module has intimate knowledge of the format of .pyc files. """ -import __builtin__ +import builtins +import errno import imp import marshal import os import sys +import tokenize import traceback MAGIC = imp.get_magic() @@ -36,16 +38,18 @@ class PyCompileError(Exception): can be accesses as class variable 'file' msg: string message to be written as error message - If no value is given, a default exception message will be given, - consistent with 'standard' py_compile output. - message (or default) can be accesses as class variable 'msg' + If no value is given, a default exception message will be + given, consistent with 'standard' py_compile output. + message (or default) can be accesses as class variable + 'msg' """ def __init__(self, exc_type, exc_value, file, msg=''): exc_type_name = exc_type.__name__ if exc_type is SyntaxError: - tbtext = ''.join(traceback.format_exception_only(exc_type, exc_value)) + tbtext = ''.join(traceback.format_exception_only( + exc_type, exc_value)) errmsg = tbtext.replace('File "<string>"', 'File "%s"' % file) else: errmsg = "Sorry: %s: %s" % (exc_type_name,exc_value) @@ -63,28 +67,30 @@ class PyCompileError(Exception): def wr_long(f, x): """Internal; write a 32-bit int to a file in little-endian order.""" - f.write(chr( x & 0xff)) - f.write(chr((x >> 8) & 0xff)) - f.write(chr((x >> 16) & 0xff)) - f.write(chr((x >> 24) & 0xff)) + f.write(bytes([x & 0xff, + (x >> 8) & 0xff, + (x >> 16) & 0xff, + (x >> 24) & 0xff])) -def compile(file, cfile=None, dfile=None, doraise=False): +def compile(file, cfile=None, dfile=None, doraise=False, optimize=-1): """Byte-compile one Python source file to Python bytecode. - Arguments: - - file: source filename - cfile: target filename; defaults to source with 'c' or 'o' appended - ('c' normally, 'o' in optimizing mode, giving .pyc or .pyo) - dfile: purported filename; defaults to source (this is the filename - that will show up in error messages) - doraise: flag indicating whether or not an exception should be - raised when a compile error is found. If an exception - occurs and this flag is set to False, a string - indicating the nature of the exception will be printed, - and the function will return to the caller. If an - exception occurs and this flag is set to True, a - PyCompileError exception will be raised. + :param file: The source file name. + :param cfile: The target byte compiled file name. When not given, this + defaults to the PEP 3147 location. + :param dfile: Purported file name, i.e. the file name that shows up in + error messages. Defaults to the source file name. + :param doraise: Flag indicating whether or not an exception should be + raised when a compile error is found. If an exception occurs and this + flag is set to False, a string indicating the nature of the exception + will be printed, and the function will return to the caller. If an + exception occurs and this flag is set to True, a PyCompileError + exception will be raised. + :param optimize: The optimization level for the compiler. Valid values + are -1, 0, 1 and 2. A value of -1 means to use the optimization + level of the current interpreter, as given by -O command line options. + + :return: Path to the resulting byte compiled file. Note that it isn't necessary to byte-compile Python modules for execution efficiency -- Python itself byte-compiles a module when @@ -101,32 +107,43 @@ def compile(file, cfile=None, dfile=None, doraise=False): See compileall.py for a script/module that uses this module to byte-compile all installed files (or all files in selected directories). - """ - with open(file, 'U') as f: + with tokenize.open(file) as f: try: - timestamp = long(os.fstat(f.fileno()).st_mtime) + timestamp = int(os.fstat(f.fileno()).st_mtime) except AttributeError: - timestamp = long(os.stat(file).st_mtime) + timestamp = int(os.stat(file).st_mtime) codestring = f.read() try: - codeobject = __builtin__.compile(codestring, dfile or file,'exec') - except Exception,err: - py_exc = PyCompileError(err.__class__,err.args,dfile or file) + codeobject = builtins.compile(codestring, dfile or file, 'exec', + optimize=optimize) + except Exception as err: + py_exc = PyCompileError(err.__class__, err, dfile or file) if doraise: raise py_exc else: sys.stderr.write(py_exc.msg + '\n') return if cfile is None: - cfile = file + (__debug__ and 'c' or 'o') + if optimize >= 0: + cfile = imp.cache_from_source(file, debug_override=not optimize) + else: + cfile = imp.cache_from_source(file) + try: + dirname = os.path.dirname(cfile) + if dirname: + os.makedirs(dirname) + except OSError as error: + if error.errno != errno.EEXIST: + raise with open(cfile, 'wb') as fc: - fc.write('\0\0\0\0') + fc.write(b'\0\0\0\0') wr_long(fc, timestamp) marshal.dump(codeobject, fc) fc.flush() fc.seek(0, 0) fc.write(MAGIC) + return cfile def main(args=None): """Compile several source files. |