[MIG] contract_price_revision: Migration to 12.0

contract_price_revision/wizards/contract_price_revision.py

@@ -0,0 +1,54 @@
+# Copyright 2019 Tecnativa - Vicent Cubells
+# Copyright 2019 Tecnativa - Carlos Dauden
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html).
+
+from dateutil.relativedelta import relativedelta
+
+from odoo import fields, models
+import odoo.addons.decimal_precision as dp
+
+
+class ContractPriceRevisionWizard(models.TransientModel):
+    """ Update contract price based on percentage variation """
+    _name = 'contract.price.revision.wizard'
+    _description = "Wizard to update price based on percentage variation"
+
+    date_start = fields.Date(
+        required=True,
+    )
+    date_end = fields.Date()
+    variation_percent = fields.Float(
+        digits=dp.get_precision('Product Price'),
+        required=True,
+        string='Variation %',
+    )
+
+    def action_apply(self):
+        ContractLine = self.env['contract.line']
+        active_ids = self.env.context['active_ids']
+        for line in self.env['contract.contract'].browse(active_ids).mapped(
+                'contract_line_ids'):
+            if (line.automatic_price or line.successor_contract_line_id or
+                    not line.recurring_next_date):
+                continue
+            line.update({
+                'date_end': self.date_start - relativedelta(days=1),
+            })
+            new_vals = line.copy_data({
+                'date_start': self.date_start,
+                'date_end': self.date_end,
+                'predecessor_contract_line_id': line.id,
+                'price_unit': line.price_unit * (
+                    1.0 + self.variation_percent / 100.0),
+            })[0]
+            tmp_line = ContractLine.new(new_vals)
+            tmp_line._onchange_date_start()
+            new_line = ContractLine.create(
+                tmp_line._convert_to_write(tmp_line._cache))
+            line.update({
+                'successor_contract_line_id': new_line.id,
+            })
+        action = self.env['ir.actions.act_window'].for_xml_id(
+            'contract', 'action_customer_contract')
+        action['domain'] = [('id', 'in', active_ids)]
+        return action