You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
1.9 KiB
44 lines
1.9 KiB
# 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)
|