From 4a7516589bdc5a79eeedaff2a450d997b3af5057 Mon Sep 17 00:00:00 2001
From: Juliana <juliana@le-filament.com>
Date: Wed, 2 Jun 2021 14:44:16 +0200
Subject: [PATCH] [ADD] Add warninf message date + fill date start on operation
 from API

---
 models/api_enedis.py                 | 2 ++
 wizards/enercoop_operation_wizard.py | 4 ++++
 2 files changed, 6 insertions(+)

diff --git a/models/api_enedis.py b/models/api_enedis.py
index 9715e23..4028851 100644
--- a/models/api_enedis.py
+++ b/models/api_enedis.py
@@ -267,6 +267,8 @@ class ApiEnedis(models.AbstractModel):
                         is_delivery = True
                     if usage_point['type'] == 'PROD':
                         is_injection = True
+                        self.date_start_contract = date.fromisoformat(usage_point['start'])
+                        self.date_end_contract = date.fromisoformat(usage_point['end'])
 
                     prm_id = self.env['enercoop.counter'].create({
                         'name': usage_point['usage_point_id'],
diff --git a/wizards/enercoop_operation_wizard.py b/wizards/enercoop_operation_wizard.py
index 908f6f0..3e31665 100644
--- a/wizards/enercoop_operation_wizard.py
+++ b/wizards/enercoop_operation_wizard.py
@@ -2,6 +2,7 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
 
 from odoo import fields, models
+from odoo.exceptions import UserError
 
 
 class EnercoopOperationWizard(models.TransientModel):
@@ -38,6 +39,9 @@ class EnercoopOperationWizard(models.TransientModel):
     # Actions
     # ------------------------------------------------------
     def get_curves(self):
+        if (self.date_end - self.date_start).days > 31:
+            raise UserError(
+                "L'intervalle de temps ne doit pas dépasser 31 Jours")
         context = dict(self._context or {})
         if context.get('active_ids', False):
             self.env['enercoop.operation'].browse(context.get('active_ids')).get_curves(self.date_start, self.date_end)
-- 
GitLab