Merge pull request #42 from toggle-corp/feat/add-pmer

feat(pmer): add pmer feature

Author: sandeshit

apps/pmer/__init__.py

@@ -0,0 +1,12 @@
+from django.contrib import admin
+
+from .models import PmerReport
+
+
+@admin.register(PmerReport)
+class PmerReportAdmin(admin.ModelAdmin):
+    list_display = ["title", "category", "report_type", "visibility"]
+    list_filter = ["category", "report_type", "visibility"]
+    search_fields = ["title"]
+    readonly_fields = ["created_at", "updated_at"]
+    ordering = ["-created_at"]

apps/pmer/admin.py

@@ -0,0 +1,7 @@
+from django.apps import AppConfig
+
+
+class PmerConfig(AppConfig):
+    default_auto_field = "django.db.models.BigAutoField"
+    name = "apps.pmer"
+    verbose_name = "PMER"

apps/pmer/apps.py

@@ -0,0 +1,19 @@
+from factory.declarations import Sequence, SubFactory
+from factory.django import DjangoModelFactory, FileField
+
+from apps.users.factories import UserFactory
+
+from .models import PmerReport
+
+
+class PmerReportFactory(DjangoModelFactory[PmerReport]):
+    title = Sequence(lambda n: f"PMER Report {n}")
+    description = Sequence(lambda n: f"PMER report description {n}")
+    file = FileField(filename="pmer-report.pdf")
+    category = PmerReport.Category.DPR
+    report_type = PmerReport.DocumentType.ANNUAL_REPORT
+    visibility = PmerReport.Visibility.PUBLIC
+    created_by = SubFactory(UserFactory)
+
+    class Meta:  # type: ignore[misc]
+        model = PmerReport

apps/pmer/factories.py

@@ -0,0 +1,17 @@
+import strawberry
+import strawberry_django
+
+from apps.pmer.models import PmerReport, PmerReportCategory, PmerReportDocumentType
+from apps.reports.models import ReportVisibility
+
+
+@strawberry_django.filters.filter(PmerReport, lookups=True)
+class PmerReportFilter:
+    id: strawberry.ID | None = strawberry.UNSET
+    category: PmerReportCategory | None = strawberry.UNSET
+    report_type: PmerReportDocumentType | None = strawberry.UNSET
+    title: strawberry.auto
+    department: strawberry.auto
+    project: strawberry.auto
+    region_id: strawberry.ID | None = strawberry.UNSET
+    visibility: ReportVisibility | None = strawberry.UNSET

apps/pmer/graphql/__init__.py

@@ -0,0 +1,34 @@
+import strawberry
+import strawberry_django
+from strawberry.file_uploads import Upload
+
+from apps.pmer.models import PmerReport, PmerReportCategory, PmerReportDocumentType
+from apps.reports.models import ReportVisibility
+
+
+@strawberry_django.input(PmerReport)
+class PmerReportCreateInput:
+    title: strawberry.auto
+    category: PmerReportCategory
+    report_type: PmerReportDocumentType
+    file: Upload
+    description: strawberry.auto
+    department: strawberry.auto
+    project: strawberry.auto
+    region: strawberry.ID | None = strawberry.UNSET
+    visibility: ReportVisibility = ReportVisibility.PUBLIC
+
+
+@strawberry_django.partial(PmerReport)
+class PmerReportUpdateInput:
+    title: strawberry.auto
+    # `strawberry.auto` cannot resolve an IntegerChoicesField to a registered
+    # GraphQL enum, so the choices enum is named explicitly here.
+    category: PmerReportCategory | None = strawberry.UNSET
+    report_type: PmerReportDocumentType | None = strawberry.UNSET
+    file: Upload | None = strawberry.UNSET
+    description: strawberry.auto
+    department: strawberry.auto
+    project: strawberry.auto
+    region: strawberry.ID | None = strawberry.UNSET
+    visibility: ReportVisibility | None = strawberry.UNSET

apps/pmer/graphql/filters.py

@@ -0,0 +1,43 @@
+import strawberry
+import strawberry_django
+
+from apps.pmer.models import PmerReport
+from apps.pmer.serializers import PmerReportSerializer
+from main.graphql.context import Info
+from main.graphql.permissions import IsAuthenticated
+from utils.graphql.mutations import ModelMutation
+from utils.graphql.types import MutationResponseType
+
+from .inputs import PmerReportCreateInput, PmerReportUpdateInput
+from .types import PmerReportType
+
+
+@strawberry.type
+class Mutation:
+    @strawberry_django.mutation(permission_classes=[IsAuthenticated])
+    async def create_pmer_report(
+        self,
+        info: Info,
+        data: PmerReportCreateInput,
+    ) -> MutationResponseType[PmerReportType]:
+        return await ModelMutation(PmerReportSerializer).handle_create_mutation(data, info)
+
+    @strawberry_django.mutation(permission_classes=[IsAuthenticated])
+    async def update_pmer_report(
+        self,
+        info: Info,
+        id: strawberry.ID,
+        data: PmerReportUpdateInput,
+    ) -> MutationResponseType[PmerReportType]:
+        instance = await PmerReport.objects.aget(id=id)
+        return await ModelMutation(PmerReportSerializer).handle_update_mutation(data, info, instance)
+
+    @strawberry_django.mutation(permission_classes=[IsAuthenticated])
+    async def delete_pmer_report(
+        self,
+        info: Info,
+        id: strawberry.ID,
+    ) -> MutationResponseType[PmerReportType]:
+        instance = await PmerReport.objects.aget(id=id)
+        await instance.adelete()
+        return MutationResponseType(ok=True)

apps/pmer/graphql/inputs.py

@@ -0,0 +1,9 @@
+import strawberry
+import strawberry_django
+
+from apps.pmer.models import PmerReport
+
+
+@strawberry_django.order_type(PmerReport)
+class PmerReportOrder:
+    title: strawberry.auto

apps/pmer/graphql/mutations.py

@@ -0,0 +1,22 @@
+import strawberry
+import strawberry_django
+from strawberry_django.pagination import OffsetPaginated
+
+from main.graphql.permissions import IsAuthenticated
+
+from .filters import PmerReportFilter
+from .orders import PmerReportOrder
+from .types import PmerReportType
+
+
+@strawberry.type
+class Query:
+    pmer_reports: OffsetPaginated[PmerReportType] = strawberry_django.offset_paginated(
+        filters=PmerReportFilter,
+        order=PmerReportOrder,
+        permission_classes=[IsAuthenticated],
+    )
+
+    pmer_report: PmerReportType = strawberry_django.field(
+        permission_classes=[IsAuthenticated],
+    )