base_time_window.py

# -*- coding: utf-8 -*-
# Copyright 2020 ACSONE SA/NV
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).

import math
from datetime import time

from psycopg2.extensions import AsIs

from odoo import _, api, fields, models
from odoo.exceptions import ValidationError


class TimeWindow(models.AbstractModel):

    _name = "time.window.mixin"
    _description = "Time Window"
    _order = "start"

    # TODO patch api.constrains with field here?
    _overlap_check_field = False

    start = fields.Float("From", required=True)
    end = fields.Float("To", required=True)
    weekday_ids = fields.Many2many(
        comodel_name="time.weekday", required=True
    )

    @api.constrains("start", "end", "weekday_ids")
    def check_window_no_overlaps(self):
        weekdays_field = self._fields["weekday_ids"]
        for record in self:
            if record.start > record.end:
                raise ValidationError(
                    _("%s must be > %s")
                    % (
                        self.float_to_time_repr(record.end),
                        self.float_to_time_repr(record.start),
                    )
                )
            # here we use a plain SQL query to benefit of the numrange
            # function available in PostgresSQL
            # (http://www.postgresql.org/docs/current/static/rangetypes.html)
            SQL = """
                SELECT
                    id
                FROM
                    %(table)s w
                    join %(relation)s as d
                    on d.%(relation_window_fkey)s = w.id
                WHERE
                    NUMRANGE(w.start::numeric, w.end::numeric) &&
                        NUMRANGE(%(start)s::numeric, %(end)s::numeric)
                    AND w.id != %(window_id)s
                    AND d.%(relation_week_day_fkey)s in %(weekday_ids)s
                    AND w.%(check_field)s = %(check_field_id)s;"""
            self.env.cr.execute(
                SQL,
                dict(
                    table=AsIs(self._table),
                    relation=AsIs(weekdays_field.relation),
                    relation_window_fkey=AsIs(weekdays_field.column1),
                    relation_week_day_fkey=AsIs(weekdays_field.column2),
                    start=record.start,
                    end=record.end,
                    window_id=record.id,
                    weekday_ids=tuple(record.weekday_ids.ids),
                    check_field=AsIs(self._overlap_check_field),
                    check_field_id=record[self._overlap_check_field].id,
                ),
            )
            res = self.env.cr.fetchall()
            if res:
                other = self.browse(res[0][0])
                raise ValidationError(
                    _("%s overlaps %s")
                    % (record.display_name, other.display_name)
                )

    @api.depends("start", "end", "weekday_ids")
    def _compute_display_name(self):
        for record in self:
            record.display_name = _("{days}: From {start} to {end}").format(
                days=", ".join(record.weekday_ids.mapped("display_name")),
                start=self.float_to_time_repr(record.start),
                end=self.float_to_time_repr(record.end),
            )

    @api.model
    def _get_hour_min_from_value(self, value):
        hour = math.floor(value)
        minute = round((value % 1) * 60)
        if minute == 60:
            minute = 0
            hour += 1
        return hour, minute

    @api.model
    def float_to_time_repr(self, value):
        pattern = "%02d:%02d"
        hour, minute = self._get_hour_min_from_value(value)
        return pattern % (hour, minute)

    @api.model
    def float_to_time(self, value):
        hour, minute = self._get_hour_min_from_value(value)
        return time(hour=hour, minute=minute)

    def get_start_time(self):
        self.float_to_time(self.start)

    def get_end_time(self):
        self.float_to_time(self.end)