Added Facility Hubs

care/facility/admin.py

@@ -6,6 +6,7 @@
 from care.facility.models.ambulance import Ambulance, AmbulanceDriver
 from care.facility.models.asset import Asset
 from care.facility.models.bed import AssetBed, Bed
+from care.facility.models.facility import FacilityHubSpoke
 from care.facility.models.file_upload import FileUpload
 from care.facility.models.patient_consultation import (
     PatientConsent,
@@ -96,6 +97,11 @@ class FacilityAdmin(DjangoQLSearchMixin, admin.ModelAdmin):
     djangoql_completion_enabled_by_default = True
 
 
+class FacilityHubSpokeAdmin(DjangoQLSearchMixin, admin.ModelAdmin):
+    search_fields = ["name"]
+    djangoql_completion_enabled_by_default = True
+
+
 class FacilityStaffAdmin(DjangoQLSearchMixin, admin.ModelAdmin):
     autocomplete_fields = ["facility", "staff"]
     djangoql_completion_enabled_by_default = True
@@ -189,6 +195,7 @@ class FacilityUserAdmin(DjangoQLSearchMixin, admin.ModelAdmin, ExportCsvMixin):
 
 
 admin.site.register(Facility, FacilityAdmin)
+admin.site.register(FacilityHubSpoke, FacilityHubSpokeAdmin)
 admin.site.register(FacilityStaff, FacilityStaffAdmin)
 admin.site.register(FacilityCapacity, FacilityCapacityAdmin)
 admin.site.register(FacilityVolunteer, FacilityVolunteerAdmin)

care/facility/api/serializers/facility.py

@@ -1,11 +1,12 @@
 import boto3
 from django.conf import settings
 from django.contrib.auth import get_user_model
+from django.db.models import Q
 from rest_framework import serializers
 
 from care.facility.models import FACILITY_TYPES, Facility, FacilityLocalGovtBody
 from care.facility.models.bed import Bed
-from care.facility.models.facility import FEATURE_CHOICES
+from care.facility.models.facility import FEATURE_CHOICES, FacilityHubSpoke
 from care.facility.models.patient import PatientRegistration
 from care.users.api.serializers.lsg import (
     DistrictSerializer,
@@ -14,6 +15,7 @@
     WardSerializer,
 )
 from care.utils.csp.config import BucketType, get_client_config
+from care.utils.serializer.external_id_field import ExternalIdSerializerField
 from config.serializers import ChoiceField
 from config.validators import MiddlewareDomainAddressValidator
 
@@ -166,6 +168,76 @@ def create(self, validated_data):
         return super().create(validated_data)
 
 
+class FacilitySpokeSerializer(serializers.ModelSerializer):
+    id = serializers.UUIDField(source="external_id", read_only=True)
+    spoke = ExternalIdSerializerField(
+        queryset=Facility.objects.all(), required=True, write_only=True
+    )
+    hub_object = FacilityBasicInfoSerializer(read_only=True, source="hub")
+    spoke_object = FacilityBasicInfoSerializer(read_only=True, source="spoke")
+
+    class Meta:
+        model = FacilityHubSpoke
+        fields = (
+            "id",
+            "spoke",
+            "hub_object",
+            "spoke_object",
+            "relationship",
+            "created_date",
+            "modified_date",
+        )
+        read_only_fields = (
+            "id",
+            "spoke_object",
+            "hub_object",
+            "created_date",
+            "modified_date",
+        )
+
+    def validate(self, data):
+        data["hub"] = self.context["facility"]
+        return data
+
+    def validate_spoke(self, spoke: Facility):
+        hub: Facility = self.context["facility"]
+        user = self.context["request"].user
+
+        if hub == spoke:
+            raise serializers.ValidationError("Cannot set a facility as it's own spoke")
+
+        if not (
+            user.is_superuser
+            or (
+                user.user_type <= User.TYPE_VALUE_MAP["LocalBodyAdmin"]
+                and spoke.state == user.state
+                and spoke.district == user.district
+                and spoke.local_body == user.local_body
+            )
+            or (
+                user.user_type > User.TYPE_VALUE_MAP["LocalBodyAdmin"]
+                and user.user_type <= User.TYPE_VALUE_MAP["DistrictAdmin"]
+                and spoke.state == user.state
+                and spoke.district == user.district
+            )
+            or (
+                user.user_type > User.TYPE_VALUE_MAP["DistrictAdmin"]
+                and user.user_type <= User.TYPE_VALUE_MAP["StateAdmin"]
+                and spoke.state == user.state
+            )
+        ):
+            raise serializers.ValidationError(
+                "You do not have permission to set this spoke"
+            )
+
+        if FacilityHubSpoke.objects.filter(
+            Q(hub=hub, spoke=spoke) | Q(hub=spoke, spoke=hub)
+        ).first():
+            raise serializers.ValidationError("Facility is already a spoke/hub")
+
+        return spoke
+
+
 class FacilityImageUploadSerializer(serializers.ModelSerializer):
     cover_image = serializers.ImageField(required=True, write_only=True)
     read_cover_image_url = serializers.URLField(read_only=True)

care/facility/api/viewsets/facility.py

@@ -1,4 +1,5 @@
 from django.conf import settings
+from django.shortcuts import get_object_or_404
 from django_filters import rest_framework as filters
 from djqscsv import render_to_csv_response
 from drf_spectacular.utils import extend_schema, extend_schema_view
@@ -14,15 +15,17 @@
     FacilityBasicInfoSerializer,
     FacilityImageUploadSerializer,
     FacilitySerializer,
+    FacilitySpokeSerializer,
 )
 from care.facility.models import (
     Facility,
     FacilityCapacity,
     FacilityPatientStatsHistory,
     HospitalDoctors,
 )
-from care.facility.models.facility import FacilityUser
+from care.facility.models.facility import FacilityHubSpoke, FacilityUser
 from care.users.models import User
+from care.utils.queryset.facility import get_facility_queryset
 
 
 class FacilityFilter(filters.FilterSet):
@@ -183,3 +186,25 @@ class AllFacilityViewSet(
     filterset_class = FacilityFilter
     lookup_field = "external_id"
     search_fields = ["name", "district__name", "state__name"]
+
+
+class FacilitySpokesViewSet(viewsets.ModelViewSet):
+    queryset = FacilityHubSpoke.objects.all().select_related("spoke", "hub")
+    serializer_class = FacilitySpokeSerializer
+    permission_classes = (IsAuthenticated,)
+    lookup_field = "external_id"
+
+    def get_queryset(self):
+        return self.queryset.filter(hub=self.get_facility())
+
+    def get_facility(self):
+        facilities = get_facility_queryset(self.request.user)
+        return get_object_or_404(
+            facilities.filter(external_id=self.kwargs["facility_external_id"])
+        )
+
+    def get_serializer_context(self):
+        facility = self.get_facility()
+        context = super().get_serializer_context()
+        context["facility"] = facility
+        return context

care/facility/migrations/0454_facilityhubspoke_and_more.py

@@ -0,0 +1,88 @@
+# Generated by Django 4.2.10 on 2024-08-29 21:13
+
+import uuid
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("facility", "0453_merge_20240824_2040"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="FacilityHubSpoke",
+            fields=[
+                (
+                    "id",
+                    models.BigAutoField(
+                        auto_created=True,
+                        primary_key=True,
+                        serialize=False,
+                        verbose_name="ID",
+                    ),
+                ),
+                (
+                    "external_id",
+                    models.UUIDField(db_index=True, default=uuid.uuid4, unique=True),
+                ),
+                (
+                    "created_date",
+                    models.DateTimeField(auto_now_add=True, db_index=True, null=True),
+                ),
+                (
+                    "modified_date",
+                    models.DateTimeField(auto_now=True, db_index=True, null=True),
+                ),
+                ("deleted", models.BooleanField(db_index=True, default=False)),
+                (
+                    "relationship",
+                    models.IntegerField(
+                        choices=[(1, "Regular Hub"), (2, "Tele ICU Hub")], default=1
+                    ),
+                ),
+                (
+                    "hub",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="hub_set",
+                        to="facility.facility",
+                    ),
+                ),
+                (
+                    "spoke",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE,
+                        related_name="spoke_set",
+                        to="facility.facility",
+                    ),
+                ),
+            ],
+        ),
+        migrations.AddConstraint(
+            model_name="facilityhubspoke",
+            constraint=models.CheckConstraint(
+                check=models.Q(("hub", models.F("spoke")), _negated=True),
+                name="hub_and_spoke_not_same",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="facilityhubspoke",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("deleted", False)),
+                fields=("hub", "spoke"),
+                name="unique_hub_spoke",
+            ),
+        ),
+        migrations.AddConstraint(
+            model_name="facilityhubspoke",
+            constraint=models.UniqueConstraint(
+                condition=models.Q(("deleted", False)),
+                fields=("spoke", "hub"),
+                name="unique_spoke_hub",
+            ),
+        ),
+    ]

care/facility/models/facility.py

@@ -3,6 +3,9 @@
 from django.contrib.postgres.fields import ArrayField
 from django.core.validators import MinValueValidator
 from django.db import models
+from django.db.models import IntegerChoices
+from django.db.models.constraints import CheckConstraint, UniqueConstraint
+from django.utils.translation import gettext_lazy as _
 from simple_history.models import HistoricalRecords
 
 from care.facility.models import FacilityBaseModel, reverse_choices
@@ -11,6 +14,7 @@
     FacilityRelatedPermissionMixin,
 )
 from care.users.models import District, LocalBody, State, Ward
+from care.utils.models.base import BaseModel
 from care.utils.models.validators import mobile_or_landline_number_validator
 
 User = get_user_model()
@@ -48,6 +52,11 @@
 ]
 
 
+class HubRelationship(IntegerChoices):
+    REGULAR_HUB = 1, _("Regular Hub")
+    TELE_ICU_HUB = 2, _("Tele ICU Hub")
+
+
 class FacilityFeature(models.IntegerChoices):
     CT_SCAN_FACILITY = 1, "CT Scan Facility"
     MATERNITY_CARE = 2, "Maternity Care"
@@ -289,6 +298,39 @@ def get_features_display(self):
     CSV_MAKE_PRETTY = {"facility_type": (lambda x: REVERSE_FACILITY_TYPES[x])}
 
 
+class FacilityHubSpoke(BaseModel):
+    hub = models.ForeignKey(Facility, on_delete=models.CASCADE, related_name="hub_set")
+    spoke = models.ForeignKey(
+        Facility, on_delete=models.CASCADE, related_name="spoke_set"
+    )
+    relationship = models.IntegerField(
+        choices=HubRelationship.choices, default=HubRelationship.REGULAR_HUB
+    )
+
+    class Meta:
+        constraints = [
+            # Ensure hub and spoke are not the same
+            CheckConstraint(
+                check=~models.Q(hub=models.F("spoke")),
+                name="hub_and_spoke_not_same",
+            ),
+            # bidirectional uniqueness
+            UniqueConstraint(
+                fields=["hub", "spoke"],
+                name="unique_hub_spoke",
+                condition=models.Q(deleted=False),
+            ),
+            UniqueConstraint(
+                fields=["spoke", "hub"],
+                name="unique_spoke_hub",
+                condition=models.Q(deleted=False),
+            ),
+        ]
+
+    def __str__(self):
+        return f"Hub: {self.hub.name} Spoke: {self.spoke.name}"
+
+
 class FacilityLocalGovtBody(models.Model):
     """
     DEPRECATED_FROM: 2020-03-29