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 osv import fields, osv
  24 from 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             item_obj = self.pool.get('product.pricelist.item')
  48             price_type_obj = self.pool.get('product.price.type')
  49             product_obj = self.pool.get('product.product')
  50             template_obj = self.pool.get('product.template')
  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_tmpl_id = product.product_tmpl_id.id
  61
  62             product_read = template_obj.read(cr, uid, product_tmpl_id, [field_name], context)
  63
  64             factor = 1.0
  65             if uom and uom != product.uom_id.id:
  66                 product_uom_obj = self.pool.get('product.uom')
  67                 uom_data = product_uom_obj.browse(cr, uid,  product.uom_id.id)
  68                 factor = uom_data.factor
  69             return product_read[field_name] * factor
  70
  71
  72         res=super(sale_order_line, self).product_id_change(cr, uid, ids, pricelist, product, qty,
  73             uom, qty_uos, uos, name, partner_id,
  74             lang, update_tax, date_order, packaging=packaging, fiscal_position=fiscal_position, flag=flag, context=context)
  75
  76         context = {'lang': lang, 'partner_id': partner_id}
  77         result=res['value']
  78         pricelist_obj=self.pool.get('product.pricelist')
  79         product_obj = self.pool.get('product.product')
  80         if product:
  81             if result.get('price_unit',False):
  82                 price=result['price_unit']
  83             else:
  84                 return res
  85
  86             product = product_obj.browse(cr, uid, product, context)
  87             list_price = pricelist_obj.price_get(cr, uid, [pricelist],
  88                     product.id, qty or 1.0, partner_id, {'uom': uom,'date': date_order })
  89
  90             pricelists = pricelist_obj.read(cr,uid,[pricelist],['visible_discount'])
  91
  92             new_list_price = get_real_price(list_price, product.id, qty, uom, pricelist)
  93             if(len(pricelists)>0 and pricelists[0]['visible_discount'] and list_price[pricelist] != 0):
  94                 discount = (new_list_price - price) / new_list_price * 100
  95                 result['price_unit'] = new_list_price
  96                 result['discount'] = discount
  97             else:
  98                 result['discount'] = 0.0
  99         return res
 100
 101 sale_order_line()
 102
 103 class account_invoice_line(osv.osv):
 104     _inherit = "account.invoice.line"
 105
 106     def product_id_change(self, cr, uid, ids, product, uom, qty=0, name='', type='out_invoice', partner_id=False, fposition_id=False, price_unit=False, address_invoice_id=False, currency_id=False, context=None, company_id=None):
 107         res = super(account_invoice_line, self).product_id_change(cr, uid, ids, product, uom, qty, name, type, partner_id, fposition_id, price_unit, address_invoice_id, currency_id, context=context, company_id=company_id)
 108
 109         def get_real_price(res_dict, product_id, qty, uom, pricelist):
 110             item_obj = self.pool.get('product.pricelist.item')
 111             price_type_obj = self.pool.get('product.price.type')
 112             product_obj = self.pool.get('product.product')
 113             template_obj = self.pool.get('product.template')
 114             field_name = 'list_price'
 115
 116             if res_dict.get('item_id',False) and res_dict['item_id'].get(pricelist,False):
 117                 item = res_dict['item_id'].get(pricelist,False)
 118                 item_base = item_obj.read(cr, uid, [item], ['base'])[0]['base']
 119                 if item_base > 0:
 120                     field_name = price_type_obj.browse(cr, uid, item_base).field
 121
 122             product = product_obj.browse(cr, uid, product_id, context)
 123             product_tmpl_id = product.product_tmpl_id.id
 124
 125             product_read = template_obj.read(cr, uid, product_tmpl_id, [field_name], context)
 126
 127             factor = 1.0
 128             if uom and uom != product.uom_id.id:
 129                 product_uom_obj = self.pool.get('product.uom')
 130                 uom_data = product_uom_obj.browse(cr, uid,  product.uom_id.id)
 131                 factor = uom_data.factor
 132             return product_read[field_name] * factor
 133
 134         if product:
 135             pricelist_obj = self.pool.get('product.pricelist')
 136             partner_obj = self.pool.get('res.partner')
 137             product = self.pool.get('product.product').browse(cr, uid, product, context=context)
 138             result = res['value']
 139             pricelist = False
 140             real_price = 0.00
 141             if type in ('in_invoice', 'in_refund'):
 142                 if not price_unit and partner_id:
 143                     pricelist =partner_obj.browse(cr, uid, partner_id).property_product_pricelist_purchase.id
 144                     if not pricelist:
 145                         raise osv.except_osv(_('No Purchase Pricelist Found!'),_("You must first define a pricelist on the supplier form!"))
 146                     price_unit_res = pricelist_obj.price_get(cr, uid, [pricelist], product.id, qty or 1.0, partner_id, {'uom': uom})
 147                     price_unit = price_unit_res[pricelist]
 148                     real_price = get_real_price(price_unit_res, product.id, qty, uom, pricelist)
 149             else:
 150                 if partner_id:
 151                     pricelist = partner_obj.browse(cr, uid, partner_id).property_product_pricelist.id
 152                     if not pricelist:
 153                         raise osv.except_osv(_('No Sale Pricelist Found!'),_("You must first define a pricelist on the customer form!"))
 154                     price_unit_res = pricelist_obj.price_get(cr, uid, [pricelist], product.id, qty or 1.0, partner_id, {'uom': uom})
 155                     price_unit = price_unit_res[pricelist]
 156
 157                     real_price = get_real_price(price_unit_res, product.id, qty, uom, pricelist)
 158             if pricelist:
 159                 pricelists=pricelist_obj.read(cr,uid,[pricelist],['visible_discount'])
 160                 if(len(pricelists)>0 and pricelists[0]['visible_discount'] and real_price != 0):
 161                     discount=(real_price-price_unit) / real_price * 100
 162                     result['price_unit'] = real_price
 163                     result['discount'] = discount
 164                 else:
 165                     result['discount']=0.0
 166         return res
 167
 168 account_invoice_line()
 169
 170 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: