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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
import sys
import os
import errno
from distutils import dir_util, file_util
from modulegraph import modulegraph
def mkdir_p(path):
try:
os.makedirs(path)
except OSError as exc:
if exc.errno == errno.EEXIST and os.path.isdir(path):
pass
else:
raise
def collect_deps(root, dest_lib_dir, path_file):
mg = modulegraph.ModuleGraph(
[sys.path[0]] +
[x.strip() for x in open(path_file, 'r').readlines()] +
sys.path[1:]) # , debug=3)
mg.import_hook("distutils")
mg.import_hook("site")
mg.import_hook("jsonschema")
mg.import_hook("scrypt")
mg.import_hook("_scrypt")
mg.import_hook("ConfigParser")
mg.import_hook("encodings.idna")
mg.import_hook("leap.soledad.client")
mg.import_hook("leap.mail")
mg.import_hook("leap.keymanager")
mg.import_hook("argparse")
mg.import_hook("srp")
mg.import_hook("pkgutil")
mg.import_hook("pkg_resources")
mg.import_hook("_sre")
mg.import_hook("zope.proxy")
mg.import_hook("tuf")
mg.import_hook("timeit")
mg.import_hook("daemon") # for leap/bitmask/util/polkit_agent.py
mg.import_hook("functools32") # jsonschema dep
# this import ensures the inclusion of the 'service-identity' dependency
# since we don't import it implicitly anywhere
mg.import_hook("service_identity")
mg.run_script(root)
packages = [mg.findNode(i) for i in ["leap.common", "leap.keymanager",
"leap.mail", "leap.soledad.client",
"leap.soledad.common", "jsonschema"]]
other = []
sorted_pkg = [(os.path.basename(m.identifier), m) for m in mg.flatten()]
sorted_pkg.sort()
for (name, pkg) in sorted_pkg:
# skip namespace packages
if name in ("leap", "leap.soledad", "google", "zope", "repoze") or \
name.endswith("leap/bitmask/app.py"):
continue
# print pkg
if isinstance(pkg, modulegraph.MissingModule):
# print "ignoring", pkg.identifier
continue
elif isinstance(pkg, modulegraph.Package):
foundpackage = False
for i in packages:
if pkg.identifier.startswith(i.identifier):
# print "skipping", pkg.identifier, "member of", i.identifier
# print " found in", i.filename
foundpackage = True
break
if foundpackage:
continue
if pkg.filename is None:
continue
if pkg not in packages:
packages.append(pkg)
else: # if isinstance(pkg, modulegraph.Extension):
foundpackage = False
for i in packages:
if pkg.identifier.startswith(i.identifier):
# print "skipping", pkg.identifier, "member of", i.identifier
# print " found in", i.filename
foundpackage = True
break
if foundpackage:
continue
if pkg.filename is None:
continue
other.append(pkg)
# print pkg.identifier
print "Packages", len(packages)
for i in sorted(packages):
# if i.identifier == "distutils":
# i.filename = distutils.__file__
print i.identifier, i.filename
if i.identifier == "leap.bitmask":
continue
parts = i.identifier.split(".")
destdir = os.path.join(*([dest_lib_dir]+parts))
mkdir_p(destdir)
dir_util.copy_tree(os.path.dirname(i.filename), destdir)
before = []
for part in parts:
before.append(part)
current = before + ["__init__.py"]
try:
with open(os.path.join(dest_lib_dir, *current), 'a'):
pass
except Exception:
pass
print "Other", len(other)
for i in sorted(other):
# if i.identifier == "site":
# i.filename = site.__file__
print i.identifier, i.filename
file_util.copy_file(i.filename, dest_lib_dir)
|
