Add no-drop parameter dbrestore (#400)

Co-authored-by: Emiliano Beronich <[email protected]>
Co-authored-by: Mark <[email protected]>

dbbackup/management/commands/dbrestore.py

@@ -15,6 +15,7 @@
 class Command(BaseDbBackupCommand):
     help = "Restore a database backup from storage, encrypted and/or compressed."
     content_type = "db"
+    no_drop = False
 
     option_list = BaseDbBackupCommand.option_list + (
         make_option("-d", "--database", help="Database to restore"),
@@ -52,6 +53,13 @@ class Command(BaseDbBackupCommand):
             default=[],
             help="Specify schema(s) to restore. Can be used multiple times.",
         ),
+        make_option(
+            "-r",
+            "--no-drop",
+            action="store_true",
+            default=False,
+            help="Don't clean (drop) the database. This only works with mongodb and postgresql.",
+        ),
     )
 
     def handle(self, *args, **options):
@@ -74,6 +82,7 @@ def handle(self, *args, **options):
                 self.input_database_name
             )
             self.storage = get_storage()
+            self.no_drop = options.get("no_drop")
             self.schemas = options.get("schema")
             self._restore_backup()
         except StorageError as err:
@@ -129,8 +138,7 @@ def _restore_backup(self):
 
         input_file.seek(0)
         self.connector = get_connector(self.database_name)
-
         if self.schemas:
             self.connector.schemas = self.schemas
-
         self.connector.restore_dump(input_file)
+        self.connector.drop = not self.no_drop

docs/changelog.rst

@@ -4,7 +4,7 @@ Changelog
 Unreleased
 ----------
 
-* Nothing (yet)!
+* Add `--no-drop` option to `dbrestore` command to prevent dropping tables before restoring data.
 
 4.2.0 (2024-08-22)
 ------------------