From b911382fbd6e73a98944eb61a81853c416125434 Mon Sep 17 00:00:00 2001
From: Julien Ortet <julien@le-filament.com>
Date: Wed, 14 Aug 2024 16:58:04 +0200
Subject: [PATCH] [ADD] create only prod or cons data

---
 wizard/acc_operation_import_wizard.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/wizard/acc_operation_import_wizard.py b/wizard/acc_operation_import_wizard.py
index b47a369..0958bb8 100644
--- a/wizard/acc_operation_import_wizard.py
+++ b/wizard/acc_operation_import_wizard.py
@@ -77,4 +77,6 @@ class AccOperationImportWizard(models.TransientModel):
         return message
 
     def create_curve(self, curve_data):
-        self.env["acc.enedis.raw.cdc"].create(curve_data)
+
+        if curve_data.get("comp_data_type") in ["prod", "cons"]:
+            self.env["acc.enedis.raw.cdc"].create(curve_data)
-- 
GitLab