From 53739aee377f7017e797842e37aae33049ddd36a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?R=C3=A9mi=20-=20Le=20Filament?= <remi@le-filament.com>
Date: Thu, 20 Mar 2025 04:45:58 +0100
Subject: [PATCH] [FIX] call functions depending on source table Update
 partner_id and generation of by-day aggregated curves should only happen for
 acc.enedis.cdc (not for raw curves for instance)

---
 wizard/acc_operation_import_wizard.py | 26 ++++++++++++++------------
 1 file changed, 14 insertions(+), 12 deletions(-)

diff --git a/wizard/acc_operation_import_wizard.py b/wizard/acc_operation_import_wizard.py
index cc16417..2c356a0 100644
--- a/wizard/acc_operation_import_wizard.py
+++ b/wizard/acc_operation_import_wizard.py
@@ -238,18 +238,20 @@ class AccOperationImportWizard(models.TransientModel):
             _logger.debug(f"Processing file {file.name} - Curves creation")
             message += "Création des courbes<br/>"
             self.create_curve(curves_to_create)
-            _logger.debug(f"Processing file {file.name} - Update partner_id on curves")
-            message += "Mise à jour du contact associé aux courbes<br/>"
-            self._update_partner_id(data_filename, counter_id)
-
-            _logger.debug(f"Processing file {file.name} - Generating daily curves")
-            message += "Génération des données agrégées au jour<br/>"
-            self.env["acc.enedis.cdc.day"].generate(
-                acc_operation_id=self.operation_id.id,
-                acc_counter_id=counter_id,
-                start_date=start_date,
-                end_date=end_date,
-            )
+
+            if self.model == "acc.enedis.cdc":
+                _logger.debug(f"Processing file {file.name} - Update partner_id on curves")
+                message += "Mise à jour du contact associé aux courbes<br/>"
+                self._update_partner_id(data_filename, counter_id)
+
+                _logger.debug(f"Processing file {file.name} - Generating daily curves")
+                message += "Génération des données agrégées au jour<br/>"
+                self.env["acc.enedis.cdc.day"].generate(
+                    acc_operation_id=self.operation_id.id,
+                    acc_counter_id=counter_id,
+                    start_date=start_date,
+                    end_date=end_date,
+                )
 
             message += "Fin de l'import des données OK<br/>"
             # Suppression du fichier après création des enregistrements
-- 
GitLab