odoo/addons/website_event/models/event.py

# -*- coding: utf-8 -*-
##############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2013-Today OpenERP SA (<http://www.openerp.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.osv import osv, fields
from openerp import SUPERUSER_ID


# defined for access rules
class product(osv.osv):
    _inherit = 'product.product'
    _columns = {
        'event_ticket_ids': fields.one2many('event.event.ticket', 'product_id', 'Event Tickets'),
    }


class event(osv.osv):
    _name = 'event.event'
    _inherit = ['event.event','website.seo.metadata']

    def _get_new_menu_pages(serf, cr, uid, event, context=None):
        context = context or {}
        todo = [
            (_('Introduction'), 'website_event.template_intro'),
            (_('Location'), 'website_event.template_location')
        ]
        web = request.registry['website']
        result = []
        for name,path in todo:
            newpath = web.new_page(request.cr, request.uid, path, request.context)
            url = "/event/event.id/page/" + newpath
            result.append((name, newpath))
        return result

    def _set_show_menu(self, cr, uid, ids, name, value, arg, context=None):
        menuobj = self.pool.get('website.menu')
        for event in self.browse(cr, uid, ids, context=context):
            if event.menu_id and not value:
                menuobj.unlink(cr, uid, [event.menu_id.id], context=context)
            elif value and not event.menu_id:
                root = menuobj.create(cr, uid, {
                    'name': event.name
                }, context=context)
                tocreate = self._get_new_menu_pages(cr, uid, event, context)
                tocreate.append((_('Register'), '/event/register/'+str(event.id)))
                sequence = 0
                for name,url in tocreate:
                    menuobj.create(cr, uid, {
                        'name': name,
                        'url': url,
                        'parent_id': root,
                        'sequence': sequence
                    }, context=context)
                    sequence += 1
        return True

    def _get_show_menu(self, cr, uid, ids, field_name, arg, context=None):
        res = dict.fromkeys(ids, '')
        for event in self.browse(cr, uid, ids, context=context):
            res[event.id] = bool(event.menu_id)
        return res

    def _website_url(self, cr, uid, ids, field_name, arg, context=None):
        res = dict.fromkeys(ids, '')
        base_url = self.pool.get('ir.config_parameter').get_param(cr, uid, 'web.base.url')
        for event in self.browse(cr, uid, ids, context=context):
            res[event.id] = "%s/event/%s/" % (base_url, event.id)
        return res

    _columns = {
        'twitter_hashtag': fields.char('Twitter Hashtag'),
        'website_published': fields.boolean('Visible in Website'),
        # TDE TODO FIXME: when website_mail/mail_thread.py inheritance work -> this field won't be necessary
        'website_message_ids': fields.one2many(
            'mail.message', 'res_id',
            domain=lambda self: [
                '&', ('model', '=', self._name), ('type', '=', 'comment')
            ],
            string='Website Messages',
            help="Website communication history",
        ),
        'website_url': fields.function(_website_url, string="Website url"),
        'show_menu': fields.function(_get_show_menu, fnct_inv=_set_show_menu, type='boolean', string='Dedicated Menu'),
        'menu_id': fields.many2one('website.menu', 'Event Menu'),
    }
    _defaults = {
        'show_menu': False,
    }

    def google_map_img(self, cr, uid, ids, zoom=8, width=298, height=298, context=None):
        partner = self.browse(cr, uid, ids[0], context=context)
        if partner.address_id:
            return self.browse(cr, SUPERUSER_ID, ids[0], context=context).address_id.google_map_img()

    def google_map_link(self, cr, uid, ids, zoom=8, context=None):
        partner = self.browse(cr, uid, ids[0], context=context)
        if partner.address_id:
            return self.browse(cr, SUPERUSER_ID, ids[0], context=context).address_id.google_map_link()

class sale_order_line(osv.osv):
    _inherit = "sale.order.line"

    def _recalculate_product_values(self, cr, uid, ids, product_id=0, context=None):
        if not ids:
            return super(sale_order_line, self)._recalculate_product_values(cr, uid, ids, product_id, context=context)

        order_line = self.browse(cr, uid, ids[0], context=context)
        product = product_id and self.pool.get('product.product').browse(cr, uid, product_id, context=context) or order_line.product_id
        res = super(sale_order_line, self)._recalculate_product_values(cr, uid, ids, product.id, context=context)
        if product.event_type_id and order_line.event_ticket_id and order_line.event_ticket_id.price != product.lst_price:
            res.update({'price_unit': order_line.event_ticket_id.price})

        return res