forked from Mapan/odoo17e
124 lines
5.1 KiB
Python
124 lines
5.1 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
import logging
|
|
|
|
from collections import defaultdict
|
|
|
|
from odoo import models, fields, api, _
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
|
|
class HrPayrollEmployeeDeclaration(models.Model):
|
|
_name = 'hr.payroll.employee.declaration'
|
|
_description = 'Payroll Employee Declaration'
|
|
_rec_name = 'employee_id'
|
|
|
|
res_model = fields.Char(
|
|
'Declaration Model Name', required=True, index=True)
|
|
res_id = fields.Many2oneReference(
|
|
'Declaration Model Id', index=True, model_field='res_model', required=True)
|
|
employee_id = fields.Many2one('hr.employee', domain="['|', ('active', '=', True), ('active', '=', False)]")
|
|
company_id = fields.Many2one('res.company', default=lambda self: self.env.company, required=True)
|
|
pdf_file = fields.Binary('PDF File', readonly=True, attachment=False)
|
|
pdf_filename = fields.Char()
|
|
pdf_to_generate = fields.Boolean()
|
|
state = fields.Selection([
|
|
('draft', 'Draft'),
|
|
('pdf_to_generate', 'Queued PDF generation'),
|
|
('pdf_generated', 'Generated PDF'),
|
|
], compute='_compute_state', store=True)
|
|
|
|
_sql_constraints = [
|
|
('unique_employee_sheet', 'unique(employee_id, res_model, res_id)', 'An employee can only have one declaration per sheet.'),
|
|
]
|
|
|
|
@api.depends('pdf_to_generate', 'pdf_file')
|
|
def _compute_state(self):
|
|
for declaration in self:
|
|
if declaration.pdf_to_generate:
|
|
declaration.state = 'pdf_to_generate'
|
|
elif declaration.pdf_file:
|
|
declaration.state = 'pdf_generated'
|
|
else:
|
|
declaration.state = 'draft'
|
|
|
|
def _generate_pdf(self):
|
|
report_sudo = self.env["ir.actions.report"].sudo()
|
|
declarations_by_sheet = defaultdict(lambda: self.env['hr.payroll.employee.declaration'])
|
|
for declaration in self:
|
|
declarations_by_sheet[(declaration.res_model, declaration.res_id)] += declaration
|
|
|
|
|
|
for (res_model, res_id), declarations in declarations_by_sheet.items():
|
|
sheet = self.env[res_model].browse(res_id)
|
|
if not sheet.exists():
|
|
_logger.warning('Sheet %s %s does not exist', res_model, res_id)
|
|
continue
|
|
report_id = sheet._get_pdf_report().id
|
|
rendering_data = sheet._get_rendering_data(declarations.employee_id)
|
|
if 'error' in rendering_data:
|
|
sheet.pdf_error = rendering_data['error']
|
|
continue
|
|
rendering_data = sheet._post_process_rendering_data_pdf(rendering_data)
|
|
|
|
pdf_files = []
|
|
sheet_count = len(rendering_data)
|
|
counter = 1
|
|
for employee, employee_data in rendering_data.items():
|
|
_logger.info('Printing %s (%s/%s)', sheet._description, counter, sheet_count)
|
|
counter += 1
|
|
sheet_filename = sheet._get_pdf_filename(employee)
|
|
sheet_file, dummy = report_sudo.with_context(lang=employee.lang or self.env.lang)._render_qweb_pdf(
|
|
report_id,
|
|
[employee.id], data={'report_data': employee_data, 'employee': employee, 'company_id': employee.company_id})
|
|
pdf_files.append((employee, sheet_filename, sheet_file))
|
|
if pdf_files:
|
|
sheet._process_files(pdf_files)
|
|
|
|
@api.model_create_multi
|
|
def create(self, vals_list):
|
|
declarations = super().create(vals_list)
|
|
if any(declaration.pdf_to_generate for declaration in declarations):
|
|
self.env.ref('hr_payroll.ir_cron_generate_payslip_pdfs')._trigger()
|
|
return declarations
|
|
|
|
def write(self, vals):
|
|
res = super().write(vals)
|
|
if vals.get('pdf_to_generate'):
|
|
self.env.ref('hr_payroll.ir_cron_generate_payslip_pdfs')._trigger()
|
|
return res
|
|
|
|
def action_generate_pdf(self):
|
|
if self:
|
|
self.write({'pdf_to_generate': True})
|
|
self.env.ref('hr_payroll.ir_cron_generate_payslip_pdfs')._trigger()
|
|
message = _("PDF generation started. It will be available shortly.")
|
|
else:
|
|
message = _("Please select the declarations for which you want to generate a PDF.")
|
|
return {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'display_notification',
|
|
'params': {
|
|
'type': 'success',
|
|
'message': message,
|
|
'next': {
|
|
'type': 'ir.actions.client',
|
|
'tag': 'reload'
|
|
}
|
|
}
|
|
}
|
|
|
|
@api.autovacuum
|
|
def _gc_orphan_declarations(self):
|
|
orphans = self.env['hr.payroll.employee.declaration']
|
|
grouped_declarations = self.read_group([], ['ids:array_agg(id)', 'res_ids:array_agg(res_id)'], ['res_model'])
|
|
for gd in grouped_declarations:
|
|
sheet_ids = self.env[gd['res_model']].browse(set(gd['res_ids'])).exists().ids
|
|
for declaration in self.browse(gd['ids']):
|
|
if declaration.res_id not in sheet_ids:
|
|
orphans += declaration
|
|
if orphans:
|
|
orphans.unlink()
|