You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
294 lines
12 KiB
294 lines
12 KiB
# -*- coding: utf-8 -*-
|
|
import json
|
|
from datetime import datetime, timedelta
|
|
|
|
from openerp import _, api, fields, models
|
|
from openerp.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')
|
|
# TODO: Maybe is_regular and is_compensation must be merged in a
|
|
# selection field as they are mutually exclusive.
|
|
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):
|
|
start_time_dt = fields.Datetime.from_string(self.start_time)
|
|
if datetime.now() < start_time_dt:
|
|
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: fields.Date
|
|
:type end_date: fields.Date
|
|
:type now: fields.Datetime
|
|
"""
|
|
if now:
|
|
if len(now) != 19:
|
|
raise UserError (_("'Now' must be a datetime."))
|
|
date_domain = [('start_time', '>', now)]
|
|
else:
|
|
today = today or fields.Date.today()
|
|
today += ' 00:00:00'
|
|
date_domain = [('start_time', '>', today)]
|
|
if end_date:
|
|
end_date += ' 23:59: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'].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[:10]
|
|
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.strftime("%Y-%m-%d 00:00:01")),
|
|
("start_time", "<", end_time.strftime("%Y-%m-%d 23:59:59")),
|
|
("worker_id", "!=", False),
|
|
("state", "=", "open"),
|
|
]
|
|
)
|
|
|
|
for rec in confirmed_tasks:
|
|
shift_summary_mail_template.send_mail(rec.id, True)
|