[sip/asterisk-graph] Supression du fichier temporaire

On écrit directement dans PIPE avec subprocess
This commit is contained in:
Valentin Samir 2013-06-21 00:06:21 +02:00
parent c8cee16a97
commit 77fa3f6404

View file

@ -1,72 +1,83 @@
#!/usr/bin/env python #!/usr/bin/env python
from __future__ import print_function # -*- coding: utf-8 -*-
from sh import neato """
from sh import date Génère un graphe de dépendance des extensions
from sh import dot du plan de numérotation d'asterisk
tmp_file = '/tmp/graph.gv' Auteur: Valentin Samir
exten_file = '/etc/asterisk/extensions.conf'
start_nodes = ["crans-sip", "crans-from-external", "ovh-sip", "crans-sms" ]
out = file(tmp_file,'w') """
arrow={}
iarrow={}
print("""digraph G { from subprocess import Popen, PIPE, STDOUT
rankdir=LR import time
edge [len=2.50, ratio=fill]
""", file=out)
for node in start_nodes:
print('"%s" [style=filled];' % node, file=out)
file=open(exten_file).read() def gen_extentions_graph(exten_file, start_nodes, svg_file):
file=file.split('\n[')
del(file[0]) out=""
for context in file: arrow={}
context=context.split('\n') iarrow={}
context_name = context[0].strip()[:-1]
arrow[context_name]= arrow.get(context_name, set()) out+="""digraph G {
iarrow[context_name]= iarrow.get(context_name, set()) rankdir=LR
del(context[0]) edge [len=2.50, ratio=fill]
print('"%s";' % context_name, file=out) """
for line in context: for node in start_nodes:
if line.startswith('include'): out+='"%s" [style=filled];' % node
dest=line.split('=>', 1)[1].strip()
iarrow[context_name].add(dest) file=open(exten_file).read()
line=line.split(',') file=file.split('\n[')
if len(line)>2 and line[2].strip().startswith('Goto('): del(file[0])
dest=line[2].strip()[5:] for context in file:
arrow[context_name].add(dest) context=context.split('\n')
context_name = context[0].strip()[:-1]
arrow[context_name]= arrow.get(context_name, set())
iarrow[context_name]= iarrow.get(context_name, set())
del(context[0])
out+='"%s";' % context_name
for line in context:
if line.startswith('include'):
dest=line.split('=>', 1)[1].strip()
iarrow[context_name].add(dest)
line=line.split(',')
if len(line)>2 and line[2].strip().startswith('Goto('):
dest=line[2].strip()[5:]
arrow[context_name].add(dest)
for (context_name, dests) in arrow.items(): for (context_name, dests) in arrow.items():
for dest in dests: for dest in dests:
print('"%s" -> "%s" [arrowhead=normal];' % (context_name, dest), file=out) out+='"%s" -> "%s" [arrowhead=normal];' % (context_name, dest)
for (context_name, dests) in iarrow.items(): for (context_name, dests) in iarrow.items():
for dest in dests: for dest in dests:
print('"%s" -> "%s" [arrowhead=crow, style=dashed];' % (context_name, dest), file=out) out+='"%s" -> "%s" [arrowhead=crow, style=dashed];' % (context_name, dest)
print(""" out+="""
node [shape=plaintext] node [shape=plaintext]
subgraph cluster_01 { subgraph cluster_01 {
key [label=<<table border="0" cellpadding="2" cellspacing="0" cellborder="0"> key [label=<<table border="0" cellpadding="2" cellspacing="0" cellborder="0">
<tr><td align="right" port="i1">B inclus dans A : A</td></tr> <tr><td align="right" port="i1">B inclus dans A : A</td></tr>
<tr><td align="right" port="i2">Saut de A vers B : A</td></tr> <tr><td align="right" port="i2">Saut de A vers B : A</td></tr>
</table>>]; </table>>];
key2 [label=<<table border="0" cellpadding="2" cellspacing="0" cellborder="0"> key2 [label=<<table border="0" cellpadding="2" cellspacing="0" cellborder="0">
<tr><td port="j1">B</td></tr> <tr><td port="j1">B</td></tr>
<tr><td port="j2">B</td></tr> <tr><td port="j2">B</td></tr>
</table>>]; </table>>];
key:i1:e -> key2:j1:e [style=dashed, arrowhead=crow] key:i1:e -> key2:j1:e [style=dashed, arrowhead=crow]
key:i2:e -> key2:j2:w [] key:i2:e -> key2:j2:w []
} }
""", file=out) """
print(""" out+="""
label = "\\n\\nAsterisk extensions\\nLes Nounous\\n%s"; label = "\\n\\nAsterisk extensions\\nLes Nounous\\n%s";
}""" % str(date())[0:-1], file=out) }""" % time.strftime('%A %d %B %Y, %H:%M:%S')
out.close()
dot("-Tsvg", tmp_file, "-o", "/usr/scripts/var/doc/asterisk/extensions.svg") p = Popen(['dot', '-Tsvg', '-o', svg_file], stdout=PIPE, stdin=PIPE, stderr=STDOUT)
#neato("-Tsvg", tmp_file, "-o", "/usr/scripts/var/doc/asterisk/extensions.svg") p.communicate(input=out)
if __name__ == '__main__' :
import sys
if len(sys.argv)>1:
import locale
locale.setlocale(locale.LC_TIME, 'fr_FR.UTF-8')
gen_extentions_graph('/etc/asterisk/extensions.conf', ["crans-sip", "crans-from-external", "ovh-sip", "crans-sms"], sys.argv[1])