summaryrefslogtreecommitdiff
path: root/src/op_mode/show_vrf.py
blob: 3c7a90205ef17db401c80b244a942d8cb3ca9665 (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
#!/usr/bin/env python3
#
# Copyright (C) 2020 VyOS maintainers and contributors
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 or later as
# published by the Free Software Foundation.
#
# This program 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 this program.  If not, see <http://www.gnu.org/licenses/>.

import argparse
import jinja2
from json import loads

from vyos.util import cmd

vrf_out_tmpl = """VRF name          state     mac address        flags                     interfaces
--------          -----     -----------        -----                     ----------
{%- for v in vrf %}
{{"%-16s"|format(v.ifname)}}  {{ "%-8s"|format(v.operstate | lower())}}  {{"%-17s"|format(v.address | lower())}}  {{ v.flags|join(',')|lower()}}  {{v.members|join(',')|lower()}}
{%- endfor %}

"""

def list_vrfs():
    command = 'ip -j -br link show type vrf'
    answer = loads(cmd(command))
    return [_ for _ in answer if _]

def list_vrf_members(vrf):
    command = f'ip -j -br link show master {vrf}'
    answer = loads(cmd(command))
    return [_ for _ in answer if _]

parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group()
group.add_argument("-e", "--extensive", action="store_true",
                   help="provide detailed vrf informatio")
parser.add_argument('interface', metavar='I', type=str, nargs='?',
                    help='interface to display')

args = parser.parse_args()

if args.extensive:
    data = { 'vrf': [] }
    for vrf in list_vrfs():
        name = vrf['ifname']
        if args.interface and name != args.interface:
            continue

        vrf['members'] = []
        for member in list_vrf_members(name):
            vrf['members'].append(member['ifname'])
        data['vrf'].append(vrf)

    tmpl = jinja2.Template(vrf_out_tmpl)
    print(tmpl.render(data))

else:
    print(" ".join([vrf['ifname'] for vrf in list_vrfs()]))