From 82092bc428d2b277b41c1fe2fb068f228b619e5e Mon Sep 17 00:00:00 2001
From: banagiri <banagiri@marvell.com>
Date: Mon, 21 Sep 2020 20:14:39 +0530
Subject: [PATCH] lldp cmds to config 802.1q tlv

---
 config/main.py | 43 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)

diff --git a/config/main.py b/config/main.py
index b605d37..11e4d72 100755
--- a/config/main.py
+++ b/config/main.py
@@ -3535,5 +3535,48 @@ def delete(ctx):
     sflow_tbl['global'].pop('agent_id')
     config_db.set_entry('SFLOW', 'global', sflow_tbl['global'])
 
+@config.group()
+@click.pass_context
+@click.option('-s', '--redis-unix-socket-path', help='unix socket path for redis connection')
+def lldp(ctx, redis_unix_socket_path):
+    """lldp related configuration tasks"""
+    kwargs = {}
+    if redis_unix_socket_path:
+        kwargs['unix_socket_path'] = redis_unix_socket_path
+    config_db = ConfigDBConnector(**kwargs)
+    config_db.connect(wait_for_init=False)
+    ctx.obj = {'db': config_db}
+    pass
+
+
+@lldp.command('dot1q_adv_type')
+@click.argument('cap_type', metavar='<type>', required=True)
+@click.argument('enable', metavar='<enable>', required=True)
+@click.argument('interface_name', metavar='<interface_name>', required=False)
+@click.pass_context
+def dot1q_adv_type(ctx, cap_type, enable, interface_name="all"):
+    print(cap_type, enable, interface_name)
+    db = ctx.obj['db']
+    print("type {}".format(cap_type))
+    if cap_type not in ["all", "name", "pvid"] :
+        ctx.fail("Invalid type {} all/name/pvid".format(cap_type))
+
+    if enable not in ["enable", "disable"] :
+        ctx.fail("Invalid option {} enable/disable".format(cap_type))
+    port_list = []
+    if interface_name == "all":
+        port_list = db.get_keys("PORT")
+    else:
+        port_list.append(interface_name)
+    print(port_list)
+    for port in port_list:
+        port_cfg = db.get_entry('CFG_LLDP_DOT1Q_CAPBILITY', port)
+        print(port_cfg)
+        if cap_type == "all" or cap_type == "name":
+            port_cfg["name"] = 1 if enable == "enable" else 0
+        if cap_type == "all" or cap_type == "pvid":
+            port_cfg["pvid"] = 1 if enable == "enable" else 0
+        db.set_entry('CFG_LLDP_DOT1Q_CAPBILITY', port, port_cfg)
+
 if __name__ == '__main__':
     config()