Add a parameter to configure if cleaning the database during a backup…

… restoration.

dbbackup/management/commands/dbrestore.py

@@ -25,12 +25,15 @@ class Command(BaseDbBackupCommand):
         make_option("-I", "--input-path", help="Specify path on local filesystem to backup from"),
         make_option("-s", "--servername",
                     help="If backup file is not specified, filter the "
-                         "existing ones with the given servername"),
+                    "existing ones with the given servername"),
         make_option("-c", "--decrypt", default=False, action='store_true',
                     help="Decrypt data before restoring"),
-        make_option("-p", "--passphrase", help="Passphrase for decrypt file", default=None),
+        make_option("-p", "--passphrase",
+                    help="Passphrase for decrypt file", default=None),
         make_option("-z", "--uncompress", action='store_true', default=False,
                     help="Uncompress gzip data before restoring")
+        make_option("-r", "--nodrop", action="store_true",
+                    default=False, help="Don't clean (drop) database",)
     )
 
     def handle(self, *args, **options):
@@ -50,6 +53,7 @@ def handle(self, *args, **options):
             self.interactive = options.get('interactive')
             self.database_name, self.database = self._get_database(options)
             self.storage = get_storage()
+            self.nodrop = options.get("nodrop")
             self._restore_backup()
         except StorageError as err:
             raise CommandError(err)
@@ -91,4 +95,5 @@ def _restore_backup(self):
 
         input_file.seek(0)
         self.connector = get_connector(self.database_name)
+        self.connector.drop = not self.nodrop
         self.connector.restore_dump(input_file)