b2f46753ec
* Add junos_system declartive module and other related change * junos_system declartive module * integration test for junos_system * integration test for net_system (junos platform) * pep8 fixes for junos modules * move to lxml from elementree for xml parsing as it support complete set of xpath api's * other minor changes * Fix CI and doc changes * Fix unit test failures * Fix typo in import * Fix import issue for py2.6 * Add missed Element in import
113 lines
3.5 KiB
Python
113 lines
3.5 KiB
Python
#
|
|
# (c) 2016 Red Hat Inc.
|
|
#
|
|
# This file is part of Ansible
|
|
#
|
|
# Ansible is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# Ansible is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with Ansible. If not, see <http://www.gnu.org/licenses/>.
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import json
|
|
import traceback
|
|
|
|
from ansible.module_utils._text import to_text
|
|
|
|
try:
|
|
from __main__ import display
|
|
except ImportError:
|
|
from ansible.utils.display import Display
|
|
display = Display()
|
|
|
|
|
|
class Rpc:
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self._rpc = set()
|
|
super(Rpc, self).__init__(*args, **kwargs)
|
|
|
|
def _exec_rpc(self, request):
|
|
method = request.get('method')
|
|
|
|
if method.startswith('rpc.') or method.startswith('_'):
|
|
error = self.invalid_request()
|
|
return json.dumps(error)
|
|
|
|
params = request.get('params')
|
|
setattr(self, '_identifier', request.get('id'))
|
|
args = []
|
|
kwargs = {}
|
|
|
|
if all((params, isinstance(params, list))):
|
|
args = params
|
|
elif all((params, isinstance(params, dict))):
|
|
kwargs = params
|
|
|
|
rpc_method = None
|
|
for obj in self._rpc:
|
|
rpc_method = getattr(obj, method, None)
|
|
if rpc_method:
|
|
break
|
|
|
|
if not rpc_method:
|
|
error = self.method_not_found()
|
|
response = json.dumps(error)
|
|
else:
|
|
try:
|
|
result = rpc_method(*args, **kwargs)
|
|
except Exception as exc:
|
|
display.display(traceback.format_exc(), log_only=True)
|
|
error = self.internal_error(data=to_text(exc, errors='surrogate_then_replace'))
|
|
response = json.dumps(error)
|
|
else:
|
|
if isinstance(result, dict) and 'jsonrpc' in result:
|
|
response = result
|
|
else:
|
|
response = self.response(result)
|
|
|
|
response = json.dumps(response)
|
|
|
|
delattr(self, '_identifier')
|
|
return response
|
|
|
|
def header(self):
|
|
return {'jsonrpc': '2.0', 'id': self._identifier}
|
|
|
|
def response(self, result=None):
|
|
response = self.header()
|
|
response['result'] = result or 'ok'
|
|
return response
|
|
|
|
def error(self, code, message, data=None):
|
|
response = self.header()
|
|
error = {'code': code, 'message': message}
|
|
if data:
|
|
error['data'] = data
|
|
response['error'] = error
|
|
return response
|
|
|
|
# json-rpc standard errors (-32768 .. -32000)
|
|
def parse_error(self, data=None):
|
|
return self.error(-32700, 'Parse error', data)
|
|
|
|
def method_not_found(self, data=None):
|
|
return self.error(-32601, 'Method not found', data)
|
|
|
|
def invalid_request(self, data=None):
|
|
return self.error(-32600, 'Invalid request', data)
|
|
|
|
def invalid_params(self, data=None):
|
|
return self.error(-32602, 'Invalid params', data)
|
|
|
|
def internal_error(self, data=None):
|
|
return self.error(-32603, 'Internal error', data)
|