projects
/
osm
/
vim-emu.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Merge pull request #224 from stevenvanrossem/master
[osm/vim-emu.git]
/
src
/
emuvim
/
api
/
rest
/
network.py
diff --git
a/src/emuvim/api/rest/network.py
b/src/emuvim/api/rest/network.py
index
65dddc4
..
1374373
100755
(executable)
--- 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
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': '*'}
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
# 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 = {}
# then no data
if data is None:
data = {}
@@
-132,11
+132,12
@@
class DrawD3jsgraph(Resource):
nodes2 = list()
links = list()
# add all DCs
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)
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
nodes.append(node_dict)
# add links between other DCs