op mode: T5262: add a warning message about unsaved config changes

on reboot and shutdown attempts

1 files changed, 12 insertions, 1 deletions

diff --git a/src/op_mode/powerctrl.py b/src/op_mode/powerctrl.py
index fd4f86d88..b517527bb 100755
--- a/src/op_mode/powerctrl.py
+++ b/src/op_mode/powerctrl.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 #
-# Copyright (C) 2018 VyOS maintainers and contributors
+# Copyright (C) 2023 VyOS maintainers and contributors
 #
 # This program is free software; you can redistribute it and/or modify
 # it under the terms of the GNU General Public License version 2 or later as
@@ -102,8 +102,19 @@ def cancel_shutdown():
     else:
         print("Reboot or poweroff is not scheduled")
 
+def check_unsaved_config():
+    from vyos.config_mgmt import ConfigMgmt
+
+    c = ConfigMgmt()
+    if c._unsaved_commits():
+        print("Warning: there are unsaved configuration changes!")
+        print("Run 'save' command if you do not want to lose those changes after reboot/shutdown.")
+    else:
+        pass
 
 def execute_shutdown(time, reboot=True, ask=True):
+    check_unsaved_config()
+
     action = "reboot" if reboot else "poweroff"
     if not ask:
         if not ask_yes_no(f"Are you sure you want to {action} this system?"):