summaryrefslogtreecommitdiff
path: root/lib/ansible/module_utils/network/nxos/facts/acl_interfaces/acl_interfaces.py
blob: 87d522df6700d64fbed38a2edec7ae274f5a6b2b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
#
# -*- coding: utf-8 -*-
# Copyright 2019 Red Hat
# GNU General Public License v3.0+
# (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
"""
The nxos acl_interfaces fact class
It is in this file the configuration is collected from the device
for a given resource, parsed, and the facts tree is populated
based on the configuration.
"""
from __future__ import absolute_import, division, print_function
__metaclass__ = type

import re
from copy import deepcopy
from ansible.module_utils.network.common import utils
from ansible.module_utils.network.nxos.argspec.acl_interfaces.acl_interfaces import Acl_interfacesArgs
from ansible.module_utils.network.nxos.utils.utils import normalize_interface


class Acl_interfacesFacts(object):
    """ The nxos acl_interfaces fact class
    """

    def __init__(self, module, subspec='config', options='options'):
        self._module = module
        self.argument_spec = Acl_interfacesArgs.argument_spec
        spec = deepcopy(self.argument_spec)
        if subspec:
            if options:
                facts_argument_spec = spec[subspec][options]
            else:
                facts_argument_spec = spec[subspec]
        else:
            facts_argument_spec = spec

        self.generated_spec = utils.generate_dict(facts_argument_spec)

    def get_device_data(self, connection):
        return connection.get('show running-config | section interface')

    def populate_facts(self, connection, ansible_facts, data=None):
        """ Populate the facts for acl_interfaces
        :param connection: the device connection
        :param ansible_facts: Facts dictionary
        :param data: previously collected conf
        :rtype: dictionary
        :returns: facts
        """
        if not data:
            data = self.get_device_data(connection)
        data = data.split('interface')

        resources = []
        for i in range(len(data)):
            intf = data[i].split('\n')
            for l in range(1, len(intf)):
                if not re.search('ip(v6)?( port)? (access-group|traffic-filter)', intf[l]):
                    intf[l] = ''
            intf = list(filter(None, intf))
            resources.append(intf)

        objs = []
        for resource in resources:
            if resource:
                obj = self.render_config(self.generated_spec, resource)
                if obj:
                    objs.append(obj)

        ansible_facts['ansible_network_resources'].pop('acl_interfaces', None)
        facts = {}
        if objs:
            params = utils.validate_config(
                self.argument_spec, {'config': objs})
            params = utils.remove_empties(params)
            facts['acl_interfaces'] = params['config']

        ansible_facts['ansible_network_resources'].update(facts)
        return ansible_facts

    def render_config(self, spec, conf):
        """
        Render config as dictionary structure and delete keys
          from spec for null values

        :param spec: The facts tree, generated from the argspec
        :param conf: The configuration
        :rtype: dictionary
        :returns: The generated config
        """
        config = deepcopy(spec)
        name = conf[0].strip()
        config['name'] = normalize_interface(name)
        config['access_groups'] = []
        v4 = {'afi': 'ipv4', 'acls': []}
        v6 = {'afi': 'ipv6', 'acls': []}
        for c in conf[1:]:
            if c:
                acl4 = re.search(r'ip( port)? access-group (\w*) (\w*)', c)
                acl6 = re.search(r'ipv6( port)? traffic-filter (\w*) (\w*)', c)
                if acl4:
                    acl = {'name': acl4.group(2).strip(
                    ), 'direction': acl4.group(3).strip()}
                    if acl4.group(1):
                        acl.update({'port': True})
                    v4['acls'].append(acl)
                elif acl6:
                    acl = {'name': acl6.group(2), 'direction': acl6.group(3)}
                    if acl6.group(1):
                        acl.update({'port': True})
                    v6['acls'].append(acl)

        if len(v4['acls']) > 0:
            config['access_groups'].append(v4)
        if len(v6['acls']) > 0:
            config['access_groups'].append(v6)

        return utils.remove_empties(config)