implement myIncidentReports query.

onehealthtoolkit · Aug 25, 2022 · 52163a8 · 52163a8
1 parent 754585d
commit 52163a8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/reports/schema/query.py b/reports/schema/query.py
@@ -34,6 +34,7 @@ class Query(graphene.ObjectType):
     category = graphene.Field(CategoryType, id=graphene.ID(required=True))
     report_type = graphene.Field(ReportTypeType, id=graphene.ID(required=True))
     incident_reports = DjangoPaginationConnectionField(IncidentReportType)
+    my_incident_reports = DjangoPaginationConnectionField(IncidentReportType)
     incident_report = graphene.Field(IncidentReportType, id=graphene.ID(required=True))
     followup_report = graphene.Field(FollowupReportType, id=graphene.ID(required=True))
     reporter_notification = graphene.Field(
@@ -117,3 +118,13 @@ def resolve_incident_reports(root, info, **kwargs):
             query = query.filter(relevant_authorities__in=child_authorities)
 
         return query
+
+    @staticmethod
+    @login_required
+    def resolve_my_incident_reports(root, info, **kwargs):
+        user = info.context.user
+        return (
+            IncidentReport.objects.filter(reported_by=user)
+            .order_by("-created_at")
+            .prefetch_related("images", "reported_by", "report_type")
+        )