# Part of Odoo. See LICENSE file for full copyright and licensing details. from datetime import datetime, time, timedelta from odoo import _, api, fields, models class ResourceResource(models.Model): _inherit = 'resource.resource' avatar_128 = fields.Image(compute='_compute_avatar_128', export_string_translation=False) role_ids = fields.Many2many('planning.role', 'resource_resource_planning_role_rel', 'resource_resource_id', 'planning_role_id', 'Roles', compute='_compute_role_ids', store=True, readonly=False) default_role_id = fields.Many2one('planning.role', string="Default Role", compute='_compute_default_role_id', groups='hr.group_hr_user', store=True, readonly=False, help="Role that will be selected by default when creating a shift for this resource.\n" "This role will also have precedence over the other roles of the resource when planning shifts.") @api.depends('role_ids') def _compute_default_role_id(self): self.env.remove_to_compute(self._fields['role_ids'], self) for resource in self: if resource.default_role_id not in resource.role_ids: resource.default_role_id = resource.role_ids[:1] @api.depends('default_role_id') def _compute_role_ids(self): self.env.remove_to_compute(self._fields['default_role_id'], self) resources_wo_default_role_ids = [] for resource in self: if resource.default_role_id: resource.role_ids |= resource.default_role_id else: resources_wo_default_role_ids.append(resource.id) self.browse(resources_wo_default_role_ids)._compute_default_role_id() def get_formview_id(self, access_uid=None): if self.env.context.get('from_planning'): return self.env.ref('planning.resource_resource_with_employee_form_view_inherit', raise_if_not_found=False).id return super().get_formview_id(access_uid) @api.model_create_multi def create(self, vals_list): resources = super().create(vals_list) if self.env.context.get('from_planning'): create_vals = [] for resource in resources.filtered(lambda r: r.resource_type == 'user'): create_vals.append({ 'name': resource.name, 'resource_id': resource.id, }) self.env['hr.employee'].sudo().with_context(from_planning=False).create(create_vals) return resources @api.depends('employee_id') @api.depends_context('show_job_title') def _compute_display_name(self): if not self.env.context.get('show_job_title'): return super()._compute_display_name() for resource in self: resource_default_role_id = resource.sudo().default_role_id if resource.resource_type == 'material' and resource_default_role_id: resource.display_name = _("%(resource_name)s (%(role)s)", resource_name=resource.name, role=resource_default_role_id.name) else: resource.display_name = resource.employee_id.display_name if resource.employee_id else resource.name def action_archive(self): res = super().action_archive() departure_date = datetime.combine(fields.Date.context_today(self) + timedelta(days=1), time.min) planning_slots = self.env['planning.slot'].sudo().search([ ('resource_id', 'in', self.ids), ('resource_type', '=', 'material'), ('end_datetime', '>=', departure_date), ]) planning_slots._manage_archived_resources(departure_date) return res @api.onchange('company_id') def _onchange_company_id(self): if self.resource_type != 'material' or not self.env.context.get('from_planning'): super()._onchange_company_id() def get_avatar_card_data(self, fields): if self.env.user.has_group('hr.group_hr_user'): fields.append('default_role_id') result = super().get_avatar_card_data(fields) if not result: return [] roles_ids = result[0].get('role_ids') if roles_ids: result[0]['role_ids'] = self.env['planning.role'].search_read( domain=[('id', 'in', roles_ids)], fields=['name', 'color'], ) return result