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