diff --git a/models/res_partner.py b/models/res_partner.py index 13d0939b5905257b3626041c19223ee3c2752c7e..0cd98e2e61651eb61976f33b871fe90823d866f3 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