blob: a61987505152cce7b988ad6c923df6bbe70cc326 [file] [log] [blame]
Khen Nursimuluaaac7ee2016-12-11 22:03:52 -05001#!/usr/bin/env python
2#
3# Copyright 2016 the original author or authors.
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16#
17from lxml import etree
18import structlog
19from netconf.nc_rpc.rpc import Rpc
20import netconf.nc_common.error as ncerror
21from netconf.constants import Constants as C
22from netconf.utils import filter_tag_match
23from twisted.internet.defer import inlineCallbacks, returnValue
24import dicttoxml
25from simplejson import dumps, load
26
27log = structlog.get_logger()
28
29
30class GetVoltha(Rpc):
31 def __init__(self, rpc_request, rpc_method, grpc_client, session):
32 super(GetVoltha, self).__init__(rpc_request, rpc_method,
33 grpc_client, session)
34 self._validate_parameters()
35
36 @inlineCallbacks
37 def execute(self):
38 log.info('get-voltha-request', session=self.session.session_id)
39 if self.rpc_response.is_error:
40 returnValue(self.rpc_response)
41
42 # Invoke voltha via the grpc client
43 res_dict = yield self.grpc_client.get_voltha_instance()
44
45 # convert dict to xml
46 xml = dicttoxml.dicttoxml(res_dict)
47 log.info('voltha-info', res=res_dict, xml=xml)
48
49 self.rpc_response.node = self.get_root_element(xml)
50 self.rpc_response.is_error = False
51
52 returnValue(self.rpc_response)
53
54
55 def _validate_parameters(self):
56 log.info('validate-parameters', session=self.session.session_id)
57 self.params = self.rpc_method.getchildren()
58 if len(self.params) > 1:
59 self.rpc_response.is_error = True
60 self.rpc_response.node = ncerror.BadMsg(self.rpc_request)
61 return
62
63 if self.params and not filter_tag_match(self.params[0], C.NC_FILTER):
64 self.rpc_response.is_error = True
65 self.rpc_response.node = ncerror.UnknownElement(
66 self.rpc_request, self.params[0])
67 return
68
69 if not self.params:
70 self.params = [None]