# -*- encoding: utf-8 -*- ############################################################################## # # OpenERP, Open Source Management Solution # Copyright (C) 2004-2008 Tiny SPRL (). All Rights Reserved # $Id$ # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU 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 General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program. If not, see . # ############################################################################## import time import netsvc from osv import fields, osv import ir class account_invoice(osv.osv): _inherit = "account.invoice" _columns = { 'price_type': fields.selection([('tax_included','Tax included'), ('tax_excluded','Tax excluded')], 'Price method', required=True, readonly=True, states={'draft':[('readonly',False)]}), } _defaults = { 'price_type': lambda *a: 'tax_excluded', } account_invoice() class account_invoice_line(osv.osv): _inherit = "account.invoice.line" def _amount_line2(self, cr, uid, ids, name, args, context={}): """ Return the subtotal excluding taxes with respect to price_type. """ res = {} tax_obj = self.pool.get('account.tax') res_init = super(account_invoice_line, self)._amount_line(cr, uid, ids, name, args, context) for line in self.browse(cr, uid, ids): res[line.id] = { 'price_subtotal': 0.0, 'price_subtotal_incl': 0.0, 'data': [] } if not line.quantity: continue if line.invoice_id: product_taxes = [] if line.product_id: if line.invoice_id.type in ('out_invoice', 'out_refund'): product_taxes = filter(lambda x: x.price_include, line.product_id.taxes_id) else: product_taxes = filter(lambda x: x.price_include, line.product_id.supplier_taxes_id) if ((set(product_taxes) == set(line.invoice_line_tax_id)) or not product_taxes) and (line.invoice_id.price_type == 'tax_included'): res[line.id]['price_subtotal_incl'] = res_init[line.id] else: res[line.id]['price_subtotal'] = res_init[line.id] for tax in tax_obj.compute_inv(cr, uid, product_taxes, res_init[line.id]/line.quantity, line.quantity): res[line.id]['price_subtotal'] = res[line.id]['price_subtotal'] - round(tax['amount'], 2) else: res[line.id]['price_subtotal'] = res_init[line.id] if res[line.id]['price_subtotal']: res[line.id]['price_subtotal_incl'] = res[line.id]['price_subtotal'] for tax in tax_obj.compute(cr, uid, line.invoice_line_tax_id, res[line.id]['price_subtotal']/line.quantity, line.quantity): res[line.id]['price_subtotal_incl'] = res[line.id]['price_subtotal_incl'] + tax['amount'] res[line.id]['data'].append( tax) else: res[line.id]['price_subtotal'] = res[line.id]['price_subtotal_incl'] for tax in tax_obj.compute_inv(cr, uid, line.invoice_line_tax_id, res[line.id]['price_subtotal_incl']/line.quantity, line.quantity): res[line.id]['price_subtotal'] = res[line.id]['price_subtotal'] - tax['amount'] res[line.id]['data'].append( tax) res[line.id]['price_subtotal']= round(res[line.id]['price_subtotal'], 2) res[line.id]['price_subtotal_incl']= round(res[line.id]['price_subtotal_incl'], 2) return res def _price_unit_default(self, cr, uid, context={}): if 'check_total' in context: t = context['check_total'] if context.get('price_type', False) == 'tax_included': for l in context.get('invoice_line', {}): if len(l) >= 3 and l[2]: p = l[2].get('price_unit', 0) * (1-l[2].get('discount', 0)/100.0) t = t - (p * l[2].get('quantity')) return t return super(account_invoice_line, self)._price_unit_default(cr, uid, context) return 0 def _get_invoice(self, cr, uid, ids, context): result = {} for inv in self.pool.get('account.invoice').browse(cr, uid, ids, context=context): for line in inv.invoice_line: result[line.id] = True return result.keys() _columns = { 'price_subtotal': fields.function(_amount_line2, method=True, string='Subtotal w/o tax', multi='amount', store={'account.invoice':(_get_invoice,['price_type'],10), 'account.invoice.line': (lambda self,cr,uid,ids,c={}: ids, None,10)}), 'price_subtotal_incl': fields.function(_amount_line2, method=True, string='Subtotal', multi='amount', store={'account.invoice':(_get_invoice,['price_type'],10), 'account.invoice.line': (lambda self,cr,uid,ids,c={}: ids, None,10)}), } _defaults = { 'price_unit': _price_unit_default, } def move_line_get_item(self, cr, uid, line, context={}): return { 'type':'src', 'name':line.name, 'price_unit':line.price_subtotal / line.quantity, 'quantity':line.quantity, 'price':line.price_subtotal, 'account_id':line.account_id.id, 'product_id': line.product_id.id, 'uos_id':line.uos_id.id, 'account_analytic_id':line.account_analytic_id.id, } # TODO: check why ? # # def product_id_change_unit_price_inv(self, cr, uid, tax_id, price_unit, qty, address_invoice_id, product, partner_id, context={}): # if context.get('price_type', False) == 'tax_included': # return {'price_unit': price_unit,'invoice_line_tax_id': tax_id} # else: # return super(account_invoice_line, self).product_id_change_unit_price_inv(cr, uid, tax_id, price_unit, qty, address_invoice_id, product, partner_id, context=context) # # def product_id_change(self, cr, uid, ids, product, uom, qty=0, name='', type='out_invoice', partner_id=False, price_unit=False, address_invoice_id=False, price_type='tax_excluded', context={}): # context.update({'price_type': price_type}) # return super(account_invoice_line, self).product_id_change(cr, uid, ids, product, uom, qty, name, type, partner_id, price_unit, address_invoice_id, context=context) account_invoice_line() class account_invoice_tax(osv.osv): _inherit = "account.invoice.tax" def compute(self, cr, uid, invoice_id, context={}): inv = self.pool.get('account.invoice').browse(cr, uid, invoice_id) line_ids = map(lambda x: x.id, inv.invoice_line) tax_grouped = {} tax_obj = self.pool.get('account.tax') cur_obj = self.pool.get('res.currency') cur = inv.currency_id for line in inv.invoice_line: data = self.pool.get('account.invoice.line')._amount_line2(cr, uid, [line.id], [], [], context)[line.id] for tax in data['data']: val={} val['invoice_id'] = inv.id val['name'] = tax['name'] val['amount'] = cur_obj.round(cr, uid, cur, tax['amount']) val['manual'] = False val['sequence'] = tax['sequence'] val['base'] = tax['price_unit'] * line['quantity'] if inv.type in ('out_invoice','in_invoice'): val['base_code_id'] = tax['base_code_id'] val['tax_code_id'] = tax['tax_code_id'] val['base_amount'] = val['base'] * tax['base_sign'] val['tax_amount'] = val['amount'] * tax['tax_sign'] val['account_id'] = tax['account_collected_id'] or line.account_id.id else: val['base_code_id'] = tax['ref_base_code_id'] val['tax_code_id'] = tax['ref_tax_code_id'] val['base_amount'] = val['base'] * tax['ref_base_sign'] val['tax_amount'] = val['amount'] * tax['ref_tax_sign'] val['account_id'] = tax['account_paid_id'] or line.account_id.id key = (val['tax_code_id'], val['base_code_id'], val['account_id']) if not key in tax_grouped: tax_grouped[key] = val else: tax_grouped[key]['amount'] += val['amount'] tax_grouped[key]['base'] += val['base'] tax_grouped[key]['base_amount'] += val['base_amount'] tax_grouped[key]['tax_amount'] += val['tax_amount'] return tax_grouped account_invoice_tax() # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: