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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
# Port of Ronny Pfannschmidt's genscript package
# https://bitbucket.org/RonnyPfannschmidt/genscript
import sys
import pickle
import bz2
import base64
import os
def find_toplevel(name):
for syspath in sys.path:
lib = os.path.join(syspath, name)
if os.path.isdir(lib):
return lib
mod = lib + '.py'
if os.path.isfile(mod):
return mod
raise LookupError(name)
def pkgname(toplevel, rootpath, path):
parts = path.split(os.sep)[len(rootpath.split(os.sep)):]
return '/'.join([toplevel] + parts)
def pkg_to_mapping(name):
toplevel = find_toplevel(name)
if os.path.isfile(toplevel):
return {name: toplevel.read()}
name2src = {}
for root, dirs, files in os.walk(toplevel):
for pyfile in files:
if os.path.splitext(pyfile)[1] in '.py .pem'.split():
pkg = pkgname(name, toplevel, os.path.join(root, pyfile))
f = open(os.path.join(root, pyfile))
try:
name2src[pkg] = f.read().encode('quoted-printable')
finally:
f.close()
return name2src
def compress_mapping(mapping):
data = pickle.dumps(mapping, 2)
data = bz2.compress(data, 9)
data = base64.encodestring(data)
data = data.decode('ascii')
return data
def compress_packages(names):
mapping = {}
for name in names:
mapping.update(pkg_to_mapping(name))
return compress_mapping(mapping)
def generate_script(entry, packages):
data = compress_packages(packages)
tmpl = open(os.path.join(os.path.dirname(__file__), 'template.py'))
exe = tmpl.read()
tmpl.close()
exe = exe.replace('@SOURCES@', data)
exe = exe.replace('@ENTRY@', entry)
return exe
|
