l10n-spain/l10n_es_aeat/models/res_company.py at 92278c4877eb3c80391653120b2507bd3efefa7d · moduon/l10n-spain · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
# Copyright 2019-2020 Tecnativa - Pedro M. Baeza
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl.html).

import logging

from odoo import api, fields, models
from odoo.fields import Domain
from odoo.tools import ormcache

_logger = logging.getLogger(__name__)


class ResCompany(models.Model):
    _inherit = "res.company"

    tax_agency_id = fields.Many2one("aeat.tax.agency", string="Tax Agency")
    representative_vat = fields.Char(
        string="L.R. VAT number",
        size=9,
        help="Legal Representative VAT number for all the AEAT reports of this "
        "company.",
    )

    @api.model_create_multi
    def create(self, vals_list):
        """Create immediately all the AEAT sequences when creating company."""
        companies = super().create(vals_list)
        models = (
            self.env["ir.model"]
            .sudo()
            .search([Domain("model", "=like", "l10n.es.aeat.%.report")])
        )
        for model in models:
            try:
                self.env[model.model]._register_hook(companies=companies)
            except Exception as e:
                _logger.debug(e)
        return companies

    @ormcache("self", "xmlid")
    def _get_tax_id_from_xmlid(self, xmlid):
        """Low level cached search for a tax given its template XML-ID and company.

        WARNING: It doesn't return equivalent taxes. Call `_get_taxes_from_xmlids` for
        that.
        """
        self.ensure_one()
        return (
            xmlid
            and self.sudo()
            .env["ir.model.data"]
            .search(
                Domain.AND(
                    [
                        Domain("model", "=", "account.tax"),
                        Domain(
                            "module", "=", "account"
                        ),  # All is registered under this module
                        Domain("name", "=", f"{self.id}_{xmlid}"),
                    ]
                ),
            )
            .res_id
            or False
        )

    def _get_taxes_from_xmlids(self, xmlids):
        """Search for the taxes (direct or equivalent), given the company and the tax
        template XML-IDs.

        :returns: Company taxes recordset.
        """
        self.ensure_one()
        tax_ids = set(self._get_tax_id_from_xmlid(x) for x in xmlids)
        if False in tax_ids:  # for avoiding that incorrect XML-IDs breaks the rest
            tax_ids.remove(False)
        tax_obj = self.env["account.tax"]
        extra_taxes = tax_obj.search([("aeat_equivalent_tax_id", "in", list(tax_ids))])
        return tax_obj.browse(tax_ids) | extra_taxes

    @ormcache("self", "xmlid")
    def _get_account_id_from_xmlid(self, xmlid):
        """Low level cached search for a tax given its account template and
        company.
        """
        self.ensure_one()
        return (
            xmlid
            and self.sudo()
            .env["ir.model.data"]
            .search(
                Domain.AND(
                    [
                        Domain("model", "=", "account.account"),
                        Domain(
                            "module", "=", "account"
                        ),  # All is registered under this module
                        Domain("name", "=", f"{self.id}_{xmlid}"),
                    ]
                ),
            )
            .res_id
            or False
        )