CLI support for SmartSwitch PMON

config/chassis_modules.py

@@ -30,8 +30,9 @@ def shutdown_chassis_module(db, chassis_module_name):
 
     if not chassis_module_name.startswith("SUPERVISOR") and \
        not chassis_module_name.startswith("LINE-CARD") and \
-       not chassis_module_name.startswith("FABRIC-CARD"):
-        ctx.fail("'module_name' has to begin with 'SUPERVISOR', 'LINE-CARD' or 'FABRIC-CARD'")
+       not chassis_module_name.startswith("FABRIC-CARD") and \
+       not chassis_module_name.startswith("DPU"):
+        ctx.fail("'module_name' has to begin with 'SUPERVISOR', 'LINE-CARD', 'FABRIC-CARD', 'DPU'")
 
     fvs = {'admin_status': 'down'}
     config_db.set_entry('CHASSIS_MODULE', chassis_module_name, fvs)

show/chassis_modules.py

@@ -37,14 +37,14 @@ def status(db, chassis_module_name):
     header = ['Name', 'Description', 'Physical-Slot', 'Oper-Status', 'Admin-Status', 'Serial']
     chassis_cfg_table = db.cfgdb.get_table('CHASSIS_MODULE')
 
-    state_db = SonicV2Connector(host="127.0.0.1")
+    state_db = SonicV2Connector(host="127.0.0.1", port="6379")
     state_db.connect(state_db.STATE_DB)
 
-    key_pattern = '*'
+    key_pattern = 'CHASSIS_MODULE_TABLE|*'
     if chassis_module_name:
-        key_pattern = '|' + chassis_module_name
+        key_pattern = 'CHASSIS_MODULE_TABLE|' + chassis_module_name
 
-    keys = state_db.keys(state_db.STATE_DB, CHASSIS_MODULE_INFO_TABLE + key_pattern)
+    keys = state_db.keys(state_db.STATE_DB, key_pattern)
     if not keys:
         print('Key {} not found in {} table'.format(key_pattern, CHASSIS_MODULE_INFO_TABLE))
         return

show/reboot_cause.py

@@ -1,6 +1,7 @@
 import json
 import os
 import sys
+import redis
 
 import click
 from tabulate import tabulate
@@ -9,7 +10,8 @@
 
 
 PREVIOUS_REBOOT_CAUSE_FILE_PATH = "/host/reboot-cause/previous-reboot-cause.json"
-
+STATE_DB = 6
+CHASSIS_STATE_DB = 13
 
 def read_reboot_cause_file():
     reboot_cause_dict = {}
@@ -24,6 +26,98 @@ def read_reboot_cause_file():
     return reboot_cause_dict
 
 
+# Function to fetch reboot cause data from database
+def fetch_data_from_db(module_name, fetch_history=False, use_chassis_db=False):
+    prefix = 'REBOOT_CAUSE|'
+    if use_chassis_db:
+        try:
+            rdb = redis.Redis(host='redis_chassis.server', port=6380, decode_responses=True, db=CHASSIS_STATE_DB)
+            table_keys = rdb.keys(prefix+'*')
+        except Exception:
+            return []
+    else:
+        rdb = SonicV2Connector(host='127.0.0.1')
+        rdb.connect(rdb.STATE_DB, False)   # Make one attempt only
+        table_keys = rdb.keys(rdb.STATE_DB, prefix+'*')
+
+    if table_keys is not None:
+        table_keys.sort(reverse=True)
+
+    table = []
+    d = []
+    for tk in table_keys:
+        r = []
+        append = False
+        if use_chassis_db:
+            entry = rdb.hgetall(tk)
+        else:
+            entry = rdb.get_all(rdb.STATE_DB, tk)
+
+        if module_name is not None:
+            if 'device' in entry:
+                if module_name != entry['device'] and module_name != "all":
+                    continue
+                if entry['device'] in d and not history:
+                    append = False
+                    continue
+                elif not entry['device'] in d or entry['device'] in d and history:
+                    append = True
+                    if not entry['device'] in d:
+                        d.append(entry['device'])
+            r.append(entry['device'] if 'device' in entry else "SWITCH")
+        suffix = ""
+        if append and "DPU" in entry['device']:
+            suffix = entry['device'] + '|'
+        r.append(tk.replace(prefix, "").replace(suffix, ""))
+        r.append(entry['cause'] if 'cause' in entry else "")
+        r.append(entry['time'] if 'time' in entry else "")
+        r.append(entry['user'] if 'user' in entry else "")
+        if append and not fetch_history:
+            table.append(r)
+        elif fetch_history:
+            r.append(entry['comment'] if 'comment' in entry else "")
+            if module_name is None or module_name == 'all' or module_name.startswith('SWITCH') or \
+               'device' in entry and module_name == entry['device']:
+                table.append(r)
+
+    return table
+
+
+# Wrapper-function to fetch reboot cause data from database
+def fetch_reboot_cause_from_db(module_name):
+    table = []
+    r = []
+
+    # Read the previous reboot cause
+    reboot_cause_dict = read_reboot_cause_file()
+    reboot_gen_time = reboot_cause_dict.get("gen_time", "N/A")
+    reboot_cause = reboot_cause_dict.get("cause", "Unknown")
+    reboot_time = reboot_cause_dict.get("time", "N/A")
+    reboot_user = reboot_cause_dict.get("user", "N/A")
+
+    r.append("SWITCH")
+    r.append(reboot_gen_time if reboot_gen_time else "")
+    r.append(reboot_cause if reboot_cause else "")
+    r.append(reboot_time if reboot_time else "")
+    r.append(reboot_user if reboot_user else "")
+    table.append(r)
+
+    table += fetch_data_from_db(module_name, fetch_history=False, use_chassis_db=True)
+    return table
+
+
+# Function to fetch reboot cause history data from database
+def fetch_reboot_cause_history_from_db(module_name):
+    if module_name == "all":
+        # Combine data from both Redis containers for "all" modules
+        data_switch = fetch_data_from_db(module_name, fetch_history=True, use_chassis_db=False)
+        data_dpu = fetch_data_from_db(module_name, fetch_history=True, use_chassis_db=True)
+        return data_switch + data_dpu
+    elif module_name is None or module_name == "SWITCH":
+        return fetch_data_from_db(module_name, fetch_history=True, use_chassis_db=False)
+    else:
+        return fetch_data_from_db(module_name, fetch_history=True, use_chassis_db=True)
+
 #
 # 'reboot-cause' group ("show reboot-cause")
 #
@@ -62,33 +156,24 @@ def reboot_cause(ctx):
         click.echo(reboot_cause_str)
 
 
-# 'history' subcommand ("show reboot-cause history")
+# 'all' command within 'reboot-cause'
 @reboot_cause.command()
-def history():
-    """Show history of reboot-cause"""
-    REBOOT_CAUSE_TABLE_NAME = "REBOOT_CAUSE"
-    TABLE_NAME_SEPARATOR = '|'
-    db = SonicV2Connector(host='127.0.0.1')
-    db.connect(db.STATE_DB, False)   # Make one attempt only
-    prefix = REBOOT_CAUSE_TABLE_NAME + TABLE_NAME_SEPARATOR
-    _hash = '{}{}'.format(prefix, '*')
-    table_keys = db.keys(db.STATE_DB, _hash)
-    if table_keys is not None:
-        table_keys.sort(reverse=True)
+def all():
+    """Show cause of most recent reboot"""
+    reboot_cause_data = fetch_reboot_cause_from_db("all")
+    header = ['Device', 'Name', 'Cause', 'Time', 'User']
+    click.echo(tabulate(reboot_cause_data, header, numalign="left"))
 
-        table = []
-        for tk in table_keys:
-            entry = db.get_all(db.STATE_DB, tk)
-            r = []
-            r.append(tk.replace(prefix, ""))
-            r.append(entry['cause'] if 'cause' in entry else "")
-            r.append(entry['time'] if 'time' in entry else "")
-            r.append(entry['user'] if 'user' in entry else "")
-            r.append(entry['comment'] if 'comment' in entry else "")
-            table.append(r)
 
-        header = ['Name', 'Cause', 'Time', 'User', 'Comment']
-        click.echo(tabulate(table, header, numalign="left"))
+# 'history' command within 'reboot-cause'
+@reboot_cause.command()
+@click.argument('module_name', required=False)
+def history(module_name):
+    """Show history of reboot-cause"""
+    reboot_cause_history = fetch_reboot_cause_history_from_db(module_name)
+    if module_name is not None:
+        header = ['Device', 'Name', 'Cause', 'Time', 'User', 'Comment']
+        click.echo(tabulate(reboot_cause_history, header, numalign="left"))
     else:
-        click.echo("Reboot-cause history is not yet available in StateDB")
-        sys.exit(1)
+        header = ['Name', 'Cause', 'Time', 'User', 'Comment']
+        click.echo(tabulate(reboot_cause_history, header, numalign="left"))