hooks.py

# Copyright (C) 2022-Today - Engenere (<https://engenere.one>).
# @author Antônio S. Pereira Neto <neto@engenere.one>
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

from odoo import SUPERUSER_ID, api, tools


def post_init_hook(cr, registry):
    env = api.Environment(cr, SUPERUSER_ID, {})
    load_simples_nacional_demo(env, registry)


def load_simples_nacional_demo(env, registry):
    """
    Load demo data for company 'Simples Nacional' with
    default user company set to this company.
    """

    # Load XML file with demo data.
    if not tools.config["without_demo"]:

        # Allow all companies for OdooBot user and set default user company
        companies = env["res.company"].search([])
        env.user.company_ids = [(6, 0, companies.ids)]
        env.user.company_id = env.ref("l10n_br_base.empresa_simples_nacional")

        tools.convert_file(
            env.cr,
            "l10n_br_account_nfe",
            "demo/account_invoice_sn_demo.xml",
            None,
            mode="init",
            noupdate=True,
            kind="demo",
        )

        # É necessário rodar os onchanges fiscais para
        # preencher os campos referentes aos Impostos
        invoice_tag_cobranca = env.ref("l10n_br_account_nfe.demo_nfe_dados_de_cobranca")
        for line in invoice_tag_cobranca.invoice_line_ids:
            line._onchange_fiscal_operation_line_id()
            line._onchange_fiscal_tax_ids()

        invoice_sem_tag_cobranca = env.ref(
            "l10n_br_account_nfe.demo_nfe_sem_dados_de_cobranca"
        )
        for line in invoice_sem_tag_cobranca.invoice_line_ids:
            line._onchange_fiscal_operation_line_id()
            line._onchange_fiscal_tax_ids()

    # back to the main company as the next modules to be installed
    # expect this to be the default company.
    env.user.company_id = env.ref("base.main_company")