[13.0][MIG] account_document_reversal

account_document_reversal/models/account_move.py

@@ -1,26 +1,79 @@
 # Copyright 2019 Eficent Business and IT Consulting Services, S.L.
 # License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl.html)
-from odoo import _, api, fields, models
-from odoo.exceptions import ValidationError
+from odoo import _, fields, models
+from odoo.exceptions import UserError
 
 
 class AccountMove(models.Model):
     _name = "account.move"
     _inherit = ["account.move", "account.document.reversal"]
 
-    is_cancel_reversal = fields.Boolean(
-        string="Use Cancel Reversal", related="journal_id.is_cancel_reversal",
+    cancel_reversal = fields.Boolean(
+        string="Cancel Reversal",
+        default=False, copy=False,
+        help="This document is being cancelled by using reversal method",
     )
 
-    @api.multi
-    def button_cancel(self):
-        """ Do not allow using this button for cancel with reversal """
-        cancel_reversal = any(self.mapped("is_cancel_reversal"))
-        if cancel_reversal:
-            raise ValidationError(
-                _(
-                    "This action is not allowed for cancel with reversal.\n"
-                    "Please use Reverse Entry."
-                )
-            )
-        return super().button_cancel()
+    def button_cancel_reversal(self):
+        return self.reverse_document_wizard()
+
+    def button_draft(self):
+        for rec in self:
+            if rec.is_cancel_reversal and rec.state != "cancel":
+                raise UserError(_("Cannot set to draft!"))
+        return super().button_draft()
+
+    def action_document_reversal(self, date=None, journal_id=None):
+        # Check document readiness
+        valid_state = len(self.mapped("state")) == 1 and \
+            list(set(self.mapped("state")))[0] == "posted"
+        if not valid_state:
+            raise UserError(
+                _("Only posted document can be cancelled (reversal)"))
+        if self.mapped("line_ids.matched_debit_ids") | self.mapped("line_ids.matched_credit_ids"):
+            raise UserError(
+                _("Only fully unpaid invoice can be cancelled.\n"
+                  "To cancel this invoice, make sure all payment(s) "
+                  "are also cancelled."))
+        # Create reverse entries
+        self._cancel_reversal(journal_id)
+        return True
+
+    def _cancel_reversal(self, journal_id):
+        self.mapped("line_ids").filtered(
+            lambda x: x.account_id.reconcile).remove_move_reconcile()
+        Reversal = self.env["account.move.reversal"]
+        ctx = {"active_ids": self.ids, "active_model": "account.move"}
+        res = Reversal.with_context(ctx).default_get([])
+        res.update({"journal_id": journal_id,
+                    "refund_method": "cancel",
+                    "move_type": "entry", })
+        reversal = Reversal.create(res)
+        reversal.with_context(cancel_reversal=True).reverse_moves()
+
+    def _reverse_moves(self, default_values_list=None, cancel=False):
+        """ Set flag on the moves and the reversal moves being reversed """
+        if self._context.get("cancel_reversal"):
+            self.write({"cancel_reversal": True})
+        reverse_moves = super()._reverse_moves(default_values_list, cancel)
+        if self._context.get("cancel_reversal"):
+            reverse_moves.write({"cancel_reversal": True})
+        return reverse_moves
+
+    def _reverse_move_vals(self, default_values, cancel=True):
+        """ Reverse with cancel reversal, always use move_type = entry """
+        if self._context.get("cancel_reversal"):
+            default_values.update({"type": "entry"})
+        return super()._reverse_move_vals(default_values, cancel)
+
+
+class AccountMoveLine(models.Model):
+    _inherit = "account.move.line"
+
+    def remove_move_reconcile(self):
+        """ For move with cancel_reversal = True, freeze it """
+        if not self._context.get("cancel_reversal") and \
+                any(self.mapped("move_id").mapped("cancel_reversal")):
+            raise UserError(_("This document was cancelled and freezed,\n"
+                              "unreconcilation not allowed."))
+        return super().remove_move_reconcile()