[FIX] sale_margin: fix
[odoo/odoo.git] / addons / sale_margin / sale_margin.py
1 ##############################################################################
2 #
3 #    OpenERP, Open Source Management Solution
4 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
5 #
6 #    This program is free software: you can redistribute it and/or modify
7 #    it under the terms of the GNU Affero General Public License as
8 #    published by the Free Software Foundation, either version 3 of the
9 #    License, or (at your option) any later version.
10 #
11 #    This program is distributed in the hope that it will be useful,
12 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
13 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 #    GNU Affero General Public License for more details.
15 #
16 #    You should have received a copy of the GNU Affero General Public License
17 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
18 #
19 ##############################################################################
20
21 from osv import fields,osv
22 import pooler
23 from tools import config
24
25 class sale_order_line(osv.osv):
26     _name = "sale.order.line"
27     _inherit = "sale.order.line"
28
29     def product_id_change(self, cr, uid, ids, pricelist, product, qty=0,
30             uom=False, qty_uos=0, uos=False, name='', partner_id=False,
31             lang=False, update_tax=True, date_order=False, packaging=False, fiscal_position=False, flag=False):
32         res = super(sale_order_line, self).product_id_change(cr, uid, ids, pricelist, product, qty=qty,
33             uom=uom, qty_uos=qty_uos, uos=uos, name=name, partner_id=partner_id,
34             lang=lang, update_tax=update_tax, date_order=date_order, packaging=packaging, fiscal_position=fiscal_position, flag=flag)
35         if product:
36             purchase_price = self.pool.get('product.product').browse(cr, uid, product).standard_price
37             res['value'].update({'purchase_price':purchase_price})
38         return res
39
40     def _product_margin(self, cr, uid, ids, field_name, arg, context=None):
41         res = {}
42         for line in self.browse(cr, uid, ids):
43             res[line.id] = 0
44             if line.product_id:
45                 if line.purchase_price:
46                     res[line.id] = round((line.price_unit*line.product_uos_qty*(100.0-line.discount)/100.0) -(line.purchase_price*line.product_uos_qty),2)
47                 else:
48                     res[line.id] = round((line.price_unit*line.product_uos_qty*(100.0-line.discount)/100.0) -(line.product_id.standard_price*line.product_uos_qty),2)
49         return res
50
51     _columns = {
52         'margin': fields.function(_product_margin, method=True, string='Margin', store=True),
53         'purchase_price': fields.float('Cost Price', digits=(16,2))
54     }
55
56 sale_order_line()
57
58 class sale_order(osv.osv):
59     _inherit = "sale.order"
60
61     def _product_margin(self, cr, uid, ids, field_name, arg, context=None):
62         result = {}
63         for sale in self.browse(cr, uid, ids, context=context):
64             result[sale.id] = 0.0
65             for line in sale.order_line:
66                 result[sale.id] += line.margin or 0.0
67         return result
68
69     _columns = {
70         'margin': fields.function(_product_margin, method=True, string='Margin', store=True),
71     }
72
73 sale_order()
74
75 class stock_picking(osv.osv):
76     _inherit = 'stock.picking'
77
78     _columns = {
79         'invoice_ids': fields.many2many('account.invoice', 'picking_invoice_rel', 'picking_id', 'invoice_id', 'Invoices', domain=[('type','=','in_invoice')]),
80     }
81
82     def create_invoice(self, cr, uid, ids, *args):
83         # need to carify with new requirement
84         invoice_ids = []
85         margin_deduce = 0.0
86         picking_obj = self.pool.get('stock.picking')
87         picking_obj.write(cr, uid, ids, {'invoice_state' : '2binvoiced'})
88         res = picking_obj.action_invoice_create(cr, uid, ids, type='out_invoice', context={})
89         invoice_ids = res.values()
90         picking_obj.write(cr, uid, ids,{'invoice_ids': [[6,0,invoice_ids]]})
91         return True
92
93 stock_picking()
94 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: