display prest type of activity instead

gip-inclusion · Nov 20, 2024 · 3d2bc61 · 3d2bc61
1 parent 9255755
commit 3d2bc61
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lemarche/siaes/models.py b/lemarche/siaes/models.py
@@ -1008,8 +1008,12 @@ def presta_type_display(self) -> str:
             return "Intérim"
         if self.kind == siae_constants.KIND_AI:
             return "Mise à disposition du personnel"
-        if self.presta_type:
-            return choice_array_to_values(siae_constants.PRESTA_CHOICES, self.presta_type)
+        if self.activities.exists():
+            presta_types = set()
+            for activity in self.activities.all():
+                if activity.presta_type:
+                    presta_types.update(activity.presta_type)
+            return choice_array_to_values(siae_constants.PRESTA_CHOICES, list(presta_types))
         return ""
 
     @property