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