summaryrefslogtreecommitdiffstats
path: root/Lib/py_compile.py
blob: 1adc3a209c608d7236ccc0aba98d0272e50e1933 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# Routine to "compile" a .py file to a .pyc file.
# This has intimate knowledge of how Python/import.c does it.
# By Sjoerd Mullender (I forced him to write it :-).

import imp
MAGIC = imp.get_magic()

def wr_long(f, x):
	f.write(chr( x        & 0xff))
	f.write(chr((x >> 8)  & 0xff))
	f.write(chr((x >> 16) & 0xff))
	f.write(chr((x >> 24) & 0xff))

def compile(file, cfile = None):
	import os, marshal, __builtin__
	f = open(file)
	try:
	    timestamp = os.fstat(file.fileno())
	except AttributeError:
	    timestamp = long(os.stat(file)[8])
	codestring = f.read()
	f.close()
	codeobject = __builtin__.compile(codestring, file, 'exec')
	if not cfile:
		cfile = file + (__debug__ and 'c' or 'o')
	fc = open(cfile, 'wb')
	fc.write('\0\0\0\0')
	wr_long(fc, timestamp)
	marshal.dump(codeobject, fc)
	fc.flush()
	fc.seek(0, 0)
	fc.write(MAGIC)
	fc.close()
	if os.name == 'mac':
		import macfs
		macfs.FSSpec(cfile).SetCreatorType('Pyth', 'PYC ')
		macfs.FSSpec(file).SetCreatorType('Pyth', 'TEXT')