Merge PR #647 into 9.0

Signed-off-by pedrobaeza

account_tax_balance/__init__.py

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

account_tax_balance/__openerp__.py

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

account_tax_balance/hooks.py

@@ -0,0 +1,44 @@
+# coding: utf-8
+# 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)