Add milestones field to project model

Author: ahmedxgouda

backend/apps/owasp/api/internal/nodes/project.py

@@ -9,7 +9,6 @@
 from apps.github.api.internal.nodes.pull_request import PullRequestNode
 from apps.github.api.internal.nodes.release import ReleaseNode
 from apps.github.api.internal.nodes.repository import RepositoryNode
-from apps.github.models.milestone import Milestone
 from apps.owasp.api.internal.nodes.common import GenericEntityNode
 from apps.owasp.api.internal.nodes.project_health_metrics import (
     ProjectHealthMetricsNode,
@@ -84,24 +83,10 @@ def recent_issues(self, root: Project) -> list[IssueNode]:
         """Resolve recent issues."""
         return root.issues.order_by("-created_at")[:RECENT_ISSUES_LIMIT]
 
-    @strawberry_django.field
+    @strawberry_django.field(prefetch_related=["recent_milestones"])
     def recent_milestones(self, root: Project, limit: int = 5) -> list[MilestoneNode]:
         """Resolve recent milestones."""
-        return (
-            Milestone.objects.filter(
-                repository__in=root.repositories.all(),
-            )
-            .select_related(
-                "repository__organization",
-                "author__owasp_profile",
-            )
-            .prefetch_related(
-                "labels",
-            )
-            .order_by("-created_at")[:limit]
-            if (limit := min(limit, MAX_LIMIT)) > 0
-            else []
-        )
+        return root.milestones.order_by("-created_at")[:limit]
 
     @strawberry_django.field
     def recent_pull_requests(self, root: Project) -> list[PullRequestNode]:

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

@@ -2,6 +2,7 @@
 
 from django.core.management.base import BaseCommand
 
+from apps.github.models.milestone import Milestone
 from apps.owasp.models.project import Project
 
 
@@ -102,6 +103,11 @@ def handle(self, *_args, **options) -> None:
             project.languages = sorted(languages)
             project.licenses = sorted(licenses)
             project.topics = sorted(topics)
+            project.milestones.set(
+                Milestone.objects.filter(
+                    repository__in=project.repositories.all(),
+                )
+            )
 
             project.has_active_repositories = project.repositories.filter(
                 is_archived=False,

backend/apps/owasp/migrations/0073_project_milestones.py

@@ -0,0 +1,20 @@
+# Generated by Django 6.0 on 2026-01-23 08:59
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("github", "0041_milestone_github_milestone_created_at_and_more"),
+        ("owasp", "0072_project_project_name_gin_idx_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="project",
+            name="milestones",
+            field=models.ManyToManyField(
+                blank=True, to="github.milestone", verbose_name="Milestones"
+            ),
+        ),
+    ]

backend/apps/owasp/models/project.py

@@ -15,7 +15,6 @@
 from apps.common.utils import get_absolute_url
 from apps.core.models.prompt import Prompt
 from apps.github.models.issue import Issue
-from apps.github.models.milestone import Milestone
 from apps.github.models.pull_request import PullRequest
 from apps.github.models.release import Release
 from apps.owasp.models.common import RepositoryBasedEntityModel
@@ -137,6 +136,12 @@ class Meta:
     )
 
     # M2Ms.
+
+    milestones = models.ManyToManyField(
+        "github.Milestone",
+        verbose_name="Milestones",
+        blank=True,
+    )
     organizations = models.ManyToManyField(
         "github.Organization",
         verbose_name="Organizations",
@@ -301,22 +306,6 @@ def published_releases(self):
             "repository__organization",
         )
 
-    @property
-    def recent_milestones(self):
-        """Return recent milestones."""
-        return (
-            Milestone.objects.filter(
-                repository__in=self.repositories.all(),
-            )
-            .select_related(
-                "author",
-                "repository",
-            )
-            .prefetch_related(
-                "labels",
-            )
-        )
-
     @property
     def recent_releases_count(self) -> int:
         """Return count of recent releases per a specific period."""

docker-compose/local/compose.yaml

@@ -66,7 +66,7 @@ services:
     networks:
       - nest-network
     volumes:
-      - db-data:/var/lib/postgresql/data
+      - db-data-milestones:/var/lib/postgresql/data
 
   docs:
     container_name: nest-docs
@@ -143,7 +143,7 @@ networks:
 volumes:
   backend-venv:
   cache-data:
-  db-data:
+  db-data-milestones:
   docs-venv:
   frontend-next:
   frontend-node-modules: