Added the ability to create timed announcements on the admin site

Author: paolo7

tmh_registry/registry/admin.py

@@ -15,7 +15,7 @@
     Zone,
     Region,
     HospitalRegionMapping,
-    RegionZoneMapping,
+    RegionZoneMapping, Announcement,
 )
 
 
@@ -67,7 +67,9 @@ class HospitalRegionMappingAdmin(ExportMixin, admin.ModelAdmin):
 class RegionZoneMappingAdmin(ExportMixin, admin.ModelAdmin):
     model = RegionZoneMapping
 
-
+@admin.register(Announcement)
+class AnnouncementAdmin(ExportMixin, admin.ModelAdmin):
+    model = Announcement
 """
 To allow CSV export on User model
 """

tmh_registry/registry/api/serializers.py

@@ -14,7 +14,7 @@
     Hospital,
     Patient,
     PatientHospitalMapping,
-    PreferredHospital,
+    PreferredHospital, Announcement,
 )
 
 
@@ -712,3 +712,8 @@ def get_patient_hospital_id(self, obj):
         ).first()
 
         return mapping.patient_hospital_id if mapping else None
+
+class AnnouncementSerializer(ModelSerializer):
+    class Meta:
+        model = Announcement
+        fields = ["id", "announcement_text", "display_from", "display_until", "created_at"]

tmh_registry/registry/api/viewsets.py

@@ -17,7 +17,8 @@
 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
+from rest_framework.viewsets import GenericViewSet, ReadOnlyModelViewSet
+from django.utils.timezone import now
 
 from ..models import (
     Discharge,
@@ -27,6 +28,7 @@
     Patient,
     PatientHospitalMapping,
     PreferredHospital,
+    Announcement,
 )
 from .serializers import (
     CreatePatientSerializer,
@@ -41,7 +43,7 @@
     PatientHospitalMappingWriteSerializer,
     PreferredHospitalReadSerializer,
     ReadPatientSerializer,
-    SurgeonEpisodeSummarySerializer, OwnedEpisodeSerializer, UnlinkedPatientSerializer,
+    SurgeonEpisodeSummarySerializer, OwnedEpisodeSerializer, UnlinkedPatientSerializer, AnnouncementSerializer,
 )
 from ...users.models import MedicalPersonnel
 
@@ -401,3 +403,13 @@ def get_serializer_context(self):
         context = super().get_serializer_context()
         context['request'] = self.request
         return context
+
+class AnnouncementViewSet(ReadOnlyModelViewSet):
+    serializer_class = AnnouncementSerializer
+
+    def get_queryset(self):
+        current_time = now()
+        return Announcement.objects.filter(
+            Q(display_from__lte=current_time) | Q(display_from__isnull=True),
+            Q(display_until__gte=current_time) | Q(display_until__isnull=True)
+        ).order_by("-created_at")

tmh_registry/registry/migrations/0039_announcement.py

@@ -0,0 +1,23 @@
+# Generated by Django 3.1.3 on 2025-05-27 07:26
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('registry', '0038_auto_20250519_0903'),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name='Announcement',
+            fields=[
+                ('id', models.AutoField(auto_created=True, primary_key=True, serialize=False, verbose_name='ID')),
+                ('announcement_text', models.TextField()),
+                ('display_from', models.DateTimeField(blank=True, null=True)),
+                ('display_until', models.DateTimeField(blank=True, null=True)),
+                ('created_at', models.DateTimeField(auto_now_add=True)),
+            ],
+        ),
+    ]

tmh_registry/registry/models.py

@@ -273,3 +273,11 @@ def __str__(self):
     class Meta:
         verbose_name_plural = "Region-Zone Mappings"
 
+class Announcement(Model):
+    announcement_text = TextField()
+    display_from = DateTimeField(null=True, blank=True)
+    display_until = DateTimeField(null=True, blank=True)
+    created_at = DateTimeField(auto_now_add=True)
+
+    def __str__(self):
+        return f"Announcement ({self.created_at}): {self.announcement_text[:50]}"

tmh_registry/registry/urls.py

@@ -10,7 +10,9 @@
     PatientViewSet,
     PreferredHospitalViewSet,
     SurgeonEpisodeSummaryViewSet,
-    OwnedEpisodesViewSet, UnlinkedPatientsViewSet,
+    OwnedEpisodesViewSet,
+    UnlinkedPatientsViewSet,
+    AnnouncementViewSet
 )
 
 router = DefaultRouter()
@@ -24,7 +26,7 @@
 router.register(r'surgeon-episode-summary', SurgeonEpisodeSummaryViewSet, basename='surgeon-episode-summary')
 router.register(r"owned-episodes", OwnedEpisodesViewSet, basename='owned-episodes')
 router.register(r'unlinked-patients', UnlinkedPatientsViewSet, basename='unlinked-patients')
-
+router.register(r"announcements", AnnouncementViewSet, basename="announcements")
 
 urlpatterns = [
     path("", include(router.urls)),