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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
|
#
# Copyright 2020 Martin Owens <doctormo@gmail.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>
#
"""
Extract information from an inx file and follow all the files.
"""
import os
import sys
from datetime import datetime
from modulefinder import ModuleFinder
from collections import defaultdict
from inkex.inx import InxFile
from inkman import __pkgname__, __version__
from inkman.utils import DATA_DIR
MODULES = (
(
"Depricated",
"Using depricated modules {names}",
"Critical",
(
"bezmisc",
"cubicsuperpath",
"simplestyle",
"cspsubdiv",
"ffgeom",
"simplepath",
"simpletransform",
),
),
("Unusual", "Unusual modules being used {names}", "Warning", ("argparse",)),
("Ignored", None, None, ("os", "sys", "lxml")),
)
class PackageInxFile(InxFile):
def __init__(self, filename):
super().__init__(filename)
self.inx_file = filename
def get_file_root(self):
return os.path.dirname(self.inx_file)
def get_script(self):
"""Return the file the INX points to"""
if self.script.get("location", None) in ("inx", None):
return os.path.join(self.get_file_root(), self.script["script"])
raise IOError("Can't find script filename")
class IncludeFile(object):
"""A file to include in the package"""
def __init__(self, root, filepath, target=None):
self.root = root
self.filepath = filepath
self.name = os.path.basename(filepath)
self.target = target or os.path.dirname(filepath)
def __repr__(self):
return f"<File name='{self.name}'>"
def file_icon(self):
"""Return a filename type"""
ext = self.filepath.rsplit(".", 1)[-1]
if ext in ("inx", "py", "txt", "svg", "png"):
return ext
return "default"
def detect_deps(self, modules=False):
if not self.filepath.endswith(".py"):
return [], []
# We want to find all modules imported by the script directly
# So we set the location to nowhere and collect the badmodules
finder = ModuleFinder("never-land")
finder.run_script(self.filepath)
# TODO: split out local modules, system modules and genuine deps
deps, mods = [], []
for key, locs in finder.badmodules.items():
if "__main__" not in locs:
continue
try:
mod_path = finder.find_module("inkman", path=sys.path)[1]
if "/python" in mod_path:
if "site-packages" in mod_path:
deps.append(key.split(".")[0])
elif mod_path.startswith(self.root):
mods.append(mod_path)
except ImportError:
continue
return mods, deps
class GeneratePackage(object):
"""A generated package based on data"""
def __init__(self, inx_files, template=os.path.join(DATA_DIR, "setup.template")):
with open(template, "r") as fhl:
self._template = fhl.read()
self.files = []
self.requires = set()
self.warnings = defaultdict(list)
self.name = ""
self.ident = ""
for x, inx_file in enumerate(inx_files):
try:
inx = PackageInxFile(inx_file)
except Exception as err:
self.add_warning(f"Can't add file {inx_file}: {err}", 5)
continue
self._add_file(inx.get_file_root(), inx.inx_file)
self._add_file(inx.get_file_root(), inx.get_script())
if not x:
self.name = inx.name or ""
self.ident = inx.ident.replace(".", "-")
def _add_file(self, root, fname):
included = IncludeFile(root, fname)
self.files.append(included)
(mods, deps) = included.detect_deps()
for dep in deps:
self._add_dep(dep)
for fname in mods:
self._add_file(root, fname)
print(f"DEP: {self.requires}")
def _add_dep(self, name):
"""Try and find the package associated with this name"""
for (list_name, warning, level, mod_list) in MODULES:
if name in mod_list:
self.warnings[list_name].append(name)
return
self.requires.add(name)
def generate_setup(self, target):
"""Generates a setup.py file contents"""
args = {
"__pkgname__": __pkgname__,
"__version__": __version__,
"year": datetime.now().year,
"readme": self.generate_readme(target),
"ident": self.generate_ident(),
"version": self.generate_version(),
"description": self.widget("entry_name").get_text(),
"author": self.widget("entry_author").get_text(),
"author_email": self.widget("entry_email").get_text(),
"url": self.widget("entry_url").get_text(),
"license": self.widget("licenses").get_active_id(),
"datafiles": self.generate_datafiles(),
"requires": self.generate_requires(),
"classifiers": "", # TODO
}
return self_template.format(**args)
def generate_ident(self):
"""Generate the ident from the text entry"""
ident = self.widget("entry_ident").get_text()
ident = ident.replace(" ", "-").lower()
self.widget("entry_ident").set_text(ident)
return "inx-" + ident
def generate_datafiles(self):
"""Generate a list of datafile and their install location"""
result = ""
template = " ('{target}', ['{file}']),\n"
# 2. Loop through all the file objects in the list
# 3. Add them to the final result
return result.strip()
def generate_requires(self):
"""Generate a list of required modules"""
# TODO: Version keeping
result = ""
# 1. Loop through all the dep objects in the list
# 2. Append them to the result
return result
if __name__ == "__main__":
gen = GeneratePackage(sys.argv[1:])
|