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
|
# Copyright (C) 2012 Codethink Limited
#
# 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; version 2 of the License.
#
# 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, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
import cliapp
import os
import morphlib
class GraphingPlugin(cliapp.Plugin):
def enable(self):
self.app.add_subcommand('graph-build-depends',
self.graph_build_depends,
arg_synopsis='REPO REF MORPHOLOGY')
def disable(self):
pass
def graph_build_depends(self, args):
for repo_name, ref, filename in self.app.itertriplets(args):
self.app.status(msg='Creating build order for '
'%(repo_name)s %(ref)s %(filename)s',
repo_name=repo_name, ref=ref, filename=filename)
builder = morphlib.buildcommand.BuildCommand(self.app)
order = builder.compute_build_order(repo_name, ref, filename)
basename, ext = os.path.splitext(filename)
dot_filename = basename + '.gv'
dep_fmt = ' "%s" -> "%s";\n'
shape_name = {
'system': 'octagon',
'stratum': 'box',
'chunk': 'ellipse',
}
self.app.status(msg='Writing DOT file to %(filename)s',
filename=dot_filename)
with open(dot_filename, 'w') as f:
f.write('digraph "%s" {\n' % basename)
for i, group in enumerate(order.groups):
for artifact in group:
f.write(
' "%s" [shape=%s];\n' %
(artifact.name,
shape_name[artifact.source.morphology['kind']]))
for dep in artifact.dependencies:
if artifact.source.morphology['kind'] == 'stratum':
if dep.dependents == [artifact]:
f.write(dep_fmt %
(artifact.name, dep.name))
else:
f.write(dep_fmt % (artifact.name, dep.name))
f.write('}\n')
|
