cooperative-market/beesdoo_shift/models/task.py

# -*- coding: utf-8 -*-
from openerp import models, fields, api, _
from openerp.exceptions import UserError
import json

class TaskStage(models.Model):
    _name = 'beesdoo.shift.stage'
    _order = 'sequence asc'

    name = fields.Char()
    sequence = fields.Integer()
    color = fields.Integer()
    code = fields.Char(readonly=True)

    @api.multi
    def unlink(self):
        raise UserError(_("You Cannot delete Task Stage"))


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'))])
    start_time = fields.Datetime(track_visibility='always', index=True)
    end_time = fields.Datetime(track_visibility='always')
    stage_id = fields.Many2one('beesdoo.shift.stage', required=True, track_visibility='onchange', default=lambda self: self.env.ref('beesdoo_shift.open'))
    super_coop_id = fields.Many2one('res.users', string="Super Cooperative", domain=[('partner_id.super', '=', True)], track_visibility='onchange')
    color = fields.Integer(related="stage_id.color", readonly=True)
    is_regular = fields.Boolean(default=False)
    replaced_id = fields.Many2one('res.partner', track_visibility='onchange', domain=[('eater', '=', 'worker_eater')])
    revert_info = fields.Text()

    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)

    @api.model
    def _read_group_stage_id(self, ids, domain, read_group_order=None, access_rights_uid=None):
        res  = self.env['beesdoo.shift.stage'].search([]).name_get()
        fold = dict.fromkeys([r[0] for r in res], False)
        return res, fold

    _group_by_full = {
        'stage_id': _read_group_stage_id,
    }

    #TODO button to replaced someone
    @api.model
    def unsubscribe_from_today(self, worker_ids, today=None, end_date=None):
        today = today or fields.Date.today()
        today += ' 00:00:00'
        if end_date:
            end_date += ' 23:59:59'
        date_domain = [('worker_id', 'in', worker_ids), ('start_time', '>=', today)]
        if end_date:
            date_domain.append(('end_time', '<=', end_date))
        to_unsubscribe = self.search([('worker_id', 'in', worker_ids)] + date_domain)

        to_unsubscribe.write({'worker_id': False, 'is_regular': False})
        #What about replacement ?
        #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, 'is_regular': False, 'replaced_id': False})

    @api.multi
    def write(self, vals):
        """
            Overwrite write to track stage change
        """
        if 'stage_id' in vals:
            for rec in self:
                if vals['stage_id'] != rec.stage_id.id:
                    rec._update_stage(rec.stage_id.id, vals['stage_id'])
        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
        try:
            data = json.loads(self.revert_info)
            self.env['cooperative.status'].browse(data['status_id'])._change_counter(data['data'])
            self.revert_info = False
        except:
            pass

    def _update_stage(self, old_stage, new_stage):
        self.ensure_one()
        self._revert()
        update = int(self.env['ir.config_parameter'].get_param('always_update', False))
        
        new_stage = self.env['beesdoo.shift.stage'].browse(new_stage)
        data = {}
        DONE = self.env.ref('beesdoo_shift.done')
        ABSENT = self.env.ref('beesdoo_shift.absent')
        EXCUSED = self.env.ref('beesdoo_shift.excused')
        NECESSITY = self.env.ref('beesdoo_shift.excused_necessity')
        
        if not (self.worker_id or self.replaced_id) and new_stage in (DONE, ABSENT, EXCUSED, NECESSITY):
            raise UserError(_("You cannot change to the status %s if the is no worker defined on the shift") % new_stage.name)
        
        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_stage == DONE and not self.is_regular:
                if status.sr < 0:
                    data['sr'] = 1
                elif status.sc < 0:
                    data['sc'] = 1
                else:
                    data['sr'] = 1
    
            if new_stage == ABSENT and not self.replaced_id:
                data['sr'] = - 1
                if status.sr <= 0:
                    data['sc'] = -1
            if new_stage == ABSENT and self.replaced_id:
                data['sr'] = -1
    
            if new_stage == EXCUSED:
                data['sr'] = -1

        elif self.worker_id.working_mode == 'irregular':
            status = self.worker_id.cooperative_status_ids[0]
            if new_stage == DONE or new_stage == NECESSITY:
                data['sr'] = 1
                data['irregular_absence_date'] = False
                data['irregular_absence_counter'] = 1 if status.irregular_absence_counter < 0 else 0
            if new_stage == ABSENT or new_stage == EXCUSED:
                data['sr'] = -2
                data['irregular_absence_date'] = self.start_time[:10]
                data['irregular_absence_counter'] = -1

        status.sudo()._change_counter(data)
        self._set_revert_info(data, status)