Added preferred hospital functionality

Author: paolo7

tmh_registry/registry/admin.py

@@ -11,6 +11,7 @@
     Hospital,
     Patient,
     PatientHospitalMapping,
+    PreferredHospital,
 )
 
 
@@ -38,6 +39,9 @@ class PatientAdmin(ExportMixin, admin.ModelAdmin):
 class PatientHospitalMappingAdmin(ExportMixin, admin.ModelAdmin):
     model = PatientHospitalMapping
 
+@admin.register(PreferredHospital)
+class PreferredHospitalAdmin(ExportMixin, admin.ModelAdmin):
+    model = PreferredHospital
 
 @admin.register(Episode)
 class EpisodeAdmin(ExportMixin, admin.ModelAdmin):

tmh_registry/registry/api/serializers.py

@@ -14,6 +14,7 @@
     Hospital,
     Patient,
     PatientHospitalMapping,
+    PreferredHospital,
 )
 
 
@@ -239,6 +240,23 @@ def to_representation(self, instance):
 
         return data
 
+class PreferredHospitalReadSerializer(ModelSerializer):
+    hospital = SerializerMethodField()
+
+    class Meta:
+        model = PreferredHospital
+        fields = ["hospital"]
+
+    def get_hospital(self, obj):
+        request = self.context.get("request")
+        if request and request.user:
+            try:
+                medical_personnel = MedicalPersonnel.objects.get(user=request.user)
+                if obj.medical_personnel == medical_personnel:
+                    return {"id": obj.hospital.id}
+            except MedicalPersonnel.DoesNotExist:
+                pass
+        return None
 
 class PatientHospitalMappingWriteSerializer(ModelSerializer):
     patient_id = PrimaryKeyRelatedField(queryset=Patient.objects.all())

tmh_registry/registry/api/viewsets.py

@@ -15,6 +15,7 @@
 from rest_framework.decorators import action
 from rest_framework.exceptions import NotFound
 from rest_framework.mixins import CreateModelMixin, RetrieveModelMixin
+from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 from rest_framework.viewsets import GenericViewSet
 
@@ -25,6 +26,7 @@
     Hospital,
     Patient,
     PatientHospitalMapping,
+    PreferredHospital,
 )
 from .serializers import (
     CreatePatientSerializer,
@@ -37,14 +39,33 @@
     HospitalSerializer,
     PatientHospitalMappingReadSerializer,
     PatientHospitalMappingWriteSerializer,
+    PreferredHospitalReadSerializer,
     ReadPatientSerializer,
 )
+from ...users.models import MedicalPersonnel
 
 
 class HospitalViewSet(viewsets.ReadOnlyModelViewSet):
     queryset = Hospital.objects.all()
     serializer_class = HospitalSerializer
 
+class PreferredHospitalViewSet(viewsets.GenericViewSet):
+    serializer_class = PreferredHospitalReadSerializer
+    permission_classes = [IsAuthenticated]
+
+    @action(detail=False, methods=['get'])
+    def retrieve_for_current_user(self, request, *args, **kwargs):
+        user = request.user
+        try:
+            medical_personnel = MedicalPersonnel.objects.get(user=user)
+        except MedicalPersonnel.DoesNotExist:
+            return Response({}, status=200)
+        try:
+            preferred_hospital = PreferredHospital.objects.get(medical_personnel=medical_personnel)
+        except PreferredHospital.DoesNotExist:
+            return Response({}, status=200)
+        serializer = self.get_serializer(preferred_hospital)
+        return Response(serializer.data)
 
 class PatientFilterSet(FilterSet):
     hospital_id = NumberFilter(

tmh_registry/registry/migrations/0037_preferredhospital.py

@@ -0,0 +1,27 @@
+# Generated by Django 3.1.3 on 2024-07-29 13:58
+
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('users', '0008_auto_20210630_1407'),
+        ('registry', '0036_followup_recurrence'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='PreferredHospital',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('hospital', models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, related_name='preferred_by_medical_personnel', to='registry.hospital')),
+                ('medical_personnel', models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, related_name='preferred_hospital', to='users.medicalpersonnel')),
+            ],
+            options={
+                'verbose_name_plural': 'Medical Personnel Preferred Hospitals',
+                'unique_together': {('medical_personnel', 'hospital')},
+            },
+        ),
+    ]

tmh_registry/registry/models.py

@@ -73,11 +73,27 @@ class Meta:
             ("patient", "hospital"),
             ("hospital", "patient_hospital_id"),
         )
-        verbose_name_plural = "Patient-Hospital mappings"
+        verbose_name_plural = "Patient-Hospital Mappings"
 
     def __str__(self):
         return f"Patient {self.patient.full_name} ({self.patient_hospital_id}) - Hospital {self.hospital.name}"
 
+class PreferredHospital(Model):
+    medical_personnel = OneToOneField(
+        MedicalPersonnel, on_delete=CASCADE, related_name="preferred_hospital"
+    )
+    hospital = ForeignKey(
+        Hospital, on_delete=CASCADE, related_name="preferred_by_medical_personnel"
+    )
+
+    class Meta:
+        unique_together = (
+            ("medical_personnel", "hospital"),
+        )
+        verbose_name_plural = "Medical Personnel Preferred Hospitals"
+
+    def __str__(self):
+        return f"{self.medical_personnel.user.first_name} {self.medical_personnel.user.last_name} ({self.medical_personnel.user.username}) - Hospital {self.hospital.name}"
 
 class Episode(Model):
     class EpisodeChoices(TextChoices):

tmh_registry/registry/urls.py

@@ -8,6 +8,7 @@
     HospitalViewSet,
     PatientHospitalMappingViewset,
     PatientViewSet,
+    PreferredHospitalViewSet,
 )
 
 router = DefaultRouter()
@@ -17,6 +18,7 @@
 router.register(r"episodes", EpisodeViewset)
 router.register(r"discharges", DischargeViewset)
 router.register(r"follow-ups", FollowUpViewset)
+router.register(r"preferred-hospital", PreferredHospitalViewSet, basename='preferred-hospital')
 
 urlpatterns = [
     path("", include(router.urls)),