feat [server]: calendar view grouped by projection month

server/myapi/views/std_views.py

@@ -1,7 +1,10 @@
 # from django.shortcuts import render
 from datetime import datetime
+from itertools import groupby
 from typing import Callable
 
+from django.db.models import Q
+from django.db.models.functions import TruncMonth
 from rest_framework import viewsets, mixins
 from rest_framework.decorators import action
 from rest_framework.renderers import StaticHTMLRenderer
@@ -50,8 +53,27 @@ class FilmViewSet(mixins.RetrieveModelMixin, mixins.ListModelMixin, GenericViewS
     serializer_class = FilmSerializer
 
     def get_queryset(self):
-        queryset = Film.objects.all().filter(is_confirmed=True).order_by("projection_date")
+        queryset = Film.objects.all().filter(is_confirmed=True)
         past = self.request.query_params.get("past", False)
-        if past:
-            return queryset.filter(projection_date__date__lt=datetime.now())
-        return queryset.filter(projection_date__date__gte=datetime.now())
+        date_filter = (
+            Q(projection_date__date__lt=datetime.now())
+            if past
+            else Q(projection_date__date__gte=datetime.now())
+        )
+        ordering = f"{'-' if past else ''}projection_date"
+        return queryset.filter(date_filter).order_by(ordering)
+
+    @action(detail=False, methods=["GET"])
+    def calendar(self, request):
+        qs = self.get_queryset().annotate(
+            projection_month=TruncMonth("projection_date")
+        )
+        grouped = groupby(qs, lambda f: f.projection_month)
+        data = [
+            {
+                "projection_month": month.strftime("%B %Y"),
+                "films": FilmSerializer(monthly_data, many=True).data,
+            }
+            for (month, monthly_data) in grouped
+        ]
+        return Response(data)