Manage siae location

lemarche/siaes/management/commands/create_siae_activities.py

@@ -1,3 +1,4 @@
+from lemarche.perimeters.models import Perimeter
 from lemarche.siaes.models import Siae, SiaeActivity
 from lemarche.utils.commands import BaseCommand
 from lemarche.utils.data import reset_app_sql_sequences
@@ -6,7 +7,11 @@
 class Command(BaseCommand):
     """
     Script to generate SiaeActivities from Siae data
-    Will delete all existing SiaeActivities !!
+    - create 1 SiaeActivity per Siae sector_group
+    - match the location on the Siae address (post_code & city)
+    - copy the presta_type, geo_range, geo_range_custom_distance from the Siae
+
+    Note: Will delete all existing SiaeActivities !!
 
     Usage:
     poetry run python manage.py create_siae_activities --dry-run
@@ -41,8 +46,11 @@ def handle(self, *args, **options):
             # Step 3: create SiaeActivities
             self.stdout_info("-" * 80)
             self.stdout_info("Creating SiaeActivities")
-            for siae in siae_qs:
-                self.create_siae_activities(siae)
+            for index, siae in enumerate(siae_qs):
+                siae_location: Perimeter = self.match_siae_location(siae)
+                self.create_siae_activities(siae, location=siae_location)
+                if (index % 500) == 0:
+                    self.stdout_info(f"{index}...")
 
             # Recap
             self.stdout_info("-" * 80)
@@ -54,7 +62,40 @@ def handle(self, *args, **options):
             ]
             self.stdout_messages_success(msg_success)
 
-    def create_siae_activities(self, siae: Siae):
+    def match_siae_location(self, siae: Siae):
+        """
+        Find the Siae's location based on the post_code (and city)
+        """
+        if siae.post_code:
+            location_results_from_siae_post_code = Perimeter.objects.post_code_search(
+                siae.post_code, include_insee_code=True
+            )
+
+            if not location_results_from_siae_post_code.exists():
+                self.stdout_warning(f"No location found for {siae} (with post_code {siae.post_code})")
+                return None
+            elif location_results_from_siae_post_code.count() == 1:
+                return location_results_from_siae_post_code.first()
+            else:
+                # found multiple locations with the post_code, try to match with the city
+                if siae.city:
+                    location_results_from_siae_city = Perimeter.objects.name_search(siae.city)
+                    if location_results_from_siae_city.count():
+                        if (
+                            location_results_from_siae_post_code.first()
+                            == location_results_from_siae_post_code.first()
+                        ):
+                            return location_results_from_siae_post_code.first()
+                        else:
+                            self.stdout_warning(
+                                f"Multiple locations found for {siae} (with post_code {siae.post_code})"
+                            )
+                            return None
+
+        self.stdout_warning(f"No location found for {siae} (post_code empty)")
+        return None
+
+    def create_siae_activities(self, siae: Siae, siae_location: Perimeter = None):
         """
         - sector_group / sectors: we look at the existing siae sectors, and create an activity per sector group
         - presta_type: we look at the existing siae presta_types
@@ -64,20 +105,17 @@ def create_siae_activities(self, siae: Siae):
             self.stdout_warning(f"No sectors for {siae}")
             return
 
-        # Get the siae location
-        # TODO
-
         siae_sector_group_ids = list(set(siae.sectors.values_list("group", flat=True)))
         # For each SectorGroup, create a SiaeActivity
         for sector_group_id in siae_sector_group_ids:
             siae_activity = SiaeActivity.objects.create(
                 siae=siae,
                 sector_group_id=sector_group_id,
                 presta_type=siae.presta_type,
-                # location=siae.location,
+                location=siae_location,
                 geo_range=siae.geo_range,
                 geo_range_custom_distance=siae.geo_range_custom_distance,
             )
             siae_activity.sectors.set(siae.sectors.filter(group_id=sector_group_id))
 
-        self.stdout_info(f"Created {len(siae_sector_group_ids)} activities for {siae}")
+        # self.stdout_info(f"Created {len(siae_sector_group_ids)} activities for {siae}")