golem/golem_season/models/golem_season.py

112 lines
4.9 KiB
Python

# -*- coding: utf-8 -*-
# Copyright 2016-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 Season management """
from odoo import models, fields, api, _
from odoo.exceptions import ValidationError
class GolemSeason(models.Model):
""" GOLEM Season """
_name = 'golem.season'
_description = 'GOLEM Season'
_order = 'date_start desc'
_sql_constraints = [('golem_season_name_uniq', 'UNIQUE (name)',
_('This season name has already been used.'))]
name = fields.Char('Season name', copy=False, required=True)
membership_ids = fields.One2many('product.template', 'membership_season_id',
string='Membership types',
domain=[('membership', '=', True)])
member_counter = fields.Integer('Counter for member number generation',
readonly=True, default=1)
date_start = fields.Date('Period start')
date_end = fields.Date('Period end')
is_default = fields.Boolean('Default season for views?', readonly=True,
copy=False)
@api.onchange('membership_ids')
def _onchange_season_dates(self):
""" Sets defaults dates according to membership type """
for season in self:
if season.membership_ids and not season.date_start:
season.update({
'date_start': season.membership_ids[0].membership_date_from,
'date_end': season.membership_ids[0].membership_date_to
})
@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 or season.date_end:
if season.date_start and not season.date_end:
raise ValidationError(_('The date end is required'))
elif season.date_end and not season.date_start:
raise ValidationError(_('The date start is required'))
if season.date_start > season.date_end:
raise ValidationError(_('Start of the period cannot be '
'after end of the period.'))
seasons = self.env['golem.season'].search([])
for eachs in seasons:
if eachs.date_start < season.date_start < eachs.date_end:
msg = _(u'Start of the period is in range of an '
'existing period.')
raise ValidationError(msg)
if eachs.date_start < season.date_end < eachs.date_end:
msg = _(u'End of the period is in range of an '
'existing period.')
raise ValidationError(msg)
if season.date_start < eachs.date_start < season.date_end:
msg = _(u'Current period cannot be included into '
'another existing period.')
raise ValidationError(msg)
@api.multi
def do_default_season(self):
""" is_default on and ensure that only one is_default is active """
self.ensure_one()
old_default_season = self.search([('is_default', '=', True)])
if old_default_season:
old_default_season.is_default = False
self.is_default = True
@api.multi
def copy(self, default=None):
""" Handles copy name """
self.ensure_one()
default = dict(default or {}, name=_('%s (copy)') % self.name)
return super(GolemSeason, self).copy(default)
@api.model
def create(self, values):
""" If the season if the first one created, it must be by default """
if self.search_count([]) == 0:
values['is_default'] = True
return super(GolemSeason, self).create(values)
@api.multi
def unlink(self):
""" Forbids default season removal """
for season in self:
if season.is_default:
emsg = _('You can\'t delete the default season')
raise ValidationError(emsg)
else:
return super(GolemSeason, self).unlink()