blob: 98c5db082ee807bc5c151f667cc823483f981c95 [file] [log] [blame]
Guido van Rossum3bb54481994-08-29 10:52:58 +00001# Routine to "compile" a .py file to a .pyc file.
2# This has intimate knowledge of how Python/import.c does it.
3# By Sjoerd Mullender (I forced him to write it :-).
4
5MAGIC = 0x999903
6
7def wr_long(f, x):
8 f.write(chr( x & 0xff))
9 f.write(chr((x >> 8) & 0xff))
10 f.write(chr((x >> 16) & 0xff))
11 f.write(chr((x >> 24) & 0xff))
12
13def compile(file, cfile = None):
14 import os, marshal, __builtin__
15 f = open(file)
16 codestring = f.read()
Guido van Rossum3bb54481994-08-29 10:52:58 +000017 f.close()
Guido van Rossum7e4b2de1995-01-27 02:41:45 +000018 timestamp = os.stat(file)[8]
Guido van Rossum3bb54481994-08-29 10:52:58 +000019 codeobject = __builtin__.compile(codestring, file, 'exec')
20 if not cfile:
21 cfile = file + 'c'
Guido van Rossum7e4b2de1995-01-27 02:41:45 +000022 fc = open(cfile, 'wb')
Guido van Rossum3bb54481994-08-29 10:52:58 +000023 wr_long(fc, MAGIC)
24 wr_long(fc, timestamp)
25 marshal.dump(codeobject, fc)
Guido van Rossum7e4b2de1995-01-27 02:41:45 +000026 fc.close()
27 if os.name == 'mac':
28 import MacOS
29 MacOS.SetFileType(cfile, 'PYC ', 'PYTH')
30 MacOS.SetFileType(file, 'TEXT', 'PYTH')