bcfg2-graph.py 1.7 KB
Newer Older
1
#!/usr/bin/env python
2
# -*- mode: python; coding: utf-8 -*-
3 4 5 6 7 8 9
"""
Génère un graphe de dépendance des groups bcfg2

Auteurs: Daniel Stan
         Valentin Samir

"""
10 11
from __future__ import print_function
import xml.dom.minidom
12
import subprocess
13

14 15 16 17
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'
18

19
out = file(TMP_FILE, 'w')
20

21
import datetime
22

23 24
# groups est le nœud racine
groups = xml.dom.minidom.parse(file(GROUPS_FILE,'r')).documentElement
25 26 27 28 29 30

print("""digraph G {
edge [len=4.50, ratio=fill]
""", file=out)

def childGroups(parent):
31 32 33
    """Récupère les groups enfants (dépendances) d'un nœud
    Attention, cette fonction travaille sur (et renvoie) des nœuds xml
    """
34 35 36
    return [ x for x in parent.childNodes if \
             x.nodeType == x.ELEMENT_NODE and x.tagName == u'Group']

37 38
# Les clients (ie des serveurs)
# sont coloriés d'une autre couleur
39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
print("""
subgraph cluster_1 {
        node [style=filled];
""", file=out)
for elem in childGroups(groups):
    if elem.hasAttribute('profile'):
        print('"%s";' % elem.getAttribute('name'), file=out)
print("""
        label = "process #2";
        color=blue
    }
""", file=out)
# Le reste

for elem in childGroups(groups):
    print('"%s" -> {%s};' % \
         (  elem.getAttribute('name'),
             " ".join( [ '"%s"' % x.getAttribute('name')
                            for x in childGroups(elem) ]),
                            ), file=out)

print("""
label = "\\n\\nBCFG2 Groups\\nLes Nounous\\n%s";
62
}""" % datetime.datetime.now().strftime('%c'), file=out)
63 64 65

out.close()

66
subprocess.Popen([GRAPH_BIN, "-Tsvg", TMP_FILE, "-o", OUTPUT_FILE]).communicate()