diff --git a/wizard/acc_operation_import_wizard.py b/wizard/acc_operation_import_wizard.py index cc16417780abf55d88b215c4f185dcce8c89b4e7..2c356a0a2ddf2d0fbe164b88b634b179489d79f0 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