Implemented a first version of a surgeon dashboard landing page

Author: paolo7

tmh_registry/registry/api/serializers.py

@@ -1,8 +1,8 @@
 from drf_yasg.utils import swagger_serializer_method
 from rest_framework.exceptions import ValidationError
-from rest_framework.fields import BooleanField, CharField, IntegerField
+from rest_framework.fields import BooleanField, CharField, IntegerField, DateField
 from rest_framework.relations import PrimaryKeyRelatedField
-from rest_framework.serializers import ModelSerializer, SerializerMethodField
+from rest_framework.serializers import ModelSerializer, SerializerMethodField, Serializer
 
 from ...common.utils.functions import get_text_choice_value_from_label
 from ...users.api.serializers import MedicalPersonnelSerializer
@@ -645,3 +645,28 @@ def create(self, validated_data):
         follow_up.attendees.set(attendees)
 
         return follow_up
+
+class SurgeonEpisodeSummarySerializer(Serializer):
+    episode_count = IntegerField()
+    last_episode_date = DateField(allow_null=True)
+
+
+class OwnedEpisodeSerializer(ModelSerializer):
+    patient_name = CharField(source='patient_hospital_mapping.patient.full_name', read_only=True)
+    patient_id = CharField(source='patient_hospital_mapping.patient.id', read_only=True)
+    hospital_id = CharField(source='patient_hospital_mapping.hospital.id', read_only=True)
+    follow_up_dates = SerializerMethodField()
+    discharge = SerializerMethodField()
+
+    class Meta:
+        model = Episode
+        fields = ['id', 'surgery_date', 'patient_name', 'discharge', 'follow_up_dates', 'patient_id', 'hospital_id']
+
+    def get_follow_up_dates(self, obj):
+        # Return only the dates of follow-up objects as a list of strings
+        return [follow_up.date for follow_up in obj.followup_set.all()]
+
+    def get_discharge(self, obj):
+        # Return the discharge date if available, otherwise None
+        discharge = getattr(obj, 'discharge', None)
+        return discharge.date if discharge else None

tmh_registry/registry/api/viewsets.py

@@ -1,5 +1,5 @@
 from django.core.exceptions import ObjectDoesNotExist
-from django.db.models import CharField, Q
+from django.db.models import CharField, Q, Count, Max
 from django.db.models.functions import Cast
 from django.utils.decorators import method_decorator
 from django_filters import (  # pylint: disable=E0401
@@ -41,6 +41,7 @@
     PatientHospitalMappingWriteSerializer,
     PreferredHospitalReadSerializer,
     ReadPatientSerializer,
+    SurgeonEpisodeSummarySerializer, OwnedEpisodeSerializer,
 )
 from ...users.models import MedicalPersonnel
 
@@ -67,6 +68,50 @@ def retrieve_for_current_user(self, request, *args, **kwargs):
         serializer = self.get_serializer(preferred_hospital)
         return Response(serializer.data)
 
+class SurgeonEpisodeSummaryViewSet(viewsets.ReadOnlyModelViewSet):
+    serializer_class = SurgeonEpisodeSummarySerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        user = self.request.user
+        try:
+            medical_personnel = MedicalPersonnel.objects.get(user=user)
+        except MedicalPersonnel.DoesNotExist:
+            return Episode.objects.none()
+
+        return Episode.objects.filter(surgeons=medical_personnel)
+
+    def list(self, request, *args, **kwargs):
+        queryset = self.get_queryset()
+
+        stats = queryset.aggregate(
+            episode_count=Count('id'),
+            last_episode_date=Max('surgery_date')
+        )
+
+        serializer = self.get_serializer(stats)
+        return Response(serializer.data)
+
+class OwnedEpisodesViewSet(viewsets.ReadOnlyModelViewSet):
+    pagination_class = None
+    serializer_class = OwnedEpisodeSerializer
+    permission_classes = [IsAuthenticated]
+
+    def get_queryset(self):
+        user = self.request.user
+        try:
+            medical_personnel = MedicalPersonnel.objects.get(user=user)
+        except MedicalPersonnel.DoesNotExist:
+            return Episode.objects.none()
+
+        episodes = (
+            Episode.objects
+            .filter(surgeons=medical_personnel)
+            .select_related("patient_hospital_mapping__patient")
+            .prefetch_related( "discharge", "followup_set")
+        )
+        return episodes
+
 class PatientFilterSet(FilterSet):
     hospital_id = NumberFilter(
         method="filter_hospital",

tmh_registry/registry/urls.py

@@ -9,6 +9,8 @@
     PatientHospitalMappingViewset,
     PatientViewSet,
     PreferredHospitalViewSet,
+    SurgeonEpisodeSummaryViewSet,
+    OwnedEpisodesViewSet,
 )
 
 router = DefaultRouter()
@@ -19,6 +21,9 @@
 router.register(r"discharges", DischargeViewset)
 router.register(r"follow-ups", FollowUpViewset)
 router.register(r"preferred-hospital", PreferredHospitalViewSet, basename='preferred-hospital')
+router.register(r'surgeon-episode-summary', SurgeonEpisodeSummaryViewSet, basename='surgeon-episode-summary')
+router.register(r"owned-episodes", OwnedEpisodesViewSet, basename='owned-episodes')
+
 
 urlpatterns = [
     path("", include(router.urls)),