PVSTP feature implementation

clear/main.py

@@ -5,6 +5,7 @@
 
 import click
 
+from . import stp
 from utilities_common import util_base
 
 from . import plugins
@@ -124,6 +125,11 @@ def cli():
     """SONiC command line - 'Clear' command"""
     pass
 
+#
+# 'STP'
+#
+cli.add_command(stp.spanning_tree)
+
 #
 # 'ip' group ###
 #

clear/stp.py

@@ -0,0 +1,41 @@
+import click
+import utilities_common.cli as clicommon
+
+#
+# This group houses Spanning_tree commands and subgroups
+#
+@click.group(cls=clicommon.AliasedGroup)
+@click.pass_context
+def spanning_tree(ctx):
+    '''Clear Spanning-tree counters'''
+    pass
+
+@spanning_tree.group('statistics', cls=clicommon.AliasedGroup, invoke_without_command=True)
+@click.pass_context
+def stp_clr_stats(ctx):
+    if ctx.invoked_subcommand is None:
+        command = 'sudo stpctl clrstsall'
+        clicommon.run_command(command)
+
+@stp_clr_stats.command('interface')
+@click.argument('interface_name', metavar='<interface_name>', required=True)
+@click.pass_context
+def stp_clr_stats_intf(ctx, interface_name):
+    command = 'sudo stpctl clrstsintf ' + interface_name
+    clicommon.run_command(command)
+
+@stp_clr_stats.command('vlan')
+@click.argument('vlan_id', metavar='<vlan_id>', required=True)
+@click.pass_context
+def stp_clr_stats_vlan(ctx, vlan_id):
+    command = 'sudo stpctl clrstsvlan ' + vlan_id
+    clicommon.run_command(command)
+
+@stp_clr_stats.command('vlan-interface')
+@click.argument('vlan_id', metavar='<vlan_id>', required=True)
+@click.argument('interface_name', metavar='<interface_name>', required=True)
+@click.pass_context
+def stp_clr_stats_vlan_intf(ctx, vlan_id, interface_name):
+    command = 'sudo stpctl clrstsvlanintf ' + vlan_id + ' ' + interface_name
+    clicommon.run_command(command)
+

config/main.py

@@ -34,6 +34,7 @@
 from . import vlan
 from . import vxlan
 from . import plugins
+from . import stp
 from .config_mgmt import ConfigMgmtDPB
 
 # mock masic APIs for unit test
@@ -875,6 +876,7 @@ def config(ctx):
 config.add_command(nat.nat)
 config.add_command(vlan.vlan)
 config.add_command(vxlan.vxlan)
+config.add_command(stp.spanning_tree)
 
 @config.command()
 @click.option('-y', '--yes', is_flag=True, callback=_abort_if_false,