EVOLUTION-MANAGER

Edit File: process.py

#
# (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 re

from ansible.module_utils.six import iteritems
from ansible.module_utils.network.common.utils import to_list
from ansible.module_utils.network.eos.providers.providers import register_provider
from ansible.module_utils.network.eos.providers.providers import CliProvider
from ansible.module_utils.network.eos.providers.cli.config.bgp.neighbors import Neighbors
from ansible.module_utils.network.eos.providers.cli.config.bgp.address_family import AddressFamily

REDISTRIBUTE_PROTOCOLS = frozenset(['ospf', 'ospf3', 'rip', 'isis', 'static', 'connected'])

@register_provider('eos', 'eos_bgp')
class Provider(CliProvider):

def render(self, config=None):
        commands = list()

existing_as = None
        if config:
            match = re.search(r'router bgp (\d+)', config, re.M)
            if match:
                existing_as = match.group(1)

operation = self.params['operation']

context = None
        if self.params['config']:
            context = 'router bgp %s' % self.get_value('config.bgp_as')

if operation == 'delete':
            if existing_as:
                commands.append('no router bgp %s' % existing_as)
            elif context:
                commands.append('no %s' % context)

else:
            self._validate_input(config)
            if operation == 'replace':
                if existing_as and int(existing_as) != self.get_value('config.bgp_as'):
                    commands.append('no router bgp %s' % existing_as)
                    config = None

elif operation == 'override':
                if existing_as:
                    commands.append('no router bgp %s' % existing_as)
                config = None

context_commands = list()

for key, value in iteritems(self.get_value('config')):
                if value is not None:
                    meth = getattr(self, '_render_%s' % key, None)
                    if meth:
                        resp = meth(config)
                        if resp:
                            context_commands.extend(to_list(resp))

if context and context_commands:
                commands.append(context)
                commands.extend(context_commands)
                commands.append('exit')
        return commands

def _render_router_id(self, config=None):
        cmd = 'router-id %s' % self.get_value('config.router_id')
        if not config or cmd not in config:
            return cmd

def _render_log_neighbor_changes(self, config=None):
        cmd = 'bgp log-neighbor-changes'
        log_neighbor_changes = self.get_value('config.log_neighbor_changes')
        if log_neighbor_changes is True:
            if not config or cmd not in config:
                return cmd
        elif log_neighbor_changes is False:
            if config and cmd in config:
                return 'no %s' % cmd

def _render_networks(self, config=None):
        commands = list()
        safe_list = list()

for entry in self.get_value('config.networks'):
            network = entry['prefix']
            if entry['masklen']:
                network = '%s/%s' % (entry['prefix'], entry['masklen'])
            safe_list.append(network)

cmd = 'network %s' % network

if entry['route_map']:
                cmd += ' route-map %s' % entry['route_map']

if not config or cmd not in config:
                commands.append(cmd)

if self.params['operation'] == 'replace':
            if config:
                matches = re.findall(r'network (\S+)', config, re.M)
                for entry in set(matches).difference(safe_list):
                    commands.append('no network %s' % entry)

return commands

def _render_redistribute(self, config=None):
        commands = list()
        safe_list = list()

for entry in self.get_value('config.redistribute'):
            option = entry['protocol']

cmd = 'redistribute %s' % entry['protocol']

if entry['route_map']:
                cmd += ' route-map %s' % entry['route_map']

if not config or cmd not in config:
                commands.append(cmd)

safe_list.append(option)

if self.params['operation'] == 'replace':
            if config:
                matches = re.findall(r'redistribute (\S+)(?:\s*)(\d*)', config, re.M)
                for i in range(0, len(matches)):
                    matches[i] = ' '.join(matches[i]).strip()
                for entry in set(matches).difference(safe_list):
                    commands.append('no redistribute %s' % entry)

return commands

def _render_neighbors(self, config):
        """ generate bgp neighbor configuration
        """
        return Neighbors(self.params).render(config)

def _render_address_family(self, config):
        """ generate address-family configuration
        """
        return AddressFamily(self.params).render(config)

def _validate_input(self, config):
        def device_has_AF(config):
            return re.search(r'address-family (?:.*)', config)

address_family = self.get_value('config.address_family')
        root_networks = self.get_value('config.networks')
        operation = self.params['operation']

if operation == 'replace' and root_networks:
            if address_family:
                for item in address_family:
                    if item['networks']:
                        raise ValueError('operation is replace but provided both root level networks and networks under %s address family'
                                         % item['afi'])

if config and device_has_AF(config):
                raise ValueError('operation is replace and device has one or more address family activated but root level network(s) provided')