PVST Feature commit

clear/main.py

@@ -12,7 +12,7 @@
 from show.plugins.pbh import read_pbh_counters
 from config.plugins.pbh import serialize_pbh_counters
 from . import plugins
-
+from . import stp
 # This is from the aliases example:
 # https://github.com/pallets/click/blob/57c6f09611fc47ca80db0bd010f05998b3c0aa95/examples/aliases/aliases.py
 class Config(object):
@@ -145,6 +145,10 @@ def ipv6():
     pass
 
 
+# 'STP'
+#
+cli.add_command(stp.spanning_tree)
+
 #
 # Inserting BGP functionality into cli's clear parse-chain.
 # BGP commands are determined by the routing-stack being elected.

clear/stp.py

@@ -0,0 +1,46 @@
+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

@@ -66,7 +66,7 @@
 from . import switchport
 from . import dns
 from . import bgp_cli
-
+from . import stp
 
 # mock masic APIs for unit test
 try:
@@ -1434,7 +1434,10 @@ def config(ctx):
 config.add_command(vlan.vlan)
 config.add_command(vxlan.vxlan)
 
-#add mclag commands
+# add stp commands
+config.add_command(stp.spanning_tree)
+
+# add mclag commands
 config.add_command(mclag.mclag)
 config.add_command(mclag.mclag_member)
 config.add_command(mclag.mclag_unique_ip)