import json from datetime import datetime, time, timedelta from odoo import _, api, fields, models from odoo.exceptions import UserError, ValidationError class Task(models.Model): _name = 'beesdoo.shift.shift' _inherit = ['mail.thread'] _order = "start_time asc" name = fields.Char(track_visibility='always') task_template_id = fields.Many2one('beesdoo.shift.template') planning_id = fields.Many2one(related='task_template_id.planning_id', store=True) task_type_id = fields.Many2one('beesdoo.shift.type', string="Task Type") worker_id = fields.Many2one('res.partner', track_visibility='onchange', domain=[ ('eater', '=', 'worker_eater'), ('working_mode', 'in', ('regular', 'irregular')), ('state', 'not in', ('unsubscribed', 'resigning')), ]) start_time = fields.Datetime(track_visibility='always', index=True, required=True) end_time = fields.Datetime(track_visibility='always', required=True) state = fields.Selection(selection=[ ("draft","Unconfirmed"), ("open","Confirmed"), ("done","Attended"), ("absent_2","Absent - 2 compensations"), ("absent_1","Absent - 1 compensation"), ("absent_0","Absent - 0 compensation"), ("cancel","Cancelled") ], default="open", required=True, store=True, track_visibility='onchange', ) color = fields.Integer(compute="_compute_color") super_coop_id = fields.Many2one('res.users', string="Super Cooperative", domain=[('partner_id.super', '=', True)], track_visibility='onchange') is_regular = fields.Boolean(default=False, string="Regular shift") is_compensation = fields.Boolean(default=False, string="Compensation shift") replaced_id = fields.Many2one('res.partner', track_visibility='onchange', domain=[ ('eater', '=', 'worker_eater'), ('working_mode', '=', 'regular'), ('state', 'not in', ('unsubscribed', 'resigning')), ]) revert_info = fields.Text(copy=False) working_mode = fields.Selection(related='worker_id.working_mode') @api.depends("state") def _compute_color(self): color_mapping = { "draft": 0, "open": 1, "done": 5, "absent_2": 2, "absent_1": 7, "absent_0": 3, "cancel": 9, } for rec in self: rec.color = color_mapping[rec.state] def _compensation_validation(self, task): """ Raise a validation error if the fields is_regular and is_compensation are not properly set. """ if (task.is_regular == task.is_compensation or not (task.is_regular or task.is_compensation)): raise ValidationError( "You must choose between Regular Shift or " "Compensation Shift." ) @api.constrains("state") def _lock_future_task(self): if datetime.now() < self.start_time: if self.state in ["done", "absent_2", "absent_1", "absent_0"]: raise UserError(_( "Shift state of a future shift " "can't be set to 'present' or 'absent'." )) @api.constrains('is_regular', 'is_compensation') def _check_compensation(self): for task in self: if task.working_mode == 'regular': self._compensation_validation(task) @api.constrains('worker_id') def _check_worker_id(self): """ When worker_id changes we need to check whether is_regular and is_compensation are set correctly. When worker_id is set to a worker that doesn't need field is_regular and is_compensation, these two fields are set to False. """ for task in self: if task.working_mode == 'regular': self._compensation_validation(task) else: task.write({ 'is_regular': False, 'is_compensation': False, }) if task.worker_id: if task.worker_id == task.replaced_id: raise UserError("A worker cannot replace himself.") def message_auto_subscribe(self, updated_fields, values=None): self._add_follower(values) return super(Task, self).message_auto_subscribe(updated_fields, values=values) def _add_follower(self, vals): if vals.get('worker_id'): worker = self.env['res.partner'].browse(vals['worker_id']) self.message_subscribe(partner_ids=worker.ids) #TODO button to replaced someone @api.model def unsubscribe_from_today(self, worker_ids, today=None, end_date=None, now=None): """ Unsubscribe workers from *worker_ids* from all shift that start *today* and later. If *end_date* is given, unsubscribe workers from shift between *today* and *end_date*. If *now* is given workers are unsubscribed from all shifts starting *now* and later. If *now* is given, *end_date* is not taken into account. :type today: date :type end_date: date :type now: datetime """ if now: if not isinstance(now, datetime): raise UserError (_("'Now' must be a datetime.")) date_domain = [('start_time', '>', now)] else: today = today or fields.Date.today() today = datetime.combine(today, time()) date_domain = [('start_time', '>', today)] if end_date: end_date = datetime.combine(end_date,time(hour=23, minute=59, second=59)) date_domain.append(('end_time', '<=', end_date)) to_unsubscribe = self.search([('worker_id', 'in', worker_ids)] + date_domain) to_unsubscribe.write({'worker_id': False}) # Remove worker, replaced_id and regular to_unsubscribe_replace = self.search([('replaced_id', 'in', worker_ids)] + date_domain) to_unsubscribe_replace.write({'worker_id': False, 'replaced_id': False}) # If worker is Super cooperator, remove it from planning super_coop_ids = self.env['res.users'].search( [('partner_id', 'in', worker_ids), ('super', '=', True)]).ids if super_coop_ids: to_unsubscribe_super_coop = self.search( [('super_coop_id', 'in', super_coop_ids)] + date_domain) to_unsubscribe_super_coop.write({'super_coop_id': False}) @api.multi def write(self, vals): """ Overwrite write to track state change If worker is changer: Revert for the current worker Change the worker info Compute state change for the new worker """ if 'worker_id' in vals: for rec in self: if rec.worker_id.id != vals['worker_id']: rec._revert() # To satisfy the constrains on worker_id, it must be # accompanied by the change in is_regular and # is_compensation field. super(Task, rec).write({ 'worker_id': vals['worker_id'], 'is_regular': vals.get('is_regular', rec.is_regular), 'is_compensation': vals.get('is_compensation', rec.is_compensation), }) rec._update_state(rec.state) if 'state' in vals: for rec in self: if vals['state'] != rec.state: rec._update_state(vals['state']) return super(Task, self).write(vals) def _set_revert_info(self, data, status): data_new = { 'status_id': status.id, 'data' : {k: data.get(k, 0) * -1 for k in ['sr', 'sc', 'irregular_absence_counter']} } if data.get('irregular_absence_date'): data_new['data']['irregular_absence_date'] = False self.write({'revert_info': json.dumps(data_new)}) def _revert(self): if not self.revert_info: return data = json.loads(self.revert_info) self.env['cooperative.status'].browse(data['status_id']).sudo()._change_counter(data['data']) self.revert_info = False def _update_state(self, new_state): self.ensure_one() self._revert() update = int(self.env['ir.config_parameter'].sudo().get_param('always_update', False)) data = {} if not (self.worker_id or self.replaced_id) and new_state in ("done", "absent_0", "absent_1", "absent_2"): raise UserError(_("You cannot change to the status %s if no worker is defined for the shift") % new_state) if update or not (self.worker_id or self.replaced_id): return if self.worker_id.working_mode == 'regular': if not self.replaced_id: #No replacement case status = self.worker_id.cooperative_status_ids[0] else: status = self.replaced_id.cooperative_status_ids[0] if new_state == "done" and not self.is_regular: # Regular counter is always updated first if status.sr < 0: data['sr'] = 1 elif status.sc < 0: data['sc'] = 1 # Bonus shift case else: data['sr'] = 1 if new_state == "absent_2": data['sr'] = -1 data['sc'] = -1 if new_state == "absent_1": data['sr'] = -1 elif self.worker_id.working_mode == 'irregular': status = self.worker_id.cooperative_status_ids[0] if new_state == "done" or new_state == "absent_0": data['sr'] = 1 data['irregular_absence_date'] = False data['irregular_absence_counter'] = 1 if status.irregular_absence_counter < 0 else 0 if new_state == "absent_2" or new_state == "absent_1": if new_state == "absent_2": data['sr'] = -1 data['irregular_absence_date'] = self.start_time.date() data['irregular_absence_counter'] = -1 else: raise UserError(_("Working mode is not properly defined. Please check if the worker is subscribed")) status.sudo()._change_counter(data) self._set_revert_info(data, status) @api.model def _cron_send_weekly_emails(self): """ Send a summary email for all workers if they have a shift planned during the week. """ tasks = self.env["beesdoo.shift.shift"] shift_summary_mail_template = self.env.ref( "beesdoo_shift.email_template_shift_summary", False ) start_time = datetime.now() + timedelta(days=1) end_time = datetime.now() + timedelta(days=7) confirmed_tasks = tasks.search( [ ("start_time", ">", start_time), ("start_time", "<", end_time), ("worker_id", "!=", False), ("state", "=", "open"), ] ) for rec in confirmed_tasks: shift_summary_mail_template.send_mail(rec.id, True)