openstack/vitrage
Code Issues Proposed changes

add graph type option for get topology

Browse Source 
Implements: blueprint get-topology-api
Change-Id: I1a0ae7a5b8e97723ec8f6fa9da607107cc497187
This commit is contained in:
Eyal 2015-12-24 16:09:10 +02:00
parent 7fb66bdc9b
commit 5d807bf05c
1 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
vitrage/api/controllers/v1/topology.py
View File

@ -14,6 +14,7 @@
import json
import pecan
from networkx.readwrite import json_graph
from oslo_log import log
from pecan.core import abort
from pecan import rest
@ -26,20 +27,27 @@ LOG = log.getLogger(__name__)
class TopologyController(rest.RestController):
@pecan.expose('json')
def index(self, edges=None, vertices=None, depth=None):
def index(self, edges=None, vertices=None, depth=None, graph_type='graph'):
enforce("get topology", pecan.request.headers,
pecan.request.enforcer, {})
LOG.info(_LI('received get topology: edges->%(edges)s vertices->%('
'vertices)s depth->%(depth)s') %
{'edges': edges, 'vertices': vertices, 'depth': depth})
'vertices)s depth->%(depth)s graph_type->%(graph_type)s')
% {'edges': edges, 'vertices': vertices, 'depth': depth,
'graph_type': graph_type})
# TODO(eyal) temporary mock
graph_file = pecan.request.cfg.find_file('graph.sample.json')
try:
with open(graph_file) as data_file:
return json.load(data_file)
graph = json.load(data_file)
if graph_type == 'graph':
return graph
if graph_type == 'tree':
return json_graph.tree_data(
json_graph.node_link_graph(graph).reverse(), root=0)
except Exception as e:
LOG.exception("failed to open file ", e)
abort(404, str(e))