from flask_restful import Resource
from flask import request
import json
+import networkx
-logging.basicConfig(level=logging.INFO)
+logging.basicConfig()
CORS_HEADER = {'Access-Control-Allow-Origin': '*'}
global net
- def put(self, vnf_src_name, vnf_dst_name):
+ def put(self):
logging.debug("REST CALL: network chain add")
command = 'add-flow'
- return self._NetworkAction(vnf_src_name, vnf_dst_name, command=command)
+ return self._NetworkAction(command=command)
- def delete(self, vnf_src_name, vnf_dst_name):
+ def delete(self):
logging.debug("REST CALL: network chain remove")
command = 'del-flows'
- return self._NetworkAction(vnf_src_name, vnf_dst_name, command=command)
+ return self._NetworkAction(command=command)
- def _NetworkAction(self, vnf_src_name, vnf_dst_name, command=None):
+ def _NetworkAction(self, command=None):
# call DCNetwork method, not really datacenter specific API for now...
# no check if vnfs are really connected to this datacenter...
try:
- # check if json data is a dict
+ # check json payload
+ logging.debug("json: {}".format(request.json))
+ logging.debug("args: {}".format(request.args))
+
data = request.json
+ if data is None:
+ data = request.args
if data is None:
data = {}
- elif type(data) is not dict:
- data = json.loads(request.json)
+ vnf_src_name = data.get("vnf_src_name")
+ vnf_dst_name = data.get("vnf_dst_name")
vnf_src_interface = data.get("vnf_src_interface")
vnf_dst_interface = data.get("vnf_dst_interface")
weight = data.get("weight")
except Exception as ex:
logging.exception("API error.")
return ex.message, 500, CORS_HEADER
+
+
+class DrawD3jsgraph(Resource):
+
+ global net
+
+ def get(self):
+ nodes = list()
+ nodes2 = list()
+ links = list()
+ # add all 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)
+ type = node_attr[node_name]
+ node_dict = {"name":node_name,"group":type}
+ nodes.append(node_dict)
+
+ # add links between other DCs
+ for node1_name in net.DCNetwork_graph.nodes():
+ node1_index = nodes2.index(node1_name)
+ for node2_name in net.DCNetwork_graph.neighbors(node1_name):
+ node2_index = nodes2.index(node2_name)
+ edge_dict = {"source": node1_index, "target": node2_index, "value": 10}
+ links.append(edge_dict)
+
+ json = {"nodes":nodes, "links":links}
+ return json, 200, CORS_HEADER