hooks.py

# Copyright 2018 Akretion (http://www.akretion.com)
# @author: Alexis de Lattre <alexis.delattre@akretion.com>
# Copyright 2019 Onestein (<https://www.onestein.eu>)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import SUPERUSER_ID, api


def set_xml_format_in_pdf_invoice_to_ubl(cr, registry):
    with api.Environment.manage():
        env = api.Environment(cr, SUPERUSER_ID, {})
        companies = env["res.company"].search([])
        companies.write({"xml_format_in_pdf_invoice": "ubl"})


def remove_ubl_xml_format_in_pdf_invoice(cr, registry):
    with api.Environment.manage():
        env = api.Environment(cr, SUPERUSER_ID, {})
        companies = env["res.company"].search(
            [("xml_format_in_pdf_invoice", "=", "ubl")]
        )
        companies.write({"xml_format_in_pdf_invoice": "none"})