119 lines
3.9 KiB
Python
119 lines
3.9 KiB
Python
#
|
|
# (c) 2020 Red Hat Inc.
|
|
#
|
|
# (c) 2020 Dell 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
|
|
|
|
DOCUMENTATION = """
|
|
---
|
|
cliconf: sonic
|
|
short_description: Use sonic cliconf to run command on Dell OS10 platform
|
|
description:
|
|
- This sonic plugin provides low level abstraction apis for
|
|
sending and receiving CLI commands from Dell OS10 network devices.
|
|
"""
|
|
|
|
import json
|
|
|
|
from itertools import chain
|
|
|
|
from ansible.errors import AnsibleConnectionFailure
|
|
from ansible.module_utils._text import to_bytes, to_text
|
|
from ansible.module_utils.common._collections_compat import Mapping
|
|
from ansible_collections.ansible.netcommon.plugins.module_utils.network.common.utils import to_list
|
|
from ansible.plugins.cliconf import CliconfBase, enable_mode
|
|
|
|
|
|
class Cliconf(CliconfBase):
|
|
|
|
def get_device_info(self):
|
|
device_info = {}
|
|
device_info['network_os'] = 'sonic'
|
|
return device_info
|
|
|
|
@enable_mode
|
|
def edit_config(self, command):
|
|
response = []
|
|
self.send_command("configure terminal")
|
|
for cmd in to_list(command):
|
|
if isinstance(cmd, dict):
|
|
resp = self.get(command=cmd["command"], prompt=cmd["prompt"], answer=cmd["answer"])
|
|
response.append(resp)
|
|
else:
|
|
response.append(self.send_command(to_bytes(cmd)))
|
|
self.send_command("end")
|
|
return response
|
|
|
|
@enable_mode
|
|
def get_config(self, source="running", flags=None, format=None):
|
|
if source not in ("running", "startup"):
|
|
raise ValueError(
|
|
"fetching configuration from %s is not supported" % source
|
|
)
|
|
if not flags:
|
|
flags = []
|
|
if source == "running":
|
|
cmd = "show running-config "
|
|
else:
|
|
cmd = "show startup-config "
|
|
|
|
cmd += " ".join(to_list(flags))
|
|
cmd = cmd.strip()
|
|
return self.send_command(cmd)
|
|
|
|
def get(self, command, prompt=None, answer=None, sendonly=False, newline=True, check_all=False):
|
|
return self.send_command(command=command, prompt=prompt, answer=answer, sendonly=sendonly, newline=newline, check_all=check_all)
|
|
|
|
def get_capabilities(self):
|
|
result = super(Cliconf, self).get_capabilities()
|
|
return json.dumps(result)
|
|
|
|
def run_commands(self, commands=None, check_rc=True):
|
|
if commands is None:
|
|
raise ValueError("'commands' value is required")
|
|
|
|
responses = list()
|
|
for cmd in to_list(commands):
|
|
if not isinstance(cmd, Mapping):
|
|
cmd = {'command': cmd}
|
|
|
|
output = cmd.pop('output', None)
|
|
if output:
|
|
raise ValueError("'output' value %s is not supported for run_commands" % output)
|
|
|
|
try:
|
|
out = self.send_command(**cmd)
|
|
except AnsibleConnectionFailure as e:
|
|
if check_rc:
|
|
raise
|
|
out = getattr(e, 'err', to_text(e))
|
|
|
|
responses.append(out)
|
|
|
|
return responses
|
|
|
|
def set_cli_prompt_context(self):
|
|
"""
|
|
Make sure we are in the operational cli mode
|
|
:return: None
|
|
"""
|
|
if self._connection.connected:
|
|
self._update_cli_prompt_context(config_context=')#')
|