X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;f=src%2Femuvim%2Fapi%2Frest%2Fnetwork.py;h=13743731e924348d4207b329f5ec61c43acbb1a7;hb=e0f920e099b91a07e7cb2f3bb3a940a71b493f08;hp=65dddc43ab53c07a14759ff2a6558c81954ca790;hpb=2542bfaa225f20272df99f1ccab5236326f16070;p=osm%2Fvim-emu.git

diff --git a/src/emuvim/api/rest/network.py b/src/emuvim/api/rest/network.py
index 65dddc4..1374373 100755
--- a/src/emuvim/api/rest/network.py
+++ b/src/emuvim/api/rest/network.py
@@ -36,8 +36,9 @@ import logging
 from flask_restful import Resource
 from flask import request
 import json
+import networkx
 
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig(level=logging.DEBUG)
 
 CORS_HEADER = {'Access-Control-Allow-Origin': '*'}
 
@@ -82,8 +83,7 @@ class NetworkAction(Resource):
             # check if json data is a dict
             data = request.args
             # try json payload
-            if data is None:
-                data = request.json
+            data = request.json
             # then no data
             if data is None:
                 data = {}
@@ -132,11 +132,12 @@ class DrawD3jsgraph(Resource):
         nodes2 = list()
         links = list()
         # add all DCs
-        #for dc in net.dcs:
+        node_attr = networkx.get_node_attributes(net.DCNetwork_graph, 'type')
         for node_name in net.DCNetwork_graph.nodes():
             nodes2.append(node_name)
             node_index = nodes2.index(node_name)
-            node_dict = {"name":node_name,"group":node_index}
+            type = node_attr[node_name]
+            node_dict = {"name":node_name,"group":type}
             nodes.append(node_dict)
 
         # add links between other DCs