[PBH] CLI for Policy Based Hashing

clear/main.py

@@ -2,11 +2,12 @@
 import os
 import subprocess
 import sys
-
 import click
+import utilities_common.cli as clicommon
+import json
 
 from utilities_common import util_base
-
+from show.plugins.pbh import read_pbh_counters
 from . import plugins
 
 
@@ -450,6 +451,30 @@ def translations():
     run_command(cmd)
 
 
+def remap_keys(dict):
+    return [{'key': k, 'value': v} for k, v in dict.items()]
+
+
+#
+# 'pbh' subcommand
+#
+@cli.command()
+@clicommon.pass_db
+def pbh(db):
+    """ Clear PBH counters
+        clear counters -- write current counters to file in /tmp
+    """
+
+    pbh_rules = db.cfgdb.get_table("PBH_RULE")
+    pbh_counters = read_pbh_counters(pbh_rules)
+
+    try:
+        with open('/tmp/.pbh_counters.txt', 'w') as fp:
+            json.dump(remap_keys(pbh_counters), fp)
+    except IOError as err:
+        pass
+
+
 # Load plugins and register them
 helper = util_base.UtilHelper()
 for plugin in helper.load_plugins(plugins):