clean up Award implementation and fix migration conflicts

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/nest/admin/badge.py

@@ -12,4 +12,4 @@ class BadgeAdmin(admin.ModelAdmin):
     list_display = ("name", "description", "weight", "css_class")
     list_filter = ("weight",)
     search_fields = ("name", "description")
-    ordering = ("weight", "name")
+    ordering = ("weight", "name")

backend/apps/nest/models/badge.py

@@ -38,4 +38,4 @@ class Meta:
 
     def __str__(self) -> str:
         """Return the badge string representation."""
-        return self.name
+        return self.name

backend/apps/owasp/Makefile

@@ -64,3 +64,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/admin/award.py

@@ -13,14 +13,12 @@ class AwardAdmin(admin.ModelAdmin):
         "name",
         "category",
         "year",
-        "award_type",
         "winner_name",
         "user",
         "nest_created_at",
         "nest_updated_at",
     )
     list_filter = (
-        "award_type",
         "category",
         "year",
     )
@@ -38,7 +36,7 @@ class AwardAdmin(admin.ModelAdmin):
     fieldsets = (
         (
             "Basic Information",
-            {"fields": ("name", "category", "award_type", "year", "description")},
+            {"fields": ("name", "category", "year", "description")},
         ),
         (
             "Winner Information",

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

@@ -0,0 +1,43 @@
+"""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
+        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 without WASPY awards
+        users_with_badge = User.objects.filter(badges=waspy_badge)
+        for user in users_with_badge:
+            if not Award.get_user_waspy_awards(user).exists():
+                user.badges.remove(waspy_badge)
+
+        self.stdout.write(f"Updated badges for {waspy_winners.count()} WASPY winners")

backend/apps/owasp/migrations/0045_award.py

@@ -25,7 +25,14 @@ class Migration(migrations.Migration):
                 (
                     "category",
                     models.CharField(
-                        help_text="Award category (e.g., 'WASPY', 'Lifetime Achievement')",
+                        choices=[
+                            ("WASPY", "WASPY"),
+                            (
+                                "Distinguished Lifetime Memberships",
+                                "Distinguished Lifetime Memberships",
+                            ),
+                        ],
+                        help_text="Award category (e.g., 'WASPY', 'Distinguished Lifetime Memberships')",
                         max_length=100,
                         verbose_name="Category",
                     ),
@@ -50,9 +57,7 @@ class Migration(migrations.Migration):
                 (
                     "year",
                     models.IntegerField(
-                        blank=True,
-                        help_text="Year the award was given (null for category definitions)",
-                        null=True,
+                        help_text="Year the award was given",
                         verbose_name="Year",
                     ),
                 ),
@@ -85,16 +90,6 @@ class Migration(migrations.Migration):
                         verbose_name="Winner Image",
                     ),
                 ),
-                (
-                    "award_type",
-                    models.CharField(
-                        choices=[("category", "Category"), ("award", "Award")],
-                        default="award",
-                        help_text="Type of entry: category definition or individual award",
-                        max_length=20,
-                        verbose_name="Award Type",
-                    ),
-                ),
                 (
                     "user",
                     models.ForeignKey(

backend/apps/owasp/migrations/0046_merge_0045_badge_0045_project_audience.py

@@ -5,7 +5,7 @@
 
 class Migration(migrations.Migration):
     dependencies = [
-        ("owasp", "0045_badge"),
+        ("owasp", "0045_award"),
         ("owasp", "0045_project_audience"),
     ]