[IMP]Replace class flow
[odoo/odoo.git] / addons / account / wizard / account_invoice_state.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 osv import osv
23 from tools.translate import _
24 import netsvc
25 import pooler
26
27 class account_invoice_confirm(osv.osv_memory):
28     """
29     This wizard will confirm the all the selected draft invoices
30     """
31
32     _name = "account.invoice.confirm"
33     _description = "Confirm the selected invoices"
34
35     def invoice_confirm(self, cr, uid, ids, context=None):
36         wf_service = netsvc.LocalService('workflow')
37         if context is None:
38             context = {}
39         pool_obj = pooler.get_pool(cr.dbname)
40         data_inv = pool_obj.get('account.invoice').read(cr, uid, context['active_ids'], ['state'], context=context)
41
42         for record in data_inv:
43             if record['state'] not in ('draft','proforma','proforma2'):
44                 raise osv.except_osv(_('Warning'), _("Selected Invoice(s) cannot be confirmed as they are not in 'Draft' or 'Pro-Forma' state!"))
45             wf_service.trg_validate(uid, 'account.invoice', record['id'], 'invoice_open', cr)
46         return {'type': 'ir.actions.act_window_close'}
47
48 account_invoice_confirm()
49
50 class account_invoice_cancel(osv.osv_memory):
51     """
52     This wizard will cancel the all the selected invoices.
53     If in the journal, the option allow cancelling entry is not selected then it will give warning message.
54     """
55
56     _name = "account.invoice.cancel"
57     _description = "Cancel the Selected Invoices"
58
59     def invoice_cancel(self, cr, uid, ids, context=None):
60         if context is None:
61             context = {}
62         wf_service = netsvc.LocalService('workflow')
63         pool_obj = pooler.get_pool(cr.dbname)
64         data_inv = pool_obj.get('account.invoice').read(cr, uid, context['active_ids'], ['state'], context=context)
65
66         for record in data_inv:
67             if record['state'] in ('cancel','paid'):
68                 raise osv.except_osv(_('Warning'), _("Selected Invoice(s) cannot be cancelled as they are already in 'Cancelled' or 'Done' state!"))
69             wf_service.trg_validate(uid, 'account.invoice', record['id'], 'invoice_cancel', cr)
70         return {'type': 'ir.actions.act_window_close'}
71
72 account_invoice_cancel()
73
74 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: