Initialised and setup for the Strawberry

Author: abhayymishraa

.pre-commit-config.yaml

@@ -47,6 +47,7 @@ repos:
           - types-python-dateutil
           - types-PyYAML
           - types-requests
+          - strawberry-graphql-django
         args:
           - --config-file
           - backend/pyproject.toml

backend/apps/common/graphql/__init__.py

@@ -1,31 +1,39 @@
 """GitHub issue GraphQL node."""
 
-import graphene
+import strawberry
+import strawberry_django
 
-from apps.common.graphql.nodes import BaseNode
+from apps.github.graphql.nodes.user import UserNode
 from apps.github.models.issue import Issue
 
 
-class IssueNode(BaseNode):
+@strawberry_django.type(
+    Issue,
+    fields=[
+        "created_at",
+        "state",
+        "title",
+        "url",
+    ],
+)
+class IssueNode:
     """GitHub issue node."""
 
-    organization_name = graphene.String()
-    repository_name = graphene.String()
+    @strawberry.field
+    def author(self) -> UserNode | None:
+        """Resolve author."""
+        return self.author
 
-    class Meta:
-        model = Issue
-        fields = (
-            "author",
-            "created_at",
-            "state",
-            "title",
-            "url",
-        )
-
-    def resolve_organization_name(self, info) -> str | None:
+    @strawberry.field
+    def organization_name(self) -> str | None:
         """Return organization name."""
-        return self.repository.organization.login if self.repository.organization else None
+        return (
+            self.repository.organization.login
+            if self.repository and self.repository.organization
+            else None
+        )
 
-    def resolve_repository_name(self, info):
+    @strawberry.field
+    def repository_name(self) -> str | None:
         """Resolve the repository name."""
-        return self.repository.name
+        return self.repository.name if self.repository else None

backend/apps/common/graphql/nodes.py

@@ -1,33 +1,40 @@
 """Github Milestone Node."""
 
-import graphene
+import strawberry
+import strawberry_django
 
-from apps.common.graphql.nodes import BaseNode
+from apps.github.graphql.nodes.user import UserNode
 from apps.github.models.milestone import Milestone
 
 
-class MilestoneNode(BaseNode):
+@strawberry_django.type(
+    Milestone,
+    fields=[
+        "closed_issues_count",
+        "created_at",
+        "open_issues_count",
+        "title",
+        "url",
+    ],
+)
+class MilestoneNode:
     """Github Milestone Node."""
 
-    organization_name = graphene.String()
-    repository_name = graphene.String()
+    @strawberry.field
+    def author(self) -> UserNode | None:
+        """Resolve author."""
+        return self.author
 
-    class Meta:
-        model = Milestone
-
-        fields = (
-            "author",
-            "created_at",
-            "title",
-            "open_issues_count",
-            "closed_issues_count",
-            "url",
+    @strawberry.field
+    def organization_name(self) -> str | None:
+        """Return organization name."""
+        return (
+            self.repository.organization.login
+            if self.repository and self.repository.organization
+            else None
         )
 
-    def resolve_repository_name(self, info):
+    @strawberry.field
+    def repository_name(self) -> str | None:
         """Resolve repository name."""
-        return self.repository.name
-
-    def resolve_organization_name(self, info):
-        """Return organization name."""
-        return self.repository.organization.login if self.repository.organization else None
+        return self.repository.name if self.repository else None

backend/apps/common/graphql/queries.py

@@ -1,47 +1,46 @@
 """GitHub organization GraphQL node."""
 
-import graphene
+import strawberry
+import strawberry_django
 from django.db import models
 
-from apps.common.graphql.nodes import BaseNode
 from apps.github.models.organization import Organization
 from apps.github.models.repository import Repository
 from apps.github.models.repository_contributor import RepositoryContributor
 
 
-class OrganizationStatsNode(graphene.ObjectType):
+@strawberry.type
+class OrganizationStatsNode:
     """Organization stats node."""
 
-    total_repositories = graphene.Int()
-    total_contributors = graphene.Int()
-    total_stars = graphene.Int()
-    total_forks = graphene.Int()
-    total_issues = graphene.Int()
+    total_contributors: int
+    total_forks: int
+    total_issues: int
+    total_repositories: int
+    total_stars: int
 
 
-class OrganizationNode(BaseNode):
+@strawberry_django.type(
+    Organization,
+    fields=[
+        "avatar_url",
+        "collaborators_count",
+        "company",
+        "created_at",
+        "description",
+        "email",
+        "followers_count",
+        "location",
+        "login",
+        "name",
+        "updated_at",
+    ],
+)
+class OrganizationNode:
     """GitHub organization node."""
 
-    stats = graphene.Field(OrganizationStatsNode)
-    url = graphene.String()
-
-    class Meta:
-        model = Organization
-        fields = (
-            "avatar_url",
-            "collaborators_count",
-            "company",
-            "created_at",
-            "description",
-            "email",
-            "followers_count",
-            "location",
-            "login",
-            "name",
-            "updated_at",
-        )
-
-    def resolve_stats(self, info):
+    @strawberry.field
+    def stats(self) -> OrganizationStatsNode:
         """Resolve organization stats."""
         repositories = Repository.objects.filter(organization=self)
 
@@ -51,9 +50,10 @@ def resolve_stats(self, info):
             total_forks=models.Sum("forks_count"),
             total_issues=models.Sum("open_issues_count"),
         )
-        total_stars = aggregated_stats["total_stars"] or 0
-        total_forks = aggregated_stats["total_forks"] or 0
-        total_issues = aggregated_stats["total_issues"] or 0
+
+        total_stars = aggregated_stats.get("total_stars") or 0
+        total_forks = aggregated_stats.get("total_forks") or 0
+        total_issues = aggregated_stats.get("total_issues") or 0
 
         unique_contributors = (
             RepositoryContributor.objects.filter(repository__in=repositories)
@@ -70,6 +70,7 @@ def resolve_stats(self, info):
             total_issues=total_issues,
         )
 
-    def resolve_url(self, info):
+    @strawberry.field
+    def url(self) -> str:
         """Resolve organization URL."""
         return self.url

backend/apps/github/graphql/nodes/issue.py

@@ -1,34 +1,38 @@
 """GitHub Pull Request Node."""
 
-import graphene
+import strawberry
+import strawberry_django
 
-from apps.common.graphql.nodes import BaseNode
+from apps.github.graphql.nodes.user import UserNode
 from apps.github.models.pull_request import PullRequest
 
 
-class PullRequestNode(BaseNode):
+@strawberry_django.type(
+    PullRequest,
+    fields=[
+        "created_at",
+        "title",
+    ],
+)
+class PullRequestNode:
     """GitHub pull request node."""
 
-    organization_name = graphene.String()
-    repository_name = graphene.String()
-    url = graphene.String()
+    @strawberry.field
+    def author(self) -> UserNode | None:
+        """Resolve author."""
+        return self.author
 
-    class Meta:
-        model = PullRequest
-        fields = (
-            "author",
-            "created_at",
-            "title",
-        )
-
-    def resolve_organization_name(self, info):
+    @strawberry.field
+    def organization_name(self) -> str | None:
         """Return organization name."""
         return self.repository.organization.login if self.repository.organization else None
 
-    def resolve_repository_name(self, info):
+    @strawberry.field
+    def repository_name(self) -> str:
         """Resolve repository name."""
         return self.repository.name
 
-    def resolve_url(self, info):
+    @strawberry.field
+    def url(self) -> str:
         """Resolve URL."""
-        return self.url
+        return str(self.url) if self.url else ""

backend/apps/github/graphql/nodes/milestone.py

@@ -1,46 +1,46 @@
 """GitHub release GraphQL node."""
 
-from __future__ import annotations
+import strawberry
+import strawberry_django
 
-import graphene
-
-from apps.common.graphql.nodes import BaseNode
 from apps.github.graphql.nodes.user import UserNode
 from apps.github.models.release import Release
 from apps.owasp.constants import OWASP_ORGANIZATION_NAME
 
 
-class ReleaseNode(BaseNode):
+@strawberry_django.type(
+    Release,
+    fields=[
+        "is_pre_release",
+        "name",
+        "published_at",
+        "tag_name",
+    ],
+)
+class ReleaseNode:
     """GitHub release node."""
 
-    author = graphene.Field(UserNode)
-    organization_name = graphene.String()
-    project_name = graphene.String()
-    repository_name = graphene.String()
-    url = graphene.String()
-
-    class Meta:
-        model = Release
-        fields = (
-            "author",
-            "is_pre_release",
-            "name",
-            "published_at",
-            "tag_name",
-        )
-
-    def resolve_organization_name(self, info) -> str | None:
+    @strawberry.field
+    def author(self) -> UserNode | None:
+        """Resolve author."""
+        return self.author
+
+    @strawberry.field
+    def organization_name(self) -> str | None:
         """Return organization name."""
         return self.repository.organization.login if self.repository.organization else None
 
-    def resolve_project_name(self, info) -> str:
+    @strawberry.field
+    def project_name(self) -> str:
         """Return project name."""
         return self.repository.project.name.lstrip(OWASP_ORGANIZATION_NAME)
 
-    def resolve_repository_name(self, info) -> str:
+    @strawberry.field
+    def repository_name(self) -> str:
         """Return repository name."""
         return self.repository.name
 
-    def resolve_url(self, info) -> str:
+    @strawberry.field
+    def url(self) -> str:
         """Return release URL."""
         return self.url