Add management command link_individual_members_to_users_by_email

Author: ertgl

members/management/commands/link_individual_members_to_users_by_email.py

@@ -0,0 +1,19 @@
+from django.core.management.base import BaseCommand
+from django.db import transaction
+
+from ...models import IndividualMember
+
+
+class Command(BaseCommand):
+    def add_arguments(self, parser):
+        parser.add_argument(
+            "--no-logging",
+            action="store_true",
+            help="Disable writing number of updated records to stdout.",
+        )
+
+    def handle(self, *args, **options):
+        with transaction.atomic():
+            count = IndividualMember.match_and_set_users_by_email()
+        if not options["no_logging"]:
+            self.stdout.write(f"Updated: {count}")

members/models.py

@@ -2,6 +2,7 @@
 from enum import StrEnum
 
 from django.conf import settings
+from django.contrib.auth import get_user_model
 from django.core import signing
 from django.core.mail import send_mail
 from django.db import models, transaction
@@ -17,6 +18,9 @@
 
 from djangoproject.thumbnails import LogoThumbnailMixin
 
+User = get_user_model()
+
+
 BRONZE_MEMBERSHIP = 1
 SILVER_MEMBERSHIP = 2
 GOLD_MEMBERSHIP = 3
@@ -83,6 +87,22 @@ class Meta:
             ),
         ]
 
+    @classmethod
+    def match_and_set_users_by_email(cls, queryset=None):
+        updated_count = 0
+        if queryset is None:
+            queryset = cls.objects.all()
+        queryset = queryset.filter(user_id__isnull=True)
+        # Wait for other active transactions to prevent race conditions.
+        queryset = queryset.select_for_update(nowait=False)
+        users_queryset = User.objects.filter(
+            email__in=queryset.values_list("email", flat=True),
+        )
+        for user in users_queryset.iterator():
+            with transaction.atomic():
+                updated_count += cls.objects.filter(email=user.email).update(user=user)
+        return updated_count
+
     @classmethod
     def send_account_invite_mails(cls, queryset):
         results = {}