feat(combined_emprise): add cache to field

MTES-MCT · Nov 5, 2024 · b8b22d7 · b8b22d7
1 parent 6c9085a
commit b8b22d7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/project/models/project_base.py b/project/models/project_base.py
@@ -77,12 +77,16 @@ class Status(models.TextChoices):
 
     @cached_property
     def combined_emprise(self) -> MultiPolygon:
-        """Return a combined MultiPolygon of all emprises."""
+        cache_key = f"project/{self.id}/combined_emprise"
+        if cache.has_key(cache_key):
+            return cache.get(cache_key)
         combined = self.emprise_set.aggregate(Union("mpoly"))
         if "mpoly__union" in combined:
-            return combined["mpoly__union"]
+            result = combined["mpoly__union"]
         else:
-            return MultiPolygon()
+            result = MultiPolygon()
+        cache.set(cache_key, result)
+        return result
 
     def __str__(self):
         return self.name