[MERGE]:trunk
[odoo/odoo.git] / addons / account / account_invoice.py
1 # -*- coding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
6 #
7 #    This program is free software: you can redistribute it and/or modify
8 #    it under the terms of the GNU Affero General Public License as
9 #    published by the Free Software Foundation, either version 3 of the
10 #    License, or (at your option) any later version.
11 #
12 #    This program is distributed in the hope that it will be useful,
13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 #    GNU Affero General Public License for more details.
16 #
17 #    You should have received a copy of the GNU Affero General Public License
18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
19 #
20 ##############################################################################
21
22 import time
23 from lxml import etree
24 import decimal_precision as dp
25
26 import netsvc
27 import pooler
28 from osv import fields, osv, orm
29 from tools.translate import _
30
31 class account_invoice(osv.osv):
32     def _amount_all(self, cr, uid, ids, name, args, context=None):
33         res = {}
34         for invoice in self.browse(cr, uid, ids, context=context):
35             res[invoice.id] = {
36                 'amount_untaxed': 0.0,
37                 'amount_tax': 0.0,
38                 'amount_total': 0.0
39             }
40             for line in invoice.invoice_line:
41                 res[invoice.id]['amount_untaxed'] += line.price_subtotal
42             for line in invoice.tax_line:
43                 res[invoice.id]['amount_tax'] += line.amount
44             res[invoice.id]['amount_total'] = res[invoice.id]['amount_tax'] + res[invoice.id]['amount_untaxed']
45         return res
46
47     def _get_journal(self, cr, uid, context=None):
48         if context is None:
49             context = {}
50         type_inv = context.get('type', 'out_invoice')
51         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
52         company_id = context.get('company_id', user.company_id.id)
53         type2journal = {'out_invoice': 'sale', 'in_invoice': 'purchase', 'out_refund': 'sale_refund', 'in_refund': 'purchase_refund'}
54         journal_obj = self.pool.get('account.journal')
55         res = journal_obj.search(cr, uid, [('type', '=', type2journal.get(type_inv, 'sale')),
56                                             ('company_id', '=', company_id)],
57                                                 limit=1)
58         return res and res[0] or False
59
60     def _get_currency(self, cr, uid, context=None):
61         res = False
62         journal_id = self._get_journal(cr, uid, context=context)
63         if journal_id:
64             journal = self.pool.get('account.journal').browse(cr, uid, journal_id, context=context)
65             res = journal.currency and journal.currency.id or journal.company_id.currency_id.id
66         return res
67
68     def _get_journal_analytic(self, cr, uid, type_inv, context=None):
69         type2journal = {'out_invoice': 'sale', 'in_invoice': 'purchase', 'out_refund': 'sale', 'in_refund': 'purchase'}
70         tt = type2journal.get(type_inv, 'sale')
71         result = self.pool.get('account.analytic.journal').search(cr, uid, [('type','=',tt)], context=context)
72         if not result:
73             raise osv.except_osv(_('No Analytic Journal !'),_("You must define an analytic journal of type '%s'!") % (tt,))
74         return result[0]
75
76     def _get_type(self, cr, uid, context=None):
77         if context is None:
78             context = {}
79         return context.get('type', 'out_invoice')
80
81     def _reconciled(self, cr, uid, ids, name, args, context=None):
82         res = {}
83         for id in ids:
84             res[id] = self.test_paid(cr, uid, [id])
85         return res
86
87     def _get_reference_type(self, cr, uid, context=None):
88         return [('none', _('Free Reference'))]
89
90     def _amount_residual(self, cr, uid, ids, name, args, context=None):
91         result = {}
92         for invoice in self.browse(cr, uid, ids, context=context):
93             result[invoice.id] = 0.0
94             if invoice.move_id:
95                 for m in invoice.move_id.line_id:
96                     if m.account_id.type in ('receivable','payable'):
97                         result[invoice.id] += m.amount_residual_currency
98         return result
99
100     # Give Journal Items related to the payment reconciled to this invoice
101     # Return ids of partial and total payments related to the selected invoices
102     def _get_lines(self, cr, uid, ids, name, arg, context=None):
103         res = {}
104         for invoice in self.browse(cr, uid, ids, context=context):
105             id = invoice.id
106             res[id] = []
107             if not invoice.move_id:
108                 continue
109             data_lines = [x for x in invoice.move_id.line_id if x.account_id.id == invoice.account_id.id]
110             partial_ids = []
111             for line in data_lines:
112                 ids_line = []
113                 if line.reconcile_id:
114                     ids_line = line.reconcile_id.line_id
115                 elif line.reconcile_partial_id:
116                     ids_line = line.reconcile_partial_id.line_partial_ids
117                 l = map(lambda x: x.id, ids_line)
118                 partial_ids.append(line.id)
119                 res[id] =[x for x in l if x <> line.id and x not in partial_ids]
120         return res
121
122     def _get_invoice_line(self, cr, uid, ids, context=None):
123         result = {}
124         for line in self.pool.get('account.invoice.line').browse(cr, uid, ids, context=context):
125             result[line.invoice_id.id] = True
126         return result.keys()
127
128     def _get_invoice_tax(self, cr, uid, ids, context=None):
129         result = {}
130         for tax in self.pool.get('account.invoice.tax').browse(cr, uid, ids, context=context):
131             result[tax.invoice_id.id] = True
132         return result.keys()
133
134     def _compute_lines(self, cr, uid, ids, name, args, context=None):
135         result = {}
136         for invoice in self.browse(cr, uid, ids, context=context):
137             src = []
138             lines = []
139             if invoice.move_id:
140                 for m in invoice.move_id.line_id:
141                     temp_lines = []
142                     if m.reconcile_id:
143                         temp_lines = map(lambda x: x.id, m.reconcile_id.line_id)
144                     elif m.reconcile_partial_id:
145                         temp_lines = map(lambda x: x.id, m.reconcile_partial_id.line_partial_ids)
146                     lines += [x for x in temp_lines if x not in lines]
147                     src.append(m.id)
148
149             lines = filter(lambda x: x not in src, lines)
150             result[invoice.id] = lines
151         return result
152
153     def _get_invoice_from_line(self, cr, uid, ids, context=None):
154         move = {}
155         for line in self.pool.get('account.move.line').browse(cr, uid, ids, context=context):
156             if line.reconcile_partial_id:
157                 for line2 in line.reconcile_partial_id.line_partial_ids:
158                     move[line2.move_id.id] = True
159             if line.reconcile_id:
160                 for line2 in line.reconcile_id.line_id:
161                     move[line2.move_id.id] = True
162         invoice_ids = []
163         if move:
164             invoice_ids = self.pool.get('account.invoice').search(cr, uid, [('move_id','in',move.keys())], context=context)
165         return invoice_ids
166
167     def _get_invoice_from_reconcile(self, cr, uid, ids, context=None):
168         move = {}
169         for r in self.pool.get('account.move.reconcile').browse(cr, uid, ids, context=context):
170             for line in r.line_partial_ids:
171                 move[line.move_id.id] = True
172             for line in r.line_id:
173                 move[line.move_id.id] = True
174
175         invoice_ids = []
176         if move:
177             invoice_ids = self.pool.get('account.invoice').search(cr, uid, [('move_id','in',move.keys())], context=context)
178         return invoice_ids
179
180     _name = "account.invoice"
181     _inherit = ['mail.thread']
182     _description = 'Invoice'
183     _order = "id desc"
184
185     _columns = {
186         'name': fields.char('Description', size=64, select=True, readonly=True, states={'draft':[('readonly',False)]}),
187         'origin': fields.char('Source Document', size=64, help="Reference of the document that produced this invoice.", readonly=True, states={'draft':[('readonly',False)]}),
188         'supplier_invoice_number': fields.char('Supplier Invoice Number', size=64, help="The reference of this invoice as provided by the supplier.", readonly=True, states={'draft':[('readonly',False)]}),
189         'type': fields.selection([
190             ('out_invoice','Customer Invoice'),
191             ('in_invoice','Supplier Invoice'),
192             ('out_refund','Customer Refund'),
193             ('in_refund','Supplier Refund'),
194             ],'Type', readonly=True, select=True, change_default=True),
195
196         'number': fields.related('move_id','name', type='char', readonly=True, size=64, relation='account.move', store=True, string='Number'),
197         'internal_number': fields.char('Invoice Number', size=32, readonly=True, help="Unique number of the invoice, computed automatically when the invoice is created."),
198         'reference': fields.char('Invoice Reference', size=64, help="The partner reference of this invoice."),
199         'reference_type': fields.selection(_get_reference_type, 'Payment Reference',
200             required=True, readonly=True, states={'draft':[('readonly',False)]}),
201         'comment': fields.text('Additional Information'),
202
203         'state': fields.selection([
204             ('draft','Draft'),
205             ('proforma','Pro-forma'),
206             ('proforma2','Pro-forma'),
207             ('open','Open'),
208             ('paid','Paid'),
209             ('cancel','Cancelled'),
210             ],'Status', select=True, readonly=True,
211             help=' * The \'Draft\' status is used when a user is encoding a new and unconfirmed Invoice. \
212             \n* The \'Pro-forma\' when invoice is in Pro-forma status,invoice does not have an invoice number. \
213             \n* The \'Open\' status is used when user create invoice,a invoice number is generated.Its in open status till user does not pay invoice. \
214             \n* The \'Paid\' status is set automatically when the invoice is paid. Its related journal entries may or may not be reconciled. \
215             \n* The \'Cancelled\' status is used when user cancel invoice.'),
216         'sent': fields.boolean('Sent', readonly=True, help="It indicates that the invoice has been sent."),
217         'date_invoice': fields.date('Invoice Date', readonly=True, states={'draft':[('readonly',False)]}, select=True, help="Keep empty to use the current date"),
218         'date_due': fields.date('Due Date', readonly=True, states={'draft':[('readonly',False)]}, select=True,
219             help="If you use payment terms, the due date will be computed automatically at the generation "\
220                 "of accounting entries. If you keep the payment term and the due date empty, it means direct payment. The payment term may compute several due dates, for example 50% now, 50% in one month."),
221         'partner_id': fields.many2one('res.partner', 'Partner', change_default=True, readonly=True, required=True, states={'draft':[('readonly',False)]}),
222         'payment_term': fields.many2one('account.payment.term', 'Payment Term',readonly=True, states={'draft':[('readonly',False)]},
223             help="If you use payment terms, the due date will be computed automatically at the generation "\
224                 "of accounting entries. If you keep the payment term and the due date empty, it means direct payment. "\
225                 "The payment term may compute several due dates, for example 50% now, 50% in one month."),
226         'period_id': fields.many2one('account.period', 'Force Period', domain=[('state','<>','done')], help="Keep empty to use the period of the validation(invoice) date.", readonly=True, states={'draft':[('readonly',False)]}),
227
228         'account_id': fields.many2one('account.account', 'Account', required=True, readonly=True, states={'draft':[('readonly',False)]}, help="The partner account used for this invoice."),
229         'invoice_line': fields.one2many('account.invoice.line', 'invoice_id', 'Invoice Lines', readonly=True, states={'draft':[('readonly',False)]}),
230         'tax_line': fields.one2many('account.invoice.tax', 'invoice_id', 'Tax Lines', readonly=True, states={'draft':[('readonly',False)]}),
231
232         'move_id': fields.many2one('account.move', 'Journal Entry', readonly=True, select=1, ondelete='restrict', help="Link to the automatically generated Journal Items."),
233         'amount_untaxed': fields.function(_amount_all, digits_compute=dp.get_precision('Account'), string='Untaxed',
234             store={
235                 'account.invoice': (lambda self, cr, uid, ids, c={}: ids, ['invoice_line'], 20),
236                 'account.invoice.tax': (_get_invoice_tax, None, 20),
237                 'account.invoice.line': (_get_invoice_line, ['price_unit','invoice_line_tax_id','quantity','discount','invoice_id'], 20),
238             },
239             multi='all'),
240         'amount_tax': fields.function(_amount_all, digits_compute=dp.get_precision('Account'), string='Tax',
241             store={
242                 'account.invoice': (lambda self, cr, uid, ids, c={}: ids, ['invoice_line'], 20),
243                 'account.invoice.tax': (_get_invoice_tax, None, 20),
244                 'account.invoice.line': (_get_invoice_line, ['price_unit','invoice_line_tax_id','quantity','discount','invoice_id'], 20),
245             },
246             multi='all'),
247         'amount_total': fields.function(_amount_all, digits_compute=dp.get_precision('Account'), string='Total',
248             store={
249                 'account.invoice': (lambda self, cr, uid, ids, c={}: ids, ['invoice_line'], 20),
250                 'account.invoice.tax': (_get_invoice_tax, None, 20),
251                 'account.invoice.line': (_get_invoice_line, ['price_unit','invoice_line_tax_id','quantity','discount','invoice_id'], 20),
252             },
253             multi='all'),
254         'currency_id': fields.many2one('res.currency', 'Currency', required=True, readonly=True, states={'draft':[('readonly',False)]}),
255         'journal_id': fields.many2one('account.journal', 'Journal', required=True, readonly=True, states={'draft':[('readonly',False)]}),
256         'company_id': fields.many2one('res.company', 'Company', required=True, change_default=True, readonly=True, states={'draft':[('readonly',False)]}),
257         'check_total': fields.float('Verification Total', digits_compute=dp.get_precision('Account'), readonly=True, states={'draft':[('readonly',False)]}),
258         'reconciled': fields.function(_reconciled, string='Paid/Reconciled', type='boolean',
259             store={
260                 'account.invoice': (lambda self, cr, uid, ids, c={}: ids, None, 50), # Check if we can remove ?
261                 'account.move.line': (_get_invoice_from_line, None, 50),
262                 'account.move.reconcile': (_get_invoice_from_reconcile, None, 50),
263             }, help="It indicates that the invoice has been paid and the journal entry of the invoice has been reconciled with one or several journal entries of payment."),
264         'partner_bank_id': fields.many2one('res.partner.bank', 'Bank Account',
265             help='Bank Account Number to which the invoice will be paid. A Company bank account if this is a Customer Invoice or Supplier Refund, otherwise a Partner bank account number.', readonly=True, states={'draft':[('readonly',False)]}),
266         'move_lines':fields.function(_get_lines, type='many2many', relation='account.move.line', string='Entry Lines'),
267         'residual': fields.function(_amount_residual, digits_compute=dp.get_precision('Account'), string='Balance',
268             store={
269                 'account.invoice': (lambda self, cr, uid, ids, c={}: ids, ['invoice_line','move_id'], 50),
270                 'account.invoice.tax': (_get_invoice_tax, None, 50),
271                 'account.invoice.line': (_get_invoice_line, ['price_unit','invoice_line_tax_id','quantity','discount','invoice_id'], 50),
272                 'account.move.line': (_get_invoice_from_line, None, 50),
273                 'account.move.reconcile': (_get_invoice_from_reconcile, None, 50),
274             },
275             help="Remaining amount due."),
276         'payment_ids': fields.function(_compute_lines, relation='account.move.line', type="many2many", string='Payments'),
277         'move_name': fields.char('Journal Entry', size=64, readonly=True, states={'draft':[('readonly',False)]}),
278         'user_id': fields.many2one('res.users', 'Salesperson', readonly=True, states={'draft':[('readonly',False)]}),
279         'fiscal_position': fields.many2one('account.fiscal.position', 'Fiscal Position', readonly=True, states={'draft':[('readonly',False)]})
280     }
281     _defaults = {
282         'type': _get_type,
283         'state': 'draft',
284         'journal_id': _get_journal,
285         'currency_id': _get_currency,
286         'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.invoice', context=c),
287         'reference_type': 'none',
288         'check_total': 0.0,
289         'internal_number': False,
290         'user_id': lambda s, cr, u, c: u,
291         'sent': False,
292     }
293     _sql_constraints = [
294         ('number_uniq', 'unique(number, company_id, journal_id, type)', 'Invoice Number must be unique per Company!'),
295     ]
296
297     def fields_view_get(self, cr, uid, view_id=None, view_type=False, context=None, toolbar=False, submenu=False):
298         journal_obj = self.pool.get('account.journal')
299         if context is None:
300             context = {}
301
302         if context.get('active_model', '') in ['res.partner'] and context.get('active_ids', False) and context['active_ids']:
303             partner = self.pool.get(context['active_model']).read(cr, uid, context['active_ids'], ['supplier','customer'])[0]
304             if not view_type:
305                 view_id = self.pool.get('ir.ui.view').search(cr, uid, [('name', '=', 'account.invoice.tree')])
306                 view_type = 'tree'
307             if view_type == 'form':
308                 if partner['supplier'] and not partner['customer']:
309                     view_id = self.pool.get('ir.ui.view').search(cr,uid,[('name', '=', 'account.invoice.supplier.form')])
310                 elif partner['customer'] and not partner['supplier']:
311                     view_id = self.pool.get('ir.ui.view').search(cr,uid,[('name', '=', 'account.invoice.form')])
312         if view_id and isinstance(view_id, (list, tuple)):
313             view_id = view_id[0]
314         res = super(account_invoice,self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar, submenu=submenu)
315
316         type = context.get('journal_type', False)
317         for field in res['fields']:
318             if field == 'journal_id' and type:
319                 journal_select = journal_obj._name_search(cr, uid, '', [('type', '=', type)], context=context, limit=None, name_get_uid=1)
320                 res['fields'][field]['selection'] = journal_select
321
322         doc = etree.XML(res['arch'])
323
324         if context.get('type', False):
325             for node in doc.xpath("//field[@name='partner_bank_id']"):
326                 if context['type'] == 'in_refund':
327                     node.set('domain', "[('partner_id.ref_companies', 'in', [company_id])]")
328                 elif context['type'] == 'out_refund':
329                     node.set('domain', "[('partner_id', '=', partner_id)]")
330             res['arch'] = etree.tostring(doc)
331
332         if view_type == 'search':
333             if context.get('type', 'in_invoice') in ('out_invoice', 'out_refund'):
334                 for node in doc.xpath("//group[@name='extended filter']"):
335                     doc.remove(node)
336             res['arch'] = etree.tostring(doc)
337
338         if view_type == 'tree':
339             partner_string = _('Customer')
340             if context.get('type', 'out_invoice') in ('in_invoice', 'in_refund'):
341                 partner_string = _('Supplier')
342                 for node in doc.xpath("//field[@name='reference']"):
343                     node.set('invisible', '0')
344             for node in doc.xpath("//field[@name='partner_id']"):
345                 node.set('string', partner_string)
346             res['arch'] = etree.tostring(doc)
347         return res
348
349     def get_log_context(self, cr, uid, context=None):
350         if context is None:
351             context = {}
352         res = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account', 'invoice_form')
353         view_id = res and res[1] or False
354         context['view_id'] = view_id
355         return context
356
357     def create(self, cr, uid, vals, context=None):
358         if context is None:
359             context = {}
360         try:
361             res = super(account_invoice, self).create(cr, uid, vals, context)
362             if res:
363                 self.create_send_note(cr, uid, [res], context=context)
364             return res
365         except Exception, e:
366             if '"journal_id" viol' in e.args[0]:
367                 raise orm.except_orm(_('Configuration Error!'),
368                      _('There is no Sale/Purchase Journal(s) defined.'))
369             else:
370                 raise orm.except_orm(_('Unknown Error!'), str(e))
371
372     def invoice_print(self, cr, uid, ids, context=None):
373         '''
374         This function prints the invoice and mark it as sent, so that we can see more easily the next step of the workflow
375         '''
376         assert len(ids) == 1, 'This option should only be used for a single id at a time.'
377         self.write(cr, uid, ids, {'sent': True}, context=context)
378         datas = {
379              'ids': ids,
380              'model': 'account.invoice',
381              'form': self.read(cr, uid, ids[0], context=context)
382         }
383         return {
384             'type': 'ir.actions.report.xml',
385             'report_name': 'account.invoice',
386             'datas': datas,
387             'nodestroy' : True
388         }
389
390     def action_invoice_sent(self, cr, uid, ids, context=None):
391         '''
392         This function opens a window to compose an email, with the edi invoice template message loaded by default
393         '''
394         mod_obj = self.pool.get('ir.model.data')
395         template = mod_obj.get_object_reference(cr, uid, 'account', 'email_template_edi_invoice')
396         template_id = template and template[1] or False
397         res = mod_obj.get_object_reference(cr, uid, 'mail', 'email_compose_message_wizard_form')
398         res_id = res and res[1] or False
399         ctx = dict(context)
400         ctx.update({
401             'default_model': 'account.invoice',
402             'default_res_id': ids[0],
403             'default_use_template': True,
404             'default_template_id': template_id,
405             })
406         return {
407             'view_type': 'form',
408             'view_mode': 'form',
409             'res_model': 'mail.compose.message',
410             'views': [(res_id, 'form')],
411             'view_id': res_id,
412             'type': 'ir.actions.act_window',
413             'target': 'new',
414             'context': ctx,
415             'nodestroy': True,
416         }
417
418     def confirm_paid(self, cr, uid, ids, context=None):
419         if context is None:
420             context = {}
421         self.write(cr, uid, ids, {'state':'paid'}, context=context)
422         self.confirm_paid_send_note(cr, uid, ids, context=context)
423         return True
424
425     def unlink(self, cr, uid, ids, context=None):
426         if context is None:
427             context = {}
428         invoices = self.read(cr, uid, ids, ['state','internal_number'], context=context)
429         unlink_ids = []
430         for t in invoices:
431             if t['state'] in ('draft', 'cancel') and t['internal_number']== False:
432                 unlink_ids.append(t['id'])
433             else:
434                 raise osv.except_osv(_('Invalid Action!'), _('You cannot delete an invoice which is open or paid. You should refund it instead.'))
435         osv.osv.unlink(self, cr, uid, unlink_ids, context=context)
436         return True
437
438     def onchange_partner_id(self, cr, uid, ids, type, partner_id,\
439             date_invoice=False, payment_term=False, partner_bank_id=False, company_id=False):
440         invoice_addr_id = False
441         partner_payment_term = False
442         acc_id = False
443         bank_id = False
444         fiscal_position = False
445
446         opt = [('uid', str(uid))]
447         if partner_id:
448
449             opt.insert(0, ('id', partner_id))
450             res = self.pool.get('res.partner').address_get(cr, uid, [partner_id], ['invoice'])
451             invoice_addr_id = res['invoice']
452             p = self.pool.get('res.partner').browse(cr, uid, partner_id)
453             if company_id:
454                 if p.property_account_receivable.company_id.id != company_id and p.property_account_payable.company_id.id != company_id:
455                     property_obj = self.pool.get('ir.property')
456                     rec_pro_id = property_obj.search(cr,uid,[('name','=','property_account_receivable'),('res_id','=','res.partner,'+str(partner_id)+''),('company_id','=',company_id)])
457                     pay_pro_id = property_obj.search(cr,uid,[('name','=','property_account_payable'),('res_id','=','res.partner,'+str(partner_id)+''),('company_id','=',company_id)])
458                     if not rec_pro_id:
459                         rec_pro_id = property_obj.search(cr,uid,[('name','=','property_account_receivable'),('company_id','=',company_id)])
460                     if not pay_pro_id:
461                         pay_pro_id = property_obj.search(cr,uid,[('name','=','property_account_payable'),('company_id','=',company_id)])
462                     rec_line_data = property_obj.read(cr,uid,rec_pro_id,['name','value_reference','res_id'])
463                     pay_line_data = property_obj.read(cr,uid,pay_pro_id,['name','value_reference','res_id'])
464                     rec_res_id = rec_line_data and rec_line_data[0].get('value_reference',False) and int(rec_line_data[0]['value_reference'].split(',')[1]) or False
465                     pay_res_id = pay_line_data and pay_line_data[0].get('value_reference',False) and int(pay_line_data[0]['value_reference'].split(',')[1]) or False
466                     if not rec_res_id and not pay_res_id:
467                         raise osv.except_osv(_('Configuration Error!'),
468                             _('Cannot find a chart of accounts for this company, you should create one.'))
469                     account_obj = self.pool.get('account.account')
470                     rec_obj_acc = account_obj.browse(cr, uid, [rec_res_id])
471                     pay_obj_acc = account_obj.browse(cr, uid, [pay_res_id])
472                     p.property_account_receivable = rec_obj_acc[0]
473                     p.property_account_payable = pay_obj_acc[0]
474
475             if type in ('out_invoice', 'out_refund'):
476                 acc_id = p.property_account_receivable.id
477             else:
478                 acc_id = p.property_account_payable.id
479             fiscal_position = p.property_account_position and p.property_account_position.id or False
480             partner_payment_term = p.property_payment_term and p.property_payment_term.id or False
481             if p.bank_ids:
482                 bank_id = p.bank_ids[0].id
483
484         result = {'value': {
485             'account_id': acc_id,
486             'payment_term': partner_payment_term,
487             'fiscal_position': fiscal_position
488             }
489         }
490
491         if type in ('in_invoice', 'in_refund'):
492             result['value']['partner_bank_id'] = bank_id
493
494         if payment_term != partner_payment_term:
495             if partner_payment_term:
496                 to_update = self.onchange_payment_term_date_invoice(
497                     cr, uid, ids, partner_payment_term, date_invoice)
498                 result['value'].update(to_update['value'])
499             else:
500                 result['value']['date_due'] = False
501
502         if partner_bank_id != bank_id:
503             to_update = self.onchange_partner_bank(cr, uid, ids, bank_id)
504             result['value'].update(to_update['value'])
505         return result
506
507     def onchange_journal_id(self, cr, uid, ids, journal_id=False, context=None):
508         result = {}
509         if journal_id:
510             journal = self.pool.get('account.journal').browse(cr, uid, journal_id, context=context)
511             currency_id = journal.currency and journal.currency.id or journal.company_id.currency_id.id
512             company_id = journal.company_id.id
513             result = {'value': {
514                     'currency_id': currency_id,
515                     'company_id': company_id,
516                     }
517                 }
518         return result
519
520     def onchange_payment_term_date_invoice(self, cr, uid, ids, payment_term_id, date_invoice):
521         res = {}
522         if not payment_term_id:
523             return res
524         if not date_invoice:
525             date_invoice = time.strftime('%Y-%m-%d')
526         pterm_list = self.pool.get('account.payment.term').compute(cr, uid, payment_term_id, value=1, date_ref=date_invoice)
527         if pterm_list:
528             pterm_list = [line[0] for line in pterm_list]
529             pterm_list.sort()
530             res = {'value':{'date_due': pterm_list[-1]}}
531         else:
532              raise osv.except_osv(_('Insufficient Data!'), _('The payment term of supplier does not have a payment term line.'))
533         return res
534
535     def onchange_invoice_line(self, cr, uid, ids, lines):
536         return {}
537
538     def onchange_partner_bank(self, cursor, user, ids, partner_bank_id=False):
539         return {'value': {}}
540
541     def onchange_company_id(self, cr, uid, ids, company_id, part_id, type, invoice_line, currency_id):
542         val = {}
543         dom = {}
544         obj_journal = self.pool.get('account.journal')
545         account_obj = self.pool.get('account.account')
546         inv_line_obj = self.pool.get('account.invoice.line')
547         if company_id and part_id and type:
548             acc_id = False
549             partner_obj = self.pool.get('res.partner').browse(cr,uid,part_id)
550             if partner_obj.property_account_payable and partner_obj.property_account_receivable:
551                 if partner_obj.property_account_payable.company_id.id != company_id and partner_obj.property_account_receivable.company_id.id != company_id:
552                     property_obj = self.pool.get('ir.property')
553                     rec_pro_id = property_obj.search(cr, uid, [('name','=','property_account_receivable'),('res_id','=','res.partner,'+str(part_id)+''),('company_id','=',company_id)])
554                     pay_pro_id = property_obj.search(cr, uid, [('name','=','property_account_payable'),('res_id','=','res.partner,'+str(part_id)+''),('company_id','=',company_id)])
555                     if not rec_pro_id:
556                         rec_pro_id = property_obj.search(cr, uid, [('name','=','property_account_receivable'),('company_id','=',company_id)])
557                     if not pay_pro_id:
558                         pay_pro_id = property_obj.search(cr, uid, [('name','=','property_account_payable'),('company_id','=',company_id)])
559                     rec_line_data = property_obj.read(cr, uid, rec_pro_id, ['name','value_reference','res_id'])
560                     pay_line_data = property_obj.read(cr, uid, pay_pro_id, ['name','value_reference','res_id'])
561                     rec_res_id = rec_line_data and rec_line_data[0].get('value_reference',False) and int(rec_line_data[0]['value_reference'].split(',')[1]) or False
562                     pay_res_id = pay_line_data and pay_line_data[0].get('value_reference',False) and int(pay_line_data[0]['value_reference'].split(',')[1]) or False
563                     if not rec_res_id and not pay_res_id:
564                         raise osv.except_osv(_('Configuration Error!'),
565                             _('Cannot find a chart of account, you should create one from Settings\Configuration\Accounting menu.'))
566                     if type in ('out_invoice', 'out_refund'):
567                         acc_id = rec_res_id
568                     else:
569                         acc_id = pay_res_id
570                     val= {'account_id': acc_id}
571             if ids:
572                 if company_id:
573                     inv_obj = self.browse(cr,uid,ids)
574                     for line in inv_obj[0].invoice_line:
575                         if line.account_id:
576                             if line.account_id.company_id.id != company_id:
577                                 result_id = account_obj.search(cr, uid, [('name','=',line.account_id.name),('company_id','=',company_id)])
578                                 if not result_id:
579                                     raise osv.except_osv(_('Configuration Error!'),
580                                         _('Cannot find a chart of account, you should create one from Settings\Configuration\Accounting menu.'))
581                                 inv_line_obj.write(cr, uid, [line.id], {'account_id': result_id[-1]})
582             else:
583                 if invoice_line:
584                     for inv_line in invoice_line:
585                         obj_l = account_obj.browse(cr, uid, inv_line[2]['account_id'])
586                         if obj_l.company_id.id != company_id:
587                             raise osv.except_osv(_('Configuration Error!'),
588                                 _('Invoice line account\'s company and invoice\'s compnay does not match.'))
589                         else:
590                             continue
591         if company_id and type:
592             if type in ('out_invoice'):
593                 journal_type = 'sale'
594             elif type in ('out_refund'):
595                 journal_type = 'sale_refund'
596             elif type in ('in_refund'):
597                 journal_type = 'purchase_refund'
598             else:
599                 journal_type = 'purchase'
600             journal_ids = obj_journal.search(cr, uid, [('company_id','=',company_id), ('type', '=', journal_type)])
601             if journal_ids:
602                 val['journal_id'] = journal_ids[0]
603             ir_values_obj = self.pool.get('ir.values')
604             res_journal_default = ir_values_obj.get(cr, uid, 'default', 'type=%s' % (type), ['account.invoice'])
605             for r in res_journal_default:
606                 if r[1] == 'journal_id' and r[2] in journal_ids:
607                     val['journal_id'] = r[2]
608             if not val.get('journal_id', False):
609                 raise osv.except_osv(_('Configuration Error!'), (_('Cannot find any account journal of %s type for this company.\n\nYou can create one in the menu: \nConfiguration\Journals\Journals.') % (journal_type)))
610             dom = {'journal_id':  [('id', 'in', journal_ids)]}
611         else:
612             journal_ids = obj_journal.search(cr, uid, [])
613
614         return {'value': val, 'domain': dom}
615
616     # go from canceled state to draft state
617     def action_cancel_draft(self, cr, uid, ids, *args):
618         self.write(cr, uid, ids, {'state':'draft'})
619         wf_service = netsvc.LocalService("workflow")
620         for inv_id in ids:
621             wf_service.trg_delete(uid, 'account.invoice', inv_id, cr)
622             wf_service.trg_create(uid, 'account.invoice', inv_id, cr)
623         return True
624
625     # Workflow stuff
626     #################
627
628     # return the ids of the move lines which has the same account than the invoice
629     # whose id is in ids
630     def move_line_id_payment_get(self, cr, uid, ids, *args):
631         if not ids: return []
632         result = self.move_line_id_payment_gets(cr, uid, ids, *args)
633         return result.get(ids[0], [])
634
635     def move_line_id_payment_gets(self, cr, uid, ids, *args):
636         res = {}
637         if not ids: return res
638         cr.execute('SELECT i.id, l.id '\
639                    'FROM account_move_line l '\
640                    'LEFT JOIN account_invoice i ON (i.move_id=l.move_id) '\
641                    'WHERE i.id IN %s '\
642                    'AND l.account_id=i.account_id',
643                    (tuple(ids),))
644         for r in cr.fetchall():
645             res.setdefault(r[0], [])
646             res[r[0]].append( r[1] )
647         return res
648
649     def copy(self, cr, uid, id, default=None, context=None):
650         default = default or {}
651         default.update({
652             'state':'draft',
653             'number':False,
654             'move_id':False,
655             'move_name':False,
656             'internal_number': False,
657             'period_id': False,
658             'sent': False,
659         })
660         if 'date_invoice' not in default:
661             default.update({
662                 'date_invoice':False
663             })
664         if 'date_due' not in default:
665             default.update({
666                 'date_due':False
667             })
668         return super(account_invoice, self).copy(cr, uid, id, default, context)
669
670     def test_paid(self, cr, uid, ids, *args):
671         res = self.move_line_id_payment_get(cr, uid, ids)
672         if not res:
673             return False
674         ok = True
675         for id in res:
676             cr.execute('select reconcile_id from account_move_line where id=%s', (id,))
677             ok = ok and  bool(cr.fetchone()[0])
678         return ok
679
680     def button_reset_taxes(self, cr, uid, ids, context=None):
681         if context is None:
682             context = {}
683         ctx = context.copy()
684         ait_obj = self.pool.get('account.invoice.tax')
685         for id in ids:
686             cr.execute("DELETE FROM account_invoice_tax WHERE invoice_id=%s AND manual is False", (id,))
687             partner = self.browse(cr, uid, id, context=ctx).partner_id
688             if partner.lang:
689                 ctx.update({'lang': partner.lang})
690             for taxe in ait_obj.compute(cr, uid, id, context=ctx).values():
691                 ait_obj.create(cr, uid, taxe)
692         # Update the stored value (fields.function), so we write to trigger recompute
693         self.pool.get('account.invoice').write(cr, uid, ids, {'invoice_line':[]}, context=ctx)
694         return True
695
696     def button_compute(self, cr, uid, ids, context=None, set_total=False):
697         self.button_reset_taxes(cr, uid, ids, context)
698         for inv in self.browse(cr, uid, ids, context=context):
699             if set_total:
700                 self.pool.get('account.invoice').write(cr, uid, [inv.id], {'check_total': inv.amount_total})
701         return True
702
703     def _convert_ref(self, cr, uid, ref):
704         return (ref or '').replace('/','')
705
706     def _get_analytic_lines(self, cr, uid, id, context=None):
707         if context is None:
708             context = {}
709         inv = self.browse(cr, uid, id)
710         cur_obj = self.pool.get('res.currency')
711
712         company_currency = inv.company_id.currency_id.id
713         if inv.type in ('out_invoice', 'in_refund'):
714             sign = 1
715         else:
716             sign = -1
717
718         iml = self.pool.get('account.invoice.line').move_line_get(cr, uid, inv.id, context=context)
719         for il in iml:
720             if il['account_analytic_id']:
721                 if inv.type in ('in_invoice', 'in_refund'):
722                     ref = inv.reference
723                 else:
724                     ref = self._convert_ref(cr, uid, inv.number)
725                 if not inv.journal_id.analytic_journal_id:
726                     raise osv.except_osv(_('No Analytic Journal !'),_("You have to define an analytic journal on the '%s' journal!") % (inv.journal_id.name,))
727                 il['analytic_lines'] = [(0,0, {
728                     'name': il['name'],
729                     'date': inv['date_invoice'],
730                     'account_id': il['account_analytic_id'],
731                     'unit_amount': il['quantity'],
732                     'amount': cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, il['price'], context={'date': inv.date_invoice}) * sign,
733                     'product_id': il['product_id'],
734                     'product_uom_id': il['uos_id'],
735                     'general_account_id': il['account_id'],
736                     'journal_id': inv.journal_id.analytic_journal_id.id,
737                     'ref': ref,
738                 })]
739         return iml
740
741     def action_date_assign(self, cr, uid, ids, *args):
742         for inv in self.browse(cr, uid, ids):
743             res = self.onchange_payment_term_date_invoice(cr, uid, inv.id, inv.payment_term.id, inv.date_invoice)
744             if res and res['value']:
745                 self.write(cr, uid, [inv.id], res['value'])
746         return True
747
748     def finalize_invoice_move_lines(self, cr, uid, invoice_browse, move_lines):
749         """finalize_invoice_move_lines(cr, uid, invoice, move_lines) -> move_lines
750         Hook method to be overridden in additional modules to verify and possibly alter the
751         move lines to be created by an invoice, for special cases.
752         :param invoice_browse: browsable record of the invoice that is generating the move lines
753         :param move_lines: list of dictionaries with the account.move.lines (as for create())
754         :return: the (possibly updated) final move_lines to create for this invoice
755         """
756         return move_lines
757
758     def check_tax_lines(self, cr, uid, inv, compute_taxes, ait_obj):
759         if not inv.tax_line:
760             for tax in compute_taxes.values():
761                 ait_obj.create(cr, uid, tax)
762         else:
763             tax_key = []
764             for tax in inv.tax_line:
765                 if tax.manual:
766                     continue
767                 key = (tax.tax_code_id.id, tax.base_code_id.id, tax.account_id.id, tax.account_analytic_id.id)
768                 tax_key.append(key)
769                 if not key in compute_taxes:
770                     raise osv.except_osv(_('Warning!'), _('Global taxes defined, but they are not in invoice lines !'))
771                 base = compute_taxes[key]['base']
772                 if abs(base - tax.base) > inv.company_id.currency_id.rounding:
773                     raise osv.except_osv(_('Warning!'), _('Tax base different!\nClick on compute to update the tax base.'))
774             for key in compute_taxes:
775                 if not key in tax_key:
776                     raise osv.except_osv(_('Warning!'), _('Taxes are missing!\nClick on compute button.'))
777
778     def compute_invoice_totals(self, cr, uid, inv, company_currency, ref, invoice_move_lines, context=None):
779         if context is None:
780             context={}
781         total = 0
782         total_currency = 0
783         cur_obj = self.pool.get('res.currency')
784         for i in invoice_move_lines:
785             if inv.currency_id.id != company_currency:
786                 context.update({'date': inv.date_invoice or time.strftime('%Y-%m-%d')})
787                 i['currency_id'] = inv.currency_id.id
788                 i['amount_currency'] = i['price']
789                 i['price'] = cur_obj.compute(cr, uid, inv.currency_id.id,
790                         company_currency, i['price'],
791                         context=context)
792             else:
793                 i['amount_currency'] = False
794                 i['currency_id'] = False
795             i['ref'] = ref
796             if inv.type in ('out_invoice','in_refund'):
797                 total += i['price']
798                 total_currency += i['amount_currency'] or i['price']
799                 i['price'] = - i['price']
800             else:
801                 total -= i['price']
802                 total_currency -= i['amount_currency'] or i['price']
803         return total, total_currency, invoice_move_lines
804
805     def inv_line_characteristic_hashcode(self, invoice, invoice_line):
806         """Overridable hashcode generation for invoice lines. Lines having the same hashcode
807         will be grouped together if the journal has the 'group line' option. Of course a module
808         can add fields to invoice lines that would need to be tested too before merging lines
809         or not."""
810         return "%s-%s-%s-%s-%s"%(
811             invoice_line['account_id'],
812             invoice_line.get('tax_code_id',"False"),
813             invoice_line.get('product_id',"False"),
814             invoice_line.get('analytic_account_id',"False"),
815             invoice_line.get('date_maturity',"False"))
816
817     def group_lines(self, cr, uid, iml, line, inv):
818         """Merge account move lines (and hence analytic lines) if invoice line hashcodes are equals"""
819         if inv.journal_id.group_invoice_lines:
820             line2 = {}
821             for x, y, l in line:
822                 tmp = self.inv_line_characteristic_hashcode(inv, l)
823
824                 if tmp in line2:
825                     am = line2[tmp]['debit'] - line2[tmp]['credit'] + (l['debit'] - l['credit'])
826                     line2[tmp]['debit'] = (am > 0) and am or 0.0
827                     line2[tmp]['credit'] = (am < 0) and -am or 0.0
828                     line2[tmp]['tax_amount'] += l['tax_amount']
829                     line2[tmp]['analytic_lines'] += l['analytic_lines']
830                 else:
831                     line2[tmp] = l
832             line = []
833             for key, val in line2.items():
834                 line.append((0,0,val))
835         return line
836
837     def action_move_create(self, cr, uid, ids, context=None):
838         """Creates invoice related analytics and financial move lines"""
839         ait_obj = self.pool.get('account.invoice.tax')
840         cur_obj = self.pool.get('res.currency')
841         period_obj = self.pool.get('account.period')
842         payment_term_obj = self.pool.get('account.payment.term')
843         journal_obj = self.pool.get('account.journal')
844         move_obj = self.pool.get('account.move')
845         if context is None:
846             context = {}
847         for inv in self.browse(cr, uid, ids, context=context):
848             if not inv.journal_id.sequence_id:
849                 raise osv.except_osv(_('Error!'), _('Please define sequence on the journal related to this invoice.'))
850             if not inv.invoice_line:
851                 raise osv.except_osv(_('No Invoice Lines !'), _('Please create some invoice lines.'))
852             if inv.move_id:
853                 continue
854
855             ctx = context.copy()
856             ctx.update({'lang': inv.partner_id.lang})
857             if not inv.date_invoice:
858                 self.write(cr, uid, [inv.id], {'date_invoice': fields.date.context_today(self,cr,uid,context=context)}, context=ctx)
859             company_currency = inv.company_id.currency_id.id
860             # create the analytical lines
861             # one move line per invoice line
862             iml = self._get_analytic_lines(cr, uid, inv.id, context=ctx)
863             # check if taxes are all computed
864             compute_taxes = ait_obj.compute(cr, uid, inv.id, context=ctx)
865             self.check_tax_lines(cr, uid, inv, compute_taxes, ait_obj)
866
867             # I disabled the check_total feature
868             group_check_total_id = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account', 'group_supplier_inv_check_total')[1]
869             group_check_total = self.pool.get('res.groups').browse(cr, uid, group_check_total_id, context=context)
870             if group_check_total and uid in [x.id for x in group_check_total.users]:
871                 if (inv.type in ('in_invoice', 'in_refund') and abs(inv.check_total - inv.amount_total) >= (inv.currency_id.rounding/2.0)):
872                     raise osv.except_osv(_('Bad total !'), _('Please verify the price of the invoice !\nThe encoded total does not match the computed total.'))
873
874             if inv.payment_term:
875                 total_fixed = total_percent = 0
876                 for line in inv.payment_term.line_ids:
877                     if line.value == 'fixed':
878                         total_fixed += line.value_amount
879                     if line.value == 'procent':
880                         total_percent += line.value_amount
881                 total_fixed = (total_fixed * 100) / (inv.amount_total or 1.0)
882                 if (total_fixed + total_percent) > 100:
883                     raise osv.except_osv(_('Error!'), _("Cannot create the invoice.\nThe related payment term is probably misconfigured as it gives a computed amount greater than the total invoiced amount. In order to avoid rounding issues, the latest line of your payment term must be of type 'balance'."))
884
885             # one move line per tax line
886             iml += ait_obj.move_line_get(cr, uid, inv.id)
887
888             entry_type = ''
889             if inv.type in ('in_invoice', 'in_refund'):
890                 ref = inv.reference
891                 entry_type = 'journal_pur_voucher'
892                 if inv.type == 'in_refund':
893                     entry_type = 'cont_voucher'
894             else:
895                 ref = self._convert_ref(cr, uid, inv.number)
896                 entry_type = 'journal_sale_vou'
897                 if inv.type == 'out_refund':
898                     entry_type = 'cont_voucher'
899
900             diff_currency_p = inv.currency_id.id <> company_currency
901             # create one move line for the total and possibly adjust the other lines amount
902             total = 0
903             total_currency = 0
904             total, total_currency, iml = self.compute_invoice_totals(cr, uid, inv, company_currency, ref, iml, context=ctx)
905             acc_id = inv.account_id.id
906
907             name = inv['name'] or '/'
908             totlines = False
909             if inv.payment_term:
910                 totlines = payment_term_obj.compute(cr,
911                         uid, inv.payment_term.id, total, inv.date_invoice or False, context=ctx)
912             if totlines:
913                 res_amount_currency = total_currency
914                 i = 0
915                 ctx.update({'date': inv.date_invoice})
916                 for t in totlines:
917                     if inv.currency_id.id != company_currency:
918                         amount_currency = cur_obj.compute(cr, uid, company_currency, inv.currency_id.id, t[1], context=ctx)
919                     else:
920                         amount_currency = False
921
922                     # last line add the diff
923                     res_amount_currency -= amount_currency or 0
924                     i += 1
925                     if i == len(totlines):
926                         amount_currency += res_amount_currency
927
928                     iml.append({
929                         'type': 'dest',
930                         'name': name,
931                         'price': t[1],
932                         'account_id': acc_id,
933                         'date_maturity': t[0],
934                         'amount_currency': diff_currency_p \
935                                 and amount_currency or False,
936                         'currency_id': diff_currency_p \
937                                 and inv.currency_id.id or False,
938                         'ref': ref,
939                     })
940             else:
941                 iml.append({
942                     'type': 'dest',
943                     'name': name,
944                     'price': total,
945                     'account_id': acc_id,
946                     'date_maturity': inv.date_due or False,
947                     'amount_currency': diff_currency_p \
948                             and total_currency or False,
949                     'currency_id': diff_currency_p \
950                             and inv.currency_id.id or False,
951                     'ref': ref
952             })
953
954             date = inv.date_invoice or time.strftime('%Y-%m-%d')
955             part = inv.partner_id.id
956
957             line = map(lambda x:(0,0,self.line_get_convert(cr, uid, x, part, date, context=ctx)),iml)
958
959             line = self.group_lines(cr, uid, iml, line, inv)
960
961             journal_id = inv.journal_id.id
962             journal = journal_obj.browse(cr, uid, journal_id, context=ctx)
963             if journal.centralisation:
964                 raise osv.except_osv(_('User Error!'),
965                         _('You cannot create an invoice on a centralized journal. Uncheck the centralized counterpart box in the related journal from the configuration menu.'))
966
967             line = self.finalize_invoice_move_lines(cr, uid, inv, line)
968
969             move = {
970                 'ref': inv.reference and inv.reference or inv.name,
971                 'line_id': line,
972                 'journal_id': journal_id,
973                 'date': date,
974                 'narration':inv.comment
975             }
976             period_id = inv.period_id and inv.period_id.id or False
977             ctx.update({'company_id': inv.company_id.id})
978             if not period_id:
979                 period_ids = period_obj.find(cr, uid, inv.date_invoice, context=ctx)
980                 period_id = period_ids and period_ids[0] or False
981             if period_id:
982                 move['period_id'] = period_id
983                 for i in line:
984                     i[2]['period_id'] = period_id
985
986             move_id = move_obj.create(cr, uid, move, context=ctx)
987             new_move_name = move_obj.browse(cr, uid, move_id, context=ctx).name
988             # make the invoice point to that move
989             self.write(cr, uid, [inv.id], {'move_id': move_id,'period_id':period_id, 'move_name':new_move_name}, context=ctx)
990             # Pass invoice in context in method post: used if you want to get the same
991             # account move reference when creating the same invoice after a cancelled one:
992             ctx.update({'invoice':inv})
993             move_obj.post(cr, uid, [move_id], context=ctx)
994         self._log_event(cr, uid, ids)
995         return True
996
997     def invoice_validate(self, cr, uid, ids, context=None):
998         self.write(cr, uid, ids, {'state':'open'}, context=context)
999         return True
1000
1001     def line_get_convert(self, cr, uid, x, part, date, context=None):
1002         return {
1003             'date_maturity': x.get('date_maturity', False),
1004             'partner_id': part,
1005             'name': x['name'][:64],
1006             'date': date,
1007             'debit': x['price']>0 and x['price'],
1008             'credit': x['price']<0 and -x['price'],
1009             'account_id': x['account_id'],
1010             'analytic_lines': x.get('analytic_lines', []),
1011             'amount_currency': x['price']>0 and abs(x.get('amount_currency', False)) or -abs(x.get('amount_currency', False)),
1012             'currency_id': x.get('currency_id', False),
1013             'tax_code_id': x.get('tax_code_id', False),
1014             'tax_amount': x.get('tax_amount', False),
1015             'ref': x.get('ref', False),
1016             'quantity': x.get('quantity',1.00),
1017             'product_id': x.get('product_id', False),
1018             'product_uom_id': x.get('uos_id', False),
1019             'analytic_account_id': x.get('account_analytic_id', False),
1020         }
1021
1022     def action_number(self, cr, uid, ids, context=None):
1023         if context is None:
1024             context = {}
1025         #TODO: not correct fix but required a frech values before reading it.
1026         self.write(cr, uid, ids, {})
1027
1028         for obj_inv in self.browse(cr, uid, ids, context=context):
1029             id = obj_inv.id
1030             invtype = obj_inv.type
1031             number = obj_inv.number
1032             move_id = obj_inv.move_id and obj_inv.move_id.id or False
1033             reference = obj_inv.reference or ''
1034
1035             self.write(cr, uid, ids, {'internal_number':number})
1036
1037             if invtype in ('in_invoice', 'in_refund'):
1038                 if not reference:
1039                     ref = self._convert_ref(cr, uid, number)
1040                 else:
1041                     ref = reference
1042             else:
1043                 ref = self._convert_ref(cr, uid, number)
1044
1045             cr.execute('UPDATE account_move SET ref=%s ' \
1046                     'WHERE id=%s AND (ref is null OR ref = \'\')',
1047                     (ref, move_id))
1048             cr.execute('UPDATE account_move_line SET ref=%s ' \
1049                     'WHERE move_id=%s AND (ref is null OR ref = \'\')',
1050                     (ref, move_id))
1051             cr.execute('UPDATE account_analytic_line SET ref=%s ' \
1052                     'FROM account_move_line ' \
1053                     'WHERE account_move_line.move_id = %s ' \
1054                         'AND account_analytic_line.move_id = account_move_line.id',
1055                         (ref, move_id))
1056
1057             for inv_id, name in self.name_get(cr, uid, [id]):
1058                 ctx = context.copy()
1059                 if obj_inv.type in ('out_invoice', 'out_refund'):
1060                     ctx = self.get_log_context(cr, uid, context=ctx)
1061                 message = _("Invoice  '%s' is validated.") % name
1062                 self.message_post(cr, uid, [inv_id], body=message, context=context)
1063         return True
1064
1065     def action_cancel(self, cr, uid, ids, *args):
1066         context = {} # TODO: Use context from arguments
1067         account_move_obj = self.pool.get('account.move')
1068         invoices = self.read(cr, uid, ids, ['move_id', 'payment_ids'])
1069         move_ids = [] # ones that we will need to remove
1070         for i in invoices:
1071             if i['move_id']:
1072                 move_ids.append(i['move_id'][0])
1073             if i['payment_ids']:
1074                 account_move_line_obj = self.pool.get('account.move.line')
1075                 pay_ids = account_move_line_obj.browse(cr, uid, i['payment_ids'])
1076                 for move_line in pay_ids:
1077                     if move_line.reconcile_partial_id and move_line.reconcile_partial_id.line_partial_ids:
1078                         raise osv.except_osv(_('Error!'), _('You cannot cancel an invoice which is partially paid. You need to unreconcile related payment entries first.'))
1079
1080         # First, set the invoices as cancelled and detach the move ids
1081         self.write(cr, uid, ids, {'state':'cancel', 'move_id':False})
1082         if move_ids:
1083             # second, invalidate the move(s)
1084             account_move_obj.button_cancel(cr, uid, move_ids, context=context)
1085             # delete the move this invoice was pointing to
1086             # Note that the corresponding move_lines and move_reconciles
1087             # will be automatically deleted too
1088             account_move_obj.unlink(cr, uid, move_ids, context=context)
1089         self._log_event(cr, uid, ids, -1.0, 'Cancel Invoice')
1090         self.invoice_cancel_send_note(cr, uid, ids, context=context)
1091         return True
1092
1093     ###################
1094
1095     def list_distinct_taxes(self, cr, uid, ids):
1096         invoices = self.browse(cr, uid, ids)
1097         taxes = {}
1098         for inv in invoices:
1099             for tax in inv.tax_line:
1100                 if not tax['name'] in taxes:
1101                     taxes[tax['name']] = {'name': tax['name']}
1102         return taxes.values()
1103
1104     def _log_event(self, cr, uid, ids, factor=1.0, name='Open Invoice'):
1105         #TODO: implement messages system
1106         return True
1107
1108     def name_get(self, cr, uid, ids, context=None):
1109         if not ids:
1110             return []
1111         types = {
1112                 'out_invoice': 'Invoice ',
1113                 'in_invoice': 'Sup. Invoice ',
1114                 'out_refund': 'Refund ',
1115                 'in_refund': 'Supplier Refund ',
1116                 }
1117         return [(r['id'], (r['number']) or types[r['type']] + (r['name'] or '')) for r in self.read(cr, uid, ids, ['type', 'number', 'name'], context, load='_classic_write')]
1118
1119     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=100):
1120         if not args:
1121             args = []
1122         if context is None:
1123             context = {}
1124         ids = []
1125         if name:
1126             ids = self.search(cr, user, [('number','=',name)] + args, limit=limit, context=context)
1127         if not ids:
1128             ids = self.search(cr, user, [('name',operator,name)] + args, limit=limit, context=context)
1129         return self.name_get(cr, user, ids, context)
1130
1131     def _refund_cleanup_lines(self, cr, uid, lines):
1132         for line in lines:
1133             del line['id']
1134             del line['invoice_id']
1135             for field in ('company_id', 'partner_id', 'account_id', 'product_id',
1136                           'uos_id', 'account_analytic_id', 'tax_code_id', 'base_code_id'):
1137                 if line.get(field):
1138                     line[field] = line[field][0]
1139             if 'invoice_line_tax_id' in line:
1140                 line['invoice_line_tax_id'] = [(6,0, line.get('invoice_line_tax_id', [])) ]
1141         return map(lambda x: (0,0,x), lines)
1142
1143     def refund(self, cr, uid, ids, date=None, period_id=None, description=None, journal_id=None):
1144         invoices = self.read(cr, uid, ids, ['name', 'type', 'number', 'reference', 'comment', 'date_due', 'partner_id', 'partner_contact', 'partner_insite', 'partner_ref', 'payment_term', 'account_id', 'currency_id', 'invoice_line', 'tax_line', 'journal_id', 'company_id'])
1145         obj_invoice_line = self.pool.get('account.invoice.line')
1146         obj_invoice_tax = self.pool.get('account.invoice.tax')
1147         obj_journal = self.pool.get('account.journal')
1148         new_ids = []
1149         for invoice in invoices:
1150             del invoice['id']
1151
1152             type_dict = {
1153                 'out_invoice': 'out_refund', # Customer Invoice
1154                 'in_invoice': 'in_refund',   # Supplier Invoice
1155                 'out_refund': 'out_invoice', # Customer Refund
1156                 'in_refund': 'in_invoice',   # Supplier Refund
1157             }
1158
1159             invoice_lines = obj_invoice_line.read(cr, uid, invoice['invoice_line'])
1160             invoice_lines = self._refund_cleanup_lines(cr, uid, invoice_lines)
1161
1162             tax_lines = obj_invoice_tax.read(cr, uid, invoice['tax_line'])
1163             tax_lines = filter(lambda l: l['manual'], tax_lines)
1164             tax_lines = self._refund_cleanup_lines(cr, uid, tax_lines)
1165             if journal_id:
1166                 refund_journal_ids = [journal_id]
1167             elif invoice['type'] == 'in_invoice':
1168                 refund_journal_ids = obj_journal.search(cr, uid, [('type','=','purchase_refund')])
1169             else:
1170                 refund_journal_ids = obj_journal.search(cr, uid, [('type','=','sale_refund')])
1171
1172             if not date:
1173                 date = time.strftime('%Y-%m-%d')
1174             invoice.update({
1175                 'type': type_dict[invoice['type']],
1176                 'date_invoice': date,
1177                 'state': 'draft',
1178                 'number': False,
1179                 'invoice_line': invoice_lines,
1180                 'tax_line': tax_lines,
1181                 'journal_id': refund_journal_ids
1182             })
1183             if period_id:
1184                 invoice.update({
1185                     'period_id': period_id,
1186                 })
1187             if description:
1188                 invoice.update({
1189                     'name': description,
1190                 })
1191             # take the id part of the tuple returned for many2one fields
1192             for field in ('partner_id', 'company_id',
1193                     'account_id', 'currency_id', 'payment_term', 'journal_id'):
1194                 invoice[field] = invoice[field] and invoice[field][0]
1195             # create the new invoice
1196             new_ids.append(self.create(cr, uid, invoice))
1197
1198         return new_ids
1199
1200     def pay_and_reconcile(self, cr, uid, ids, pay_amount, pay_account_id, period_id, pay_journal_id, writeoff_acc_id, writeoff_period_id, writeoff_journal_id, context=None, name=''):
1201         if context is None:
1202             context = {}
1203         #TODO check if we can use different period for payment and the writeoff line
1204         assert len(ids)==1, "Can only pay one invoice at a time."
1205         invoice = self.browse(cr, uid, ids[0], context=context)
1206         src_account_id = invoice.account_id.id
1207         # Take the seq as name for move
1208         types = {'out_invoice': -1, 'in_invoice': 1, 'out_refund': 1, 'in_refund': -1}
1209         direction = types[invoice.type]
1210         #take the choosen date
1211         if 'date_p' in context and context['date_p']:
1212             date=context['date_p']
1213         else:
1214             date=time.strftime('%Y-%m-%d')
1215
1216         # Take the amount in currency and the currency of the payment
1217         if 'amount_currency' in context and context['amount_currency'] and 'currency_id' in context and context['currency_id']:
1218             amount_currency = context['amount_currency']
1219             currency_id = context['currency_id']
1220         else:
1221             amount_currency = False
1222             currency_id = False
1223
1224         pay_journal = self.pool.get('account.journal').read(cr, uid, pay_journal_id, ['type'], context=context)
1225         if invoice.type in ('in_invoice', 'out_invoice'):
1226             if pay_journal['type'] == 'bank':
1227                 entry_type = 'bank_pay_voucher' # Bank payment
1228             else:
1229                 entry_type = 'pay_voucher' # Cash payment
1230         else:
1231             entry_type = 'cont_voucher'
1232         if invoice.type in ('in_invoice', 'in_refund'):
1233             ref = invoice.reference
1234         else:
1235             ref = self._convert_ref(cr, uid, invoice.number)
1236         # Pay attention to the sign for both debit/credit AND amount_currency
1237         l1 = {
1238             'debit': direction * pay_amount>0 and direction * pay_amount,
1239             'credit': direction * pay_amount<0 and - direction * pay_amount,
1240             'account_id': src_account_id,
1241             'partner_id': invoice.partner_id.id,
1242             'ref':ref,
1243             'date': date,
1244             'currency_id':currency_id,
1245             'amount_currency':amount_currency and direction * amount_currency or 0.0,
1246             'company_id': invoice.company_id.id,
1247         }
1248         l2 = {
1249             'debit': direction * pay_amount<0 and - direction * pay_amount,
1250             'credit': direction * pay_amount>0 and direction * pay_amount,
1251             'account_id': pay_account_id,
1252             'partner_id': invoice.partner_id.id,
1253             'ref':ref,
1254             'date': date,
1255             'currency_id':currency_id,
1256             'amount_currency':amount_currency and - direction * amount_currency or 0.0,
1257             'company_id': invoice.company_id.id,
1258         }
1259
1260         if not name:
1261             name = invoice.invoice_line and invoice.invoice_line[0].name or invoice.number
1262         l1['name'] = name
1263         l2['name'] = name
1264
1265         lines = [(0, 0, l1), (0, 0, l2)]
1266         move = {'ref': ref, 'line_id': lines, 'journal_id': pay_journal_id, 'period_id': period_id, 'date': date}
1267         move_id = self.pool.get('account.move').create(cr, uid, move, context=context)
1268
1269         line_ids = []
1270         total = 0.0
1271         line = self.pool.get('account.move.line')
1272         move_ids = [move_id,]
1273         if invoice.move_id:
1274             move_ids.append(invoice.move_id.id)
1275         cr.execute('SELECT id FROM account_move_line '\
1276                    'WHERE move_id IN %s',
1277                    ((move_id, invoice.move_id.id),))
1278         lines = line.browse(cr, uid, map(lambda x: x[0], cr.fetchall()) )
1279         for l in lines+invoice.payment_ids:
1280             if l.account_id.id == src_account_id:
1281                 line_ids.append(l.id)
1282                 total += (l.debit or 0.0) - (l.credit or 0.0)
1283
1284         inv_id, name = self.name_get(cr, uid, [invoice.id], context=context)[0]
1285         if (not round(total,self.pool.get('decimal.precision').precision_get(cr, uid, 'Account'))) or writeoff_acc_id:
1286             self.pool.get('account.move.line').reconcile(cr, uid, line_ids, 'manual', writeoff_acc_id, writeoff_period_id, writeoff_journal_id, context)
1287         else:
1288             code = invoice.currency_id.symbol
1289             # TODO: use currency's formatting function
1290             msg = _("Invoice '%s' is paid partially: %s%s of %s%s (%s%s remaining).") % \
1291                     (name, pay_amount, code, invoice.amount_total, code, total, code)
1292             self.message_post(cr, uid, [inv_id], body=msg, context=context)
1293             self.pool.get('account.move.line').reconcile_partial(cr, uid, line_ids, 'manual', context)
1294
1295         # Update the stored value (fields.function), so we write to trigger recompute
1296         self.pool.get('account.invoice').write(cr, uid, ids, {}, context=context)
1297         return True
1298
1299     # -----------------------------------------
1300     # OpenChatter notifications and need_action
1301     # -----------------------------------------
1302
1303     def _get_document_type(self, type):
1304         type_dict = {
1305                 # Translation markers will have no effect at runtime, only used to properly flag export
1306                 'out_invoice': _('Customer invoice'),
1307                 'in_invoice': _('Supplier invoice'),
1308                 'out_refund': _('Customer Refund'),
1309                 'in_refund': _('Supplier Refund'),
1310         }
1311         return type_dict.get(type, 'Invoice')
1312
1313     def create_send_note(self, cr, uid, ids, context=None):
1314         for obj in self.browse(cr, uid, ids, context=context):
1315             self.message_post(cr, uid, [obj.id], body=_("%s <b>created</b>.") % (self._get_document_type(obj.type)),
1316                 subtype="account.mt_invoice_new", context=context)
1317
1318     def confirm_paid_send_note(self, cr, uid, ids, context=None):
1319          for obj in self.browse(cr, uid, ids, context=context):
1320             self.message_post(cr, uid, [obj.id], body=_("%s <b>paid</b>.") % (self._get_document_type(obj.type)),
1321                 subtype="account.mt_invoice_paid", context=context)
1322
1323     def invoice_cancel_send_note(self, cr, uid, ids, context=None):
1324         for obj in self.browse(cr, uid, ids, context=context):
1325             self.message_post(cr, uid, [obj.id], body=_("%s <b>cancelled</b>.") % (self._get_document_type(obj.type)),
1326                 context=context)
1327
1328
1329 class account_invoice_line(osv.osv):
1330
1331     def _amount_line(self, cr, uid, ids, prop, unknow_none, unknow_dict):
1332         res = {}
1333         tax_obj = self.pool.get('account.tax')
1334         cur_obj = self.pool.get('res.currency')
1335         for line in self.browse(cr, uid, ids):
1336             price = line.price_unit * (1-(line.discount or 0.0)/100.0)
1337             taxes = tax_obj.compute_all(cr, uid, line.invoice_line_tax_id, price, line.quantity, product=line.product_id, partner=line.invoice_id.partner_id)
1338             res[line.id] = taxes['total']
1339             if line.invoice_id:
1340                 cur = line.invoice_id.currency_id
1341                 res[line.id] = cur_obj.round(cr, uid, cur, res[line.id])
1342         return res
1343
1344     def _price_unit_default(self, cr, uid, context=None):
1345         if context is None:
1346             context = {}
1347         if context.get('check_total', False):
1348             t = context['check_total']
1349             for l in context.get('invoice_line', {}):
1350                 if isinstance(l, (list, tuple)) and len(l) >= 3 and l[2]:
1351                     tax_obj = self.pool.get('account.tax')
1352                     p = l[2].get('price_unit', 0) * (1-l[2].get('discount', 0)/100.0)
1353                     t = t - (p * l[2].get('quantity'))
1354                     taxes = l[2].get('invoice_line_tax_id')
1355                     if len(taxes[0]) >= 3 and taxes[0][2]:
1356                         taxes = tax_obj.browse(cr, uid, list(taxes[0][2]))
1357                         for tax in tax_obj.compute_all(cr, uid, taxes, p,l[2].get('quantity'), l[2].get('product_id', False), context.get('partner_id', False))['taxes']:
1358                             t = t - tax['amount']
1359             return t
1360         return 0
1361
1362     _name = "account.invoice.line"
1363     _description = "Invoice Line"
1364     _columns = {
1365         'name': fields.text('Description', required=True),
1366         'origin': fields.char('Source Document', size=256, help="Reference of the document that produced this invoice."),
1367         'sequence': fields.integer('Sequence', help="Gives the sequence of this line when displaying the invoice."),
1368         'invoice_id': fields.many2one('account.invoice', 'Invoice Reference', ondelete='cascade', select=True),
1369         'uos_id': fields.many2one('product.uom', 'Unit of Measure', ondelete='set null'),
1370         'product_id': fields.many2one('product.product', 'Product', ondelete='set null'),
1371         'account_id': fields.many2one('account.account', 'Account', required=True, domain=[('type','<>','view'), ('type', '<>', 'closed')], help="The income or expense account related to the selected product."),
1372         'price_unit': fields.float('Unit Price', required=True, digits_compute= dp.get_precision('Product Price')),
1373         'price_subtotal': fields.function(_amount_line, string='Subtotal', type="float",
1374             digits_compute= dp.get_precision('Account'), store=True),
1375         'quantity': fields.float('Quantity', digits_compute= dp.get_precision('Product Unit of Measure'), required=True),
1376         'discount': fields.float('Discount (%)', digits_compute= dp.get_precision('Discount')),
1377         'invoice_line_tax_id': fields.many2many('account.tax', 'account_invoice_line_tax', 'invoice_line_id', 'tax_id', 'Taxes', domain=[('parent_id','=',False)]),
1378         'account_analytic_id':  fields.many2one('account.analytic.account', 'Analytic Account'),
1379         'company_id': fields.related('invoice_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
1380         'partner_id': fields.related('invoice_id','partner_id',type='many2one',relation='res.partner',string='Partner',store=True)
1381     }
1382
1383     def _default_account_id(self, cr, uid, context=None):
1384         # XXX this gets the default account for the user's company,
1385         # it should get the default account for the invoice's company
1386         # however, the invoice's company does not reach this point
1387         prop = self.pool.get('ir.property').get(cr, uid, 'property_account_income_categ', 'product.category', context=context)
1388         return prop and prop.id or False
1389
1390     _defaults = {
1391         'quantity': 1,
1392         'discount': 0.0,
1393         'price_unit': _price_unit_default,
1394         'account_id': _default_account_id,
1395     }
1396
1397     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
1398         if context is None:
1399             context = {}
1400         res = super(account_invoice_line,self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar, submenu=submenu)
1401         if context.get('type', False):
1402             doc = etree.XML(res['arch'])
1403             for node in doc.xpath("//field[@name='product_id']"):
1404                 if context['type'] in ('in_invoice', 'in_refund'):
1405                     node.set('domain', "[('purchase_ok', '=', True)]")
1406                 else:
1407                     node.set('domain', "[('sale_ok', '=', True)]")
1408             res['arch'] = etree.tostring(doc)
1409         return res
1410
1411     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, currency_id=False, context=None, company_id=None):
1412         if context is None:
1413             context = {}
1414         company_id = company_id if company_id != None else context.get('company_id',False)
1415         context = dict(context)
1416         context.update({'company_id': company_id, 'force_company': company_id})
1417         if not partner_id:
1418             raise osv.except_osv(_('No Partner Defined !'),_("You must first select a partner !") )
1419         if not product:
1420             if type in ('in_invoice', 'in_refund'):
1421                 return {'value': {}, 'domain':{'product_uom':[]}}
1422             else:
1423                 return {'value': {'price_unit': 0.0}, 'domain':{'product_uom':[]}}
1424         part = self.pool.get('res.partner').browse(cr, uid, partner_id, context=context)
1425         fpos_obj = self.pool.get('account.fiscal.position')
1426         fpos = fposition_id and fpos_obj.browse(cr, uid, fposition_id, context=context) or False
1427
1428         if part.lang:
1429             context.update({'lang': part.lang})
1430         result = {}
1431         res = self.pool.get('product.product').browse(cr, uid, product, context=context)
1432
1433         if type in ('out_invoice','out_refund'):
1434             a = res.product_tmpl_id.property_account_income.id
1435             if not a:
1436                 a = res.categ_id.property_account_income_categ.id
1437         else:
1438             a = res.product_tmpl_id.property_account_expense.id
1439             if not a:
1440                 a = res.categ_id.property_account_expense_categ.id
1441         a = fpos_obj.map_account(cr, uid, fpos, a)
1442         if a:
1443             result['account_id'] = a
1444
1445         if type in ('out_invoice', 'out_refund'):
1446             taxes = res.taxes_id and res.taxes_id or (a and self.pool.get('account.account').browse(cr, uid, a, context=context).tax_ids or False)
1447         else:
1448             taxes = res.supplier_taxes_id and res.supplier_taxes_id or (a and self.pool.get('account.account').browse(cr, uid, a, context=context).tax_ids or False)
1449         tax_id = fpos_obj.map_tax(cr, uid, fpos, taxes)
1450
1451         if type in ('in_invoice', 'in_refund'):
1452             result.update( {'price_unit': price_unit or res.standard_price,'invoice_line_tax_id': tax_id} )
1453         else:
1454             result.update({'price_unit': res.list_price, 'invoice_line_tax_id': tax_id})
1455         result['name'] = res.partner_ref
1456
1457         domain = {}
1458         result['uos_id'] = res.uom_id.id or uom or False
1459         if res.description:
1460             result['name'] += '\n'+res.description
1461         if result['uos_id']:
1462             res2 = res.uom_id.category_id.id
1463             if res2:
1464                 domain = {'uos_id':[('category_id','=',res2 )]}
1465
1466         res_final = {'value':result, 'domain':domain}
1467
1468         if not company_id or not currency_id:
1469             return res_final
1470
1471         company = self.pool.get('res.company').browse(cr, uid, company_id, context=context)
1472         currency = self.pool.get('res.currency').browse(cr, uid, currency_id, context=context)
1473
1474         if company.currency_id.id != currency.id:
1475             if type in ('in_invoice', 'in_refund'):
1476                 res_final['value']['price_unit'] = res.standard_price
1477             new_price = res_final['value']['price_unit'] * currency.rate
1478             res_final['value']['price_unit'] = new_price
1479
1480         if uom:
1481             uom = self.pool.get('product.uom').browse(cr, uid, uom, context=context)
1482             if res.uom_id.category_id.id == uom.category_id.id:
1483                 new_price = res_final['value']['price_unit'] * uom.factor_inv
1484                 res_final['value']['price_unit'] = new_price
1485         return res_final
1486
1487     def uos_id_change(self, cr, uid, ids, product, uom, qty=0, name='', type='out_invoice', partner_id=False, fposition_id=False, price_unit=False, currency_id=False, context=None, company_id=None):
1488         if context is None:
1489             context = {}
1490         company_id = company_id if company_id != None else context.get('company_id',False)
1491         context = dict(context)
1492         context.update({'company_id': company_id})
1493         warning = {}
1494         res = self.product_id_change(cr, uid, ids, product, uom, qty, name, type, partner_id, fposition_id, price_unit, currency_id, context=context)
1495         if 'uos_id' in res['value']:
1496             del res['value']['uos_id']
1497         if not uom:
1498             res['value']['price_unit'] = 0.0
1499         if product and uom:
1500             prod = self.pool.get('product.product').browse(cr, uid, product, context=context)
1501             prod_uom = self.pool.get('product.uom').browse(cr, uid, uom, context=context)
1502             if prod.uom_id.category_id.id != prod_uom.category_id.id:
1503                 warning = {
1504                     'title': _('Warning!'),
1505                     'message': _('The selected unit of measure is not compatible with the unit of measure of the product.')
1506                 }
1507                 res['value'].update({'uos_id': prod.uom_id.id})
1508             return {'value': res['value'], 'warning': warning}
1509         return res
1510
1511     def move_line_get(self, cr, uid, invoice_id, context=None):
1512         res = []
1513         tax_obj = self.pool.get('account.tax')
1514         cur_obj = self.pool.get('res.currency')
1515         if context is None:
1516             context = {}
1517         inv = self.pool.get('account.invoice').browse(cr, uid, invoice_id, context=context)
1518         company_currency = inv.company_id.currency_id.id
1519
1520         for line in inv.invoice_line:
1521             mres = self.move_line_get_item(cr, uid, line, context)
1522             if not mres:
1523                 continue
1524             res.append(mres)
1525             tax_code_found= False
1526             for tax in tax_obj.compute_all(cr, uid, line.invoice_line_tax_id,
1527                     (line.price_unit * (1.0 - (line['discount'] or 0.0) / 100.0)),
1528                     line.quantity, line.product_id,
1529                     inv.partner_id)['taxes']:
1530
1531                 if inv.type in ('out_invoice', 'in_invoice'):
1532                     tax_code_id = tax['base_code_id']
1533                     tax_amount = line.price_subtotal * tax['base_sign']
1534                 else:
1535                     tax_code_id = tax['ref_base_code_id']
1536                     tax_amount = line.price_subtotal * tax['ref_base_sign']
1537
1538                 if tax_code_found:
1539                     if not tax_code_id:
1540                         continue
1541                     res.append(self.move_line_get_item(cr, uid, line, context))
1542                     res[-1]['price'] = 0.0
1543                     res[-1]['account_analytic_id'] = False
1544                 elif not tax_code_id:
1545                     continue
1546                 tax_code_found = True
1547
1548                 res[-1]['tax_code_id'] = tax_code_id
1549                 res[-1]['tax_amount'] = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, tax_amount, context={'date': inv.date_invoice})
1550         return res
1551
1552     def move_line_get_item(self, cr, uid, line, context=None):
1553         return {
1554             'type':'src',
1555             'name': line.name.split('\n')[0][:64],
1556             'price_unit':line.price_unit,
1557             'quantity':line.quantity,
1558             'price':line.price_subtotal,
1559             'account_id':line.account_id.id,
1560             'product_id':line.product_id.id,
1561             'uos_id':line.uos_id.id,
1562             'account_analytic_id':line.account_analytic_id.id,
1563             'taxes':line.invoice_line_tax_id,
1564         }
1565     #
1566     # Set the tax field according to the account and the fiscal position
1567     #
1568     def onchange_account_id(self, cr, uid, ids, product_id, partner_id, inv_type, fposition_id, account_id):
1569         if not account_id:
1570             return {}
1571         unique_tax_ids = []
1572         fpos = fposition_id and self.pool.get('account.fiscal.position').browse(cr, uid, fposition_id) or False
1573         account = self.pool.get('account.account').browse(cr, uid, account_id)
1574         if not product_id:
1575             taxes = account.tax_ids
1576             unique_tax_ids = self.pool.get('account.fiscal.position').map_tax(cr, uid, fpos, taxes)
1577         else:
1578             product_change_result = self.product_id_change(cr, uid, ids, product_id, False, type=inv_type,
1579                 partner_id=partner_id, fposition_id=fposition_id,
1580                 company_id=account.company_id.id)
1581             if product_change_result and 'value' in product_change_result and 'invoice_line_tax_id' in product_change_result['value']:
1582                 unique_tax_ids = product_change_result['value']['invoice_line_tax_id']
1583         return {'value':{'invoice_line_tax_id': unique_tax_ids}}
1584
1585 account_invoice_line()
1586
1587 class account_invoice_tax(osv.osv):
1588     _name = "account.invoice.tax"
1589     _description = "Invoice Tax"
1590
1591     def _count_factor(self, cr, uid, ids, name, args, context=None):
1592         res = {}
1593         for invoice_tax in self.browse(cr, uid, ids, context=context):
1594             res[invoice_tax.id] = {
1595                 'factor_base': 1.0,
1596                 'factor_tax': 1.0,
1597             }
1598             if invoice_tax.amount <> 0.0:
1599                 factor_tax = invoice_tax.tax_amount / invoice_tax.amount
1600                 res[invoice_tax.id]['factor_tax'] = factor_tax
1601
1602             if invoice_tax.base <> 0.0:
1603                 factor_base = invoice_tax.base_amount / invoice_tax.base
1604                 res[invoice_tax.id]['factor_base'] = factor_base
1605
1606         return res
1607
1608     _columns = {
1609         'invoice_id': fields.many2one('account.invoice', 'Invoice Line', ondelete='cascade', select=True),
1610         'name': fields.char('Tax Description', size=64, required=True),
1611         'account_id': fields.many2one('account.account', 'Tax Account', required=True, domain=[('type','<>','view'),('type','<>','income'), ('type', '<>', 'closed')]),
1612         'account_analytic_id': fields.many2one('account.analytic.account', 'Analytic account'),
1613         'base': fields.float('Base', digits_compute=dp.get_precision('Account')),
1614         'amount': fields.float('Amount', digits_compute=dp.get_precision('Account')),
1615         'manual': fields.boolean('Manual'),
1616         'sequence': fields.integer('Sequence', help="Gives the sequence order when displaying a list of invoice tax."),
1617         'base_code_id': fields.many2one('account.tax.code', 'Base Code', help="The account basis of the tax declaration."),
1618         'base_amount': fields.float('Base Code Amount', digits_compute=dp.get_precision('Account')),
1619         'tax_code_id': fields.many2one('account.tax.code', 'Tax Code', help="The tax basis of the tax declaration."),
1620         'tax_amount': fields.float('Tax Code Amount', digits_compute=dp.get_precision('Account')),
1621         'company_id': fields.related('account_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True),
1622         'factor_base': fields.function(_count_factor, string='Multipication factor for Base code', type='float', multi="all"),
1623         'factor_tax': fields.function(_count_factor, string='Multipication factor Tax code', type='float', multi="all")
1624     }
1625
1626     def base_change(self, cr, uid, ids, base, currency_id=False, company_id=False, date_invoice=False):
1627         cur_obj = self.pool.get('res.currency')
1628         company_obj = self.pool.get('res.company')
1629         company_currency = False
1630         factor = 1
1631         if ids:
1632             factor = self.read(cr, uid, ids[0], ['factor_base'])['factor_base']
1633         if company_id:
1634             company_currency = company_obj.read(cr, uid, [company_id], ['currency_id'])[0]['currency_id'][0]
1635         if currency_id and company_currency:
1636             base = cur_obj.compute(cr, uid, currency_id, company_currency, base*factor, context={'date': date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1637         return {'value': {'base_amount':base}}
1638
1639     def amount_change(self, cr, uid, ids, amount, currency_id=False, company_id=False, date_invoice=False):
1640         cur_obj = self.pool.get('res.currency')
1641         company_obj = self.pool.get('res.company')
1642         company_currency = False
1643         factor = 1
1644         if ids:
1645             factor = self.read(cr, uid, ids[0], ['factor_tax'])['factor_tax']
1646         if company_id:
1647             company_currency = company_obj.read(cr, uid, [company_id], ['currency_id'])[0]['currency_id'][0]
1648         if currency_id and company_currency:
1649             amount = cur_obj.compute(cr, uid, currency_id, company_currency, amount*factor, context={'date': date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1650         return {'value': {'tax_amount': amount}}
1651
1652     _order = 'sequence'
1653     _defaults = {
1654         'manual': 1,
1655         'base_amount': 0.0,
1656         'tax_amount': 0.0,
1657     }
1658     def compute(self, cr, uid, invoice_id, context=None):
1659         tax_grouped = {}
1660         tax_obj = self.pool.get('account.tax')
1661         cur_obj = self.pool.get('res.currency')
1662         inv = self.pool.get('account.invoice').browse(cr, uid, invoice_id, context=context)
1663         cur = inv.currency_id
1664         company_currency = inv.company_id.currency_id.id
1665
1666         for line in inv.invoice_line:
1667             for tax in tax_obj.compute_all(cr, uid, line.invoice_line_tax_id, (line.price_unit* (1-(line.discount or 0.0)/100.0)), line.quantity, line.product_id, inv.partner_id)['taxes']:
1668                 val={}
1669                 val['invoice_id'] = inv.id
1670                 val['name'] = tax['name']
1671                 val['amount'] = tax['amount']
1672                 val['manual'] = False
1673                 val['sequence'] = tax['sequence']
1674                 val['base'] = cur_obj.round(cr, uid, cur, tax['price_unit'] * line['quantity'])
1675
1676                 if inv.type in ('out_invoice','in_invoice'):
1677                     val['base_code_id'] = tax['base_code_id']
1678                     val['tax_code_id'] = tax['tax_code_id']
1679                     val['base_amount'] = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, val['base'] * tax['base_sign'], context={'date': inv.date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1680                     val['tax_amount'] = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, val['amount'] * tax['tax_sign'], context={'date': inv.date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1681                     val['account_id'] = tax['account_collected_id'] or line.account_id.id
1682                     val['account_analytic_id'] = tax['account_analytic_collected_id']
1683                 else:
1684                     val['base_code_id'] = tax['ref_base_code_id']
1685                     val['tax_code_id'] = tax['ref_tax_code_id']
1686                     val['base_amount'] = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, val['base'] * tax['ref_base_sign'], context={'date': inv.date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1687                     val['tax_amount'] = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, val['amount'] * tax['ref_tax_sign'], context={'date': inv.date_invoice or time.strftime('%Y-%m-%d')}, round=False)
1688                     val['account_id'] = tax['account_paid_id'] or line.account_id.id
1689                     val['account_analytic_id'] = tax['account_analytic_paid_id']
1690
1691                 key = (val['tax_code_id'], val['base_code_id'], val['account_id'], val['account_analytic_id'])
1692                 if not key in tax_grouped:
1693                     tax_grouped[key] = val
1694                 else:
1695                     tax_grouped[key]['amount'] += val['amount']
1696                     tax_grouped[key]['base'] += val['base']
1697                     tax_grouped[key]['base_amount'] += val['base_amount']
1698                     tax_grouped[key]['tax_amount'] += val['tax_amount']
1699
1700         for t in tax_grouped.values():
1701             t['base'] = cur_obj.round(cr, uid, cur, t['base'])
1702             t['amount'] = cur_obj.round(cr, uid, cur, t['amount'])
1703             t['base_amount'] = cur_obj.round(cr, uid, cur, t['base_amount'])
1704             t['tax_amount'] = cur_obj.round(cr, uid, cur, t['tax_amount'])
1705         return tax_grouped
1706
1707     def move_line_get(self, cr, uid, invoice_id):
1708         res = []
1709         cr.execute('SELECT * FROM account_invoice_tax WHERE invoice_id=%s', (invoice_id,))
1710         for t in cr.dictfetchall():
1711             if not t['amount'] \
1712                     and not t['tax_code_id'] \
1713                     and not t['tax_amount']:
1714                 continue
1715             res.append({
1716                 'type':'tax',
1717                 'name':t['name'],
1718                 'price_unit': t['amount'],
1719                 'quantity': 1,
1720                 'price': t['amount'] or 0.0,
1721                 'account_id': t['account_id'],
1722                 'tax_code_id': t['tax_code_id'],
1723                 'tax_amount': t['tax_amount'],
1724                 'account_analytic_id': t['account_analytic_id'],
1725             })
1726         return res
1727
1728 account_invoice_tax()
1729
1730
1731 class res_partner(osv.osv):
1732     """ Inherits partner and adds invoice information in the partner form """
1733     _inherit = 'res.partner'
1734     _columns = {
1735         'invoice_ids': fields.one2many('account.invoice.line', 'partner_id', 'Invoices', readonly=True),
1736     }
1737
1738     def copy(self, cr, uid, id, default=None, context=None):
1739         default = default or {}
1740         default.update({'invoice_ids' : []})
1741         return super(res_partner, self).copy(cr, uid, id, default, context)
1742
1743 res_partner()
1744
1745 class mail_message(osv.osv):
1746     _name = 'mail.message'
1747     _inherit = 'mail.message'
1748
1749     def _postprocess_sent_message(self, cr, uid, message, context=None):
1750         if message.model == 'account.invoice':
1751             self.pool.get('account.invoice').write(cr, uid, [message.res_id], {'sent':True}, context=context)
1752         return super(mail_message, self)._postprocess_sent_message(cr, uid, message=message, context=context)
1753
1754 mail_message()
1755 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: