Allow DjangoConnector to restore for any database engine

Author: Archmonger

dbbackup/management/commands/dbrestore.py

@@ -143,8 +143,9 @@ def _check_metadata(self, filename):
 
         backup_engine = metadata.get("engine")
         current_engine = settings.DATABASES[self.database_name]["ENGINE"]
+        backup_connector = metadata.get("connector")
 
-        if backup_engine != current_engine:
+        if backup_engine != current_engine and backup_connector != "dbbackup.db.django.DjangoConnector":
             msg = (
                 f"Backup file '{filename}' was created with database engine '{backup_engine}', "
                 f"but you are restoring to a database using '{current_engine}'. "

tests/commands/test_dbrestore_metadata.py

@@ -72,6 +72,17 @@ def test_local_file_metadata_mismatch(self):
             with pytest.raises(CommandError):
                 self.command._check_metadata("local_backup.dump")
 
+    def test_django_connector_mismatch_allowed(self):
+        # Setup metadata with different engine but DjangoConnector
+        metadata = {
+            "engine": "django.db.backends.postgresql",
+            "connector": "dbbackup.db.django.DjangoConnector",
+        }
+        self.command.storage.read_file.return_value = Mock(read=lambda: json.dumps(metadata))
+
+        # Should not raise
+        self.command._check_metadata("backup.dump")
+
 
 class DbrestoreConnectorOverrideTest(TestCase):
     def setUp(self):