pre_init_hook.py

# Copyright 2017-2022 Akretion France
# @author: Alexis de Lattre <alexis.delattre@akretion.com>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

import logging

from lxml import etree

from odoo import SUPERUSER_ID, api
from odoo.tools import file_open

logger = logging.getLogger(__name__)

# This code is designed to avoid a problem in the following scenario:
# On a new database, the administrator executes these steps in this order:
# 1) installs base_address_extended_geonames_import and run the geonames import wizard
#  for France
# -> it creates all the French regions (without xmlid)
# 2) installs l10n_fr_state
# -> it tries to create the French regions, but it fails due to the unicity
#  constraint unique(country_id, code) of res.country.state.


def create_fr_state_xmlid(cr):
    generic_create_state_xmlid(cr, "l10n_fr_state", "data/res_country_state.xml")


def generic_create_state_xmlid(cr, module_name, data_file):
    """This method is also used by l10n_fr_state and l10n_fr_department_oversea"""
    f = file_open("%s/%s" % (module_name, data_file), "rb")
    xml_root = etree.parse(f)
    data = {}  # key = xmlid, value = {"code": "GP", "country_id": "base.gp"}
    for record in xml_root.xpath("//record"):
        xmlid = record.attrib["id"]
        data[xmlid] = {}
        for xfield in record.xpath("field"):
            xfield_dict = xfield.attrib
            data[xmlid][xfield_dict["name"]] = xfield_dict.get("ref") or xfield.text
    logger.debug("generic_create_state_xmlid data=%s", data)
    env = api.Environment(cr, SUPERUSER_ID, {})
    for xmlid, state_data in data.items():
        country_id = env.ref(state_data["country_id"]).id
        state = env["res.country.state"].search(
            [("code", "=", state_data["code"]), ("country_id", "=", country_id)],
            limit=1,
        )
        if state:
            env["ir.model.data"].create(
                {
                    "name": xmlid,
                    "res_id": state.id,
                    "module": module_name,
                    "model": "res.country.state",
                }
            )
            logger.info(
                "XMLID %s.%s created for state %s code %s",
                module_name,
                xmlid,
                state.name,
                state.code,
            )