From d75e85c17c067772cb5cc92e48404a4b9f0fd278 Mon Sep 17 00:00:00 2001
From: Juliana <juliana@le-filament.com>
Date: Mon, 31 Jul 2023 17:51:49 +0200
Subject: [PATCH] [ADD]Add template type to check show data

---
 models/acc_operation.py | 20 +++++++++++++-------
 1 file changed, 13 insertions(+), 7 deletions(-)

diff --git a/models/acc_operation.py b/models/acc_operation.py
index 6684bd1..99da47f 100644
--- a/models/acc_operation.py
+++ b/models/acc_operation.py
@@ -1086,6 +1086,7 @@ class AccOperation(models.Model):
         date_end=None,
         partner_id=None,
         prm_id=None,
+        data_type=None
     ):
         """
         Fonction appelée pour l'affichage des courbes consommation
@@ -1111,15 +1112,20 @@ class AccOperation(models.Model):
             date_start=date_start, date_end=date_end, scale=scale
         )
 
-        # Get PRM ids
-        if prm_id:
-            acc_counter_ids = self.env["acc.counter"].browse(prm_id)
-        elif partner_id:
-            acc_counter_ids = self.env["acc.counter"].search(
-                [("partner_id", "=", partner_id)]
+        if data_type == "pmo":
+            acc_counter_ids = self.env["acc.counter"].sudo().search(
+                [("acc_operation_id", "=", self.id)]
             )
         else:
-            acc_counter_ids = self.acc_delivery_ids + self.acc_injection_ids
+            # Get PRM ids
+            if prm_id:
+                acc_counter_ids = self.env["acc.counter"].browse(prm_id)
+            elif partner_id:
+                acc_counter_ids = self.env["acc.counter"].search(
+                    [("partner_id", "=", partner_id)]
+                )
+            else:
+                acc_counter_ids = self.acc_delivery_ids + self.acc_injection_ids
 
         if scale == "week" or scale == "day":
             is_curve_line = True
-- 
GitLab