addons/product_visible_discount/product_visible_discount.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
   6 #    $Id$
   7 #
   8 #    This program is free software: you can redistribute it and/or modify
   9 #    it under the terms of the GNU Affero General Public License as published by
  10 #    the Free Software Foundation, either version 3 of the License, or
  11 #    (at your option) any later version.
  12 #
  13 #    This program is distributed in the hope that it will be useful,
  14 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #    GNU Affero General Public License for more details.
  17 #
  18 #    You should have received a copy of the GNU Affero General Public License
  19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21 ##############################################################################
  22
  23 from openerp.osv import fields, osv
  24 from openerp.tools.translate import _
  25
  26 class product_pricelist(osv.osv):
  27     _inherit = 'product.pricelist'
  28
  29     _columns ={
  30         'visible_discount': fields.boolean('Visible Discount'),
  31     }
  32     _defaults = {
  33          'visible_discount': True,
  34     }
  35
  36 product_pricelist()
  37
  38 class sale_order_line(osv.osv):
  39     _inherit = "sale.order.line"
  40
  41     def product_id_change(self, cr, uid, ids, pricelist, product, qty=0,
  42             uom=False, qty_uos=0, uos=False, name='', partner_id=False,
  43             lang=False, update_tax=True, date_order=False, packaging=False,
  44             fiscal_position=False, flag=False, context=None):
  45
  46         def get_real_price(res_dict, product_id, qty, uom, pricelist):
  47             """Retrieve the price before applying the pricelist"""
  48             item_obj = self.pool.get('product.pricelist.item')
  49             price_type_obj = self.pool.get('product.price.type')
  50             product_obj = self.pool.get('product.product')
  51             field_name = 'list_price'
  52
  53             if res_dict.get('item_id',False) and res_dict['item_id'].get(pricelist,False):
  54                 item = res_dict['item_id'].get(pricelist,False)
  55                 item_base = item_obj.read(cr, uid, [item], ['base'])[0]['base']
  56                 if item_base > 0:
  57                     field_name = price_type_obj.browse(cr, uid, item_base).field
  58
  59             product = product_obj.browse(cr, uid, product_id, context)
  60             product_read = product_obj.read(cr, uid, product_id, [field_name], context=context)
  61
  62             factor = 1.0
  63             if uom and uom != product.uom_id.id:
  64                 # the unit price is in a different uom
  65                 factor = self.pool['product.uom']._compute_qty(cr, uid, uom, 1.0, product.uom_id.id)
  66             return product_read[field_name] * factor
  67
  68
  69         res=super(sale_order_line, self).product_id_change(cr, uid, ids, pricelist, product, qty,
  70             uom, qty_uos, uos, name, partner_id,
  71             lang, update_tax, date_order, packaging=packaging, fiscal_position=fiscal_position, flag=flag, context=context)
  72
  73         context = {'lang': lang, 'partner_id': partner_id}
  74         result=res['value']
  75         pricelist_obj=self.pool.get('product.pricelist')
  76         product_obj = self.pool.get('product.product')
  77         if product and pricelist:
  78             if result.get('price_unit',False):
  79                 price=result['price_unit']
  80             else:
  81                 return res
  82             uom = result.get('product_uom', uom)
  83             product = product_obj.browse(cr, uid, product, context)
  84             list_price = pricelist_obj.price_get(cr, uid, [pricelist],
  85                     product.id, qty or 1.0, partner_id, {'uom': uom,'date': date_order })
  86
  87             so_pricelist = pricelist_obj.browse(cr, uid, pricelist, context=context)
  88
  89             new_list_price = get_real_price(list_price, product.id, qty, uom, pricelist)
  90             if so_pricelist.visible_discount and list_price[pricelist] != 0 and new_list_price != 0:
  91                 if product.company_id and so_pricelist.currency_id.id != product.company_id.currency_id.id:
  92                     # new_list_price is in company's currency while price in pricelist currency
  93                     ctx = context.copy()
  94                     ctx['date'] = date_order
  95                     new_list_price = self.pool['res.currency'].compute(cr, uid,
  96                         product.company_id.currency_id.id, so_pricelist.currency_id.id,
  97                         new_list_price, context=ctx)
  98                 discount = (new_list_price - price) / new_list_price * 100
  99                 if discount > 0:
 100                     result['price_unit'] = new_list_price
 101                     result['discount'] = discount
 102                 else:
 103                     result['discount'] = 0.0
 104             else:
 105                 result['discount'] = 0.0
 106         else:
 107             result['discount'] = 0.0
 108         return res