Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 1 | # 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 | |
| 5 | MAGIC = 0x999903 |
| 6 | |
| 7 | def 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 | |
| 13 | def compile(file, cfile = None): |
| 14 | import os, marshal, __builtin__ |
| 15 | f = open(file) |
| 16 | codestring = f.read() |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 17 | f.close() |
Guido van Rossum | 7e4b2de | 1995-01-27 02:41:45 +0000 | [diff] [blame] | 18 | timestamp = os.stat(file)[8] |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 19 | codeobject = __builtin__.compile(codestring, file, 'exec') |
| 20 | if not cfile: |
| 21 | cfile = file + 'c' |
Guido van Rossum | 7e4b2de | 1995-01-27 02:41:45 +0000 | [diff] [blame] | 22 | fc = open(cfile, 'wb') |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 23 | wr_long(fc, MAGIC) |
| 24 | wr_long(fc, timestamp) |
| 25 | marshal.dump(codeobject, fc) |
Guido van Rossum | 7e4b2de | 1995-01-27 02:41:45 +0000 | [diff] [blame] | 26 | fc.close() |
| 27 | if os.name == 'mac': |
Jack Jansen | e99c824 | 1995-04-23 22:06:57 +0000 | [diff] [blame] | 28 | import macfs |
| 29 | macfs.FSSpec(cfile).SetCreatorType('PYTH', 'PYC ') |
| 30 | macfs.FSSpec(file).SetCreatorType('PYTH', 'TEXT') |