Waspy Award Winner badge integration

Author: trucodd

backend/Makefile

@@ -109,6 +109,7 @@ shell-db:
 sync-data: \
 	update-data \
 	enrich-data \
+	owasp-update-badges \
 	index-data
 
 test-backend:

backend/apps/owasp/Makefile

@@ -67,3 +67,7 @@ owasp-update-events:
 owasp-update-sponsors:
 	@echo "Getting OWASP sponsors data"
 	@CMD="python manage.py owasp_update_sponsors" $(MAKE) exec-backend-command
+
+owasp-update-badges:
+	@echo "Updating OWASP user badges"
+	@CMD="python manage.py owasp_update_badges" $(MAKE) exec-backend-command

backend/apps/owasp/management/commands/owasp_update_badges.py

@@ -0,0 +1,45 @@
+"""Update user badges based on OWASP awards."""
+
+from django.core.management.base import BaseCommand
+
+from apps.github.models.user import User
+from apps.nest.models.badge import Badge
+from apps.owasp.models.award import Award
+
+
+class Command(BaseCommand):
+    """Update user badges based on OWASP awards."""
+
+    help = "Update user badges based on OWASP awards"
+
+    def handle(self, *args, **options):
+        """Handle the command execution."""
+        # Get or create WASPY badge
+        waspy_badge, created = Badge.objects.get_or_create(
+            name="WASPY Award Winner",
+            defaults={
+                "description": "Recipient of WASPY award from OWASP",
+                "css_class": "badge-waspy",
+                "weight": 10,
+            },
+        )
+
+        if created:
+            self.stdout.write(f"Created badge: {waspy_badge.name}")
+
+        # Get users with WASPY awards using the model method
+        waspy_winners = Award.get_waspy_award_winners()
+
+        # Add badge to WASPY winners
+        for user in waspy_winners:
+            user.badges.add(waspy_badge)
+
+        # Remove badge from users no longer on the WASPY winners list
+        users_with_badge = User.objects.filter(badges=waspy_badge)
+        waspy_winner_ids = set(waspy_winners.values_list("id", flat=True))
+
+        for user in users_with_badge:
+            if user.id not in waspy_winner_ids:
+                user.badges.remove(waspy_badge)
+
+        self.stdout.write(f"Updated badges for {waspy_winners.count()} WASPY winners")

backend/apps/owasp/models/award.py

@@ -105,7 +105,9 @@ def get_waspy_award_winners(cls):
         """
         from apps.github.models.user import User
 
-        return User.objects.filter(awards__category=cls.Category.WASPY).distinct()
+        return User.objects.filter(
+            awards__category=cls.Category.WASPY, awards__is_reviewed=True
+        ).distinct()
 
     @classmethod
     def get_user_waspy_awards(cls, user):