12.0 refactor cotiz

models/account_invoice.py

@@ -25,6 +25,14 @@ class ScopAuraAccountInvoice(models.Model):
         string="Montant calculé annuel", currency_field='company_currency_id',
         compute='_compute_amount_aura_calculated', store=True, readonly=True)
 
+    nb_quarter = fields.Selection(
+        string='Nombre de trimestres de cotisation',
+        selection=[(1, '1'),
+                   (2, '2'),
+                   (3, '3'),
+                   (4, '4')],
+        default=4,
+        required=True)
     amount_aura_prorata = fields.Monetary(
         string="Montant calculé proratisé",
         currency_field='company_currency_id',
@@ -81,23 +89,23 @@ class ScopAuraAccountInvoice(models.Model):
     # ------------------------------------------------------
     # Override parent
     # ------------------------------------------------------
-    @api.multi
-    def set_scop_contribution(self):
-        contrib_id = super(ScopAuraAccountInvoice, self).set_scop_contribution()
-        if self.cotisation_aura_id:
-            i = 4 - self.nb_quarter
-            schedule = {
-                'quarter_1': self.cotisation_aura_id.trimester_1,
-                'quarter_2': self.cotisation_aura_id.trimester_2,
-                'quarter_3': self.cotisation_aura_id.trimester_3,
-                'quarter_4': self.cotisation_aura_id.trimester_4,
-            }
-            while i != 0:
-                key = 'quarter_' + str(i)
-                schedule.pop(key, None)
-                i -= 1
-            contrib_id.update(schedule)
-        return contrib_id
+    # @api.multi
+    # def set_scop_contribution(self):
+    #     contrib_id = super(ScopAuraAccountInvoice, self).set_scop_contribution()
+    #     if self.cotisation_aura_id:
+    #         i = 4 - self.nb_quarter
+    #         schedule = {
+    #             'quarter_1': self.cotisation_aura_id.trimester_1,
+    #             'quarter_2': self.cotisation_aura_id.trimester_2,
+    #             'quarter_3': self.cotisation_aura_id.trimester_3,
+    #             'quarter_4': self.cotisation_aura_id.trimester_4,
+    #         }
+    #         while i != 0:
+    #             key = 'quarter_' + str(i)
+    #             schedule.pop(key, None)
+    #             i -= 1
+    #         contrib_id.update(schedule)
+    #     return contrib_id
 
     # ------------------------------------------------------
     # Global functions