chore(admin): add management command to create dummy contributor user

- Add link to view contributor users in team row.
- Allow to update only team in contributor admin form.

Author: Rup-Narayan-Rajbanshi

apps/contributor/admin.py

@@ -3,6 +3,8 @@
 
 from django.contrib import admin
 from django.db import transaction
+from django.urls import reverse
+from django.utils.html import format_html
 from djangoql.admin import DjangoQLSearchMixin
 
 from apps.common.admin import ArchivableResourceAdmin
@@ -19,6 +21,15 @@ class ContributorUserAdmin(DjangoQLSearchMixin, admin.ModelAdmin):
         "created_at",
         "modified_at",
     )
+    readonly_fields = (
+        "old_id",
+        "user_id",
+        "username",
+        "firebase_last_pushed",
+        "firebase_push_status",
+        "created_at",
+        "modified_at",
+    )
     list_filter = ("team",)
 
     @typing.override
@@ -42,6 +53,7 @@ class ContributorTeamAdmin(ArchivableResourceAdmin, DjangoQLSearchMixin, admin.M
         "is_archived",
         "created_at",
         "modified_at",
+        "view_contributor_users",
     )
     list_filter = ("is_archived",)
 
@@ -59,6 +71,10 @@ def save_model(self, request, obj, form, change):
         super().save_model(request, obj, form, change)  # type: ignore[reportAttributeAccessIssue]
         transaction.on_commit(lambda: FirebaseContributorTeam.task.delay(obj.id))
 
+    def view_contributor_users(self, obj):
+        url = reverse("admin:contributor_contributoruser_changelist") + f"?team__id__exact={obj.id}"
+        return format_html('<a href="{}">View Contributors</a>', url)
+
 
 @admin.register(ContributorUserGroupMembership)
 class ContributorUserGroupMembershipAdmin(DjangoQLSearchMixin, admin.ModelAdmin):

apps/contributor/factories.py

@@ -2,6 +2,7 @@
 # pyright: reportIncompatibleVariableOverride=false
 # pyright: reportMissingTypeArgument=false
 import typing
+import uuid
 
 import factory
 from factory.django import DjangoModelFactory
@@ -20,7 +21,7 @@ class ContributorUserFactory(DjangoModelFactory):
     class Meta:
         model = ContributorUser
 
-    user_id = factory.Sequence(lambda n: f"unique-contributor-user-id-{n}")
+    user_id = factory.LazyAttribute(lambda _: f"contributor-user-id-{uuid.uuid4().int % 1000000}")
     username = factory.Sequence(lambda n: f"Contributor User {n}")
 
 

apps/contributor/management/__init__.py

@@ -0,0 +1,38 @@
+import logging
+import typing
+import uuid
+
+from django.conf import settings
+from django.core.management.base import BaseCommand
+from django.db import transaction
+
+from apps.contributor.factories import ContributorTeamFactory, ContributorUserFactory
+from apps.contributor.firebase import FirebaseContributorTeam
+from apps.user.factories import UserFactory
+
+logger = logging.getLogger(__name__)
+
+
+class Command(BaseCommand):
+    help = "Create dummy contributor users. Also sync these users to firebase"
+
+    @typing.override
+    def handle(self, *args, **options):
+        if not settings.ENABLE_DANGER_MODE:
+            logger.warning("Dummy data generation is disabled")
+            return
+
+        user = UserFactory.create(email=f"user-{uuid.uuid4()}@mapwsipe.com")
+        user_resources = dict(
+            created_by=user,
+            modified_by=user,
+        )
+        team_a, team_b = ContributorTeamFactory.create_batch(2, **user_resources)
+        for team in [team_a, team_b]:
+            transaction.on_commit(lambda team_id=team.pk: FirebaseContributorTeam.task.delay(team_id))
+
+        ContributorUserFactory.create_batch(5, team=team_a)
+        ContributorUserFactory.create_batch(5, team=team_b)
+        ContributorUserFactory.create_batch(20)
+
+        logger.info("Contributor users created successfully")

apps/contributor/management/commands/__init__.py

@@ -98,8 +98,12 @@
     GOOGLE_APPLICATION_CREDENTIALS=str,
     # Pytest
     PYTEST_XDIST_WORKER=(str, None),
+    # Test
+    ENABLE_DANGER_MODE=(bool, False),
 )
 
+ENABLE_DANGER_MODE = env("ENABLE_DANGER_MODE")
+
 GIT_HELPER = GitHelper(BASE_DIR)
 
 # Quick-start development settings - unsuitable for production