GOLEM/golem_season/models/golem_season.py

64 lines
2.7 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2016 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/>.
from openerp import models, fields, api, _
class GolemSeason(models.Model):
""" GOLEM Season """
_name = 'golem.season'
_description = 'GOLEM Season'
_order = 'date_start desc'
name = fields.Char('Season name')
date_start = fields.Date('Period start')
date_end = fields.Date('Period end')
@api.constrains('date_start', 'date_end')
def _check_period(self):
""" Check if end date if after start date and if there is no conflict
with existing periods """
for season in self:
if season.date_start > season.date_end:
raise models.ValidationError(_('Start of the period cannot be '
'after end of the period.'))
else:
seasons = self.env['golem.season'].search([])
for s in seasons:
if s.date_start < season.date_start < s.date_end:
msg = _('Start of the period is in range of an '
'existing period {}'.format(s.name))
raise models.ValidationError(msg)
if s.date_start < season.date_end < s.date_end:
msg = _('End of the period is in range of an '
'existing period {}'.format(s.name))
raise models.ValidationError(msg)
if season.date_start < s.date_start < season.date_end:
msg = _('Period {} cannot be included into current '
'period'.format(s.name))
raise models.ValidationError(msg)
@api.multi
def write(self, values):
""" Extends write to recomputes all current members in case of date
changes """
date_start = values.get('date_start')
date_end = values.get('date_end')
if date_start or date_end:
self.env['golem.member']._compute_is_current()
return super(GolemSeason, self).write(values)