53 lines
2.1 KiB
Python
53 lines
2.1 KiB
Python
#-*- coding:utf-8 -*-
|
|
|
|
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.
|
|
|
|
from datetime import datetime
|
|
from dateutil.relativedelta import relativedelta
|
|
|
|
from flectra import api, fields, models, _
|
|
from flectra.exceptions import UserError
|
|
|
|
|
|
class ContributionRegisterReport(models.AbstractModel):
|
|
_name = 'report.hr_payroll.report_contributionregister'
|
|
|
|
def _get_payslip_lines(self, register_ids, date_from, date_to):
|
|
result = {}
|
|
self.env.cr.execute("""
|
|
SELECT pl.id from hr_payslip_line as pl
|
|
LEFT JOIN hr_payslip AS hp on (pl.slip_id = hp.id)
|
|
WHERE (hp.date_from >= %s) AND (hp.date_to <= %s)
|
|
AND pl.register_id in %s
|
|
AND hp.state = 'done'
|
|
ORDER BY pl.slip_id, pl.sequence""",
|
|
(date_from, date_to, tuple(register_ids)))
|
|
line_ids = [x[0] for x in self.env.cr.fetchall()]
|
|
for line in self.env['hr.payslip.line'].browse(line_ids):
|
|
result.setdefault(line.register_id.id, self.env['hr.payslip.line'])
|
|
result[line.register_id.id] += line
|
|
return result
|
|
|
|
@api.model
|
|
def get_report_values(self, docids, data=None):
|
|
if not data.get('form'):
|
|
raise UserError(_("Form content is missing, this report cannot be printed."))
|
|
|
|
register_ids = self.env.context.get('active_ids', [])
|
|
contrib_registers = self.env['hr.contribution.register'].browse(register_ids)
|
|
date_from = data['form'].get('date_from', fields.Date.today())
|
|
date_to = data['form'].get('date_to', str(datetime.now() + relativedelta(months=+1, day=1, days=-1))[:10])
|
|
lines_data = self._get_payslip_lines(register_ids, date_from, date_to)
|
|
lines_total = {}
|
|
for register in contrib_registers:
|
|
lines = lines_data.get(register.id)
|
|
lines_total[register.id] = lines and sum(lines.mapped('total')) or 0.0
|
|
return {
|
|
'doc_ids': register_ids,
|
|
'doc_model': 'hr.contribution.register',
|
|
'docs': contrib_registers,
|
|
'data': data,
|
|
'lines_data': lines_data,
|
|
'lines_total': lines_total
|
|
}
|