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 | |
Sjoerd Mullender | 2e5168c | 1995-07-19 11:21:47 +0000 | [diff] [blame] | 5 | import imp |
| 6 | MAGIC = imp.get_magic() |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 7 | |
| 8 | def wr_long(f, x): |
| 9 | f.write(chr( x & 0xff)) |
| 10 | f.write(chr((x >> 8) & 0xff)) |
| 11 | f.write(chr((x >> 16) & 0xff)) |
| 12 | f.write(chr((x >> 24) & 0xff)) |
| 13 | |
| 14 | def compile(file, cfile = None): |
| 15 | import os, marshal, __builtin__ |
| 16 | f = open(file) |
| 17 | codestring = f.read() |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 18 | f.close() |
Guido van Rossum | 9b67621 | 1996-10-24 13:42:30 +0000 | [diff] [blame] | 19 | timestamp = long(os.stat(file)[8]) |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 20 | codeobject = __builtin__.compile(codestring, file, 'exec') |
| 21 | if not cfile: |
| 22 | cfile = file + 'c' |
Guido van Rossum | 7e4b2de | 1995-01-27 02:41:45 +0000 | [diff] [blame] | 23 | fc = open(cfile, 'wb') |
Sjoerd Mullender | 2e5168c | 1995-07-19 11:21:47 +0000 | [diff] [blame] | 24 | fc.write(MAGIC) |
Guido van Rossum | 3bb5448 | 1994-08-29 10:52:58 +0000 | [diff] [blame] | 25 | wr_long(fc, timestamp) |
| 26 | marshal.dump(codeobject, fc) |
Guido van Rossum | 7e4b2de | 1995-01-27 02:41:45 +0000 | [diff] [blame] | 27 | fc.close() |
| 28 | if os.name == 'mac': |
Jack Jansen | e99c824 | 1995-04-23 22:06:57 +0000 | [diff] [blame] | 29 | import macfs |
Guido van Rossum | dc42b8a | 1996-05-28 23:01:05 +0000 | [diff] [blame] | 30 | macfs.FSSpec(cfile).SetCreatorType('Pyth', 'PYC ') |
| 31 | macfs.FSSpec(file).SetCreatorType('Pyth', 'TEXT') |