Refactor restore.py to improve restore command handling

Heavybullets8 · May 28, 2024 · f8e00f2 · f8e00f2
1 parent 3300658
commit f8e00f2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/functions/backup_restore/database/restore.py b/functions/backup_restore/database/restore.py
@@ -258,6 +258,8 @@ def _execute_restore_command(self, retries=3, wait=5) -> Dict[str, str]:
         }
         self.logger.debug(f"Executing restore command on pod: {self.primary_pod} with dump file: {self.backup_file}")
 
+        self.command, self.open_mode = self._get_restore_command()
+
         for attempt in range(retries):
             try:
                 if self.backup_file.suffix == '.gz':