blob: 98b3b21f1a9e18cf2b760f42dffb9c11f8eac63c [file] [log] [blame]
Guido van Rossum63566e21998-01-19 04:01:26 +00001"""Routine to "compile" a .py file to a .pyc (or .pyo) file.
2
3This module has intimate knowledge of the format of .pyc files.
4"""
Guido van Rossum3bb54481994-08-29 10:52:58 +00005
Sjoerd Mullender2e5168c1995-07-19 11:21:47 +00006import imp
7MAGIC = imp.get_magic()
Guido van Rossum3bb54481994-08-29 10:52:58 +00008
9def wr_long(f, x):
Guido van Rossum63566e21998-01-19 04:01:26 +000010 "Internal; write a 32-bit int to a file in little-endian order."
11 f.write(chr( x & 0xff))
12 f.write(chr((x >> 8) & 0xff))
13 f.write(chr((x >> 16) & 0xff))
14 f.write(chr((x >> 24) & 0xff))
Guido van Rossum3bb54481994-08-29 10:52:58 +000015
Guido van Rossum63566e21998-01-19 04:01:26 +000016def compile(file, cfile=None, dfile=None):
17 """Byte-compile one Python source file to Python bytecode.
18
19 Arguments:
20
21 file: source filename
22 cfile: target filename; defaults to source with 'c' or 'o' appended
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000023 ('c' normally, 'o' in optimizing mode, giving .pyc or .pyo)
Guido van Rossum63566e21998-01-19 04:01:26 +000024 dfile: purported filename; defaults to source (this is the filename
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000025 that will show up in error messages)
Guido van Rossum63566e21998-01-19 04:01:26 +000026
27 Note that it isn't necessary to byte-compile Python modules for
28 execution efficiency -- Python itself byte-compiles a module when
29 it is loaded, and if it can, writes out the bytecode to the
30 corresponding .pyc (or .pyo) file.
31
32 However, if a Python installation is shared between users, it is a
33 good idea to byte-compile all modules upon installation, since
34 other users may not be able to write in the source directories,
35 and thus they won't be able to write the .pyc/.pyo file, and then
36 they would be byte-compiling every module each time it is loaded.
37 This can slow down program start-up considerably.
38
39 See compileall.py for a script/module that uses this module to
40 byte-compile all installed files (or all files in selected
41 directories).
42
43 """
44 import os, marshal, __builtin__
45 f = open(file)
46 try:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000047 timestamp = os.fstat(file.fileno())
Guido van Rossum63566e21998-01-19 04:01:26 +000048 except AttributeError:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000049 timestamp = long(os.stat(file)[8])
Guido van Rossum63566e21998-01-19 04:01:26 +000050 codestring = f.read()
51 f.close()
52 if codestring and codestring[-1] != '\n':
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000053 codestring = codestring + '\n'
Guido van Rossum63566e21998-01-19 04:01:26 +000054 codeobject = __builtin__.compile(codestring, dfile or file, 'exec')
55 if not cfile:
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000056 cfile = file + (__debug__ and 'c' or 'o')
Guido van Rossum63566e21998-01-19 04:01:26 +000057 fc = open(cfile, 'wb')
58 fc.write('\0\0\0\0')
59 wr_long(fc, timestamp)
60 marshal.dump(codeobject, fc)
61 fc.flush()
62 fc.seek(0, 0)
63 fc.write(MAGIC)
64 fc.close()
65 if os.name == 'mac':
Guido van Rossum45e2fbc1998-03-26 21:13:24 +000066 import macfs
67 macfs.FSSpec(cfile).SetCreatorType('Pyth', 'PYC ')