[FIX] stock_invoice_directly: if picking is not done while doing partial picking...
[odoo/odoo.git] / addons / stock_invoice_directly / wizard / stock_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 from openerp.osv import fields, osv
23
24 class invoice_directly(osv.osv_memory):
25     _inherit = 'stock.partial.picking'
26
27     def do_partial(self, cr, uid, ids, context=None):
28         """Launch Create invoice wizard if invoice state is To be Invoiced,
29            after processing the partial picking.
30         """
31         if context is None: context = {}
32         result = super(invoice_directly, self).do_partial(cr, uid, ids, context)
33         partial = self.browse(cr, uid, ids[0], context)
34         if partial.picking_id.state != 'done' and partial.picking_id.backorder_id:
35             # delivery is not finished, opening invoice on backorder
36             picking = partial.picking_id.backorder_id
37         else:
38             picking = partial.picking_id
39         context.update(active_model='stock.picking',
40                        active_ids=[picking.id])
41         if picking.invoice_state == '2binvoiced':
42             return {
43                 'name': 'Create Invoice',
44                 'view_type': 'form',
45                 'view_mode': 'form',
46                 'res_model': 'stock.invoice.onshipping',
47                 'type': 'ir.actions.act_window',
48                 'target': 'new',
49                 'context': context
50             }
51         return {'type': 'ir.actions.act_window_close'}
52
53 invoice_directly()
54
55 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: