From 5fc57b88afee91237fa8c56ba2678095c8b18a72 Mon Sep 17 00:00:00 2001
From: benjamin <benjamin@le-filament.com>
Date: Wed, 8 Dec 2021 16:52:09 +0100
Subject: [PATCH] [add] function to get contribution type

---
 models/res_partner.py | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/models/res_partner.py b/models/res_partner.py
index 3125436..7beb53b 100644
--- a/models/res_partner.py
+++ b/models/res_partner.py
@@ -606,6 +606,19 @@ class ScopPartner(models.Model):
         })
         partner.scop_period_ids = new_period
 
+    def get_partner_contribution_type(self):
+        """
+        Returns list of contribution type for partner
+        """
+        contribution_type = [self.env.ref("cgscop_partner.riga_14397")]
+        if self.ur_id in [self.env.ref("cgscop_partner.riga_14232"), self.env.ref("cgscop_partner.riga_14243")]:
+            contribution_type.append(self.env.ref("cgscop_partner.riga_14399"))
+        if self.is_federation_com:
+            contribution_type.append(self.env.ref("cgscop_partner.riga_14398"))
+        if self.is_federation_cae:
+            contribution_type.append(self.env.ref("cgscop_partner.cotiz_fede_cae"))
+        return contribution_type
+
     # ------------------------------------------------------
     # Override ORM
     # ------------------------------------------------------
@@ -1068,7 +1081,6 @@ class ScopPartner(models.Model):
             'target': 'new',
         }
 
-
     # ------------------------------------------------------
     # Affichage des coop avec filtre par défaut
     # ------------------------------------------------------
@@ -1192,7 +1204,6 @@ class ScopPartner(models.Model):
         }
 
 
-
 class ResPartneCertification(models.Model):
     _name = "res.partner.certification"
     _description = "Certification"
-- 
GitLab