oca-ocb-project/odoo-bringout-oca-ocb-project_timesheet_holidays/project_timesheet_holidays/models/project_task.py
Ernad Husremovic 4a4d12c333 19.0 vanilla
2026-03-09 09:31:56 +01:00

42 lines
1.9 KiB
Python

# Part of Odoo. See LICENSE file for full copyright and licensing details.
from odoo import fields, models
from odoo.fields import Domain
from odoo.tools import OrderedSet
class ProjectTask(models.Model):
_inherit = 'project.task'
leave_types_count = fields.Integer(compute='_compute_leave_types_count', string="Time Off Types Count")
is_timeoff_task = fields.Boolean("Is Time off Task", compute="_compute_is_timeoff_task", search="_search_is_timeoff_task", export_string_translation=False, groups="hr_timesheet.group_hr_timesheet_user")
def _compute_leave_types_count(self):
timesheet_read_group = self.env['account.analytic.line']._read_group(
[('task_id', 'in', self.ids), '|', ('holiday_id', '!=', False), ('global_leave_id', '!=', False)],
['task_id'],
['__count'],
)
timesheet_count_per_task = {timesheet_task.id: count for timesheet_task, count in timesheet_read_group}
for task in self:
task.leave_types_count = timesheet_count_per_task.get(task.id, 0)
def _compute_is_timeoff_task(self):
timeoff_tasks = self.filtered(lambda task: task.leave_types_count or task.company_id.leave_timesheet_task_id == task)
timeoff_tasks.is_timeoff_task = True
(self - timeoff_tasks).is_timeoff_task = False
def _search_is_timeoff_task(self, operator, value):
if operator != 'in':
return NotImplemented
timeoff_tasks_ids = {row[0] for row in self.env.execute_query(
self.env['account.analytic.line']._search(
[('task_id', '!=', False), '|', ('holiday_id', '!=', False), ('global_leave_id', '!=', False)],
).select('DISTINCT task_id')
)}
if self.env.company.leave_timesheet_task_id:
timeoff_tasks_ids.add(self.env.company.leave_timesheet_task_id.id)
return Domain('id', 'in', tuple(timeoff_tasks_ids))