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