12.0 refactor cotiz

models/account_invoice.py

@@ -25,14 +25,10 @@ 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(
+    nb_quarter_aura = fields.Selection(
         string='Nombre de trimestres de cotisation',
-        selection=[(1, '1'),
-                   (2, '2'),
-                   (3, '3'),
-                   (4, '4')],
-        default=4,
-        required=True)
+        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',
@@ -58,15 +54,16 @@ class ScopAuraAccountInvoice(models.Model):
                     'scop.cotisation.aura'].get_cotiz_aura(
                     partner_id=i.partner_id, wage_cg=i.wage_cg_retenu)
 
-    @api.depends('amount_aura_calculated', 'nb_quarter')
+    @api.depends('amount_aura_calculated', 'nb_quarter_aura')
     @api.multi
     def _compute_amount_aura_prorata(self):
         for i in self:
-            if i.cotisation_aura_id:
-                prorata = int(i.nb_quarter) / 4
+            if i.cotisation_aura_id \
+                    and i.amount_aura_calculated and i.nb_quarter_aura:
+                prorata = i.nb_quarter_aura / 4
                 i.amount_aura_prorata = i.cotisation_aura_id.\
                     round_to_closest_multiple(
-                        i.amount_aura_calculated * prorata, int(i.nb_quarter))
+                        i.amount_aura_calculated * prorata, i.nb_quarter_aura)
 
     # ------------------------------------------------------
     # Onchange