trytond-account_es/tax.py

# The COPYRIGHT file at the top level of this repository contains the full
# copyright notices and license terms.
from trytond.model import fields
from trytond.pool import PoolMeta, Pool
from trytond.pyson import Eval
from trytond.transaction import Transaction

__all__ = ['TaxCodeTemplate', 'TaxRuleTemplate', 'TaxRuleLineTemplate',
    'TaxTemplate', 'Tax']


class TaxCodeTemplate(metaclass=PoolMeta):
    __name__ = 'account.tax.code.template'

    @classmethod
    def check_xml_record(cls, records, values):
        return True


class TaxRuleTemplate(metaclass=PoolMeta):
    __name__ = 'account.tax.rule.template'

    @classmethod
    def check_xml_record(cls, records, values):
        return True


class TaxRuleLineTemplate(metaclass=PoolMeta):
    __name__ = 'account.tax.rule.line.template'

    @classmethod
    def check_xml_record(cls, records, values):
        return True


class TaxTemplate(metaclass=PoolMeta):
    __name__ = 'account.tax.template'

    report_description = fields.Text('Report Description')
    recargo_equivalencia = fields.Boolean('Recargo Equivalencia',
        help='Indicates if the tax is Recargo de Equivalencia')
    recargo_equivalencia_related_tax = fields.Many2One(
        'account.tax.template', 'Recargo Equivalencia Related Tax',
        domain=[
            ('recargo_equivalencia', '=', True),
            ], depends=['recargo_equivalencia'],
        help='The possible Recargo Equivalencia related to this tax')
    deducible = fields.Boolean('Deducible',
        help='Indicates if the tax is deductible')
    code_lines = fields.One2Many('account.tax.code.line.template', 'tax',
        'Code Lines')
    isp = fields.Boolean('Inversion del Sujeto Pasivo',
        help='Indicates if the tax is Inversion del sujeto Pasivo')
    service = fields.Boolean('Service Rate',
        help='Indicates if the tax is used for services')

    @classmethod
    def __setup__(cls):
        super(TaxTemplate, cls).__setup__()
        cls.invoice_account.domain.remove(('type.statement', '=', 'balance'))
        cls.credit_note_account.domain.remove(
            ('type.statement', '=', 'balance'))

    @classmethod
    def __register__(cls, module_name):
        pool = Pool()
        ModelData = pool.get('ir.model.data')
        table = cls.__table__()
        model_data = ModelData.__table__()

        super().__register__(module_name)

        transaction = Transaction()
        cursor = transaction.connection.cursor()

        # Update IRPF tax use. Migrate from old versions to +5.4
        # Ensure that group field is null in all IRPF tax templates,
        # becasue the IRPF account tax rules not use group any more.
        cursor.execute(*model_data.select(
                model_data.db_id,
                where=(model_data.fs_id.like('irpf_%')) &
                    (model_data.model == 'account.tax.template')))
        for db_id, in cursor.fetchall():
            cursor.execute(*table.select(
                    table.group,
                    where=table.id == db_id))
            group, = cursor.fetchone()
            if group:
                cursor.execute(*table.update(
                        [table.group], [None],
                        where=table.id == db_id))

    def _get_tax_value(self, tax=None):
        res = super(TaxTemplate, self)._get_tax_value(tax)

        if not tax or tax.report_description != self.report_description:
            res['report_description'] = self.report_description
        if not tax or tax.recargo_equivalencia != self.recargo_equivalencia:
            res['recargo_equivalencia'] = self.recargo_equivalencia
        if not tax or tax.deducible != self.deducible:
            res['deducible'] = self.deducible
        if not tax or tax.isp != self.isp:
            res['isp'] = self.isp
        if not tax or tax.service != self.service:
            res['service'] = self.service
        return res

    @classmethod
    def update_recargo_equivalencia_related_tax(cls, template2tax):
        pool = Pool()
        Tax = pool.get('account.tax')

        templates = cls.search([
            ('recargo_equivalencia_related_tax', '!=', None),
            ])
        for template in templates:
            if template2tax.get(template.id):
                tax = Tax(template2tax[template.id])
                tax.recargo_equivalencia_related_tax = template2tax[
                    template.recargo_equivalencia_related_tax.id]
                tax.save()

    @classmethod
    def update_tax(cls, company_id, template2account, template2tax=None):
        super(TaxTemplate, cls).update_tax(company_id,
            template2account, template2tax)
        cls.update_recargo_equivalencia_related_tax(template2tax)

    @classmethod
    def create_tax(cls, account_id, company_id, template2account,
            template2tax):
        super(TaxTemplate, cls).create_tax(account_id, company_id,
            template2account, template2tax)
        cls.update_recargo_equivalencia_related_tax(template2tax)

    @classmethod
    def check_xml_record(cls, records, values):
        return True

    @fields.depends('childs', 'isp')
    def on_change_isp(self):
        for child in self.childs:
            child.isp = self.isp

    @fields.depends('childs', 'service')
    def on_change_service(self):
        for child in self.childs:
            child.service = self.service


class Tax(metaclass=PoolMeta):
    __name__ = 'account.tax'

    report_description = fields.Text('Report Description', translate=True)
    recargo_equivalencia = fields.Boolean('Recargo Equivalencia',
        help='Indicates if the tax is Recargo de Equivalencia')
    recargo_equivalencia_related_tax = fields.Many2One(
        'account.tax', 'Recargo Equivalencia Related Tax',
        domain=[
            ('recargo_equivalencia', '=', True),
            ('company', '=', Eval('company', -1)),
            ], depends=['recargo_equivalencia', 'company'],
        help='The possible Recargo Equivalencia related to this tax')
    deducible = fields.Boolean('Deducible',
        help='Indicates if the tax is deductible')
    code_lines = fields.One2Many('account.tax.code.line', 'tax',
        'Code Lines')
    isp = fields.Boolean('Inversion del Sujeto Pasivo',
        help='Indicates if the tax is Inversion del sujeto Pasivo')
    service = fields.Boolean('Service Rate',
        help='Indicates if the tax is used for services')

    @classmethod
    def __setup__(cls):
        super(Tax, cls).__setup__()
        cls.invoice_account.domain.remove(('type.statement', '=', 'balance'))
        cls.credit_note_account.domain.remove(
            ('type.statement', '=', 'balance'))

    @staticmethod
    def default_deducible():
        return True

    @fields.depends('childs', 'isp')
    def on_change_isp(self):
        for child in self.childs:
            child.isp = self.isp

    @fields.depends('childs', 'service')
    def on_change_service(self):
        for child in self.childs:
            child.service = self.service