migration repaire

apps/customer/migrations/0008_customer_mobile_unique.py

@@ -1,4 +1,5 @@
 from django.db import migrations, models
+from apps.customer.mobile_migration_repair import repair_conflicts
 
 
 def canonical_mobile(value):
@@ -10,11 +11,13 @@ def canonical_mobile(value):
     return value
 
 
-def normalized_rows(model, alias, per_workspace=False):
+def normalized_rows(model, alias, per_workspace=False, skip_ids=None):
     seen = {}
     updates = []
     fields = ('pk', 'mobile', 'sys_ws_id') if per_workspace else ('pk', 'mobile')
     for row in model.objects.using(alias).only(*fields).iterator():
+        if skip_ids and row.pk in skip_ids:
+            continue
         mobile = canonical_mobile(row.mobile)
         key = (row.sys_ws_id, mobile) if per_workspace else mobile
         if mobile is not None and key in seen:
@@ -35,9 +38,10 @@ def check_and_normalize_mobiles(apps, schema_editor):
     customer = apps.get_model('customer', 'Customer')
     user = apps.get_model('cuser', 'CUser')
     role = apps.get_model('cuser', 'CuserRole')
-    # Audit both tables before writing. A collision never silently merges people.
+    # Quarantine legacy conflicts automatically, retaining a persistent audit.
+    skipped_users = repair_conflicts(apps, schema_editor, canonical_mobile)
     customer_updates = normalized_rows(customer, alias, per_workspace=True)
-    user_updates = normalized_rows(user, alias)
+    user_updates = normalized_rows(user, alias, skip_ids=skipped_users)
     role_updates = []
     for row in role.objects.using(alias).only('pk', 'mobile').iterator():
         normalized = canonical_mobile(row.mobile)
@@ -47,6 +51,11 @@ def check_and_normalize_mobiles(apps, schema_editor):
     customer.objects.using(alias).bulk_update(customer_updates, ['mobile'], batch_size=500)
     user.objects.using(alias).bulk_update(user_updates, ['mobile'], batch_size=500)
     role.objects.using(alias).bulk_update(role_updates, ['mobile'], batch_size=500)
+    if schema_editor.connection.vendor == 'postgresql':
+        # Drain deferred FK checks from panel detachment before AddConstraint
+        # alters the same table in this atomic migration.
+        with schema_editor.connection.cursor() as cursor:
+            cursor.execute('SET CONSTRAINTS ALL IMMEDIATE')
 
 
 class Migration(migrations.Migration):