cooperative-market/beesdoo_shift/models/planning.py

# -*- coding: utf-8 -*-

from openerp import models, fields, api, _
from openerp.exceptions import UserError

from pytz import timezone, UTC
import math
from datetime import datetime, timedelta


def float_to_time(f):
    decimal, integer = math.modf(f)
    return "%s:%s" % (str(int(integer)).zfill(2), str(int(round(decimal * 60))).zfill(2))

def floatime_to_hour_minute(f):
    decimal, integer = math.modf(f)
    return int(integer), int(round(decimal * 60))

def get_first_day_of_week():
    today = datetime.now()
    return (datetime.now() - timedelta(days=today.weekday())).strftime("%Y-%m-%d")

class TaskType(models.Model):
    _name = 'beesdoo.shift.type'

    name = fields.Char()
    description = fields.Text()
    active = fields.Boolean(default=True)

class DayNumber(models.Model):
    _name = 'beesdoo.shift.daynumber'

    _order = 'number asc'

    name = fields.Char()
    number = fields.Integer("Day Number", help="From 1 to N, When you will instanciate your planning, Day 1 will be the start date of the instance, Day 2 the second, etc...")
    active = fields.Boolean(default=True)

class Planning(models.Model):
    _name = 'beesdoo.shift.planning'

    _order = 'sequence asc'

    sequence = fields.Integer()
    name = fields.Char()
    task_template_ids = fields.One2many('beesdoo.shift.template', 'planning_id')

    @api.model
    def _get_next_planning(self, sequence):
        next_planning = self.search([('sequence', '>', sequence)])
        if not next_planning:
            return self.search([])[0]
        return next_planning[0]

    @api.multi
    def _get_next_planning_date(self, date):
        self.ensure_one()
        nb_of_day = max(self.task_template_ids.mapped('day_nb_id.number'))
        return fields.Date.to_string(fields.Date.from_string(date) + timedelta(days=nb_of_day))

    @api.model
    def _generate_next_planning(self):
        config = self.env['ir.config_parameter']
        last_seq = int(config.get_param('last_planning_seq', 0))
        date = config.get_param('next_planning_date', 0)

        planning = self._get_next_planning(last_seq)
        planning = planning.with_context(visualize_date=date)
        planning.task_template_ids._generate_task_day()

        next_date = planning._get_next_planning_date(date)
        config.set_param('last_planning_seq', planning.sequence)
        config.set_param('next_planning_date', next_date)

class TaskTemplate(models.Model):
    _name = 'beesdoo.shift.template'

    _order = 'start_time'

    name = fields.Char(required=True)
    planning_id = fields.Many2one('beesdoo.shift.planning', required=True)
    day_nb_id = fields.Many2one('beesdoo.shift.daynumber', string='Day', required=True)
    task_type_id = fields.Many2one('beesdoo.shift.type', string="Type")
    attendance_sheet_id = fields.Many2one('beesdoo.shift.sheet', string="Attendance Sheet")
    start_time = fields.Float(required=True)
    end_time = fields.Float(required=True)
    super_coop_id = fields.Many2one('res.users', string="Super Cooperative", domain=[('partner_id.super', '=', True)])

    duration = fields.Float(help="Duration in Hour")
    worker_nb = fields.Integer(string="Number of worker", help="Max number of worker for this task", default=1)
    worker_ids = fields.Many2many('res.partner', string="Recurrent worker assigned", domain=[('eater', '=', 'worker_eater'), ('working_mode', '=', 'regular')])
    remaining_worker = fields.Integer(compute="_get_remaining", store=True, string="Remaining Place")
    active = fields.Boolean(default=True)
    #For Kanban View Only
    color = fields.Integer('Color Index')
    worker_name = fields.Char(compute="_get_worker_name")
    #For calendar View
    start_date = fields.Datetime(compute="_get_fake_date", search="_dummy_search")
    end_date = fields.Datetime(compute="_get_fake_date", search="_dummy_search")

    def _get_utc_date(self, day, hour, minute):
        #Don't catch error since the error should be raise on the log as an error
        #because generate time with UTC timezone is worse than not generate them
        context_tz = timezone(self._context.get('tz') or self.env.user.tz)
        day_time = day.replace(hour=hour, minute=minute)
        day_local_time = context_tz.localize(day_time)
        day_utc_time = day_local_time.astimezone(UTC)
        return day_utc_time


    @api.depends('start_time', 'end_time')
    def _get_fake_date(self):
        today = self._context.get('visualize_date', get_first_day_of_week())
        today = datetime.strptime(today, '%Y-%m-%d')
        for rec in self:
            # Find the day of this task template 'rec'.
            day = today + timedelta(days=rec.day_nb_id.number - 1)
            # Compute the beginning and ending time according to the
            # context timezone.
            h_begin, m_begin = floatime_to_hour_minute(rec.start_time)
            h_end, m_end = floatime_to_hour_minute(rec.end_time)
            rec.start_date = self._get_utc_date(day, h_begin, m_begin)
            rec.end_date = self._get_utc_date(day, h_end, m_end)

    def _dummy_search(self, operator, value):
        return []

    @api.depends('worker_ids', 'worker_nb')
    def _get_remaining(self):
        for rec in self:
            rec.remaining_worker =  rec.worker_nb - len(rec.worker_ids)

    @api.depends("worker_ids")
    def _get_worker_name(self):
        for rec in self:
            rec.worker_name = ','.join(rec.worker_ids.mapped('display_name'))

    @api.constrains('worker_nb', 'worker_ids')
    def _nb_worker_max(self):
        for rec in self:
            if len(rec.worker_ids) > rec.worker_nb:
                raise UserError(_('You cannot assign more workers than the maximal number defined on template.'))


    @api.onchange('start_time', 'end_time')
    def _get_duration(self):
        if self.start_time and self.end_time:
            self.duration = self.end_time - self.start_time

    @api.onchange('duration')
    def _set_duration(self):
        if self.start_time:
            self.end_time = self.start_time +self.duration

    def _generate_task_day(self):
        tasks = self.env['beesdoo.shift.shift']
        for rec in self:
            for i in xrange(0, rec.worker_nb):
                worker_id = rec.worker_ids[i] if len(rec.worker_ids) > i else False
                #remove worker in holiday and temporary exempted
                if worker_id and worker_id.cooperative_status_ids:
                    status = worker_id.cooperative_status_ids[0]
                    if status.holiday_start_time and status.holiday_end_time and \
                         status.holiday_start_time <= rec.start_date[:10] and status.holiday_end_time >= rec.end_date[:10]:
                        worker_id = False
                    if status.temporary_exempt_start_date and status.temporary_exempt_end_date and \
                         status.temporary_exempt_start_date <= rec.start_date[:10] and status.temporary_exempt_end_date >= rec.end_date[:10]:
                        worker_id = False
                tasks |= tasks.create({
                    'name' :  "[%s] %s %s (%s - %s) [%s]" % (
                        rec.start_date[:10],
                        rec.planning_id.name,
                        rec.day_nb_id.name,
                        float_to_time(rec.start_time),
                        float_to_time(rec.end_time),
                        i,
                    ),
                    'task_template_id' : rec.id,
                    'task_type_id' : rec.task_type_id.id,
                    'super_coop_id': rec.super_coop_id.id,
                    'worker_id' : worker_id and worker_id.id or False,
                    'is_regular': True if worker_id else False,
                    'start_time' : rec.start_date,
                    'end_time' :  rec.end_date,
                    'state': 'open',
                })

        return tasks

    @api.onchange('worker_ids')
    def check_for_multiple_shifts(self):
        original_ids = {worker.id for worker in self._origin.worker_ids}

        warnings = []
        for worker in self.worker_ids:
            if worker.id not in original_ids:
                shifts = [shift.name for shift in worker.subscribed_shift_ids if shift.id != self.id]
                if shifts:
                    warnings.append(
                        worker.name + _(' is already assigned to ') + ", ".join(shifts))

        if warnings:
            return {
                'warning': {
                    'title': _("Warning"),
                    'message': "\n".join(warnings)
                }
            }