1 files changed, 117 insertions, 0 deletions
diff --git a/addons/membership/models/account_move.py b/addons/membership/models/account_move.py
new file mode 100644
index 00000000000..62b7526eae5
--- /dev/null
+++ b/addons/membership/models/account_move.py
@@ -0,0 +1,117 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models
+
+from datetime import date
+
+
+class AccountMove(models.Model):
+    _inherit = 'account.move'
+
+    @api.multi
+    def button_draft(self):
+        # OVERRIDE to update the cancel date.
+        res = super(AccountMove, self).button_draft()
+        if self.type == 'out_invoice':
+            self.env['membership.membership_line'].search([
+                ('account_invoice_line', 'in', self.mapped('invoice_line_ids').ids)
+            ]).write({'date_cancel': False})
+        return res
+
+    @api.multi
+    def button_cancel(self):
+        # OVERRIDE to update the cancel date.
+        res = super(AccountMove, self).button_cancel()
+        if self.type == 'out_invoice':
+            self.env['membership.membership_line'].search([
+                ('account_invoice_line', 'in', self.mapped('invoice_line_ids').ids)
+            ]).write({'date_cancel': fields.Date.today()})
+        return res
+
+    @api.multi
+    def write(self, vals):
+        # OVERRIDE to write the partner on the membership lines.
+        res = super(AccountMove, self).write(vals)
+        if 'partner_id' in vals:
+            self.env['membership.membership_line'].search([
+                ('account_invoice_line', 'in', self.mapped('invoice_line_ids').ids)
+            ]).write({'partner': vals['partner_id']})
+        return res
+
+
+class AccountMoveLine(models.Model):
+    _inherit = 'account.move.line'
+
+    @api.multi
+    def write(self, vals):
+        # OVERRIDE
+        res = super(AccountMoveLine, self).write(vals)
+
+        to_process = self.filtered(lambda line: line.move_id.type == 'out_invoice' and line.product_id.membership)
+
+        # Nothing to process, break.
+        if not to_process:
+            return res
+
+        existing_memberships = self.env['membership.membership_line'].search([
+            ('account_invoice_line', 'in', to_process.ids)])
+        to_process = to_process - existing_memberships.mapped('account_invoice_line')
+
+        # All memberships already exist, break.
+        if not to_process:
+            return res
+
+        memberships_vals = []
+        for line in to_process:
+            date_from = line.product_id.membership_date_from
+            date_to = line.product_id.membership_date_to
+            if (date_from and date_from < (line.move_id.invoice_date or date.min) < (date_to or date.min)):
+                date_from = line.move_id.invoice_date
+            memberships_vals.append({
+                'partner': line.move_id.partner_id.id,
+                'membership_id': line.product_id.id,
+                'member_price': line.price_unit,
+                'date': fields.Date.today(),
+                'date_from': date_from,
+                'date_to': date_to,
+                'account_invoice_line': line.id,
+            })
+        self.env['membership.membership_line'].create(memberships_vals)
+        return res
+
+    @api.model_create_multi
+    def create(self, vals_list):
+        # OVERRIDE
+        lines = super(AccountMoveLine, self).create(vals_list)
+        to_process = lines.filtered(lambda line: line.move_id.type == 'out_invoice' and line.product_id.membership)
+
+        # Nothing to process, break.
+        if not to_process:
+            return lines
+
+        existing_memberships = self.env['membership.membership_line'].search([
+            ('account_invoice_line', 'in', to_process.ids)])
+        to_process = to_process - existing_memberships.mapped('account_invoice_line')
+
+        # All memberships already exist, break.
+        if not to_process:
+            return lines
+
+        memberships_vals = []
+        for line in to_process:
+            date_from = line.product_id.membership_date_from
+            date_to = line.product_id.membership_date_to
+            if (date_from and date_from < (line.move_id.invoice_date or date.min) < (date_to or date.min)):
+                date_from = line.move_id.invoice_date
+            memberships_vals.append({
+                'partner': line.move_id.partner_id.id,
+                'membership_id': line.product_id.id,
+                'member_price': line.price_unit,
+                'date': fields.Date.today(),
+                'date_from': date_from,
+                'date_to': date_to,
+                'account_invoice_line': line.id,
+            })
+        self.env['membership.membership_line'].create(memberships_vals)
+        return lines