/Tools/freeze/makefreeze.py
http://unladen-swallow.googlecode.com/ · Python · 90 lines · 68 code · 10 blank · 12 comment · 13 complexity · 817a00cde90dfb39ee285f34aa6b5979 MD5 · raw file
- import marshal
- import bkfile
- # Write a file containing frozen code for the modules in the dictionary.
- header = """
- #include "Python.h"
- static struct _frozen _PyImport_FrozenModules[] = {
- """
- trailer = """\
- {0, 0, 0} /* sentinel */
- };
- """
- # if __debug__ == 0 (i.e. -O option given), set Py_OptimizeFlag in frozen app.
- default_entry_point = """
- int
- main(int argc, char **argv)
- {
- extern int Py_FrozenMain(int, char **);
- """ + ((not __debug__ and """
- Py_OptimizeFlag++;
- """) or "") + """
- PyImport_FrozenModules = _PyImport_FrozenModules;
- return Py_FrozenMain(argc, argv);
- }
- """
- def makefreeze(base, dict, debug=0, entry_point=None, fail_import=()):
- if entry_point is None: entry_point = default_entry_point
- done = []
- files = []
- mods = dict.keys()
- mods.sort()
- for mod in mods:
- m = dict[mod]
- mangled = "__".join(mod.split("."))
- if m.__code__:
- file = 'M_' + mangled + '.c'
- outfp = bkfile.open(base + file, 'w')
- files.append(file)
- if debug:
- print "freezing", mod, "..."
- str = marshal.dumps(m.__code__)
- size = len(str)
- if m.__path__:
- # Indicate package by negative size
- size = -size
- done.append((mod, mangled, size))
- writecode(outfp, mangled, str)
- outfp.close()
- if debug:
- print "generating table of frozen modules"
- outfp = bkfile.open(base + 'frozen.c', 'w')
- for mod, mangled, size in done:
- outfp.write('extern unsigned char M_%s[];\n' % mangled)
- outfp.write(header)
- for mod, mangled, size in done:
- outfp.write('\t{"%s", M_%s, %d},\n' % (mod, mangled, size))
- outfp.write('\n')
- # The following modules have a NULL code pointer, indicating
- # that the prozen program should not search for them on the host
- # system. Importing them will *always* raise an ImportError.
- # The zero value size is never used.
- for mod in fail_import:
- outfp.write('\t{"%s", NULL, 0},\n' % (mod,))
- outfp.write(trailer)
- outfp.write(entry_point)
- outfp.close()
- return files
- # Write a C initializer for a module containing the frozen python code.
- # The array is called M_<mod>.
- def writecode(outfp, mod, str):
- outfp.write('unsigned char M_%s[] = {' % mod)
- for i in range(0, len(str), 16):
- outfp.write('\n\t')
- for c in str[i:i+16]:
- outfp.write('%d,' % ord(c))
- outfp.write('\n};\n')
- ## def writecode(outfp, mod, str):
- ## outfp.write('unsigned char M_%s[%d] = "%s";\n' % (mod, len(str),
- ## '\\"'.join(map(lambda s: repr(s)[1:-1], str.split('"')))))