Merge pull request #211 from Mosquito-Alert/api_reports_filters

Added is_deleted, is_last_version in reports endpoint.

Author: aescobarr

tigaserver_app/managers.py

@@ -0,0 +1,36 @@
+from django.db import models
+
+class ReportQuerySet(models.QuerySet):
+
+    def __version_subquery(self):
+        # NOTE: need to force using "objects" (default manager)
+        return self.model._default_manager.filter(
+            user=models.OuterRef('user'), 
+            report_id=models.OuterRef('report_id'), 
+            type=models.OuterRef('type')
+        )
+
+    def __latest_version_subquery(self):
+        return self.__version_subquery().order_by('-version_number', '-server_upload_time')
+
+    def last_version_of_each(self, state: bool=True):
+        return self.filter(
+            models.Q(
+                pk__in=models.Subquery(
+                    self.__latest_version_subquery().values('pk')[:1]
+                ),
+                _negated=not state
+            )
+        )
+    
+    def deleted(self, state: bool = True):
+        return self.annotate(
+            has_negative_version=models.Exists(
+                self.__version_subquery().filter(version_number=-1).values('pk')
+            )
+        ).filter(
+            has_negative_version=state
+        )
+
+
+ReportManager = models.Manager.from_queryset(ReportQuerySet)

tigaserver_app/models.py

@@ -32,7 +32,7 @@
 import tigaserver_project.settings as conf
 from tigacrafting.messaging import send_message_android, send_message_ios
 
-
+from .managers import ReportManager
 from .mixins import TimeZoneModelMixin
 
 logger_report_geolocation = logging.getLogger('mosquitoalert.location.report_location')
@@ -630,6 +630,7 @@ class Report(TimeZoneModelMixin, models.Model):
     )
 
     # Object Manager
+    objects = ReportManager()
 
     # Custom Properties
     @property

tigaserver_app/views.py

@@ -283,10 +283,42 @@ class ReportViewSet(ReadWriteOnlyModelViewSet):
 * report_id: The 4-digit report ID.
 * type: The report type (adult, site, or mission).
     """
-    queryset = Report.objects.all()
+    queryset = Report.objects.all().prefetch_related("responses")
     serializer_class = ReportSerializer
     filter_fields = ('user', 'version_number', 'report_id', 'type')
 
+    def get_queryset(self):
+        queryset = super().get_queryset()
+
+        # Get the filters from the request
+        is_deleted = self.request.query_params.get('is_deleted', None)
+        is_last_version = self.request.query_params.get('is_last_version', None)
+
+        # Apply additional filters if provided
+        if is_deleted is not None:
+            if is_deleted.lower() not in ['true', 'false']:
+                raise ParseError("Invalid value for 'is_deleted'. It should be 'true', 'false', or not provided.")
+
+            if is_deleted.lower() == 'true':
+                # Filter queryset to include only deleted records
+                queryset = queryset.deleted(state=True)
+            elif is_deleted.lower() == 'false':
+                # Filter queryset to exclude deleted records
+                queryset = queryset.deleted(state=False)
+
+        if is_last_version is not None:
+            if is_last_version.lower() not in ['true', 'false']:
+                raise ParseError("Invalid value for 'is_last_version'. It should be 'true', 'false', or not provided.")
+
+            if is_last_version.lower() == 'true':
+                # Filter queryset to include only deleted records
+                queryset = queryset.last_version_of_each(state=True)
+            elif is_last_version.lower() == 'false':
+                # Filter queryset to exclude deleted records
+                queryset = queryset.last_version_of_each(state=False)
+
+        return queryset
+
 
 # For production version, substitute WriteOnlyModelViewSet
 class PhotoViewSet(ReadWriteOnlyModelViewSet):