GOLEM/golem_resource/models/golem_resource_reservation.py

255 lines
12 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2018 Youssef El Ouahby <youssef@yaltik.com>
# Copyright 2018 Fabien Bourgeois <fabien@yaltik.com>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
""" GOLEM Resource Reservation """
from math import modf
from datetime import timedelta
from odoo import models, fields, api, _
from odoo.exceptions import ValidationError
import logging
_logger = logging.getLogger(__name__)
class GolemResourceReservation(models.Model):
""" GOLEM Resource Reservation Model """
_name = 'golem.resource.reservation'
_description = 'GOLEM Reservation Model'
_inherit = 'mail.thread'
_order = 'date_start desc'
name = fields.Char(compute='_compute_name', store=True)
# TODO: handle multiple days reservation
date_start = fields.Datetime('Start date',required=True,
index=True, readonly=True,
states={'draft': [('readonly', False)]})
date_stop = fields.Datetime('Stop date', required=True,
index=True, readonly=True,
states={'draft': [('readonly', False)]})
#date = fields.Date(required=True, index=True, readonly=True,
# states={'draft': [('readonly', False)]})
#hour_start = fields.Float('Start hour', required=True, readonly=True,
# states={'draft': [('readonly', False)]})
#hour_stop = fields.Float('Stop hour', required=True, readonly=True,
# states={'draft': [('readonly', False)]})
#date_start = fields.Datetime(compute='_compute_date_start', store=True, index=True)
#date_stop = fields.Datetime(compute='_compute_date_stop', store=True, index=True)
resource_id = fields.Many2one('golem.resource', required=True, index=True,
string='Resource', readonly=True,
track_visibility='onchange',
states={'draft': [('readonly', False)]})
resource_avaibility_start = fields.Date(related='resource_id.avaibility_start')
resource_avaibility_stop = fields.Date(related='resource_id.avaibility_stop')
resource_timetable_ids = fields.One2many(related='resource_id.timetable_ids')
note = fields.Text(help='Notes, optional subject for the reservation, reason')
user_id = fields.Many2one('res.users', required=True, index=True, readonly=True,
string='User', default=lambda self: self.env.user,
states={'draft': [('readonly', False)]})
partner_id = fields.Many2one('res.partner', string='On behalf of',
required=True, index=True, readonly=True,
track_visibility='onchange',
states={'draft': [('readonly', False)]})
state = fields.Selection([('canceled', 'Canceled'),
('draft', 'Draft'),
('confirmed', 'Confirmed'),
('validated', 'Validated'),
('rejected', 'Rejected')],
default='draft', track_visibility='onchange')
rejection_reason = fields.Text(readonly=True, track_visibility='onchange')
resource_reservation_count = fields.Integer(compute='_reservation_count')
@api.depends('resource_id', 'date_start')
def _compute_name(self):
""" Computes reservation name """
for reservation in self:
reservation.name = u'{}/{}'.format(reservation.resource_id.name,
reservation.date_start)
#@api.depends('date', 'hour_start')
#def _compute_date_start(self):
# """ Computes Date start """
# for reservation in self:
# minute_start, hour_start = modf(reservation.hour_start)
# hour_start = int(hour_start)
# minute_start = int(round(minute_start * 60))
# reservation.date_start = u'{} {}:{}:00'.format(reservation.date,
# hour_start, minute_start)
#@api.depends('date', 'hour_stop')
#def _compute_date_stop(self):
# """ Computes Date stop """
# for reservation in self:
# minute_stop, hour_stop = modf(reservation.hour_stop)
# hour_stop = int(hour_stop)
# minute_stop = int(round(minute_stop * 60))
# reservation.date_stop = u'{} {}:{}:00'.format(reservation.date,
# hour_stop, minute_stop)
@api.onchange('date_start')
def onchange_date_start(self):
""" Propose automatically stop date after start date had been filled """
for reservation in self:
if reservation.date_start:
start = fields.Datetime.from_string(reservation.date_start)
duration = timedelta(hours=1)
reservation.date_stop = start + duration
@api.constrains('date_start', 'date_stop')
def _check_date_consistency(self):
""" Checks date consistency """
for reservation in self:
if reservation.date_stop <= reservation.date_start:
raise ValidationError(_('Stop date should be after start date'))
@api.multi
def state_draft(self):
""" Status to draft """
self.write({'state': 'draft'})
@api.multi
def state_confirm(self):
""" Confirms reservation, or validates it if not workflow is involved """
for reservation in self:
# Needed, for constraint checking
reservation.state = 'confirmed'
if not reservation.resource_id.validation_required:
reservation.state = 'validated'
@api.multi
def state_canceled(self):
""" Status to cancel """
self.write({'state': 'canceled'})
@api.multi
def state_validated(self):
""" Status to validated """
self.write({'state': 'validated'})
@api.multi
def state_rejected(self):
""" Wizard call for reservation reject """
self.ensure_one()
reservation_id = self[0]
return {'name' : _('Please enter the rejection reason'),
'type' : 'ir.actions.act_window',
'res_model' : 'golem.reservation.rejection.wizard',
'context': {'default_reservation_id': reservation_id.id},
'view_mode': 'form',
'target': 'new'}
@api.constrains('state')
def check_access(self):
""" Checks access when state is updated """
reservation = self[0]
if reservation.state in ('rejected', 'validated'):
if not self.env.user.has_group('golem_base.group_golem_manager'):
verr = _('You do not have permissions to validate or reject a reservation.')
raise ValidationError(verr)
"""
@api.constrains('state')
def check_confirmed(self):"""
""" Check date coherence on reservation confirmation """
"""" for reservation in self:
if reservation.state == 'confirmed':
# Check is reservation is not taking place out of the resource avaibility period
if reservation.date < reservation.resource_id.avaibility_start or \
reservation.date > reservation.resource_id.avaibility_stop:
verr = _('Not allowed, the resource is not available in '
'this period, please choose another périod before '
'confirming')
raise ValidationError(verr)
#check if the resource hasn't a total availibility
if not reservation.resource_id.availibility_24_7:
# Check if reservation is not taking place out the avaibility timetables
is_day_allowed = False
for timetable in reservation.resource_id.timetable_ids:
# Check for the time according to resource timetable avaibility
date = fields.Datetime.from_string(reservation.date)
if int(timetable.weekday) == date.weekday():
is_day_allowed = True
if reservation.hour_start < timetable.time_start or \
reservation.hour_stop > timetable.time_stop:
verr = _('Not allowed, the resource is not available '
'during this period, please choose another '
'time before confirming.')
raise ValidationError(verr)
if not is_day_allowed:
verr = _('Not allowed, the resource is not available '
'this day. Please choose another date.')
raise ValidationError(verr)
# Check if the resource is already taken during this period
# PERF : check the date, not iterate over all reservations
domain = [('resource_id', '=', reservation.resource_id.id),
('date', '=', reservation.date),
('state', 'in', ('confirmed', 'validated')),
('id', '!=', reservation.id)]
reservations = self.env['golem.resource.reservation'].search(domain)
for other_res in reservations:
if (other_res.hour_start < reservation.hour_start < other_res.hour_stop) or \
(other_res.hour_start < reservation.hour_stop < other_res.hour_stop):
verr = _('Not allowed, the resource is already taken '
'during this period : from {} to {} this day, '
'please choose another périod before confirming.')
raise ValidationError(verr.format(other_res.date_start,
other_res.date_stop))
"""
@api.multi
def reservation_calendar(self):
""" current resource reservation calendar """
self.ensure_one()
calendar_view = {
'name': ('Resource Reservation list'),
'view_mode': 'calendar',
'res_model': 'golem.resource.reservation',
'view_id': False,
'domain': [('resource_id', '=', self.resource_id.id)],
'type': 'ir.actions.act_window',
'target':'current'
}
return calendar_view
@api.multi
def reserveration_list(self):
""" current resource reservation list """
self.ensure_one()
tree_view = {
'name': ('Resource Reservation list'),
'view_mode': 'tree',
'res_model': 'golem.resource.reservation',
'view_id': False,
'domain': [('resource_id', '=', self.resource_id.id)],
'type': 'ir.actions.act_window',
'target':'current'
}
return tree_view
@api.multi
@api.depends('resource_id')
def _reservation_count(self):
for reservation in self:
reservation.resource_reservation_count = reservation.search_count([
('resource_id', '=', reservation.resource_id.id)])