commit f371201f8a205911398ee81359b30afcba526038
author stevenvanrossem <steven.vanrossem@intec.ugent.be> Thu May 04 00:01:52 2017 +0200
committer stevenvanrossem <steven.vanrossem@intec.ugent.be> Thu May 04 00:01:52 2017 +0200
tree edecd0c7b056d2b30a0ce15c9cd7266083021a5c
parent 68a0ba9dc7fe9ca4f0caf512cf0d4ecdb7a179d5

update dashboard, to display placement, colored by node type

src/emuvim/api/rest/network.py

diff --git a/src/emuvim/api/rest/network.py b/src/emuvim/api/rest/network.py
index 65dddc4..84db87e 100755
--- a/src/emuvim/api/rest/network.py
+++ b/src/emuvim/api/rest/network.py
@@ -36,6 +36,7 @@
 from flask_restful import Resource
 from flask import request
 import json
+import networkx
 
 logging.basicConfig(level=logging.INFO)
 
@@ -132,11 +133,12 @@
         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