trytonpsk-staff_payroll_co/uvt.py

#This file is part of Tryton.  The COPYRIGHT file at the top level of
#this repository contains the full copyright notices and license terms.
from decimal import Decimal
from trytond.model import ModelView, ModelSQL, fields
from trytond.pool import Pool

__all__ = ['UvtWithholding']

UVT_SUM_FORMULA = {
    28: 10,
    33: 69,
    35: 162,
    37: 268,
    39: 770
}


class UvtWithholding(ModelSQL, ModelView):
    "UVT Withholding"
    __name__ = 'staff.payroll.uvt_withholding'
    start_range = fields.Numeric('Start Range', digits=(16, 0), required=True)
    end_range = fields.Numeric('End Range', digits=(16, 0))
    rate = fields.Numeric('Rate', digits=(16, 0), required=True)

    @classmethod
    def __setup__(cls):
        super(UvtWithholding, cls).__setup__()

    @classmethod
    def compute_withholding(cls, payment):
        """
        Compute unit price for application of withholdings tax to employee
        """
        res = Decimal('0.0')
        Configuration = Pool().get('staff.configuration')
        configuration = Configuration(1)
        if not configuration.uvt_value:
            return res
        uvt_config = Decimal(configuration.uvt_value)

        percent_exent = Decimal('0.75')
        payment2uvt = Decimal(payment * percent_exent) / uvt_config
        uvt_withholdings = cls.search([
            ['AND', ['OR',
                     [
                        ('start_range', '<', payment2uvt),
                        ('end_range', '>=', payment2uvt)
                     ],
                     [
                        ('start_range', '<', payment2uvt),
                        ('end_range', '=', None)
                     ],
                     ]]
            ])
        if uvt_withholdings:
            uvt_wh = uvt_withholdings[0]
            rate = uvt_wh.rate
            add_sum = UVT_SUM_FORMULA[rate] if UVT_SUM_FORMULA.get(rate) else 0
            print(payment)
            print(payment2uvt, rate)
            value = (payment2uvt - uvt_wh.start_range) * Decimal(rate / 100) + add_sum
            print(value)
            res = round(value, 1) * uvt_config
            print(res)
            res = Decimal(round(res, 2))
        return res