from odoo import models


class Payslip(models.Model):
    _inherit = 'hr.payslip'

    def ytd(self, code, allow_draft=False):
        to_date = self.date_to
        from_date = str(self.date_to.year) + '-01-01'
        state_allowed = ('done',) if not allow_draft else ('done', 'verify', 'draft')
        self.env.cr.execute("""
            SELECT sum(pi.total) as total,
                   sum(pi.quantity) as quantity,
                   sum(pi.amount) as amount
            FROM hr_payslip as hp
            JOIN hr_payslip_line as pi ON hp.id = pi.slip_id
            WHERE hp.employee_id = %s 
                  AND hp.state in %s
                  AND hp.date_to >= %s AND hp.date_to <= %s AND pi.code = %s""",
                            (self.employee_id.id, state_allowed, from_date, to_date, code))
        res = self.env.cr.dictfetchone()
        if res:
            # Can return dictionary with NULL aka Nones
            for key in res:
                res[key] = res[key] or 0.0
        else:
            res = {'total': 0.0, 'quantity': 0.0, 'amount': 0.0}
        return res

    def worked_ytd(self, code, allow_draft=False):
        to_date = self.date_to
        from_date = str(self.date_to.year) + '-01-01'
        state_allowed = ('done',) if not allow_draft else ('done', 'verify', 'draft')
        self.env.cr.execute("""
                    SELECT sum(wd.amount) as amount
                    FROM hr_payslip as hp
                    JOIN hr_payslip_worked_days as wd ON hp.id = wd.payslip_id
                    JOIN hr_work_entry_type as wt ON wd.work_entry_type_id = wt.id
                    WHERE hp.employee_id = %s 
                          AND hp.state in %s
                          AND hp.date_to >= %s AND hp.date_to <= %s AND wt.code = %s""",
                            (self.employee_id.id, state_allowed, from_date, to_date, code))
        res = self.env.cr.dictfetchone()
        if res:
            # Can return dictionary with NULL aka Nones
            res['amount'] = res.get('amount', 0.0)
        else:
            res = {'amount': 0.0}
        return res