Merge 5ed241bc90 into 0aa3d6c1ae

account_tax_balance/__init__.py

@@ -2,3 +2,4 @@
 
 from . import models
 from . import wizard
+from .hooks import pre_init_hook

account_tax_balance/__manifest__.py

@@ -25,5 +25,6 @@
     ],
     "images": [
         'images/tax_balance.png',
-    ]
+    ],
+    "pre_init_hook": "pre_init_hook",
 }

account_tax_balance/hooks.py

@@ -0,0 +1,43 @@
+# Copyright 2020 Opener B.V. <https://opener.amsterdam>
+# Copyright 2020 Tecnativa - Pedro M. Baeza
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+import logging
+from psycopg2 import sql
+
+
+def pre_init_hook(cr):
+    """ Precreate move_type and fill with appropriate values to prevent
+    a MemoryError when the ORM attempts to call its compute method on a large
+    amount of preexisting moves. Note that the order of the mapping is
+    important as one move can have move lines on accounts of multiple types
+    and the move type is set in the order of precedence. """
+    logger = logging.getLogger(__name__)
+    logger.info("Add account_move.move_type column if it does not yet exist")
+    cr.execute(
+        "ALTER TABLE account_move ADD COLUMN IF NOT EXISTS move_type VARCHAR")
+    MAPPING = [
+        ("liquidity", "liquidity", False),
+        ("payable", "payable", "AND aml.balance < 0"),
+        ("payable_refund", "payable", "AND aml.balance >= 0"),
+        ("receivable", "receivable", "AND aml.balance > 0"),
+        ("receivable_refund", "receivable", "AND aml.balance <= 0"),
+        ("other", False, False),
+    ]
+    for move_type, internal_type, extra_where in MAPPING:
+        args = [move_type]
+        query = sql.SQL("UPDATE account_move am SET move_type = %s")
+        if internal_type:
+            query += sql.SQL(
+                """FROM account_move_line aml
+                WHERE aml.account_id IN (
+                    SELECT id FROM account_account
+                    WHERE internal_type = %s)
+                AND aml.move_id = am.id AND am.move_type IS NULL
+                """)
+            args.append(internal_type)
+        else:
+            query += sql.SQL("WHERE am.move_type IS NULL")
+        if extra_where:
+            query += sql.SQL(extra_where)
+        cr.execute(query, tuple(args))
+        logger.info("%s move set to type %s", move_type, cr.rowcount)