hooks.py

import logging

from odoo import SUPERUSER_ID, api

_logger = logging.getLogger(__name__)

try:
    from odoo.addons.base_multi_company import hooks
except ImportError:
    _logger.info("Cannot find `base_multi_company` module in addons path.")


def post_init_hook(cr, registry):
    hooks.post_init_hook(
        cr,
        "base.res_partner_rule",
        "res.partner",
    )


def uninstall_hook(cr, registry):
    """
    Restore original domain for base.res_partner_rule

    Args:
        cr (Cursor): Database cursor to use for operation.
    """
    with api.Environment.manage():
        env = api.Environment(cr, SUPERUSER_ID, {})
        rule = env.ref("base.res_partner_rule")
        rule.write(
            {
                "active": False,
                "domain_force": (
                    "['|', '|', ('partner_share', '=', False), "
                    "('company_id', 'in', company_ids), "
                    "('company_id', '=', False)]"
                ),
            }
        )