[13.0][MIG] account_spread_cost_revenue

account_spread_cost_revenue/models/account_move.py

@@ -0,0 +1,34 @@
+# Copyright 2016-2020 Onestein (<https://www.onestein.eu>)
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).
+
+from odoo import api, models
+
+
+class AccountMove(models.Model):
+    _inherit = "account.move"
+
+    def action_post(self):
+        """Invoked when validating the invoices."""
+        self.mapped("invoice_line_ids").create_auto_spread()
+        res = super().action_post()
+        spreads = self.mapped("invoice_line_ids.spread_id")
+        spreads.compute_spread_board()
+        spreads.reconcile_spread_moves()
+        return res
+
+    def button_cancel(self):
+        """Cancel the spread lines and their related moves when
+        the invoice is canceled."""
+        spread_lines = self.mapped("invoice_line_ids.spread_id.line_ids")
+        moves = spread_lines.mapped("move_id")
+        moves.line_ids.remove_move_reconcile()
+        moves.filtered(lambda move: move.state == "posted").button_draft()
+        moves.with_context(force_delete=True).unlink()
+        spread_lines.unlink()
+        res = super().button_cancel()
+        return res
+
+    @api.constrains("name", "journal_id", "state")
+    def _check_unique_sequence_number(self):
+        if not self.env.context.get("skip_unique_sequence_number"):
+            return super()._check_unique_sequence_number()