From 62040d7136dfc4270c8ba94fc90854250a92c688 Mon Sep 17 00:00:00 2001
From: benjamin <benjamin@le-filament.com>
Date: Wed, 8 Jun 2022 12:29:47 +0200
Subject: [PATCH] [update] move get_partner_contribution_type from
 cgscop_partner

---
 models/res_partner.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/models/res_partner.py b/models/res_partner.py
index 13d0939..0cd98e2 100644
--- a/models/res_partner.py
+++ b/models/res_partner.py
@@ -12,3 +12,19 @@ class ResPartner(models.Model):
         inverse_name="partner_id",
         string="Cotisations",
     )
+
+    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
-- 
GitLab