Add an action to restore in-progress motd message on finishing stage

It could be useful if we owerride in progress message on signal handling

pleskdistup/actions/common.py

@@ -59,13 +59,7 @@ def _revert_action(self) -> action.ActionResult:
         return action.ActionResult()
 
 
-class AddInProgressSshLoginMessage(action.ActiveAction):
-    in_progress_message: str
-
-    def __init__(self, new_os: str) -> None:
-        self.name = "add in progress SSH login message"
-        path_to_util = os.path.abspath(sys.argv[0])
-        self.in_progress_message = f"""
+IN_PROGRESS_MESSAGE_FORMAT = """
 ===============================================================================
 Message from the Plesk dist-upgrader tool:
 The server is being converted to {new_os}. Please wait. During the conversion the
@@ -75,6 +69,15 @@ def __init__(self, new_os: str) -> None:
 ===============================================================================
 """
 
+
+class AddInProgressSshLoginMessage(action.ActiveAction):
+    in_progress_message: str
+
+    def __init__(self, new_os: str) -> None:
+        self.name = "add in progress SSH login message"
+        path_to_util = os.path.abspath(sys.argv[0])
+        self.in_progress_message = IN_PROGRESS_MESSAGE_FORMAT.format(new_os=new_os, path_to_util=path_to_util)
+
     def _prepare_action(self) -> action.ActionResult:
         log.debug("Adding 'in progress' login message...")
         motd.restore_ssh_login_message()
@@ -89,6 +92,27 @@ def _revert_action(self) -> action.ActionResult:
         return action.ActionResult()
 
 
+class RestoreInProgressSshLoginMessage(action.ActiveAction):
+    in_progress_message: str
+
+    def __init__(self, new_os: str) -> None:
+        self.name = "restore in progress SSH login message"
+        path_to_util = os.path.abspath(sys.argv[0])
+        self.in_progress_message = IN_PROGRESS_MESSAGE_FORMAT.format(new_os=new_os, path_to_util=path_to_util)
+
+    def _prepare_action(self) -> action.ActionResult:
+        return action.ActionResult()
+
+    def _post_action(self) -> action.ActionResult:
+        log.debug("Restore 'in progress' login message...")
+        motd.restore_ssh_login_message()
+        motd.add_inprogress_ssh_login_message(self.in_progress_message)
+        return action.ActionResult()
+
+    def _revert_action(self) -> action.ActionResult:
+        return action.ActionResult()
+
+
 class DisablePleskSshBanner(action.ActiveAction):
     banner_command_path: str