41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
# (c) 2019, Ansible by Red Hat, inc
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
|
|
import json
|
|
|
|
from ansible.module_utils.connection import Connection, ConnectionError
|
|
from ansible.module_utils._text import to_text
|
|
|
|
|
|
def get_capabilities(module):
|
|
if hasattr(module, '_frr_capabilities'):
|
|
return module._frr_capabilities
|
|
try:
|
|
capabilities = Connection(module._socket_path).get_capabilities()
|
|
except ConnectionError as exc:
|
|
module.fail_json(msg=to_text(exc, errors='surrogate_then_replace'))
|
|
module._frr_capabilities = json.loads(capabilities)
|
|
return module._frr_capabilities
|
|
|
|
|
|
def run_commands(module, commands, check_rc=True):
|
|
connection = get_connection(module)
|
|
try:
|
|
return connection.run_commands(commands=commands, check_rc=check_rc)
|
|
except ConnectionError as exc:
|
|
module.fail_json(msg=to_text(exc))
|
|
|
|
|
|
def get_connection(module):
|
|
if hasattr(module, '_frr_connection'):
|
|
return module._frr_connection
|
|
|
|
capabilities = get_capabilities(module)
|
|
network_api = capabilities.get('network_api')
|
|
if network_api == 'cliconf':
|
|
module._frr_connection = Connection(module._socket_path)
|
|
else:
|
|
module.fail_json(msg='Invalid connection type %s' % network_api)
|
|
|
|
return module._frr_connection
|