enh: endpoint for raster DL (#538)

Author: ludovicdmt

back/api/urls.py

@@ -25,7 +25,7 @@
         TileView.as_view(),
         name="retrieve-tile",
     ),
-    # L'URL spécifique doit venir AVANT l'URL générale avec <id>
+    # Specific URL should be BEFORE general URL with <id>
     path(
         "tiles/<datatype>/in-polygon/",
         ScoresInPolygonView.as_view(),
@@ -44,9 +44,9 @@
     path("feedback/", FeedbackView.as_view(), name="create-feedback"),
     path("qpv/", QPVListView.as_view(), name="qpv-list"),
     path(
-        "rasters/plantability/",
+        "rasters/<str:raster_type>/",
         RasterDownloadView.as_view(),
-        name="download-plantability-raster",
+        name="download-raster",
     ),
     path("dashboard/", DashboardView.as_view(), name="dashboard"),
     path("", include(router.urls)),

back/api/views/raster_views.py

@@ -3,31 +3,40 @@
 from django.conf import settings
 from rest_framework.views import APIView
 
+RASTER_MAP = {
+    "plantability": ("rasters/plantability.tif", "plantability_2025.tif"),
+    "vegestrate": (
+        "rasters/vegestrate_lyonmetro_02m_postprocessed.tif",
+        "vegestrate_2023_02m.tif",
+    ),
+}
+
 
 class RasterDownloadView(APIView):
-    """
-    API endpoint to download the plantability raster file.
-    Provides a stable URL for accessing the raster regardless of internal file structure.
+    """API endpoint to download the plantability and vegestrate raster files.
 
-    Example: GET /api/rasters/plantability/
+    Example: GET /api/rasters/plantability/ or GET /api/rasters/vegestrate/
     """
 
-    def get(self, request):
-        """Serve the plantability raster file as a download."""
-        raster_path = os.path.join(settings.MEDIA_ROOT, "rasters", "plantability.tif")
+    def get(self, request, raster_type):
+        if raster_type not in RASTER_MAP:
+            raise Http404(
+                "Raster does not exist, only plantability or vegestrate are available."
+            )
 
-        if not os.path.exists(raster_path):
-            raise Http404("Plantability raster file not found")
+        relative_path, filename = RASTER_MAP[raster_type]
+        raster_path = os.path.join(settings.MEDIA_ROOT, relative_path)
 
-        file_handle = open(raster_path, "rb")
+        if not os.path.exists(raster_path):
+            raise Http404(
+                "Raster file not found. Please send an email to contact@telescoop.fr"
+            )
 
         response = FileResponse(
-            file_handle,
+            open(raster_path, "rb"),
             content_type="image/tiff",
             as_attachment=True,
-            filename="plantability.tif",
+            filename=filename,
         )
-
         response["Cache-Control"] = "public, max-age=3600"
-
         return response

front/cypress/components/SidebarDownload.cy.ts

@@ -1,6 +1,6 @@
 import MapSidePanelDownload from "@/components/map/panels/sidepanel/MapSidePanelDownload.vue"
 import { useMapStore } from "@/stores/map"
-import { DataType } from "@/utils/enum"
+import { DataType, DataTypeToDownloadLink } from "@/utils/enum"
 
 describe("MapSidePanelDownload", () => {
   beforeEach(() => {
@@ -28,7 +28,7 @@ describe("MapSidePanelDownload", () => {
 
     cy.get("@windowOpen").should(
       "have.been.calledWith",
-      "https://data.grandlyon.com/portail/en/jeux-de-donnees/calque-plantabilite-metropole-lyon/info",
+      DataTypeToDownloadLink[DataType.PLANTABILITY],
       "_blank"
     )
   })
@@ -47,7 +47,7 @@ describe("MapSidePanelDownload", () => {
 
     cy.get("@windowOpen").should(
       "have.been.calledWith",
-      "https://www.data.gouv.fr/datasets/cartographie-des-zones-climatiques-locales-lcz-des-88-aires-urbaines-de-plus-de-50-000-habitants-de-france-metropolitaine/#/resources/e0c0f5e4-c8bb-4d33-aec9-ba16b5736102",
+      DataTypeToDownloadLink[DataType.CLIMATE_ZONE],
       "_blank"
     )
   })
@@ -66,7 +66,45 @@ describe("MapSidePanelDownload", () => {
 
     cy.get("@windowOpen").should(
       "have.been.calledWith",
-      "https://data.grandlyon.com/portail/en/jeux-de-donnees/exposition-et-vulnerabilite-aux-fortes-chaleurs-dans-la-metropole-de-lyon/info",
+      DataTypeToDownloadLink[DataType.VULNERABILITY],
+      "_blank"
+    )
+  })
+
+  it("should open correct download link for PLANTABILITY_VULNERABILITY data type", () => {
+    cy.window().then(() => {
+      const store = useMapStore()
+      store.selectedDataType = DataType.PLANTABILITY_VULNERABILITY
+    })
+
+    cy.window().then((win) => {
+      cy.stub(win, "open").as("windowOpen")
+    })
+
+    cy.get('[data-cy="download-data"]').click()
+
+    cy.get("@windowOpen").should(
+      "have.been.calledWith",
+      DataTypeToDownloadLink[DataType.PLANTABILITY_VULNERABILITY],
+      "_blank"
+    )
+  })
+
+  it("should open correct download link for VEGETATION data type", () => {
+    cy.window().then(() => {
+      const store = useMapStore()
+      store.selectedDataType = DataType.VEGETATION
+    })
+
+    cy.window().then((win) => {
+      cy.stub(win, "open").as("windowOpen")
+    })
+
+    cy.get('[data-cy="download-data"]').click()
+
+    cy.get("@windowOpen").should(
+      "have.been.calledWith",
+      DataTypeToDownloadLink[DataType.VEGETATION],
       "_blank"
     )
   })

front/src/utils/enum.ts

@@ -98,13 +98,12 @@ export const VulnerabilityCategoryOrder = [
 ]
 
 export const DataTypeToDownloadLink: Record<DataType, string> = {
-  [DataType.PLANTABILITY]:
-    "https://data.grandlyon.com/portail/en/jeux-de-donnees/calque-plantabilite-metropole-lyon/info",
+  [DataType.PLANTABILITY]: "https://carte.iarbre.fr/api/rasters/plantability/",
   [DataType.CLIMATE_ZONE]:
     "https://www.data.gouv.fr/datasets/cartographie-des-zones-climatiques-locales-lcz-des-88-aires-urbaines-de-plus-de-50-000-habitants-de-france-metropolitaine/#/resources/e0c0f5e4-c8bb-4d33-aec9-ba16b5736102",
   [DataType.VULNERABILITY]:
     "https://data.grandlyon.com/portail/en/jeux-de-donnees/exposition-et-vulnerabilite-aux-fortes-chaleurs-dans-la-metropole-de-lyon/info",
   [DataType.PLANTABILITY_VULNERABILITY]:
     "https://data.grandlyon.com/portail/en/jeux-de-donnees/calque-plantabilite-metropole-lyon/info",
-  [DataType.VEGETATION]: "https://github.com/IGNF/FLAIR-HUB"
+  [DataType.VEGETATION]: "https://carte.iarbre.fr/api/rasters/vegestrate/"
 }