Limiter les résultats de l'export pour l'utilisateur

Le but de ce commit est de limiter les fiches exportables par un
utilisateur/utilisatrice donné à ce qui est effectivement visible dans
l'outil. Ainsi la prise en compte des règles de visibilité empéche un
agent de voir dans l'export des fiches qu'il ne verrait pas dans
l'outil.

Au cours de mes tests je me suis rendu compte que l'export était
relativement lent même pour une base relativement petite (~800 lignes
dans le fichier CSV). J'ai optimisé et rendu les tests plus réalistes
avec l'utilisation de _fill_optional.

sv/export.py

@@ -47,9 +47,19 @@ class FicheDetectionExport:
     def _clean_field_name(self, field):
         return field.replace("_", " ").title()
 
-    def get_queryset(self):
-        queryset = FicheDetection.objects.select_related("etat", "numero")
-        queryset = queryset.prefetch_related("lieux", "lieux__prelevements", "lieux__prelevements__structure_preleveur")
+    def get_queryset(self, user):
+        queryset = FicheDetection.objects.all().get_fiches_user_can_view(user).optimized_for_details()
+        queryset = queryset.prefetch_related(
+            "lieux",
+            "lieux__prelevements",
+            "lieux__departement",
+            "lieux__prelevements__structure_preleveur",
+            "lieux__prelevements__espece_echantillon",
+            "lieux__prelevements__matrice_prelevee",
+            "lieux__prelevements__site_inspection",
+            "lieux__prelevements__laboratoire_agree",
+            "lieux__prelevements__laboratoire_confirmation_officielle",
+        )
         return queryset
 
     def get_fieldnames(self):
@@ -86,8 +96,8 @@ def get_lines_from_instance(self, fiche_detection):
         else:
             yield self.get_fiche_data(fiche_detection)
 
-    def export(self, stream):
-        queryset = self.get_queryset()
+    def export(self, stream, user):
+        queryset = self.get_queryset(user)
         writer = csv.DictWriter(stream, fieldnames=self.get_fieldnames())
         writer.writeheader()
 

sv/managers.py

@@ -55,5 +55,10 @@ def with_first_region_name(self):
     def optimized_for_list(self):
         return self.select_related("etat", "numero", "organisme_nuisible", "createur")
 
+    def optimized_for_details(self):
+        return self.select_related(
+            "statut_reglementaire", "etat", "numero", "contexte", "createur", "statut_evenement", "organisme_nuisible"
+        )
+
     def get_all_not_in_fiche_zone_delimitee(self):
         return self.filter(zone_infestee__isnull=True, hors_zone_infestee__isnull=True).order_by("numero")

sv/tests/test_export.py

@@ -2,12 +2,14 @@
 import pytest
 from model_bakery import baker
 from unittest import mock
+
+from core.models import Visibilite
 from sv.export import FicheDetectionExport
 from sv.models import Etat, FicheDetection, NumeroFiche, Lieu, Prelevement, StructurePreleveur
 import datetime
 
 
-def _create_fiche_with_lieu_and_prelevement(numero=123):
+def _create_fiche_with_lieu_and_prelevement(numero=123, fill_optional=False):
     etat = Etat.objects.get(id=Etat.get_etat_initial())
     numero = NumeroFiche.objects.create(annee=2024, numero=numero)
     mocked = datetime.datetime(2024, 8, 1, 0, 0, 0, tzinfo=datetime.timezone.utc)
@@ -24,6 +26,10 @@ def _create_fiche_with_lieu_and_prelevement(numero=123):
             mesures_consignation="MC",
             mesures_phytosanitaires="MP",
             mesures_surveillance_specifique="MSP",
+            visibilite=Visibilite.NATIONAL,
+            hors_zone_infestee=None,
+            zone_infestee=None,
+            _fill_optional=fill_optional,
         )
     lieu = baker.make(
         Lieu,
@@ -36,6 +42,7 @@ def _create_fiche_with_lieu_and_prelevement(numero=123):
         adresse_lieu_dit="L'angle",
         commune="Saint-Pierre",
         code_insee="12345",
+        _fill_optional=fill_optional,
     )
     structure = StructurePreleveur.objects.create(nom="My structure")
     baker.make(
@@ -47,14 +54,15 @@ def _create_fiche_with_lieu_and_prelevement(numero=123):
         numero_phytopass="Phyto123",
         resultat="detecte",
         structure_preleveur=structure,
+        _fill_optional=fill_optional,
     )
 
 
 @pytest.mark.django_db
-def test_export_fiche_detection_content():
+def test_export_fiche_detection_content(mocked_authentification_user):
     stream = StringIO()
     _create_fiche_with_lieu_and_prelevement()
-    FicheDetectionExport().export(stream=stream)
+    FicheDetectionExport().export(stream=stream, user=mocked_authentification_user)
 
     stream.seek(0)
     lines = stream.readlines()
@@ -104,37 +112,37 @@ def test_export_fiche_detection_content():
 
 
 @pytest.mark.django_db
-def test_export_fiche_detection_performance(django_assert_num_queries):
+def test_export_fiche_detection_performance(django_assert_num_queries, mocked_authentification_user):
     stream = StringIO()
-    _create_fiche_with_lieu_and_prelevement()
+    _create_fiche_with_lieu_and_prelevement(fill_optional=True)
 
-    with django_assert_num_queries(4):
-        FicheDetectionExport().export(stream=stream)
+    with django_assert_num_queries(10):
+        FicheDetectionExport().export(stream=stream, user=mocked_authentification_user)
 
     stream.seek(0)
     lines = stream.readlines()
     assert len(lines) == 2
 
     stream = StringIO()
-    _create_fiche_with_lieu_and_prelevement(numero=4)
-    _create_fiche_with_lieu_and_prelevement(numero=5)
-    _create_fiche_with_lieu_and_prelevement(numero=6)
-    with django_assert_num_queries(4):
-        FicheDetectionExport().export(stream=stream)
+    _create_fiche_with_lieu_and_prelevement(numero=4, fill_optional=True)
+    _create_fiche_with_lieu_and_prelevement(numero=5, fill_optional=True)
+    _create_fiche_with_lieu_and_prelevement(numero=6, fill_optional=True)
+    with django_assert_num_queries(10):
+        FicheDetectionExport().export(stream=stream, user=mocked_authentification_user)
 
     stream.seek(0)
     lines = stream.readlines()
     assert len(lines) == 5
 
 
 @pytest.mark.django_db
-def test_export_fiche_detection_numbers_of_lines(django_assert_num_queries):
+def test_export_fiche_detection_numbers_of_lines(django_assert_num_queries, mocked_authentification_user):
     stream = StringIO()
     etat = Etat.objects.get(id=Etat.get_etat_initial())
     numero = NumeroFiche.objects.create(annee=2024, numero=123)
     mocked = datetime.datetime(2024, 8, 1, 0, 0, 0, tzinfo=datetime.timezone.utc)
     with mock.patch("django.utils.timezone.now", mock.Mock(return_value=mocked)):
-        fiche = baker.make(FicheDetection, etat=etat, numero=numero)
+        fiche = baker.make(FicheDetection, etat=etat, numero=numero, visibilite=Visibilite.NATIONAL)
     _lieu_without_prelevement = baker.make(Lieu, fiche_detection=fiche)
     _lieu_without_prelevement_2 = baker.make(Lieu, fiche_detection=fiche)
     lieu = baker.make(
@@ -152,7 +160,7 @@ def test_export_fiche_detection_numbers_of_lines(django_assert_num_queries):
     baker.make(Prelevement, lieu=lieu)
     baker.make(Prelevement, lieu=lieu)
 
-    FicheDetectionExport().export(stream=stream)
+    FicheDetectionExport().export(stream=stream, user=mocked_authentification_user)
 
     stream.seek(0)
     lines = stream.readlines()

sv/views.py

@@ -89,9 +89,7 @@ class FicheDetectionDetailView(
     DetailView,
 ):
     model = FicheDetection
-    queryset = FicheDetection.objects.select_related(
-        "statut_reglementaire", "etat", "numero", "contexte", "createur", "statut_evenement", "organisme_nuisible"
-    )
+    queryset = FicheDetection.objects.all().optimized_for_details()
 
     def get_object(self, queryset=None):
         if hasattr(self, "object"):
@@ -580,7 +578,7 @@ class FicheDetectionExportView(View):
 
     def post(self, request):
         response = HttpResponse(content_type="text/csv")
-        FicheDetectionExport().export(stream=response)
+        FicheDetectionExport().export(stream=response, user=request.user)
         response["Content-Disposition"] = "attachment; filename=export_fiche_detection.csv"
         return response