odoo/addons/hr_timesheet_sheet/models/hr_department.py

20 lines
802 B
Python

# -*- coding: utf-8 -*-
# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import api, fields, models
class HrDepartment(models.Model):
_inherit = 'hr.department'
timesheet_to_approve_count = fields.Integer(
compute='_compute_timesheet_to_approve', string='Timesheet to Approve')
@api.multi
def _compute_timesheet_to_approve(self):
timesheet_data = self.env['hr_timesheet_sheet.sheet'].read_group(
[('department_id', 'in', self.ids), ('state', '=', 'confirm')], ['department_id'], ['department_id'])
result = dict((data['department_id'][0], data['department_id_count']) for data in timesheet_data)
for department in self:
department.timesheet_to_approve_count = result.get(department.id, 0)