generadorDot.py

import pydot
import os
import sys
import pika
import logging


def callback(self, ch, method, properties, body):
    self.crearGrafo(body)



def crearGrafo(self, msg):

    g=pydot.Dot(graph_type='graph')
    nodos=[]
    edges=[]
    i=-1

    for line in msg:
        origen=line.split(":")[0]
        nodo_origen=pydot.Node(origen, style='filled', fillcolor='#598de0')
        if nodo_origen not in nodos:
            nodos.append(nodo_origen)
            i=i+1
        for val in line.split(":")[1].split("/"):
            nodo=pydot.Node(val, style='filled', fillcolor='#598de0')
            if nodo not in nodos:
                nodos.append(nodo)
                i=i+1
                edge=pydot.Edge(nodos[nodos.index(nodo_origen)],nodos[i])
                edges.append(edge)

    for nodo in nodos:
        g.add_node(nodo) 

    for edge in edges:
        g.add_edge(edge)

    g.write_png(str('diagram.png'))

    file = open("diagram.png", "rb")
    data = file.read()
    channel.basic_publish(exchange="", routing_key="graficosRelacion", body=data)
    file.close()



url = os.environ.get('CLOUDAMQP_URL', 'amqp://guest:guest@localhost/%2f')
params = pika.URLParameters(url)
params.socket_timeout = 5
connection = pika.BlockingConnection(params)

channel = connection.channel()
channel.queue_declare(queue='relacionNodos') # Cola de lectura
channel.queue_declare(queue='graficosRelacion') # Cola de escritura

print("Esperando para consumir...")
channel.basic_consume(callback, queue='relacionNodos', no_ack=True)
channel.start_consuming()
channel.close()