Improved the "/sponsors" command for slack

backend/Makefile

@@ -102,6 +102,10 @@ owasp-update-events:
 	@echo "Getting OWASP events data"
 	@CMD="python manage.py owasp_update_events" $(MAKE) exec-backend-command
 
+owasp-update-sponsors:
+	@echo "Getting OWASP sponsors data"
+	@CMD="python manage.py owasp_update_sponsors" $(MAKE) exec-backend-command
+
 purge-data:
 	@CMD="python manage.py purge_data" $(MAKE) exec-backend-command
 
@@ -136,4 +140,5 @@ update-data: \
 	owasp-scrape-projects \
 	github-update-project-related-repositories \
 	owasp-aggregate-projects \
-	owasp-update-events
+	owasp-update-events \
+	owasp-update-sponsors

backend/apps/owasp/admin.py

@@ -10,6 +10,7 @@
 from apps.owasp.models.project_health_metrics import ProjectHealthMetrics
 from apps.owasp.models.project_health_requirements import ProjectHealthRequirements
 from apps.owasp.models.snapshot import Snapshot
+from apps.owasp.models.sponsors import Sponsor
 
 
 class GenericEntityAdminMixin:
@@ -141,10 +142,41 @@ class SnapshotAdmin(admin.ModelAdmin):
     )
 
 
+class SponsorAdmin(admin.ModelAdmin):
+    """Admin configuration for Sponsor model."""
+
+    list_display = (
+        "name",
+        "sort_name",
+        "sponsor_type",
+        "is_member",
+        "member_type",
+    )
+
+    search_fields = (
+        "name",
+        "sort_name",
+        "description",
+    )
+
+    list_filter = (
+        "sponsor_type",
+        "is_member",
+        "member_type",
+    )
+
+    fieldsets = (
+        ("Basic Information", {"fields": ("name", "sort_name", "description")}),
+        ("URLs and Images", {"fields": ("url", "job_url", "image_url")}),
+        ("Status", {"fields": ("is_member", "member_type", "sponsor_type")}),
+    )
+
+
 admin.site.register(Chapter, ChapterAdmin)
 admin.site.register(Committee, CommitteeAdmin)
 admin.site.register(Event, EventAdmin)
 admin.site.register(Project, ProjectAdmin)
 admin.site.register(ProjectHealthMetrics)
 admin.site.register(ProjectHealthRequirements)
 admin.site.register(Snapshot, SnapshotAdmin)
+admin.site.register(Sponsor, SponsorAdmin)

backend/apps/owasp/constants.py

@@ -1,3 +1,4 @@
 """OWASP app constants."""
 
 OWASP_ORGANIZATION_NAME = "OWASP"
+OWASP_ORGANIZATION_DATA_URL = "https://raw.githubusercontent.com/OWASP/owasp.github.io/main"

backend/apps/owasp/graphql/nodes/sponsors.py

@@ -0,0 +1,15 @@
+"""OWASP sponsors GraphQL node."""
+
+from apps.common.graphql.nodes import BaseNode
+from apps.owasp.models.sponsors import Sponsor
+
+
+class SponsorNode(BaseNode):
+    """Sponsor node."""
+
+    class Meta:
+        model = Sponsor
+        fields = (
+            "name",
+            "image_url",
+        )

backend/apps/owasp/graphql/queries/__init__.py

@@ -5,10 +5,11 @@
 from .event import EventQuery
 from .project import ProjectQuery
 from .snapshot import SnapshotQuery
+from .sponsors import SponsorQuery
 from .stats import StatsQuery
 
 
 class OwaspQuery(
-    ChapterQuery, CommitteeQuery, EventQuery, ProjectQuery, SnapshotQuery, StatsQuery
+    ChapterQuery, CommitteeQuery, EventQuery, ProjectQuery, SnapshotQuery, SponsorQuery, StatsQuery
 ):
     """OWASP queries."""

backend/apps/owasp/graphql/queries/sponsors.py

@@ -0,0 +1,17 @@
+"""OWASP sponsors GraphQL queries."""
+
+import graphene
+
+from apps.common.graphql.queries import BaseQuery
+from apps.owasp.graphql.nodes.sponsors import SponsorNode
+from apps.owasp.models.sponsors import Sponsor
+
+
+class SponsorQuery(BaseQuery):
+    """Sponsor queries."""
+
+    sponsors = graphene.List(SponsorNode)
+
+    def resolve_sponsors(root, info):
+        """Resolve sponsors."""
+        return Sponsor.objects.all()

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

@@ -0,0 +1,20 @@
+"""A command to add OWASP sponsors data."""
+
+import yaml
+from django.core.management.base import BaseCommand
+
+from apps.github.utils import get_repository_file_content
+from apps.owasp.models.sponsors import Sponsor
+
+
+class Command(BaseCommand):
+    help = "Import sponsors from the provided YAML file"
+
+    def handle(self, *args, **kwargs):
+        data = yaml.safe_load(
+            get_repository_file_content(
+                "https://raw.githubusercontent.com/OWASP/owasp.github.io/main/_data/corp_members.yml"
+            ).expandtabs()
+        )
+
+        Sponsor.bulk_save([Sponsor.update_data(sponsor_data) for sponsor_data in data])

backend/apps/owasp/migrations/0022_sponsor.py

@@ -0,0 +1,85 @@
+# Generated by Django 5.1.5 on 2025-03-04 15:28
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("owasp", "0021_alter_snapshot_key"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Sponsor",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                ("nest_created_at", models.DateTimeField(auto_now_add=True)),
+                ("nest_updated_at", models.DateTimeField(auto_now=True)),
+                (
+                    "description",
+                    models.TextField(blank=True, verbose_name="Description"),
+                ),
+                (
+                    "key",
+                    models.CharField(max_length=100, unique=True, verbose_name="Key"),
+                ),
+                ("name", models.CharField(max_length=255, verbose_name="Name")),
+                (
+                    "sort_name",
+                    models.CharField(max_length=255, verbose_name="Sort Name"),
+                ),
+                ("url", models.URLField(blank=True, verbose_name="Website URL")),
+                ("job_url", models.URLField(blank=True, verbose_name="Job URL")),
+                (
+                    "image_url",
+                    models.CharField(blank=True, max_length=255, verbose_name="Image Path"),
+                ),
+                (
+                    "is_member",
+                    models.BooleanField(default=False, verbose_name="Is Corporate Sponsor"),
+                ),
+                (
+                    "member_type",
+                    models.CharField(
+                        blank=True,
+                        choices=[
+                            ("Platinum", "Platinum"),
+                            ("Gold", "Gold"),
+                            ("Silver", "Silver"),
+                        ],
+                        default="Silver",
+                        max_length=20,
+                        verbose_name="Member Type",
+                    ),
+                ),
+                (
+                    "sponsor_type",
+                    models.CharField(
+                        choices=[
+                            ("Diamond", "Diamond"),
+                            ("Platinum", "Platinum"),
+                            ("Gold", "Gold"),
+                            ("Silver", "Silver"),
+                            ("Supporter", "Supporter"),
+                            ("Not a Sponsor", "Not Sponsor"),
+                        ],
+                        default="Not a Sponsor",
+                        max_length=20,
+                        verbose_name="Sponsor Type",
+                    ),
+                ),
+            ],
+            options={
+                "verbose_name_plural": "Sponsors",
+                "db_table": "owasp_sponsors",
+            },
+        ),
+    ]

backend/apps/owasp/models/sponsors.py

@@ -0,0 +1,134 @@
+"""OWASP app sponsor models."""
+
+from django.db import models
+
+from apps.common.models import BulkSaveModel, TimestampedModel
+from apps.common.utils import slugify
+from apps.github.utils import normalize_url
+from apps.owasp.constants import OWASP_ORGANIZATION_DATA_URL
+
+
+class Sponsor(BulkSaveModel, TimestampedModel):
+    """Sponsor model."""
+
+    objects = models.Manager()
+
+    class Meta:
+        db_table = "owasp_sponsors"
+        verbose_name_plural = "Sponsors"
+
+    class SponsorType(models.TextChoices):
+        DIAMOND = "Diamond"
+        PLATINUM = "Platinum"
+        GOLD = "Gold"
+        SILVER = "Silver"
+        SUPPORTER = "Supporter"
+        NOT_SPONSOR = "Not a Sponsor"
+
+    class MemberType(models.TextChoices):
+        PLATINUM = "Platinum"
+        GOLD = "Gold"
+        SILVER = "Silver"
+
+    # Basic information
+    description = models.TextField(verbose_name="Description", blank=True)
+    key = models.CharField(verbose_name="Key", max_length=100, unique=True)
+    name = models.CharField(verbose_name="Name", max_length=255)
+    sort_name = models.CharField(verbose_name="Sort Name", max_length=255)
+
+    # URLs and images
+    url = models.URLField(verbose_name="Website URL", blank=True)
+    job_url = models.URLField(verbose_name="Job URL", blank=True)
+    image_url = models.CharField(verbose_name="Image Path", max_length=255, blank=True)
+
+    # Status fields
+    is_member = models.BooleanField(verbose_name="Is Corporate Sponsor", default=False)
+    member_type = models.CharField(
+        verbose_name="Member Type",
+        max_length=20,
+        choices=MemberType.choices,
+        default=MemberType.SILVER,
+        blank=True,
+    )
+    sponsor_type = models.CharField(
+        verbose_name="Sponsor Type",
+        max_length=20,
+        choices=SponsorType.choices,
+        default=SponsorType.NOT_SPONSOR,
+    )
+
+    def __str__(self):
+        """Sponsor human readable representation."""
+        return f"{self.name}"
+
+    @property
+    def readable_member_type(self):
+        """Get human-readable member type."""
+        return self.MemberType(self.member_type).label
+
+    @property
+    def readable_sponsor_type(self):
+        """Get human-readable sponsor type."""
+        return self.SponsorType(self.sponsor_type).label
+
+    @staticmethod
+    def bulk_save(sponsors, fields=None):
+        """Bulk save sponsors."""
+        BulkSaveModel.bulk_save(Sponsor, sponsors, fields=fields)
+
+    @staticmethod
+    def update_data(data, save=True):
+        """Update sponsor data."""
+        key = slugify(data["name"])
+        try:
+            sponsor = Sponsor.objects.get(key=key)
+        except Sponsor.DoesNotExist:
+            sponsor = Sponsor(key=key)
+
+        sponsor.from_dict(data)
+
+        if save:
+            sponsor.save()
+
+        return sponsor
+
+    def from_dict(self, data):
+        """Update instance based on the dict data."""
+        image_path = data.get("image", "").lstrip("/")
+        image_url = f"{OWASP_ORGANIZATION_DATA_URL}/{image_path}"
+
+        sponsor_key = str(data.get("sponsor", "-1"))
+        member_key = str(data.get("membertype", "4"))
+
+        member_type_mapping = {
+            "2": self.MemberType.PLATINUM,
+            "3": self.MemberType.GOLD,
+            "4": self.MemberType.SILVER,
+        }
+
+        sponsor_type_mapping = {
+            "1": self.SponsorType.DIAMOND,
+            "2": self.SponsorType.PLATINUM,
+            "3": self.SponsorType.GOLD,
+            "4": self.SponsorType.SILVER,
+            "5": self.SponsorType.SUPPORTER,
+            "-1": self.SponsorType.NOT_SPONSOR,
+        }
+
+        sponsor_type_label = sponsor_type_mapping.get(sponsor_key, self.SponsorType.NOT_SPONSOR)
+        member_type_label = member_type_mapping.get(member_key, self.MemberType.SILVER)
+
+        fields = {
+            "name": data.get("name", ""),
+            "sort_name": data.get("sortname", "").capitalize(),
+            "description": data.get("description", ""),
+            "url": normalize_url(data.get("url", "")) or "",
+            "job_url": normalize_url(data.get("job_url", "")) or "",
+            "image_url": image_url,
+            "is_member": bool(data.get("member", False)),
+            "sponsor_type": sponsor_type_label,
+            "member_type": member_type_label,
+        }
+
+        for key, value in fields.items():
+            setattr(self, key, value)