Update migration

Author: Marishka17

cvat/apps/iam/migrations/0002_delete_dummy_email_addresses_and_fix_case_related_mismatches.py

@@ -1,38 +1,52 @@
 # Generated by Django 4.2.21 on 2025-07-10 17:28
 
+from django.conf import settings
 from django.db import migrations
-from django.db.models import Count
+from django.db.models import Exists, F, OuterRef
 
 
 def delete_email_addresses_for_dummy_users(apps, schema_editor):
     EmailAddress = apps.get_model("account", "EmailAddress")
+    SocialAccount = apps.get_model("socialaccount", "SocialAccount")
 
+    social_accounts = SocialAccount.objects.filter(user=OuterRef("user"))
     # delete EmailAddress records for dummy users that were created by invitations
-    EmailAddress.objects.filter(verified=False).annotate(
-        social_accounts_count=Count("user__socialaccount")
-    ).filter(
-        social_accounts_count=0, user__password__startswith="!"  # nosec
+    EmailAddress.objects.filter(
+        ~Exists(social_accounts), verified=False, user__password__startswith="!"  # nosec
     ).delete()
 
 
-def normalize_email_case_mismatches(apps, schema_editor):
+def restore_email_addresses_for_dummy_users(apps, schema_editor):
+    User = apps.get_model(settings.AUTH_USER_MODEL)
     EmailAddress = apps.get_model("account", "EmailAddress")
+    SocialAccount = apps.get_model("socialaccount", "SocialAccount")
 
-    # fix case-related mismatches between User.email and EmailAddress.email
-    email_addresses = (
-        EmailAddress.objects.filter(verified=True)
-        .annotate(social_accounts_count=Count("user__socialaccount"))
-        .filter(social_accounts_count=0)
-        .select_related("user")
+    social_accounts = SocialAccount.objects.filter(user=OuterRef("user"))
+    email_addresses = EmailAddress.objects.filter(user=OuterRef("user"))
+    dummy_users = (
+        User.objects.filter(password__startswith="!")
+        .annotate(
+            has_social_account=Exists(social_accounts), has_email_address=Exists(email_addresses)
+        )
+        .filter(has_social_account=False, has_email_address=False)
     )
-    email_addresses_to_update = []
 
-    for email_address in email_addresses:
-        user_email = email_address.user.email
-        if email_address.email.lower() == user_email.lower() and email_address.email != user_email:
-            email_address.email = user_email
-            email_addresses_to_update.append(email_address)
-    EmailAddress.objects.bulk_update(email_addresses_to_update, ["email"])
+    email_addresses_to_create = [
+        EmailAddress(email=user.email, user=user, primary=True, verified=False)
+        for user in dummy_users
+    ]
+    EmailAddress.objects.bulk_create(email_addresses_to_create)
+
+
+def normalize_email_case_mismatches(apps, schema_editor):
+    EmailAddress = apps.get_model("account", "EmailAddress")
+    SocialAccount = apps.get_model("socialaccount", "SocialAccount")
+
+    social_accounts = SocialAccount.objects.filter(user=OuterRef("user"))
+    # fix case-related mismatches between User.email and EmailAddress.email
+    EmailAddress.objects.filter(
+        ~Exists(social_accounts), verified=True, email__iexact=F("user__email")
+    ).exclude(email=F("user__email")).update(email=F("user__email"))
 
 
 class Migration(migrations.Migration):
@@ -44,7 +58,7 @@ class Migration(migrations.Migration):
     operations = [
         migrations.RunPython(
             delete_email_addresses_for_dummy_users,
-            reverse_code=migrations.RunPython.noop,
+            reverse_code=restore_email_addresses_for_dummy_users,
         ),
         migrations.RunPython(
             normalize_email_case_mismatches,