crans/scripts
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[bcfg2-graph] comments+avoid sh module

Browse source
This commit is contained in:
Daniel STAN 2013-06-20 16:04:41 +02:00
parent 23a0a30678
commit 5253e7340c
1 changed files with 23 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

31
bcfg2/bcfg2-graph.py
View file

@ -1,26 +1,41 @@
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
"""
Génère un graphe de dépendance des groups bcfg2
Auteurs: Daniel Stan
Valentin Samir
"""
from __future__ import print_function from __future__ import print_function
import xml.dom.minidom import xml.dom.minidom
import subprocess
tmp_file = '/tmp/graph.gv' GRAPH_BIN = '/usr/bin/neato'
TMP_FILE = '/tmp/graph.gv'
GROUPS_FILE = '/var/lib/bcfg2/Metadata/groups.xml'
OUTPUT_FILE = '/usr/scripts/var/doc/bcfg2/groups.svg'
out = file(tmp_file,'w') out = file(TMP_FILE, 'w')
from sh import neato import datetime
from sh import date
groups = xml.dom.minidom.parse(file('/var/lib/bcfg2/Metadata/groups.xml','r')).documentElement # groups est le nœud racine
groups = xml.dom.minidom.parse(file(GROUPS_FILE,'r')).documentElement
print("""digraph G { print("""digraph G {
edge [len=4.50, ratio=fill] edge [len=4.50, ratio=fill]
""", file=out) """, file=out)
def childGroups(parent): def childGroups(parent):
"""Récupère les groups enfants (dépendances) d'un nœud
Attention, cette fonction travaille sur (et renvoie) des nœuds xml
"""
return [ x for x in parent.childNodes if \ return [ x for x in parent.childNodes if \
x.nodeType == x.ELEMENT_NODE and x.tagName == u'Group'] x.nodeType == x.ELEMENT_NODE and x.tagName == u'Group']
# Les clients # Les clients (ie des serveurs)
# sont coloriés d'une autre couleur
print(""" print("""
subgraph cluster_1 { subgraph cluster_1 {
node [style=filled]; node [style=filled];
@ -44,8 +59,8 @@ for elem in childGroups(groups):
print(""" print("""
label = "\\n\\nBCFG2 Groups\\nLes Nounous\\n%s"; label = "\\n\\nBCFG2 Groups\\nLes Nounous\\n%s";
}""" % str(date())[0:-1], file=out) }""" % datetime.datetime.now().strftime('%c'), file=out)
out.close() out.close()
neato("-Tsvg", tmp_file, "-o", "/usr/scripts/var/doc/bcfg2/groups.svg") subprocess.Popen([GRAPH_BIN, "-Tsvg", TMP_FILE, "-o", OUTPUT_FILE]).communicate()