odoo/addons/website_sale_quote/models/order.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
import hashlib
import time


class sale_quote_template(osv.osv):
    _name = "sale.quote.template"
    _description = "Sale Quotation Template"
    _columns = {
        'name': fields.char('Quotation Template', size=256, required=True),
        'website_description': fields.html('Description'),
        'quote_line': fields.one2many('sale.quote.line', 'quote_id', 'Quote Template Lines'),
        'note': fields.text('Terms and conditions'),
        'template_url': fields.char('URL', readonly=True),
    }

    def _get_signup_url(self, cr, uid, quote_id, context=None):
        base_url = self.pool.get('ir.config_parameter').get_param(cr, uid, 'web.base.url', default='http://localhost:8069', context=context)
        url = "%s/template/%s" % (base_url, quote_id)
        return url

    def create(self, cr, uid, vals, context=None): 
        new_id = super(sale_quote_template, self).create(cr, uid, vals, context)
        self.write(cr, uid, new_id, {'template_url': self._get_signup_url(cr, uid, new_id, context=context)})
        return new_id

class sale_quote_line(osv.osv):
    _name = "sale.quote.line"
    _description = "Quotation Template Lines"
    _columns = {
        'quote_id': fields.many2one('sale.quote.template', 'Quotation Template Reference', required=True, ondelete='cascade', select=True),
        'name': fields.text('Description', required=True),
        'product_id': fields.many2one('product.product', 'Product', domain=[('sale_ok', '=', True)], change_default=True),
        'website_description': fields.html('Line Description'),
        'price_unit': fields.float('Unit Price', required=True),
        'product_uom_qty': fields.float('Quantity', required=True),
    }

    _defaults = {
        'product_uom_qty': 1,
    }

    def on_change_product_id(self, cr, uid, ids, product, context=None):
        vals = {}
        product_obj = self.pool.get('product.product').browse(cr, uid, product, context=context)
        vals.update({
            'price_unit': product_obj.list_price,
            'website_description': product_obj.website_description,
            'name': product_obj.name,
        })
        return {'value': vals}


class sale_order_line(osv.osv):
    _inherit = "sale.order.line"
    _description = "Sales Order Line"
    _columns = {
        'website_description': fields.html('Line Description'),
    }

    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):
        res = super(sale_order_line, self).product_id_change(cr, uid, ids, pricelist, product, qty, uom, qty_uos, uos, name, partner_id, lang, update_tax, date_order, packaging, fiscal_position, flag, context)
        if product:
            desc = self.pool.get('product.product').browse(cr, uid, product, context).website_description
            res.get('value').update({'website_description': desc})
        return res


class sale_order(osv.osv):
    _inherit = 'sale.order'
    _columns = {
        'quote_url': fields.char('URL', readonly=True),
        'access_token': fields.char('Quotation Token', size=256),
        'template_id': fields.many2one('sale.quote.template', 'Quote Template'),
        'website_description': fields.html('Description'),
    }

    def _get_token(self, cr, uid, ids, context=None):
        """
            Generate token for sale order on action_quotation_send , send it to customer.
        """
        db_uuid = self.pool.get('ir.config_parameter').get_param(cr, uid, 'database.uuid')
        for id in ids:
            token = hashlib.sha256('%s-%s-%s' % (time.time(), db_uuid, id)).hexdigest()
        return token

    def _get_signup_url(self, cr, uid, ids, token=False, context=None):
        base_url = self.pool.get('ir.config_parameter').get_param(cr, uid, 'web.base.url', default='http://localhost:8069', context=context)
        for id in ids:
            url = "%s/quote/%s/%s" % (base_url, id, token)
        return url

    def action_quotation_send(self, cr, uid, ids, context=None):
        self._create_portal_user(cr, uid, ids, context=context)
        token = self._get_token(cr, uid, ids, context)
        url = self._get_signup_url(cr, uid, ids, token, context)
        self.write(cr, uid, ids, {'access_token': token, 'quote_url': url})
        res = super(sale_order, self).action_quotation_send(cr, uid, ids, context=context)
        return res

    def _create_portal_user(self, cr, uid, ids, context=None):
        """
            create portal user of customer in quotation , when action_quotation_send perform.
        """
        user = []
        portal_ids = self.pool.get('res.groups').search(cr, uid, [('is_portal', '=', True)])
        user_wizard_pool = self.pool.get('portal.wizard.user')
        for order in self.browse(cr, uid, ids, context=context):
            wizard_id = self.pool.get('portal.wizard').create(cr, uid, {'portal_id': portal_ids and portal_ids[0] or False})
            user.append(user_wizard_pool.create(cr, uid, {
                'wizard_id': wizard_id,
                'partner_id': order.partner_id.id,
                'email': order.partner_id.email,
                'in_portal': True}))
        return user_wizard_pool.action_apply(cr, uid, user, context=context)

    def _get_sale_order_line(self, cr, uid, template_id, context=None):
        """create order line from selected quote template line."""
        lines = []
        quote_template = self.pool.get('sale.quote.template').browse(cr, uid, template_id, context)
        for line in quote_template.quote_line:
            lines.append((0, 0, {
            'name': line.name,
            'price_unit': line.price_unit,
            'product_uom_qty': line.product_uom_qty,
            'product_id': line.product_id.id,
            'website_description': line.website_description,
            'state': 'draft',
            }))
        return {'order_line': lines, 'website_description': quote_template.website_description, 'note': quote_template.note}

    def onchange_template_id(self, cr, uid, ids, template_id, context=None):
        data = self._get_sale_order_line(cr, uid, template_id, context)
        return {'value': data}

    def recommended_products(self, cr, uid, ids, context=None):
        order_line = self.browse(cr, uid, ids[0], context=context).order_line
        product_pool = self.pool.get('product.product')
        product_ids = []
        for line in order_line:
            query = """
                SELECT      sol.product_id
                FROM        sale_order_line as my
                LEFT JOIN   sale_order_line as sol
                ON          sol.order_id = my.order_id
                WHERE       my.product_id in (%s)
                AND         sol.product_id not in (%s)
                GROUP BY    sol.product_id
                ORDER BY    COUNT(sol.order_id) DESC
                LIMIT 10
            """
            cr.execute(query, (line.product_id.id, line.product_id.id))
            for p in cr.fetchall():
                product_ids.append(p[0])
        product_ids = product_pool.search(cr, uid, [("id", "in", product_ids)], limit=3)
        return product_pool.browse(cr, uid, product_ids)