From 68a2624bedd0ed0a1860c86064ec29321fb94e76 Mon Sep 17 00:00:00 2001
From: "Danny W. Adair" <danny.adair@unfold.co.nz>
Date: Fri, 24 Feb 2023 11:10:17 +1300
Subject: [PATCH] Add pre_init_hook to add computed columns

Add columns to avoid Memory error on an existing Odoo instance
with lots of data.

The payment_mode_id fields are introduced by this module and computed only
from each other or the also newly introduced supplier_payment_mode_id and
customer_payment_mode_id on res.partner, so they can stay NULL, nothing
to compute.
---
 account_payment_partner/__init__.py     |  1 +
 account_payment_partner/__manifest__.py |  1 +
 account_payment_partner/hooks.py        | 22 ++++++++++++++++++++++
 3 files changed, 24 insertions(+)
 create mode 100644 account_payment_partner/hooks.py

diff --git a/account_payment_partner/__init__.py b/account_payment_partner/__init__.py
index 58da23772..570623a1e 100644
--- a/account_payment_partner/__init__.py
+++ b/account_payment_partner/__init__.py
@@ -1,4 +1,5 @@
 # License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).
 
+from .hooks import pre_init_hook
 from . import models
 from . import reports
diff --git a/account_payment_partner/__manifest__.py b/account_payment_partner/__manifest__.py
index a145f0650..6e25c104c 100644
--- a/account_payment_partner/__manifest__.py
+++ b/account_payment_partner/__manifest__.py
@@ -24,4 +24,5 @@
     ],
     "demo": ["demo/partner_demo.xml"],
     "installable": True,
+    "pre_init_hook": "pre_init_hook",
 }
diff --git a/account_payment_partner/hooks.py b/account_payment_partner/hooks.py
new file mode 100644
index 000000000..f2a1b840c
--- /dev/null
+++ b/account_payment_partner/hooks.py
@@ -0,0 +1,22 @@
+import logging
+
+from odoo.tools import sql
+
+logger = logging.getLogger(__name__)
+
+
+def pre_init_hook(cr):
+    """Prepare new payment_mode fields.
+
+    Add columns to avoid Memory error on an existing Odoo instance
+    with lots of data.
+
+    The payment_mode_id fields are introduced by this module and computed only
+    from each other or the also newly introduced supplier_payment_mode_id and
+    customer_payment_mode_id on res.partner, so they can stay NULL, nothing
+    to compute.
+    """
+    if not sql.column_exists(cr, "account_move", "payment_mode_id"):
+        sql.create_column(cr, "account_move", "payment_mode_id", "int4")
+    if not sql.column_exists(cr, "account_move_line", "payment_mode_id"):
+        sql.create_column(cr, "account_move_line", "payment_mode_id", "int4")