260 lines
13 KiB
Python
260 lines
13 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
|
#
|
|
# 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.addons.event.event import event_event as Event
|
|
from openerp.osv import fields, osv
|
|
from openerp.tools.translate import _
|
|
|
|
class product_template(osv.osv):
|
|
_inherit = 'product.template'
|
|
_columns = {
|
|
'event_ok': fields.boolean('Event Subscription', help='Determine if a product needs to create automatically an event registration at the confirmation of a sales order line.'),
|
|
'event_type_id': fields.many2one('event.type', 'Type of Event', help='Select event types so when we use this product in sales order lines, it will filter events of this type only.'),
|
|
}
|
|
|
|
def onchange_event_ok(self, cr, uid, ids, type, event_ok, context=None):
|
|
if event_ok:
|
|
return {'value': {'type': 'service'}}
|
|
return {}
|
|
|
|
class product(osv.osv):
|
|
_inherit = 'product.product'
|
|
_columns = {
|
|
'event_ticket_ids': fields.one2many('event.event.ticket', 'product_id', 'Event Tickets'),
|
|
}
|
|
|
|
def onchange_event_ok(self, cr, uid, ids, type, event_ok, context=None):
|
|
# cannot directly forward to product.template as the ids are theoretically different
|
|
if event_ok:
|
|
return {'value': {'type': 'service'}}
|
|
return {}
|
|
|
|
|
|
class sale_order_line(osv.osv):
|
|
_inherit = 'sale.order.line'
|
|
_columns = {
|
|
'event_id': fields.many2one('event.event', 'Event',
|
|
help="Choose an event and it will automatically create a registration for this event."),
|
|
'event_ticket_id': fields.many2one('event.event.ticket', 'Event Ticket',
|
|
help="Choose an event ticket and it will automatically create a registration for this event ticket."),
|
|
#those 2 fields are used for dynamic domains and filled by onchange
|
|
'event_type_id': fields.related('product_id','event_type_id', type='many2one', relation="event.type", string="Event Type"),
|
|
'event_ok': fields.related('product_id', 'event_ok', string='event_ok', type='boolean'),
|
|
}
|
|
|
|
def product_id_change(self, cr, uid, ids,
|
|
pricelist,
|
|
product,
|
|
qty=0,
|
|
uom=False,
|
|
qty_uos=0,
|
|
uos=False,
|
|
name='',
|
|
partner_id=False,
|
|
lang=False,
|
|
update_tax=True,
|
|
date_order=False,
|
|
packaging=False,
|
|
fiscal_position=False,
|
|
flag=False, context=None):
|
|
"""
|
|
check product if event type
|
|
"""
|
|
res = super(sale_order_line,self).product_id_change(cr, uid, ids, pricelist, product, qty=qty, uom=uom, qty_uos=qty_uos, uos=uos, name=name, partner_id=partner_id, lang=lang, update_tax=update_tax, date_order=date_order, packaging=packaging, fiscal_position=fiscal_position, flag=flag, context=context)
|
|
if product:
|
|
product_res = self.pool.get('product.product').browse(cr, uid, product, context=context)
|
|
if product_res.event_ok:
|
|
res['value'].update({'event_type_id': product_res.event_type_id.id, 'event_ok':product_res.event_ok})
|
|
return res
|
|
|
|
def button_confirm(self, cr, uid, ids, context=None):
|
|
'''
|
|
create registration with sales order
|
|
'''
|
|
if context is None:
|
|
context = {}
|
|
registration_obj = self.pool.get('event.registration')
|
|
for order_line in self.browse(cr, uid, ids, context=context):
|
|
if order_line.event_id:
|
|
dic = {
|
|
'name': order_line.order_id.partner_invoice_id.name,
|
|
'partner_id': order_line.order_id.partner_id.id,
|
|
'nb_register': int(order_line.product_uom_qty),
|
|
'email': order_line.order_id.partner_id.email,
|
|
'phone': order_line.order_id.partner_id.phone,
|
|
'origin': order_line.order_id.name,
|
|
'event_id': order_line.event_id.id,
|
|
'event_ticket_id': order_line.event_ticket_id and order_line.event_ticket_id.id or None,
|
|
}
|
|
|
|
if order_line.event_ticket_id:
|
|
message = _("The registration has been created for event <i>%s</i> with the ticket <i>%s</i> from the Sale Order %s. ") % (order_line.event_id.name, order_line.event_ticket_id.name, order_line.order_id.name)
|
|
else:
|
|
message = _("The registration has been created for event <i>%s</i> from the Sale Order %s. ") % (order_line.event_id.name, order_line.order_id.name)
|
|
|
|
context.update({'mail_create_nolog': True})
|
|
registration_id = registration_obj.create(cr, uid, dic, context=context)
|
|
registration_obj.message_post(cr, uid, [registration_id], body=message, context=context)
|
|
return super(sale_order_line, self).button_confirm(cr, uid, ids, context=context)
|
|
|
|
def onchange_event_ticket_id(self, cr, uid, ids, event_ticket_id=False, context=None):
|
|
price = event_ticket_id and self.pool.get("event.event.ticket").browse(cr, uid, event_ticket_id, context=context).price or False
|
|
return {'value': {'price_unit': price}}
|
|
|
|
|
|
class event_event(osv.osv):
|
|
_inherit = 'event.event'
|
|
|
|
def _get_seats_max(self, cr, uid, ids, field_name, arg, context=None):
|
|
result = dict.fromkeys(ids, 0)
|
|
for rec in self.browse(cr, uid, ids, context=context):
|
|
result[rec.id] = sum([ticket.seats_max for ticket in rec.event_ticket_ids])
|
|
return result
|
|
|
|
def _get_tickets(self, cr, uid, context={}):
|
|
try:
|
|
product = self.pool.get('ir.model.data').get_object(cr, uid, 'event_sale', 'product_product_event')
|
|
return [{
|
|
'name': _('Subscription'),
|
|
'product_id': product.id,
|
|
'price': 0,
|
|
}]
|
|
except ValueError:
|
|
pass
|
|
return []
|
|
|
|
def _get_ticket_events(self, cr, uid, ids, context=None):
|
|
# `self` is the event.event.ticket model when called by ORM!
|
|
return list(set(ticket.event_id.id
|
|
for ticket in self.browse(cr, uid, ids, context)))
|
|
|
|
# proxy method, can't import parent method directly as unbound_method: it would receive
|
|
# an invalid `self` <event_registration> when called by ORM
|
|
def _events_from_registrations(self, cr, uid, ids, context=None):
|
|
# `self` is the event.registration model when called by ORM
|
|
return self.pool['event.event']._get_events_from_registrations(cr, uid, ids, context=context)
|
|
|
|
_columns = {
|
|
'event_ticket_ids': fields.one2many('event.event.ticket', "event_id", "Event Ticket"),
|
|
'seats_max': fields.function(_get_seats_max,
|
|
string='Maximum Avalaible Seats',
|
|
help="The maximum registration level is equal to the sum of the maximum registration of event ticket." +
|
|
"If you have too much registrations you are not able to confirm your event. (0 to ignore this rule )",
|
|
type='integer',
|
|
readonly=True),
|
|
'seats_available': fields.function(Event._get_seats, oldname='register_avail', string='Available Seats',
|
|
type='integer', multi='seats_reserved',
|
|
store={
|
|
'event.registration': (_events_from_registrations, ['state'], 10),
|
|
'event.event': (lambda self, cr, uid, ids, c = {}: ids,
|
|
['seats_max', 'registration_ids'], 20),
|
|
'event.event.ticket': (_get_ticket_events, ['seats_max'], 10),
|
|
}),
|
|
}
|
|
_defaults = {
|
|
'event_ticket_ids': _get_tickets
|
|
}
|
|
|
|
class event_ticket(osv.osv):
|
|
_name = 'event.event.ticket'
|
|
|
|
def _get_seats(self, cr, uid, ids, fields, args, context=None):
|
|
"""Get reserved, available, reserved but unconfirmed and used seats for each event tickets.
|
|
@return: Dictionary of function field values.
|
|
"""
|
|
res = dict([(id, {}) for id in ids])
|
|
for ticket in self.browse(cr, uid, ids, context=context):
|
|
res[ticket.id]['seats_reserved'] = sum(reg.nb_register for reg in ticket.registration_ids if reg.state == "open")
|
|
res[ticket.id]['seats_used'] = sum(reg.nb_register for reg in ticket.registration_ids if reg.state == "done")
|
|
res[ticket.id]['seats_unconfirmed'] = sum(reg.nb_register for reg in ticket.registration_ids if reg.state == "draft")
|
|
res[ticket.id]['seats_available'] = ticket.seats_max - \
|
|
(res[ticket.id]['seats_reserved'] + res[ticket.id]['seats_used']) \
|
|
if ticket.seats_max > 0 else None
|
|
return res
|
|
|
|
def _is_expired(self, cr, uid, ids, field_name, args, context=None):
|
|
# FIXME: A ticket is considered expired when the deadline is passed. The deadline should
|
|
# be considered in the timezone of the event, not the timezone of the user!
|
|
# Until we add a TZ on the event we'll use the context's current date, more accurate
|
|
# than using UTC all the time.
|
|
current_date = fields.date.context_today(self, cr, uid, context=context)
|
|
return {ticket.id: ticket.deadline and ticket.deadline < current_date
|
|
for ticket in self.browse(cr, uid, ids, context=context)}
|
|
|
|
|
|
_columns = {
|
|
'name': fields.char('Name', size=64, required=True, translate=True),
|
|
'event_id': fields.many2one('event.event', "Event", required=True, ondelete='cascade'),
|
|
'product_id': fields.many2one('product.product', 'Product', required=True, domain=[("event_type_id", "!=", False)]),
|
|
'registration_ids': fields.one2many('event.registration', 'event_ticket_id', 'Registrations'),
|
|
'deadline': fields.date("Sales End"),
|
|
'is_expired': fields.function(_is_expired, type='boolean', string='Is Expired'),
|
|
'price': fields.float('Price'),
|
|
'seats_max': fields.integer('Maximum Avalaible Seats', oldname='register_max', help="You can for each event define a maximum registration level. If you have too much registrations you are not able to confirm your event. (put 0 to ignore this rule )"),
|
|
'seats_reserved': fields.function(_get_seats, string='Reserved Seats', type='integer', multi='seats_reserved'),
|
|
'seats_available': fields.function(_get_seats, string='Available Seats', type='integer', multi='seats_reserved'),
|
|
'seats_unconfirmed': fields.function(_get_seats, string='Unconfirmed Seat Reservations', type='integer', multi='seats_reserved'),
|
|
'seats_used': fields.function(_get_seats, string='Number of Participations', type='integer', multi='seats_reserved'),
|
|
}
|
|
|
|
def _default_product_id(self, cr, uid, context={}):
|
|
imd = self.pool.get('ir.model.data')
|
|
try:
|
|
product = imd.get_object(cr, uid, 'event_sale', 'product_product_event')
|
|
except ValueError:
|
|
return False
|
|
return product.id
|
|
|
|
_defaults = {
|
|
'product_id': _default_product_id
|
|
}
|
|
|
|
def _check_seats_limit(self, cr, uid, ids, context=None):
|
|
for ticket in self.browse(cr, uid, ids, context=context):
|
|
if ticket.seats_max and ticket.seats_available < 0:
|
|
return False
|
|
return True
|
|
|
|
_constraints = [
|
|
(_check_seats_limit, 'No more available tickets.', ['registration_ids','seats_max']),
|
|
]
|
|
|
|
def onchange_product_id(self, cr, uid, ids, product_id=False, context=None):
|
|
return {'value': {'price': self.pool.get("product.product").browse(cr, uid, product_id).list_price or 0}}
|
|
|
|
|
|
class event_registration(osv.osv):
|
|
"""Event Registration"""
|
|
_inherit= 'event.registration'
|
|
_columns = {
|
|
'event_ticket_id': fields.many2one('event.event.ticket', 'Event Ticket'),
|
|
}
|
|
|
|
def _check_ticket_seats_limit(self, cr, uid, ids, context=None):
|
|
for registration in self.browse(cr, uid, ids, context=context):
|
|
if registration.event_ticket_id.seats_max and registration.event_ticket_id.seats_available < 0:
|
|
return False
|
|
return True
|
|
|
|
_constraints = [
|
|
(_check_ticket_seats_limit, 'No more available tickets.', ['event_ticket_id','nb_register','state']),
|
|
]
|