Merge PR #479 into 14.0

Signed-off-by grindtildeath

account_mass_reconcile/models/base_advanced_reconciliation.py

@@ -5,7 +5,7 @@
 import logging
 from itertools import product
 
-from odoo import models
+from odoo import api, models, registry
 from odoo.tools.translate import _
 
 _logger = logging.getLogger(__name__)
@@ -219,10 +219,66 @@ class MassReconcileAdvanced(models.AbstractModel):
         """
         return False
 
+    def _rec_group(self, reconcile_groups, lines_by_id):
+        reconciled_ids = []
+        for group_count, reconcile_group_ids in enumerate(reconcile_groups, start=1):
+
+            _logger.debug(
+                "Reconciling group %d/%d with ids %s",
+                group_count,
+                len(reconcile_groups),
+                reconcile_group_ids,
+            )
+            group_lines = [lines_by_id[lid] for lid in reconcile_group_ids]
+            reconciled, full = self._reconcile_lines(group_lines, allow_partial=True)
+            if reconciled and full:
+                reconciled_ids += reconcile_group_ids
+        return reconciled_ids
+
+    def _rec_group_by_chunk(self, reconcile_groups, lines_by_id, chunk_size):
+        """Commit after each chunk
+
+        :param dict reconcile_grous: all groups to reconcile, will be splitted
+        by chunk
+        :param list lines_by_id: list of dict of move lines values,
+          the move lines we want to search for
+        :return: list of reconciled lines
+        """
+        reconciled_ids = []
+
+        _logger.info("Reconciling by chunk of %d", chunk_size)
+
+        # Copy and commit current transient model before creating a new cursor
+        # This is required to avoid CacheMiss when using data from `self`
+        # which is created during current transaction.
+        with api.Environment.manage():
+            with registry(self.env.cr.dbname).cursor() as new_cr:
+                new_env = api.Environment(new_cr, self.env.uid, self.env.context)
+                self_env = self.with_env(new_env)
+                rec = self_env.create(self.copy_data())
+        for i in range(0, len(reconcile_groups), chunk_size):
+            chunk = reconcile_groups[i : i + chunk_size]
+            _logger.debug("Reconcile group chunk %s", chunk)
+            try:
+                with api.Environment.manage():
+                    with registry(self.env.cr.dbname).cursor() as new_cr:
+                        new_env = api.Environment(
+                            new_cr, self.env.uid, self.env.context
+                        )
+                        # Re-use the commited transient we just commited
+                        self_env = self.with_env(new_env).browse(rec.id)
+                        reconciled_ids += self_env._rec_group(chunk, lines_by_id)
+            except Exception as e:
+                msg = "Reconciliation failed for group chunk %s with error:\n%s"
+                _logger.exception(msg, chunk, e)
+        return reconciled_ids
+
     def _rec_auto_lines_advanced(self, credit_lines, debit_lines):
         """Advanced reconciliation main loop"""
         # pylint: disable=invalid-commit
         reconciled_ids = []
+        for rec in self:
+            commit_every = rec.account_id.company_id.reconciliation_commit_every
             reconcile_groups = []
             _logger.info("%d credit lines to reconcile", len(credit_lines))
             for idx, credit_line in enumerate(credit_lines, start=1):
@@ -235,7 +291,7 @@ class MassReconcileAdvanced(models.AbstractModel):
                 opposite_lines = self._search_opposites(credit_line, debit_lines)
                 if not opposite_lines:
                     continue
-            opposite_ids = [line["id"] for line in opposite_lines]
+                opposite_ids = [opp["id"] for opp in opposite_lines]
                 line_ids = opposite_ids + [credit_line["id"]]
                 for group in reconcile_groups:
                     if any([lid in group for lid in opposite_ids]):
@@ -251,23 +307,11 @@ class MassReconcileAdvanced(models.AbstractModel):
                     reconcile_groups.append(set(line_ids))
             lines_by_id = {line["id"]: line for line in credit_lines + debit_lines}
             _logger.info("Found %d groups to reconcile", len(reconcile_groups))
-        for group_count, reconcile_group_ids in enumerate(reconcile_groups, start=1):
+            if commit_every:
-            _logger.debug(
+                reconciled_ids = self._rec_group_by_chunk(
-                "Reconciling group %d/%d with ids %s",
+                    reconcile_groups, lines_by_id, commit_every
-                group_count,
-                len(reconcile_groups),
-                reconcile_group_ids,
                 )
-            group_lines = [lines_by_id[lid] for lid in reconcile_group_ids]
+            else:
-            reconciled, full = self._reconcile_lines(group_lines, allow_partial=True)
+                reconciled_ids = self._rec_group(reconcile_groups, lines_by_id)
-            if reconciled and full:
-                reconciled_ids += reconcile_group_ids
-
-            if (
-                self.env.context.get("commit_every", 0)
-                and group_count % self.env.context["commit_every"] == 0
-            ):
-                self.env.cr.commit()
-                _logger.info("Commit the reconciliations after %d groups", group_count)
             _logger.info("Reconciliation is over")
         return reconciled_ids

account_mass_reconcile_as_job/data/queue_job_function_data.xml

@@ -10,4 +10,15 @@
         />
         <field name="method">reconcile_as_job</field>
     </record>
+    <record id="job_function_reconcile_lines_as_job" model="queue.job.function">
+        <field
+            name="model_id"
+            ref="account_mass_reconcile_as_job.model_account_mass_reconcile"
+        />
+        <field
+            name="channel_id"
+            ref="account_mass_reconcile_as_job.channel_mass_reconcile"
+        />
+        <field name="method">reconcile_lines_as_job</field>
+    </record>
 </odoo>

account_mass_reconcile_as_job/models/__init__.py

@@ -1 +1,2 @@
 from . import mass_reconcile
+from . import base_reconciliation

account_mass_reconcile_as_job/models/base_reconciliation.py

@@ -0,0 +1,45 @@
+# Copyright 2022 Camptocamp SA
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+import ast
+import logging
+
+from odoo import api, models
+
+_logger = logging.getLogger(__name__)
+
+
+class MassReconcileBase(models.AbstractModel):
+    _inherit = "mass.reconcile.base"
+
+    def _reconcile_lines(self, lines, allow_partial=False):
+        as_job = (
+            self.env["ir.config_parameter"]
+            .sudo()
+            .get_param("account.mass.reconcile.lines.as.job", default=False)
+        )
+        try:
+            as_job = ast.literal_eval(as_job) if as_job else False
+        except ValueError:
+            as_job = False
+
+        if as_job and self.env.context.get("reconcile_lines_as_job", True):
+            wiz_data = self.copy_data()[0]
+            self.with_delay().reconcile_lines_as_job(
+                lines,
+                allow_partial=allow_partial,
+                wiz_creation_data=(self._name, wiz_data),
+            )
+            # Report is not available with reconcile jobs
+            return False, False
+        else:
+            return super()._reconcile_lines(lines, allow_partial=allow_partial)
+
+    @api.model
+    def reconcile_lines_as_job(
+        self, lines, allow_partial=False, wiz_creation_data=False
+    ):
+        new_wiz = self.env[wiz_creation_data[0]].create(wiz_creation_data[1])
+        return new_wiz.with_context(reconcile_lines_as_job=False)._reconcile_lines(
+            lines, allow_partial=allow_partial
+        )

account_mass_reconcile_as_job/models/mass_reconcile.py

@@ -6,6 +6,8 @@ import logging
 
 from odoo import models
 
+from odoo.addons.queue_job.job import identity_exact
+
 _logger = logging.getLogger(__name__)
 
 
@@ -25,7 +27,8 @@ class AccountMassReconcile(models.Model):
 
         if as_job and self.env.context.get("mass_reconcile_as_job", True):
             for rec in self:
-                rec.with_delay().reconcile_as_job()
+                job_options = {"identity_key": identity_exact}
+                rec.with_delay(**job_options).reconcile_as_job()
             return True
         else:
             return super().run_reconcile()

account_mass_reconcile_as_job/tests/__init__.py

@@ -1 +1,2 @@
 from . import test_mass_reconcile_as_job
+from . import test_scenario_reconcile_as_job

account_mass_reconcile_as_job/tests/test_scenario_reconcile_as_job.py

@@ -0,0 +1,112 @@
+# Copyright 2022 Camptocamp SA
+# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl)
+from odoo.tests import tagged
+
+from odoo.addons.account_mass_reconcile.tests.test_scenario_reconcile import (
+    TestScenarioReconcile,
+)
+from odoo.addons.queue_job.tests.common import trap_jobs
+
+
+@tagged("post_install", "-at_install")
+class TestScenarioReconcileAsJob(TestScenarioReconcile):
+    def test_scenario_reconcile_as_job(self):
+        self.env["ir.config_parameter"].sudo().set_param(
+            "account.mass.reconcile.as.job", True
+        )
+        invoice = self.create_invoice()
+        self.assertEqual("posted", invoice.state)
+
+        receivalble_account_id = invoice.partner_id.property_account_receivable_id.id
+        # create payment
+        payment = self.env["account.payment"].create(
+            {
+                "partner_type": "customer",
+                "payment_type": "inbound",
+                "partner_id": invoice.partner_id.id,
+                "destination_account_id": receivalble_account_id,
+                "amount": 50.0,
+                "journal_id": self.bank_journal.id,
+            }
+        )
+        payment.action_post()
+
+        # create the mass reconcile record
+        mass_rec = self.mass_rec_obj.create(
+            {
+                "name": "mass_reconcile_1",
+                "account": invoice.partner_id.property_account_receivable_id.id,
+                "reconcile_method": [(0, 0, {"name": "mass.reconcile.simple.partner"})],
+            }
+        )
+        with trap_jobs() as trap:
+            # call the automatic reconcilation method
+            mass_rec.run_reconcile()
+            trap.assert_jobs_count(1)
+            trap.assert_enqueued_job(
+                self.env["account.mass.reconcile"].reconcile_as_job,
+                args=(),
+            )
+            job = trap.enqueued_jobs[0]
+            self.assertEqual(job.state, "pending")
+            trap.perform_enqueued_jobs()
+            self.assertEqual("paid", invoice.payment_state)
+
+    def test_scenario_reconcile_lines_as_job(self):
+        self.env["ir.config_parameter"].sudo().set_param(
+            "account.mass.reconcile.as.job", True
+        )
+        self.env["ir.config_parameter"].sudo().set_param(
+            "account.mass.reconcile.lines.as.job", True
+        )
+        invoice = self.create_invoice()
+        self.assertEqual("posted", invoice.state)
+
+        receivalble_account_id = invoice.partner_id.property_account_receivable_id.id
+        # create payment
+        payment = self.env["account.payment"].create(
+            {
+                "partner_type": "customer",
+                "payment_type": "inbound",
+                "partner_id": invoice.partner_id.id,
+                "destination_account_id": receivalble_account_id,
+                "amount": 50.0,
+                "journal_id": self.bank_journal.id,
+            }
+        )
+        payment.action_post()
+
+        # create the mass reconcile record
+        mass_rec = self.mass_rec_obj.create(
+            {
+                "name": "mass_reconcile_1",
+                "account": invoice.partner_id.property_account_receivable_id.id,
+                "reconcile_method": [(0, 0, {"name": "mass.reconcile.simple.partner"})],
+            }
+        )
+        with trap_jobs() as trap:
+            self.assertFalse(self.env["mass.reconcile.simple.partner"].search([]))
+            # call the automatic reconcilation method
+            mass_rec.run_reconcile()
+            trap.assert_jobs_count(1)
+            trap.assert_enqueued_job(
+                self.env["account.mass.reconcile"].reconcile_as_job,
+                args=(),
+            )
+            job = trap.enqueued_jobs[0]
+            self.assertEqual(job.state, "pending")
+            self.assertFalse(self.env["mass.reconcile.simple.partner"].search([]))
+            trap.perform_enqueued_jobs()
+            trap.assert_jobs_count(2)
+            job_2 = trap.enqueued_jobs[1]
+            # Cannot use assert_enqueue_job with all the parameters
+            self.assertEqual(job_2.model_name, "mass.reconcile.simple.partner")
+            self.assertEqual(job_2.method_name, "reconcile_lines_as_job")
+            self.assertEqual(job_2.state, "pending")
+            # Delete existing wizard to make sure the job can still after after
+            #  the wizard is garbage collected
+            wiz = self.env["mass.reconcile.simple.partner"].search([])
+            self.assertTrue(wiz)
+            wiz.unlink()
+            trap.perform_enqueued_jobs()
+            self.assertEqual("paid", invoice.payment_state)