Make migration 0020 multi-database friendly

WhyNotHugo · Nov 19, 2020 · f54cf28 · f54cf28
1 parent 3dc7cd7
commit f54cf28
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/django_mercadopago/migrations/0002_auto_20150923_2328.py b/django_mercadopago/migrations/0002_auto_20150923_2328.py
@@ -4,6 +4,7 @@
 
 def fake_data(apps, schema_editor):
     db_alias = schema_editor.connection.alias
+
     Account = apps.get_model("mp", "Account")
     Notification = apps.get_model("mp", "Notification")
     Payment = apps.get_model("mp", "Payment")
@@ -18,7 +19,11 @@ def fake_data(apps, schema_editor):
         return
 
     account = Account.objects.using(db_alias).create(
-        name="Auto-migrated", slug="_", app_id="_", secret_key="_", sandbox=True,
+        name="Auto-migrated",
+        slug="_",
+        app_id="_",
+        secret_key="_",
+        sandbox=True,
     )
     Notification.objects.using(db_alias).update(owner=account)
 

diff --git a/django_mercadopago/migrations/0020_move_preference_items.py b/django_mercadopago/migrations/0020_move_preference_items.py
@@ -3,11 +3,13 @@
 
 def move_item_data(apps, schema_editor):
     """Move Item data from Preference into the new model."""
+    db_alias = schema_editor.connection.alias
+
     Preference = apps.get_model("mp", "Preference")
     Item = apps.get_model("mp", "Item")
 
-    for preference in Preference.objects.all():
-        Item.objects.create(
+    for preference in Preference.objects.using(db_alias).all():
+        Item.objects.using(db_alias).create(
             preference=preference,
             title=preference.title,
             description=preference.description,