[fix] added exception in o2m when we query the value and the view is not yet intialized
[odoo/odoo.git] / addons / account_voucher / wizard / account_statement_from_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
24 from osv import fields, osv
25 from tools.translate import _
26
27 class account_statement_from_invoice_lines(osv.osv_memory):
28     """
29     Generate Entries by Statement from Invoices
30     """
31     _name = "account.statement.from.invoice.lines"
32     _description = "Entries by Statement from Invoices"
33     _columns = {
34         'line_ids': fields.many2many('account.move.line', 'account_move_line_relation', 'move_id', 'line_id', 'Invoices'),
35     }
36
37     def populate_statement(self, cr, uid, ids, context=None):
38         if context is None:
39             context = {}
40         statement_id = context.get('statement_id', False)
41         if not statement_id:
42             return {'type': 'ir.actions.act_window_close'}
43         data =  self.read(cr, uid, ids, context=context)[0]
44         line_ids = data['line_ids']
45         if not line_ids:
46             return {'type': 'ir.actions.act_window_close'}
47
48         line_obj = self.pool.get('account.move.line')
49         statement_obj = self.pool.get('account.bank.statement')
50         statement_line_obj = self.pool.get('account.bank.statement.line')
51         currency_obj = self.pool.get('res.currency')
52         voucher_obj = self.pool.get('account.voucher')
53         voucher_line_obj = self.pool.get('account.voucher.line')
54         line_date = time.strftime('%Y-%m-%d')
55         statement = statement_obj.browse(cr, uid, statement_id, context=context)
56
57         # for each selected move lines
58         for line in line_obj.browse(cr, uid, line_ids, context=context):
59             voucher_res = {}
60             ctx = context.copy()
61             #  take the date for computation of currency => use payment date
62             ctx['date'] = line_date
63             amount = 0.0
64
65             if line.debit > 0:
66                 amount = line.debit
67             elif line.credit > 0:
68                 amount = -line.credit
69
70             if line.amount_currency:
71                 amount = currency_obj.compute(cr, uid, line.currency_id.id,
72                     statement.currency.id, line.amount_currency, context=ctx)
73             elif (line.invoice and line.invoice.currency_id.id <> statement.currency.id):
74                 amount = currency_obj.compute(cr, uid, line.invoice.currency_id.id,
75                     statement.currency.id, amount, context=ctx)
76
77             context.update({'move_line_ids': [line.id]})
78             result = voucher_obj.onchange_partner_id(cr, uid, [], partner_id=line.partner_id.id, journal_id=statement.journal_id.id, amount=abs(amount), currency_id= statement.currency.id, ttype=(amount < 0 and 'payment' or 'receipt'), date=line_date, context=context)
79             voucher_res = { 'type':(amount < 0 and 'payment' or 'receipt'),
80                             'name': line.name,
81                             'partner_id': line.partner_id.id,
82                             'journal_id': statement.journal_id.id,
83                             'account_id': result.get('account_id', statement.journal_id.default_credit_account_id.id), # improve me: statement.journal_id.default_credit_account_id.id
84                             'company_id':statement.company_id.id,
85                             'currency_id':statement.currency.id,
86                             'date':line.date,
87                             'amount':abs(amount),
88                             'period_id':statement.period_id.id}
89             voucher_id = voucher_obj.create(cr, uid, voucher_res, context=context)
90
91             voucher_line_dict =  {}
92             if result['value']['line_ids']:
93                 for line_dict in result['value']['line_ids']:
94                     move_line = line_obj.browse(cr, uid, line_dict['move_line_id'], context)
95                     if line.move_id.id == move_line.move_id.id:
96                         voucher_line_dict = line_dict
97
98             if voucher_line_dict:
99                 voucher_line_dict.update({'voucher_id': voucher_id})
100                 voucher_line_obj.create(cr, uid, voucher_line_dict, context=context)
101             if line.journal_id.type == 'sale':
102                 type = 'customer'
103             elif line.journal_id.type == 'purchase':
104                 type = 'supplier'
105             else:
106                 type = 'general'
107             statement_line_obj.create(cr, uid, {
108                 'name': line.name or '?',
109                 'amount': amount,
110                 'type': type,
111                 'partner_id': line.partner_id.id,
112                 'account_id': line.account_id.id,
113                 'statement_id': statement_id,
114                 'ref': line.ref,
115                 'voucher_id': voucher_id,
116                 'date': time.strftime('%Y-%m-%d'), #time.strftime('%Y-%m-%d'), #line.date_maturity or,
117             }, context=context)
118         return {'type': 'ir.actions.act_window_close'}
119
120 account_statement_from_invoice_lines()
121
122 class account_statement_from_invoice(osv.osv_memory):
123     """
124     Generate Entries by Statement from Invoices
125     """
126     _name = "account.statement.from.invoice"
127     _description = "Entries by Statement from Invoices"
128     _columns = {
129         'date': fields.date('Date payment',required=True),
130         'journal_ids': fields.many2many('account.journal', 'account_journal_relation', 'account_id', 'journal_id', 'Journal'),
131         'line_ids': fields.many2many('account.move.line', 'account_move_line_relation', 'move_id', 'line_id', 'Invoices'),
132     }
133     _defaults = {
134         'date': lambda *a: time.strftime('%Y-%m-%d'),
135     }
136
137     def search_invoices(self, cr, uid, ids, context=None):
138         if context is None:
139             context = {}
140         line_obj = self.pool.get('account.move.line')
141         statement_obj = self.pool.get('account.bank.statement')
142         journal_obj = self.pool.get('account.journal')
143         mod_obj = self.pool.get('ir.model.data')
144         statement_id = 'statement_id' in context and context['statement_id']
145
146         data =  self.read(cr, uid, ids, context=context)[0]
147         statement = statement_obj.browse(cr, uid, statement_id, context=context)
148         args_move_line = []
149         repeated_move_line_ids = []
150         # Creating a group that is unique for importing move lines(move lines, once imported into statement lines, should not appear again)
151         for st_line in statement.line_ids:
152             args_move_line = []
153             args_move_line.append(('name', '=', st_line.name))
154             args_move_line.append(('ref', '=', st_line.ref))
155             if st_line.partner_id:
156                 args_move_line.append(('partner_id', '=', st_line.partner_id.id))
157             args_move_line.append(('account_id', '=', st_line.account_id.id))
158
159             move_line_id = line_obj.search(cr, uid, args_move_line, context=context)
160             if move_line_id:
161                 repeated_move_line_ids += move_line_id
162
163         journal_ids = data['journal_ids']
164         if journal_ids == []:
165             journal_ids = journal_obj.search(cr, uid, [('type', 'in', ('sale', 'cash', 'purchase'))], context=context)
166
167         args = [
168             ('reconcile_id', '=', False),
169             ('journal_id', 'in', journal_ids),
170             ('account_id.reconcile', '=', True)]
171
172         if repeated_move_line_ids:
173             args.append(('id', 'not in', repeated_move_line_ids))
174
175         line_ids = line_obj.search(cr, uid, args,
176             context=context)
177
178         model_data_ids = mod_obj.search(cr, uid, [('model', '=', 'ir.ui.view'), ('name', '=', 'view_account_statement_from_invoice_lines')], context=context)
179         resource_id = mod_obj.read(cr, uid, model_data_ids, fields=['res_id'], context=context)[0]['res_id']
180         return {
181             'domain': "[('id','in', ["+','.join([str(x) for x in line_ids])+"])]",
182             'name': _('Import Entries'),
183             'context': context,
184             'view_type': 'form',
185             'view_mode': 'form',
186             'res_model': 'account.statement.from.invoice.lines',
187             'views': [(resource_id,'form')],
188             'type': 'ir.actions.act_window',
189             'target': 'new',
190         }
191
192 account_statement_from_invoice()
193 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: