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
|
import getopt, popen2, lxml.etree
import Bcfg2.Server.Admin
class Viz(Bcfg2.Server.Admin.Mode):
__shorthelp__ = '''bcfg2-admin viz [--includehosts] [--includebundles] [--includekey] [-o output.png] [--raw]'''
__longhelp__ = __shorthelp__ + '\n\tProduce graphviz diagrams of metadata structures'
colors = ['steelblue1', 'chartreuse', 'gold', 'magenta',
'indianred1', 'limegreen', 'orange1', 'lightblue2',
'green1', 'blue1', 'yellow1', 'darkturquoise', 'gray66']
def __init__(self, cfile):
Bcfg2.Server.Admin.Mode.__init__(self, cfile)
def __call__(self, args):
Bcfg2.Server.Admin.Mode.__call__(self, args)
# First get options to the 'viz' subcommand
try:
opts, args = getopt.getopt(args, 'rhbko:',
['raw', 'includehosts', 'includebundles',
'includekey', 'outfile='])
except getopt.GetoptError, msg:
print msg
raise SystemExit(1)
rset = False
hset = False
bset = False
kset = False
outputfile = False
for opt, arg in opts:
if opt in ("-r", "--raw"):
rset = True
elif opt in ("-h", "--includehosts"):
hset = True
elif opt in ("-b", "--includebundles"):
bset = True
elif opt in ("-k", "--includekey"):
kset = True
elif opt in ("-o", "--outfile"):
outputfile = arg
data = self.Visualize(self.get_repo_path(), rset, hset, bset,
kset, outputfile)
print data
def Visualize(self, repopath, raw=False, hosts=False,
bundles=False, key=False, output=False):
'''Build visualization of groups file'''
groupdata = lxml.etree.parse(repopath + '/Metadata/groups.xml')
groupdata.xinclude()
groups = groupdata.getroot()
if raw:
cmd = "dd bs=4M"
if output:
cmd += " of=%s" % output
else:
cmd = "dot -Tpng"
if output:
cmd += " -o %s" % output
dotpipe = popen2.Popen4(cmd)
categories = {'default':'grey83'}
instances = {}
egroups = groups.findall("Group") + groups.findall('.//Groups/Group')
for group in egroups:
if not categories.has_key(group.get('category')):
categories[group.get('category')] = self.colors.pop()
group.set('color', categories[group.get('category')])
if None in categories:
del categories[None]
try:
dotpipe.tochild.write("digraph groups {\n")
except:
print "write to dot process failed. Is graphviz installed?"
raise SystemExit(1)
dotpipe.tochild.write('\trankdir="LR";\n')
if hosts:
clients = lxml.etree.parse(repopath + \
'/Metadata/clients.xml').getroot()
for client in clients.findall('Client'):
if instances.has_key(client.get('profile')):
instances[client.get('profile')].append(client.get('name'))
else:
instances[client.get('profile')] = [client.get('name')]
for profile, clist in instances.iteritems():
clist.sort()
dotpipe.tochild.write(
'''\t"%s-instances" [ label="%s", shape="record" ];\n''' \
% (profile, '|'.join(clist)))
dotpipe.tochild.write('''\t"%s-instances" -> "group-%s";\n''' \
% (profile, profile))
if bundles:
bundles = []
[bundles.append(bund.get('name')) \
for bund in groups.findall('.//Bundle')
if bund.get('name') not in bundles]
bundles.sort()
for bundle in bundles:
dotpipe.tochild.write(
'''\t"bundle-%s" [ label="%s", shape="septagon"];\n''' \
% (bundle, bundle))
gseen = []
for group in egroups:
if group.get('profile', 'false') == 'true':
style = "filled, bold"
else:
style = "filled"
gseen.append(group.get('name'))
dotpipe.tochild.write(
'\t"group-%s" [label="%s", style="%s", fillcolor=%s];\n' %
(group.get('name'), group.get('name'), style, group.get('color')))
if bundles:
for bundle in group.findall('Bundle'):
dotpipe.tochild.write('\t"group-%s" -> "bundle-%s";\n' %
(group.get('name'), bundle.get('name')))
gfmt = '\t"group-%s" [label="%s", style="filled", fillcolor="grey83"];\n'
for group in egroups:
for parent in group.findall('Group'):
if parent.get('name') not in gseen:
dotpipe.tochild.write(gfmt % (parent.get('name'),
parent.get('name')))
gseen.append(parent.get("name"))
dotpipe.tochild.write('\t"group-%s" -> "group-%s" ;\n' %
(group.get('name'), parent.get('name')))
if key:
dotpipe.tochild.write("\tsubgraph cluster_key {\n")
dotpipe.tochild.write('''\tstyle="filled";\n''')
dotpipe.tochild.write('''\tcolor="lightblue";\n''')
dotpipe.tochild.write('''\tBundle [ shape="septagon" ];\n''')
dotpipe.tochild.write('''\tGroup [shape="ellipse"];\n''')
dotpipe.tochild.write('''\tProfile [style="bold", shape="ellipse"];\n''')
dotpipe.tochild.write('''\tHblock [label="Host1|Host2|Host3", shape="record"];\n''')
for category in categories:
dotpipe.tochild.write(
'''\t"''' + category + '''" [label="''' + category + \
'''", shape="record", style="filled", fillcolor=''' + \
categories[category] + '''];\n''')
dotpipe.tochild.write('''\tlabel="Key";\n''')
dotpipe.tochild.write("\t}\n")
dotpipe.tochild.write("}\n")
dotpipe.tochild.close()
return dotpipe.fromchild.read()
|
