Add migration script for data from version ⩽15

migrations/16.0.1.0.0/post-migrate.py

+##############################################################################
+#
+#    Accounting Periods, for Odoo
+#    Copyright © 2024 XCG Consulting <https://xcg-consulting.fr/>
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+import json
+
+from odoo.tools.sql import SQL, column_exists, create_column, set_not_null, table_exists
+
+
+def migrate(cr, version):
+    # Migrate table data from <15.0
+    if table_exists(cr, "account_fiscalyear"):
+        dict_migrated_fiscal_years = {}
+        # Migrate fiscal years (account.fiscalyear to account.fiscal_year)
+        cr.execute(
+            SQL(
+                "SELECT id, name, company_id, date_start, date_stop, create_date, "
+                "create_uid, write_date, write_uid from account_fiscalyear"
+            )
+        )
+        fiscalyears = {}
+        for (
+            id,
+            name,
+            company_id,
+            date_start,
+            date_stop,
+            create_date,
+            create_uid,
+            write_date,
+            write_uid,
+        ) in cr.fetchall():
+            fiscalyears[id] = SQL(
+                "INSERT INTO account_fiscal_year (name, company_id, date_from, date_to,"
+                "create_date, create_uid, write_date, write_uid) VALUES "
+                "(%s, %s, %s, %s, %s, %s, %s, %s) RETURNING id",
+                name,
+                company_id,
+                date_start,
+                date_stop,
+                create_date,
+                create_uid,
+                write_date,
+                write_uid,
+            )
+        for id, query in fiscalyears.items():
+            cr.execute(query)
+            dict_migrated_fiscal_years[id] = cr.fetchall()[0][0]
+        # Migrate analytic_id in fiscal year if it exists
+        if column_exists(cr, "account_fiscalyear", "analytic_id"):
+            # create the column to avoid losing the value when dropping data
+            if not column_exists(cr, "account_fiscal_year", "analytic_id"):
+                create_column(cr, "account_fiscal_year", "analytic_id", "int4")
+            cr.execute(SQL("SELECT id, analytic_id FROM account_fiscalyear"))
+            update_data = [row for row in cr.fetchall()]
+            for id, analytic_id in update_data:
+                cr.execute(
+                    SQL(
+                        "UPDATE account_fiscal_year SET analytic_id = %s WHERE id = %s",
+                        analytic_id,
+                        dict_migrated_fiscal_years[id],
+                    )
+                )
+        # clean up the now useless table
+        cr.execute(SQL("DROP TABLE %s;", SQL.identifier("account_fiscalyear")))
+
+        # search the date range type to use for periods (from this module data)
+        cr.execute(
+            SQL(
+                "SELECT res_id from ir_model_data WHERE name='period_type' AND "
+                "module='account_period'"
+            )
+        )
+        date_range_type_id = cr.fetchall()[0][0]
+        # search for installed languages
+        cr.execute(SQL("SELECT code from res_lang WHERE active"))
+        languages_code = [row[0] for row in cr.fetchall()]
+        cr.execute(
+            SQL(
+                "SELECT id, company_id, code, name, date_start, date_stop, create_date"
+                ", create_uid, write_date, write_uid, fiscalyear_id FROM account_period"
+                " WHERE date_range_id is null;"
+            )
+        )
+        creation = {}
+        for row in cr.fetchall():
+            jsonb_name = json.dumps({lang: row[2] for lang in languages_code})
+            jsonb_type_name = json.dumps({})
+            creation[row[0]] = (
+                SQL(
+                    "INSERT INTO date_range (company_id, type_id, date_start, date_end,"
+                    " create_date, create_uid, write_date, write_uid, name, type_name, "
+                    "active) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, TRUE) "
+                    "RETURNING id",
+                    row[1],
+                    date_range_type_id,
+                    row[4],
+                    row[5],
+                    row[6],
+                    row[7],
+                    row[8],
+                    row[9],
+                    jsonb_name,
+                    jsonb_type_name,
+                ),
+                row[10],
+            )
+
+        update_query = (
+            "UPDATE account_period SET date_range_id = %s, fiscal_year_id = %s "
+            "WHERE id = %s"
+        )
+        for period_id, (query, fiscalyear_id) in creation.items():
+            cr.execute(query)
+            created_date_range_id = cr.fetchall()[0][0]
+            cr.execute(
+                SQL(
+                    update_query,
+                    created_date_range_id,
+                    dict_migrated_fiscal_years[fiscalyear_id],
+                    period_id,
+                )
+            )
+        for column in ("date_range_id", "fiscal_year_id"):
+            set_not_null(cr, "account_period", column)
+        # clean up the period table
+        for column in (
+            "fiscalyear_id",
+            "date_start",
+            "date_stop",
+            "company_id",
+            "name",
+            "code",
+        ):
+            cr.execute(
+                SQL("ALTER TABLE account_period DROP COLUMN %s", SQL.identifier(column))
+            )