Add show reboot-history

doc/Command-Reference.md

@@ -541,6 +541,26 @@ This command displays the cause of the previous reboot
   User issued reboot command [User: admin, Time: Mon Mar 25 01:02:03 UTC 2019]
   ```
 
+**show reboot-cause history**
+
+This command displays the history of the previous reboots up to 10 entry
+
+- Usage:
+  ```
+  show reboot-cause history
+  ```
+
+- Example:
+  ```
+  admin@sonic:~$ show reboot-cause history
+  name                 cause        time                          user    comment
+  -------------------  -----------  ----------------------------  ------  ---------
+  2020_10_09_02_33_06  reboot       Fri Oct  9 02:29:44 UTC 2020  admin
+  2020_10_09_01_56_59  reboot       Fri Oct  9 01:53:49 UTC 2020  admin
+  2020_10_09_02_00_53  fast-reboot  Fri Oct  9 01:58:04 UTC 2020  admin
+  2020_10_09_04_53_58  warm-reboot  Fri Oct  9 04:51:47 UTC 2020  admin
+  ```
+
 **show uptime**
 
 This command displays the current system uptime

show/aliases.ini

@@ -3,3 +3,4 @@ running-configuration=runningconfiguration
 running-config=runningconfiguration
 startup-configuration=startupconfiguration
 startup-config=startupconfiguration
+reboot-cause=reboot cause

show/main.py

@@ -1803,20 +1803,74 @@ def mmu():
 # 'reboot-cause' command ("show reboot-cause")
 #
 @cli.command('reboot-cause')
-def reboot_cause():
-    """Show cause of most recent reboot"""
-    PREVIOUS_REBOOT_CAUSE_FILE = "/host/reboot-cause/previous-reboot-cause.txt"
-
-    # At boot time, PREVIOUS_REBOOT_CAUSE_FILE is generated based on
-    # the contents of the 'reboot cause' file as it was left when the device
-    # went down for reboot. This file should always be created at boot,
-    # but check first just in case it's not present.
-    if not os.path.isfile(PREVIOUS_REBOOT_CAUSE_FILE):
-        click.echo("Unable to determine cause of previous reboot\n")
+@click.argument('history', required=False, type=click.Choice(["history"]))
+def reboot_cause(history):
+    """Show cause of reboot"""
+    REBOOT_CAUSE_DIR = "/host/reboot-cause/"
+    PREVIOUS_REBOOT_CAUSE_FILE = REBOOT_CAUSE_DIR + "previous-reboot-cause.json"
+    USER_ISSUE_REBOOT_CAUSE_REGEX ="User issued \'{}\' command [User: {}, Time: {}]"
+    REBOOT_CAUSE_UNKNOWN = "Unknown"
+
+    def read_last_reboot_cause():
+        last_reboot_cause = REBOOT_CAUSE_UNKNOWN
+        # Read the last previous reboot cause
+        if os.path.exists(PREVIOUS_REBOOT_CAUSE_FILE):
+            with open(PREVIOUS_REBOOT_CAUSE_FILE, "r") as last_cause_file:
+                data = json.load(last_cause_file)
+                if data['user']:
+                    last_reboot_cause = USER_ISSUE_REBOOT_CAUSE_REGEX.format(data['cause'], data['user'], data['time'])
+                else:
+                    last_reboot_cause = "{}".format(data['cause'])
+        return last_reboot_cause
+
+    def read_reboot_cause_dbs():
+        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)
+        table_keys.sort(reverse=True)
+
+        def remove_prefix(text, prefix):
+            if text.startswith(prefix):
+                return text[len(prefix):]
+            return text
+
+        table = []
+        for tk in table_keys:
+            entry = db.get_all(db.STATE_DB, tk)
+            r = []
+            r.append(remove_prefix(tk, prefix))
+            if 'cause' not in entry:
+                r.append("")
+            else:
+                r.append(entry['cause'])
+
+            if 'time' not in entry:
+                r.append("")
+            else:
+                r.append(entry['time'])
+
+            if 'user' not in entry:
+                r.append("")
+            else:
+                r.append(entry['user'])
+
+            if 'comment' not in entry:
+                r.append("")
+            else:
+                r.append(entry['comment'])
+            table.append(r)
+        return table
+
+    if not history:
+        click.echo(read_last_reboot_cause())
     else:
-        cmd = "cat {}".format(PREVIOUS_REBOOT_CAUSE_FILE)
-        proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
-        click.echo(proc.stdout.read())
+        table = read_reboot_cause_dbs()
+        header = ['name', 'cause', 'time', 'user', 'comment']
+        click.echo(tabulate(table, header))
 
 
 #

tests/mock_tables/state_db.json

@@ -203,5 +203,17 @@
         "speed_target": "50",
         "led_status": "green",
         "timestamp": "20200813 01:32:30"
+    },
+    "REBOOT_CAUSE|2020_10_09_04_53_58": {
+        "cause": "warm-reboot",
+        "time": "Fri Oct  9 04:51:47 UTC 2020",
+        "user": "admin",
+        "comment": ""
+    },
+    "REBOOT_CAUSE|2020_10_09_02_33_06": {
+        "cause": "reboot",
+        "time": "Fri Oct  9 02:29:44 UTC 2020",
+        "user": "admin",
+        "comment": ""
     }
 }

tests/show_reboot_cause_test.py

@@ -0,0 +1,58 @@
+import os
+import sys
+import textwrap
+
+import mock
+from click.testing import CliRunner
+
+test_path = os.path.dirname(os.path.abspath(__file__))
+modules_path = os.path.dirname(test_path)
+sys.path.insert(0, modules_path)
+
+import show.main as show
+
+
+TEST_REBOOT_CAUSE = "warm-reboot"
+TEST_USER = "admin"
+TEST_REBOOT_TIME = "Fri Oct  9 04:51:47 UTC 2020"
+
+
+"""
+    Note: The following 'show reboot-cause' commands simply call other SONiC
+    CLI utilities, so the unit tests for the other utilities are expected
+    to cover testing their functionality:
+
+        show reboot-cause
+        show reboot-cause history
+"""
+
+class TestShowRebootCause(object):
+    @classmethod
+    def setup_class(cls):
+        print("SETUP")
+        os.environ["UTILITIES_UNIT_TESTING"] = "1"
+
+    # Test 'show reboot-cause'
+    def test_reboot_cause(self):
+        expected_output = "Unknown\n"
+        runner = CliRunner()
+        result = runner.invoke(show.cli.commands["reboot-cause"], [])
+        assert result.output == expected_output
+
+    # Test 'show reboot-cause history'
+    def test_reboot_cause_history(self):
+        expected_output = """\
+name                 cause        time                          user    comment
+-------------------  -----------  ----------------------------  ------  ---------
+2020_10_09_04_53_58  warm-reboot  Fri Oct  9 04:51:47 UTC 2020  admin
+2020_10_09_02_33_06  reboot       Fri Oct  9 02:29:44 UTC 2020  admin
+"""
+        runner = CliRunner()
+        result = runner.invoke(show.cli.commands["reboot-cause"], ["history"])
+        assert result.output == expected_output
+
+    @classmethod
+    def teardown_class(cls):
+        print("TEARDOWN")
+        os.environ["PATH"] = os.pathsep.join(os.environ["PATH"].split(os.pathsep)[:-1])
+        os.environ["UTILITIES_UNIT_TESTING"] = "0"