addons/account_tax_include/invoice_tax_incl.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 # Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
   5 #
   6 # $Id: account.py 1005 2005-07-25 08:41:42Z nicoe $
   7 #
   8 # WARNING: This program as such is intended to be used by professional
   9 # programmers who take the whole responsability of assessing all potential
  10 # consequences resulting from its eventual inadequacies and bugs
  11 # End users who are looking for a ready-to-use solution with commercial
  12 # garantees and support are strongly adviced to contract a Free Software
  13 # Service Company
  14 #
  15 # This program is Free Software; you can redistribute it and/or
  16 # modify it under the terms of the GNU General Public License
  17 # as published by the Free Software Foundation; either version 2
  18 # of the License, or (at your option) any later version.
  19 #
  20 # This program is distributed in the hope that it will be useful,
  21 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 # GNU General Public License for more details.
  24 #
  25 # You should have received a copy of the GNU General Public License
  26 # along with this program; if not, write to the Free Software
  27 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  28 #
  29 ##############################################################################
  30
  31 import time
  32 import netsvc
  33 from osv import fields, osv
  34 import ir
  35
  36 class account_invoice(osv.osv):
  37         def _amount_untaxed(self, cr, uid, ids, name, args, context={}):
  38                 res = {}
  39                 for invoice in self.browse(cr,uid,ids):
  40                         if invoice.price_type == 'tax_included':
  41                                 res[invoice.id]= invoice.amount_total - invoice.amount_tax
  42                         else:
  43                                 res[invoice.id] = super(account_invoice, self)._amount_untaxed(cr, uid, [invoice.id], name, args, context)[invoice.id]
  44                 return res
  45
  46         def _amount_tax(self, cr, uid, ids, name, args, context={}):
  47                 res = {}
  48                 for invoice in self.browse(cr,uid,ids):
  49                         if invoice.price_type == 'tax_included':
  50                                 res[invoice.id] = reduce( lambda x, y: x+y.amount, invoice.tax_line,0)
  51                         else:
  52                                 res[invoice.id] = super(account_invoice, self)._amount_tax(cr, uid, [invoice.id], name, args, context)[invoice.id]
  53                 return res
  54
  55         def _amount_total(self, cr, uid, ids, name, args, context={}):
  56                 res = {}
  57                 for invoice in self.browse(cr,uid,ids):
  58                         if invoice.price_type == 'tax_included':
  59                                 res[invoice.id]= reduce( lambda x, y: x+y.price_subtotal_incl, invoice.invoice_line,0)
  60                         else:
  61                                 res[invoice.id] = super(account_invoice, self)._amount_total(cr, uid, [invoice.id], name, args, context)[invoice.id]
  62                 return res
  63
  64         _inherit = "account.invoice"
  65         _columns = {
  66                 'price_type': fields.selection([('tax_included','Tax included'),
  67                                                                                 ('tax_excluded','Tax excluded')],
  68                                                                                 'Price method', required=True, readonly=True,
  69                                                                                 states={'draft':[('readonly',False)]}),
  70                 'amount_untaxed': fields.function(_amount_untaxed, digits=(16,2), method=True,string='Untaxed Amount'),
  71                 'amount_tax': fields.function(_amount_tax, method=True, string='Tax', store=True),
  72                 'amount_total': fields.function(_amount_total, method=True, string='Total', store=True),
  73         }
  74         _defaults = {
  75                 'price_type': lambda *a: 'tax_excluded',
  76         }
  77 account_invoice()
  78
  79 class account_invoice_line(osv.osv):
  80         _inherit = "account.invoice.line"
  81         def _amount_line(self, cr, uid, ids, name, args, context={}):
  82                 """
  83                 Return the subtotal excluding taxes with respect to price_type.
  84                 """
  85                 res = {}
  86                 tax_obj = self.pool.get('account.tax')
  87                 res = super(account_invoice_line, self)._amount_line(cr, uid, ids, name, args, context)
  88                 res2 = res.copy()
  89                 for line in self.browse(cr, uid, ids):
  90                         if line.invoice_id.price_type == 'tax_included':
  91                                 product_taxes = None
  92                                 if line.product_id:
  93                                         if line.invoice_id.type in ('out_invoice', 'out_refund'):
  94                                                 product_taxes = line.product_id.taxes_id
  95                                         else:
  96                                                 product_taxes = line.product_id.supplier_taxes_id
  97                                 if product_taxes:
  98                                         for tax in tax_obj.compute_inv(cr, uid, product_taxes, res[line.id]/line.quantity, line.quantity):
  99                                                 res[line.id] = res[line.id] - tax['amount']
 100                                 else:
 101                                         for tax in tax_obj.compute_inv(cr, uid,line.invoice_line_tax_id, res[line.id]/line.quantity, line.quantity):
 102                                                 res[line.id] = res[line.id] - tax['amount']
 103                         if name == 'price_subtotal_incl' and line.invoice_id.price_type == 'tax_included':
 104                                 prod_taxe_ids = None
 105                                 line_taxe_ids = None
 106                                 if product_taxes:
 107                                         prod_taxe_ids = [ t.id for t in product_taxes ]
 108                                         prod_taxe_ids.sort()
 109                                         line_taxe_ids = [ t.id for t in line.invoice_line_tax_id ]
 110                                         line_taxe_ids.sort()
 111                                 if product_taxes and prod_taxe_ids == line_taxe_ids:
 112                                         res[line.id] = res2[line.id]
 113                                 elif not line.product_id:
 114                                         res[line.id] = res2[line.id]
 115                                 else:
 116                                         for tax in tax_obj.compute(cr, uid, line.invoice_line_tax_id, res[line.id]/line.quantity, line.quantity):
 117                                                 res[line.id] = res[line.id] + tax['amount']
 118                         res[line.id]= round(res[line.id], 2)
 119                 return res
 120
 121         def _price_unit_default(self, cr, uid, context={}):
 122                 if 'check_total' in context:
 123                         t = context['check_total']
 124                         if context.get('price_type', False) == 'tax_included':
 125                                 for l in context.get('invoice_line', {}):
 126                                         if len(l) >= 3 and l[2]:
 127                                                 p = l[2].get('price_unit', 0) * (1-l[2].get('discount', 0)/100.0)
 128                                                 t = t - (p * l[2].get('quantity'))
 129                                 return t
 130                         return super(account_invoice_line, self)._price_unit_default(cr, uid, context)
 131                 return 0
 132
 133         _columns = {
 134                 'price_subtotal': fields.function(_amount_line, method=True, string='Subtotal w/o tax'),
 135                 'price_subtotal_incl': fields.function(_amount_line, method=True, string='Subtotal'),
 136         }
 137
 138         _defaults = {
 139                 'price_unit': _price_unit_default,
 140         }
 141
 142         #
 143         # Compute a tax amount for each kind of tax :
 144         #
 145         def move_line_get(self, cr, uid, invoice_id, context={}):
 146                 inv = self.pool.get('account.invoice').browse(cr, uid, invoice_id)
 147                 if inv.price_type=='tax_excluded':
 148                         return super(account_invoice_line,self).move_line_get(cr, uid, invoice_id)
 149
 150                 res = []
 151                 tax_grouped = {}
 152                 tax_obj = self.pool.get('account.tax')
 153                 cur_obj = self.pool.get('res.currency')
 154                 ait_obj = self.pool.get('account.invoice.tax')
 155                 cur = inv.currency_id
 156
 157                 for line in inv.invoice_line:
 158                         res.append( self.move_line_get_item(cr, uid, line, context))
 159                         for tax in tax_obj.compute(cr, uid, line.invoice_line_tax_id, (line.price_unit *(1.0-(line['discount'] or 0.0)/100.0)), line.quantity, inv.address_invoice_id.id, line.product_id, inv.partner_id):
 160                                 if inv.type in ('out_invoice', 'in_invoice'):
 161                                         res[-1]['tax_code_id'] = tax['base_code_id']
 162                                         res[-1]['tax_amount'] = tax['price_unit'] * line['quantity'] * tax['base_sign']
 163                                 else:
 164                                         res[-1]['ta_code_id'] = tax['ref_base_code_id']
 165                                         res[-1]['tax_amount'] = tax['price_unit'] * line['quantity'] * tax['ref_base_sign']
 166                 return res
 167
 168         def move_line_get_item(self, cr, uid, line, context={}):
 169                 return {
 170                                 'type':'src',
 171                                 'name':line.name,
 172                                 'price_unit':line.price_unit,
 173                                 'quantity':line.quantity,
 174                                 'price':line.price_subtotal,
 175                                 'account_id':line.account_id.id,
 176                                 'product_id': line.product_id.id,
 177                                 'uos_id':line.uos_id.id,
 178                                 'account_analytic_id':line.account_analytic_id.id,
 179                         }
 180
 181         def product_id_change_unit_price_inv(self, cr, uid, tax_id, price_unit, qty, address_invoice_id, product, partner_id, context={}):
 182                 if context.get('price_type', False) == 'tax_included':
 183                         return {'price_unit': price_unit,'invoice_line_tax_id': tax_id}
 184                 else:
 185                         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)
 186
 187         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={}):
 188                 context.update({'price_type': price_type})
 189                 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)
 190 account_invoice_line()
 191
 192 class account_invoice_tax(osv.osv):
 193         _inherit = "account.invoice.tax"
 194
 195         def compute(self, cr, uid, invoice_id):
 196                 tax_grouped = {}
 197                 tax_obj = self.pool.get('account.tax')
 198                 cur_obj = self.pool.get('res.currency')
 199                 inv = self.pool.get('account.invoice').browse(cr, uid, invoice_id)
 200                 cur = inv.currency_id
 201
 202                 if inv.price_type=='tax_excluded':
 203                         return super(account_invoice_tax,self).compute(cr, uid, invoice_id)
 204
 205                 for line in inv.invoice_line:
 206                         for tax in tax_obj.compute_inv(cr, uid, line.invoice_line_tax_id, (line.price_unit * (1-(line.discount or 0.0)/100.0)), line.quantity, inv.address_invoice_id.id, line.product_id, inv.partner_id):
 207                                 val={}
 208                                 val['invoice_id'] = inv.id
 209                                 val['name'] = tax['name']
 210                                 val['amount'] = cur_obj.round(cr, uid, cur, tax['amount'])
 211                                 val['manual'] = False
 212                                 val['sequence'] = tax['sequence']
 213                                 val['base'] = tax['price_unit'] * line['quantity']
 214
 215                                 if inv.type in ('out_invoice','in_invoice'):
 216                                         val['base_code_id'] = tax['base_code_id']
 217                                         val['tax_code_id'] = tax['tax_code_id']
 218                                         val['base_amount'] = val['base'] * tax['base_sign']
 219                                         val['tax_amount'] = val['amount'] * tax['tax_sign']
 220                                         val['account_id'] = tax['account_collected_id'] or line.account_id.id
 221                                 else:
 222                                         val['base_code_id'] = tax['ref_base_code_id']
 223                                         val['tax_code_id'] = tax['ref_tax_code_id']
 224                                         val['base_amount'] = val['base'] * tax['ref_base_sign']
 225                                         val['tax_amount'] = val['amount'] * tax['ref_tax_sign']
 226                                         val['account_id'] = tax['account_paid_id'] or line.account_id.id
 227
 228                                 key = (val['tax_code_id'], val['base_code_id'], val['account_id'])
 229                                 if not key in tax_grouped:
 230                                         tax_grouped[key] = val
 231                                 else:
 232                                         tax_grouped[key]['amount'] += val['amount']
 233                                         tax_grouped[key]['base'] += val['base']
 234                                         tax_grouped[key]['base_amount'] += val['base_amount']
 235                                         tax_grouped[key]['tax_amount'] += val['tax_amount']
 236
 237                 return tax_grouped
 238 account_invoice_tax()
 239