fix(migration): when there are duplicated keys, the last unique const…

…raint will fail
hardcoretech · Jul 29, 2024 · 7f53a9f · 7f53a9f
1 parent 77f04b4
commit 7f53a9f
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/rest_framework_idempotency_key/migrations/0003_remove_user.py b/rest_framework_idempotency_key/migrations/0003_remove_user.py
@@ -2,6 +2,14 @@
 
 from django.db import migrations, models
 
+def remove_duplicated_records(apps, schema_editor):
+    IdempotencyKey = apps.get_model('rest_framework_idempotency_key', 'IdempotencyKey')
+    existed_keys = set()
+    for id_key in IdempotencyKey.objects.all():
+        if id_key.idempotency_key in existed_keys:
+            id_key.delete()
+        else:
+            existed_keys.add(id_key.idempotency_key)
 
 class Migration(migrations.Migration):
 
@@ -10,6 +18,7 @@ class Migration(migrations.Migration):
     ]
 
     operations = [
+        migrations.RunPython(remove_duplicated_records, migrations.RunPython.noop),
         migrations.AlterField(
             model_name='idempotencykey',
             name='user',