golem/golem_activity_registration_state/models/golem_activity_registration.py

104 lines
3.7 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
# Copyright 2017-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 Activity Registration State """
from odoo import models, fields, api, _
from odoo.exceptions import UserError
class GolemMember(models.Model):
""" GOLEM Member adaptations """
_inherit = 'golem.member'
has_draft_registrations = fields.Boolean(
'Has draft registrations ?',
compute='_compute_has_draft_reg')
@api.depends('activity_registration_ids', 'activity_registration_ids.state')
def _compute_has_draft_reg(self):
""" Check if there are draft states in member activities """
for member in self:
regis = member.activity_registration_ids
regis = regis.filtered(lambda r: r.state == 'draft')
member.has_draft_registrations = bool(len(regis))
@api.multi
def do_validate_registrations(self):
""" Validate all draft registrations """
self.ensure_one()
member = self[0]
member.activity_registration_ids.filtered(
lambda r: r.state == 'draft'
).write({'state': 'confirmed'})
class GolemActivity(models.Model):
""" GOLEM Activity adaptations """
_inherit = 'golem.activity'
@api.multi
@api.depends('activity_registration_ids')
def compute_places_used(self):
""" Computes used places """
for activity in self:
rids = activity.activity_registration_ids
activity.places_used = len(rids.filtered(lambda r: r.state == 'confirmed'))
class GolemActivityRegistration(models.Model):
""" GOLEM Activity Registration adaptations """
_inherit = 'golem.activity.registration'
state = fields.Selection([('draft', 'Draft'), ('confirmed', 'Confirmed'),
('canceled', 'Canceled')], required=True,
default='draft')
@api.multi
def state_draft(self):
""" Set registration to state draft """
self.write({'state': 'draft'})
@api.multi
def state_confirm(self):
""" Set registration to state confirmed """
if self.filtered(lambda r: r.state == 'canceled'):
uerr = _('You can not confirm a canceled registration.')
raise UserError(uerr)
self.write({'state': 'confirmed'})
@api.multi
def state_cancel(self):
""" Set registration to state canceled """
self.write({'state': 'canceled'})
@api.multi
def state_remove(self):
""" Remove registrations, only if canceled """
if self.filtered(lambda r: r.state != 'canceled'):
uerr = _('You can not confirm a canceled registration.')
raise UserError(uerr)
self.unlink()
@api.multi
def write(self, values):
""" Recomputes values linked to registrations when state change """
res = super(GolemActivityRegistration, self).write(values)
if values.get('state'):
for registration in self:
registration.activity_id.compute_places_used()
return res