From 464cde563d3aa1c75e3788e0f92f35bf994a1e3b Mon Sep 17 00:00:00 2001
From: Raphael Odini <raphodn@users.noreply.github.com>
Date: Mon, 11 Mar 2024 11:54:26 +0100
Subject: [PATCH] =?UTF-8?q?feat(SIAE):=20script=20pour=20aider=20le=20ratt?=
 =?UTF-8?q?achement=20=C3=A0=20leur=20r=C3=A9seau=20(gr=C3=A2ce=20=C3=A0?=
 =?UTF-8?q?=20des=20import=20csv)=20(#1109)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../commands/link_siae_to_network.py          | 66 +++++++++++++++++++
 1 file changed, 66 insertions(+)
 create mode 100644 lemarche/networks/management/commands/link_siae_to_network.py

diff --git a/lemarche/networks/management/commands/link_siae_to_network.py b/lemarche/networks/management/commands/link_siae_to_network.py
new file mode 100644
index 000000000..0045c310e
--- /dev/null
+++ b/lemarche/networks/management/commands/link_siae_to_network.py
@@ -0,0 +1,66 @@
+import csv
+
+from lemarche.networks.models import Network
+from lemarche.siaes.models import Siae
+from lemarche.utils.commands import BaseCommand
+
+
+class Command(BaseCommand):
+    """
+    poetry run python manage.py link_siae_to_network --file file.csv --network 8 --siret-column 'SIRET' --dry-run
+    poetry run python manage.py link_siae_to_network --file file.csv --network 8 --siret-column 'SIRET'
+    """
+
+    def add_arguments(self, parser):
+        parser.add_argument("--file", type=str, help="Path to the CSV file")
+        parser.add_argument("--network", type=int, help="Network ID")
+        parser.add_argument("--siret-column", dest="siret_column", help="Name of the SIRET column")
+        parser.add_argument("--dry-run", dest="dry_run", action="store_true", help="Dry run (no changes to the DB)")
+
+    def handle(self, *args, **options):
+        siae_list = list()
+        network = Network.objects.get(id=options["network"])
+
+        self.stdout_info("-" * 80)
+        self.stdout_info("Reading file...")
+
+        with open(options["file"]) as csv_file:
+            csvreader = csv.DictReader(csv_file, delimiter=",")
+            for index, row in enumerate(csvreader):
+                row["index"] = index + 2
+                siae_list.append(row)
+
+        self.stdout_info(f"Found {len(siae_list)} siae.")
+
+        self.stdout_info("-" * 80)
+        self.stdout_info(f"Network to attach: {network.name}")
+        self.stdout_info(f"Siae already linked: {network.siaes.count()}")
+
+        self.stdout_info("-" * 80)
+        self.stdout_info("Running for loop...")
+
+        siae_missing = 0
+        siae_has_network = 0
+        siae_network_added = 0
+
+        for index, s in enumerate(siae_list):
+            siae_qs = Siae.objects.filter(siret=s[options["siret_column"]])
+            if siae_qs.count():
+                for siae in siae_qs:
+                    if network in siae.networks.all():
+                        # Siae Network already linked
+                        siae_has_network += 1
+                    else:
+                        # Siae Network new link
+                        siae_network_added += 1
+                        if not options["dry_run"]:
+                            siae.networks.add(network)
+            else:
+                # Siae missing
+                siae_missing += 1
+
+        self.stdout_info("-" * 80)
+        self.stdout_info("Recap")
+        self.stdout_info(f"Siae not found: {siae_missing}")
+        self.stdout_info(f"Siae already linked: {siae_has_network}")
+        self.stdout_info(f"Siae new links: {siae_network_added}")