Added data preview/plot/export section to the signal set detailed view page

Author: dmytrotsko

src/rest_api/serializers.py

@@ -1,6 +1,6 @@
 from rest_framework.serializers import ModelSerializer, SlugRelatedField
 
-from signals.models import Signal
+from signals.models import Signal, Geography
 
 
 class SignalSerializer(ModelSerializer):
@@ -14,9 +14,14 @@ class SignalSerializer(ModelSerializer):
 
     class Meta:
         model = Signal
-        fields = [
-            "name",
-            "source",
-            "signal_type",
-            "available_geography",
-        ]
+        fields = ["name", "source", "signal_type", "available_geography", "time_type"]
+
+
+class AvailableGeographySerializer(ModelSerializer):
+    """
+    Serializer for the AvailableGeography model.
+    """
+
+    class Meta:
+        model = Geography
+        fields = ["name", "display_name"]

src/rest_api/urls.py

@@ -1,9 +1,12 @@
 from django.urls import path
 from django.urls.resolvers import URLPattern
 
-from rest_api.views import api_signal_detail_view
+from rest_api.views import api_signal_detail_view, api_available_geography_view
 
 
 urlpatterns: list[URLPattern] = [
-    path("signal/<int:pk>", api_signal_detail_view, name="api_signal_detail"),
+    path("rest_api/signal/<int:pk>", api_signal_detail_view, name="api_signal_detail"),
+    path("rest_api/signal/", api_signal_detail_view, name="api_signal_detail"),
+    path("rest_api/geo_level/<int:pk>", api_available_geography_view, name="api_available_geography_detail"),
+    path("rest_api/geo_level/", api_available_geography_view, name="api_available_geography_detail")
 ]

src/rest_api/views.py

@@ -1,8 +1,8 @@
 from rest_framework.decorators import api_view
 from rest_framework.response import Response
 
-from rest_api.serializers import SignalSerializer
-from signals.models import Signal
+from rest_api.serializers import SignalSerializer, AvailableGeographySerializer
+from signals.models import Signal, Geography
 
 
 @api_view(
@@ -30,3 +30,30 @@ def api_signal_detail_view(request, pk):
     if request.method == "GET":
         serializer = SignalSerializer(signal)
         return Response(serializer.data)
+
+
+@api_view(
+    [
+        "GET",
+    ]
+)
+def api_available_geography_view(request, pk):
+    """
+    API view for getting a single Geography.
+
+    Args:
+        request: The request object.
+        pk: The primary key of the Geography object.
+
+    Returns:
+        Response: The response object.
+    """
+
+    try:
+        geography = Geography.objects.get(pk=pk)
+    except Geography.DoesNotExist:
+        return Response(status=404)
+
+    if request.method == "GET":
+        serializer = AvailableGeographySerializer(geography)
+        return Response(serializer.data)

src/signal_sets/views.py

@@ -1,9 +1,12 @@
 import logging
 from typing import Any, Dict
 
+from django.conf import settings
 from django.db.models.query import QuerySet
 from django.views.generic import ListView, DetailView
 
+
+from signals.models import Geography
 from signal_sets.models import SignalSet
 from signal_sets.filters import SignalSetFilter
 from signal_sets.forms import SignalSetFilterForm
@@ -40,7 +43,9 @@ def get_url_params(self):
                 else None
             ),
             "data_source": [el for el in self.request.GET.getlist("data_source")],
-            "temporal_granularity": [el for el in self.request.GET.getlist("temporal_granularity")],
+            "temporal_granularity": [
+                el for el in self.request.GET.getlist("temporal_granularity")
+            ],
         }
         url_params_str = ""
         for param_name, param_value in url_params_dict.items():
@@ -72,4 +77,22 @@ class SignalSetDetailedView(DetailView):
 
     def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
         context: Dict[str, Any] = super().get_context_data(**kwargs)
+        context["epivis_url"] = settings.EPIVIS_URL
+        context["data_export_url"] = settings.DATA_EXPORT_URL
+        context["covidcast_url"] = settings.COVIDCAST_URL
+        context["data_source"] = (
+            self.object.signals.all()
+            .values_list("source__name", flat=True)
+            .distinct()
+            .order_by()
+            .first()
+        )
+        context["available_geographies"] = Geography.objects.filter(
+            id__in=self.object.signals.all()
+            .values_list("available_geography")
+            .distinct()
+            .order_by()
+        )
+        context["time_type"] = self.object.signals.all().values_list("time_type", flat=True).distinct().order_by().first()
+
         return context