[FIX] account: manual skip REMOVE commande as not supported by resolve_o2m_commands_t...
[odoo/odoo.git] / addons / account / account.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 datetime import datetime
24 from dateutil.relativedelta import relativedelta
25 from operator import itemgetter
26
27 import netsvc
28 import pooler
29 from osv import fields, osv
30 import decimal_precision as dp
31 from tools.translate import _
32
33 def check_cycle(self, cr, uid, ids, context=None):
34     """ climbs the ``self._table.parent_id`` chains for 100 levels or
35     until it can't find any more parent(s)
36
37     Returns true if it runs out of parents (no cycle), false if
38     it can recurse 100 times without ending all chains
39     """
40     level = 100
41     while len(ids):
42         cr.execute('SELECT DISTINCT parent_id '\
43                     'FROM '+self._table+' '\
44                     'WHERE id IN %s '\
45                     'AND parent_id IS NOT NULL',(tuple(ids),))
46         ids = map(itemgetter(0), cr.fetchall())
47         if not level:
48             return False
49         level -= 1
50     return True
51
52 class account_payment_term(osv.osv):
53     _name = "account.payment.term"
54     _description = "Payment Term"
55     _columns = {
56         'name': fields.char('Payment Term', size=64, translate=True, required=True),
57         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the payment term without removing it."),
58         'note': fields.text('Description', translate=True),
59         'line_ids': fields.one2many('account.payment.term.line', 'payment_id', 'Terms'),
60     }
61     _defaults = {
62         'active': 1,
63     }
64     _order = "name"
65
66     def compute(self, cr, uid, id, value, date_ref=False, context=None):
67         if not date_ref:
68             date_ref = datetime.now().strftime('%Y-%m-%d')
69         pt = self.browse(cr, uid, id, context=context)
70         amount = value
71         result = []
72         obj_precision = self.pool.get('decimal.precision')
73         for line in pt.line_ids:
74             prec = obj_precision.precision_get(cr, uid, 'Account')
75             if line.value == 'fixed':
76                 amt = round(line.value_amount, prec)
77             elif line.value == 'procent':
78                 amt = round(value * line.value_amount, prec)
79             elif line.value == 'balance':
80                 amt = round(amount, prec)
81             if amt:
82                 next_date = (datetime.strptime(date_ref, '%Y-%m-%d') + relativedelta(days=line.days))
83                 if line.days2 < 0:
84                     next_first_date = next_date + relativedelta(day=1,months=1) #Getting 1st of next month
85                     next_date = next_first_date + relativedelta(days=line.days2)
86                 if line.days2 > 0:
87                     next_date += relativedelta(day=line.days2, months=1)
88                 result.append( (next_date.strftime('%Y-%m-%d'), amt) )
89                 amount -= amt
90         return result
91
92 account_payment_term()
93
94 class account_payment_term_line(osv.osv):
95     _name = "account.payment.term.line"
96     _description = "Payment Term Line"
97     _columns = {
98         'name': fields.char('Line Name', size=32, required=True),
99         'sequence': fields.integer('Sequence', required=True, help="The sequence field is used to order the payment term lines from the lowest sequences to the higher ones"),
100         'value': fields.selection([('procent', 'Percent'),
101                                    ('balance', 'Balance'),
102                                    ('fixed', 'Fixed Amount')], 'Valuation',
103                                    required=True, help="""Select here the kind of valuation related to this payment term line. Note that you should have your last line with the type 'Balance' to ensure that the whole amount will be threated."""),
104
105         'value_amount': fields.float('Amount To Pay', digits_compute=dp.get_precision('Payment Term'), help="For percent enter a ratio between 0-1."),
106         'days': fields.integer('Number of Days', required=True, help="Number of days to add before computation of the day of month." \
107             "If Date=15/01, Number of Days=22, Day of Month=-1, then the due date is 28/02."),
108         'days2': fields.integer('Day of the Month', required=True, help="Day of the month, set -1 for the last day of the current month. If it's positive, it gives the day of the next month. Set 0 for net days (otherwise it's based on the beginning of the month)."),
109         'payment_id': fields.many2one('account.payment.term', 'Payment Term', required=True, select=True),
110     }
111     _defaults = {
112         'value': 'balance',
113         'sequence': 5,
114         'days2': 0,
115     }
116     _order = "sequence"
117
118     def _check_percent(self, cr, uid, ids, context=None):
119         obj = self.browse(cr, uid, ids[0], context=context)
120         if obj.value == 'procent' and ( obj.value_amount < 0.0 or obj.value_amount > 1.0):
121             return False
122         return True
123
124     _constraints = [
125         (_check_percent, 'Percentages for Payment Term Line must be between 0 and 1, Example: 0.02 for 2% ', ['value_amount']),
126     ]
127
128 account_payment_term_line()
129
130 class account_account_type(osv.osv):
131     _name = "account.account.type"
132     _description = "Account Type"
133
134     def _get_current_report_type(self, cr, uid, ids, name, arg, context=None):
135         obj_data = self.pool.get('ir.model.data')
136         obj_financial_report = self.pool.get('account.financial.report') 
137         res = {}
138         financial_report_ref = {
139             'asset': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_assets0')[1], context=context),
140             'liability': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_liability0')[1], context=context),
141             'income': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_income0')[1], context=context),
142             'expense': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_expense0')[1], context=context),
143         }
144         for record in self.browse(cr, uid, ids, context=context):
145             res[record.id] = 'none'
146             for key, financial_report in financial_report_ref.items():
147                 list_ids = [x.id for x in financial_report.account_type_ids]
148                 if record.id in list_ids:
149                     res[record.id] = key
150         return res
151
152     def _save_report_type(self, cr, uid, account_type_id, field_name, field_value, arg, context=None):
153         obj_data = self.pool.get('ir.model.data')
154         obj_financial_report = self.pool.get('account.financial.report') 
155         #unlink if it exists somewhere in the financial reports related to BS or PL
156         financial_report_ref = {
157             'asset': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_assets0')[1], context=context),
158             'liability': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_liability0')[1], context=context),
159             'income': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_income0')[1], context=context),
160             'expense': obj_financial_report.browse(cr, uid, obj_data.get_object_reference(cr, uid, 'account','account_financial_report_expense0')[1], context=context),
161         }
162         for key, financial_report in financial_report_ref.items():
163             list_ids = [x.id for x in financial_report.account_type_ids]
164             if account_type_id in list_ids:
165                 obj_financial_report.write(cr, uid, [financial_report.id], {'account_type_ids': [(3, account_type_id)]})
166         #write it in the good place
167         if field_value != 'none':
168             return obj_financial_report.write(cr, uid, [financial_report_ref[field_value].id], {'account_type_ids': [(4, account_type_id)]})
169
170     _columns = {
171         'name': fields.char('Account Type', size=64, required=True, translate=True),
172         'code': fields.char('Code', size=32, required=True, select=True),
173         'close_method': fields.selection([('none', 'None'), ('balance', 'Balance'), ('detail', 'Detail'), ('unreconciled', 'Unreconciled')], 'Deferral Method', required=True, help="""Set here the method that will be used to generate the end of year journal entries for all the accounts of this type.
174
175  'None' means that nothing will be done.
176  'Balance' will generally be used for cash accounts.
177  'Detail' will copy each existing journal item of the previous year, even the reconciled ones.
178  'Unreconciled' will copy only the journal items that were unreconciled on the first day of the new fiscal year."""),
179         'report_type': fields.function(_get_current_report_type, fnct_inv=_save_report_type, type='selection', string='P&L / BS Category', 
180             selection= [('none','/'),
181                         ('income', _('Profit & Loss (Income account)')),
182                         ('expense', _('Profit & Loss (Expense account)')),
183                         ('asset', _('Balance Sheet (Asset account)')),
184                         ('liability', _('Balance Sheet (Liability account)'))], help="This field is used to generate legal reports: profit and loss, balance sheet.", required=True),
185         'note': fields.text('Description'),
186     }
187     _defaults = {
188         'close_method': 'none',
189         'report_type': 'none',
190     }
191     _order = "code"
192
193 account_account_type()
194
195 def _code_get(self, cr, uid, context=None):
196     acc_type_obj = self.pool.get('account.account.type')
197     ids = acc_type_obj.search(cr, uid, [])
198     res = acc_type_obj.read(cr, uid, ids, ['code', 'name'], context=context)
199     return [(r['code'], r['name']) for r in res]
200
201 #----------------------------------------------------------
202 # Accounts
203 #----------------------------------------------------------
204
205 class account_tax(osv.osv):
206     _name = 'account.tax'
207 account_tax()
208
209 class account_account(osv.osv):
210     _order = "parent_left"
211     _parent_order = "code"
212     _name = "account.account"
213     _description = "Account"
214     _parent_store = True
215     logger = netsvc.Logger()
216
217     def search(self, cr, uid, args, offset=0, limit=None, order=None,
218             context=None, count=False):
219         if context is None:
220             context = {}
221         pos = 0
222
223         while pos < len(args):
224
225             if args[pos][0] == 'code' and args[pos][1] in ('like', 'ilike') and args[pos][2]:
226                 args[pos] = ('code', '=like', str(args[pos][2].replace('%', ''))+'%')
227             if args[pos][0] == 'journal_id':
228                 if not args[pos][2]:
229                     del args[pos]
230                     continue
231                 jour = self.pool.get('account.journal').browse(cr, uid, args[pos][2], context=context)
232                 if (not (jour.account_control_ids or jour.type_control_ids)) or not args[pos][2]:
233                     args[pos] = ('type','not in',('consolidation','view'))
234                     continue
235                 ids3 = map(lambda x: x.id, jour.type_control_ids)
236                 ids1 = super(account_account, self).search(cr, uid, [('user_type', 'in', ids3)])
237                 ids1 += map(lambda x: x.id, jour.account_control_ids)
238                 args[pos] = ('id', 'in', ids1)
239             pos += 1
240
241         if context and context.has_key('consolidate_children'): #add consolidated children of accounts
242             ids = super(account_account, self).search(cr, uid, args, offset, limit,
243                 order, context=context, count=count)
244             for consolidate_child in self.browse(cr, uid, context['account_id'], context=context).child_consol_ids:
245                 ids.append(consolidate_child.id)
246             return ids
247
248         return super(account_account, self).search(cr, uid, args, offset, limit,
249                 order, context=context, count=count)
250
251     def _get_children_and_consol(self, cr, uid, ids, context=None):
252         #this function search for all the children and all consolidated children (recursively) of the given account ids
253         ids2 = self.search(cr, uid, [('parent_id', 'child_of', ids)], context=context)
254         ids3 = []
255         for rec in self.browse(cr, uid, ids2, context=context):
256             for child in rec.child_consol_ids:
257                 ids3.append(child.id)
258         if ids3:
259             ids3 = self._get_children_and_consol(cr, uid, ids3, context)
260         return ids2 + ids3
261
262     def __compute(self, cr, uid, ids, field_names, arg=None, context=None,
263                   query='', query_params=()):
264         """ compute the balance, debit and/or credit for the provided
265         account ids
266         Arguments:
267         `ids`: account ids
268         `field_names`: the fields to compute (a list of any of
269                        'balance', 'debit' and 'credit')
270         `arg`: unused fields.function stuff
271         `query`: additional query filter (as a string)
272         `query_params`: parameters for the provided query string
273                         (__compute will handle their escaping) as a
274                         tuple
275         """
276         mapping = {
277             'balance': "COALESCE(SUM(l.debit),0) - COALESCE(SUM(l.credit), 0) as balance",
278             'debit': "COALESCE(SUM(l.debit), 0) as debit",
279             'credit': "COALESCE(SUM(l.credit), 0) as credit",
280             # by convention, foreign_balance is 0 when the account has no secondary currency, because the amounts may be in different currencies
281             'foreign_balance': "(SELECT CASE WHEN currency_id IS NULL THEN 0 ELSE COALESCE(SUM(l.amount_currency), 0) END FROM account_account WHERE id IN (l.account_id)) as foreign_balance",
282         }
283         #get all the necessary accounts
284         children_and_consolidated = self._get_children_and_consol(cr, uid, ids, context=context)
285         #compute for each account the balance/debit/credit from the move lines
286         accounts = {}
287         res = {}
288         null_result = dict((fn, 0.0) for fn in field_names)
289         if children_and_consolidated:
290             aml_query = self.pool.get('account.move.line')._query_get(cr, uid, context=context)
291
292             wheres = [""]
293             if query.strip():
294                 wheres.append(query.strip())
295             if aml_query.strip():
296                 wheres.append(aml_query.strip())
297             filters = " AND ".join(wheres)
298             self.logger.notifyChannel('addons.'+self._name, netsvc.LOG_DEBUG,
299                                       'Filters: %s'%filters)
300             # IN might not work ideally in case there are too many
301             # children_and_consolidated, in that case join on a
302             # values() e.g.:
303             # SELECT l.account_id as id FROM account_move_line l
304             # INNER JOIN (VALUES (id1), (id2), (id3), ...) AS tmp (id)
305             # ON l.account_id = tmp.id
306             # or make _get_children_and_consol return a query and join on that
307             request = ("SELECT l.account_id as id, " +\
308                        ', '.join(mapping.values()) +
309                        " FROM account_move_line l" \
310                        " WHERE l.account_id IN %s " \
311                             + filters +
312                        " GROUP BY l.account_id")
313             params = (tuple(children_and_consolidated),) + query_params
314             cr.execute(request, params)
315             self.logger.notifyChannel('addons.'+self._name, netsvc.LOG_DEBUG,
316                                       'Status: %s'%cr.statusmessage)
317
318             for res in cr.dictfetchall():
319                 accounts[res['id']] = res
320
321             # consolidate accounts with direct children
322             children_and_consolidated.reverse()
323             brs = list(self.browse(cr, uid, children_and_consolidated, context=context))
324             sums = {}
325             currency_obj = self.pool.get('res.currency')
326             while brs:
327                 current = brs.pop(0)
328 #                can_compute = True
329 #                for child in current.child_id:
330 #                    if child.id not in sums:
331 #                        can_compute = False
332 #                        try:
333 #                            brs.insert(0, brs.pop(brs.index(child)))
334 #                        except ValueError:
335 #                            brs.insert(0, child)
336 #                if can_compute:
337                 for fn in field_names:
338                     sums.setdefault(current.id, {})[fn] = accounts.get(current.id, {}).get(fn, 0.0)
339                     for child in current.child_id:
340                         if child.company_id.currency_id.id == current.company_id.currency_id.id:
341                             sums[current.id][fn] += sums[child.id][fn]
342                         else:
343                             sums[current.id][fn] += currency_obj.compute(cr, uid, child.company_id.currency_id.id, current.company_id.currency_id.id, sums[child.id][fn], context=context)
344
345                 # as we have to relay on values computed before this is calculated separately than previous fields
346                 if current.currency_id and current.exchange_rate and \
347                             ('adjusted_balance' in field_names or 'unrealized_gain_loss' in field_names):
348                     # Computing Adjusted Balance and Unrealized Gains and losses
349                     # Adjusted Balance = Foreign Balance / Exchange Rate
350                     # Unrealized Gains and losses = Adjusted Balance - Balance
351                     adj_bal = sums[current.id].get('foreign_balance', 0.0) / current.exchange_rate
352                     sums[current.id].update({'adjusted_balance': adj_bal, 'unrealized_gain_loss': adj_bal - sums[current.id].get('balance', 0.0)})
353
354             for id in ids:
355                 res[id] = sums.get(id, null_result)
356         else:
357             for id in ids:
358                 res[id] = null_result
359         return res
360
361     def _get_company_currency(self, cr, uid, ids, field_name, arg, context=None):
362         result = {}
363         for rec in self.browse(cr, uid, ids, context=context):
364             result[rec.id] = (rec.company_id.currency_id.id,rec.company_id.currency_id.symbol)
365         return result
366
367     def _get_child_ids(self, cr, uid, ids, field_name, arg, context=None):
368         result = {}
369         for record in self.browse(cr, uid, ids, context=context):
370             if record.child_parent_ids:
371                 result[record.id] = [x.id for x in record.child_parent_ids]
372             else:
373                 result[record.id] = []
374
375             if record.child_consol_ids:
376                 for acc in record.child_consol_ids:
377                     if acc.id not in result[record.id]:
378                         result[record.id].append(acc.id)
379
380         return result
381
382     def _get_level(self, cr, uid, ids, field_name, arg, context=None):
383         res = {}
384         for account in self.browse(cr, uid, ids, context=context):
385             #we may not know the level of the parent at the time of computation, so we
386             # can't simply do res[account.id] = account.parent_id.level + 1
387             level = 0
388             parent = account.parent_id
389             while parent:
390                 level += 1
391                 parent = parent.parent_id
392             res[account.id] = level
393         return res
394
395     def _set_credit_debit(self, cr, uid, account_id, name, value, arg, context=None):
396         if context.get('config_invisible', True):
397             return True
398
399         account = self.browse(cr, uid, account_id, context=context)
400         diff = value - getattr(account,name)
401         if not diff:
402             return True
403
404         journal_obj = self.pool.get('account.journal')
405         jids = journal_obj.search(cr, uid, [('type','=','situation'),('centralisation','=',1),('company_id','=',account.company_id.id)], context=context)
406         if not jids:
407             raise osv.except_osv(_('Error!'),_("You need an Opening journal with centralisation checked to set the initial balance!"))
408
409         period_obj = self.pool.get('account.period')
410         pids = period_obj.search(cr, uid, [('special','=',True),('company_id','=',account.company_id.id)], context=context)
411         if not pids:
412             raise osv.except_osv(_('Error!'),_("No opening/closing period defined, please create one to set the initial balance!"))
413
414         move_obj = self.pool.get('account.move.line')
415         move_id = move_obj.search(cr, uid, [
416             ('journal_id','=',jids[0]),
417             ('period_id','=',pids[0]),
418             ('account_id','=', account_id),
419             (name,'>', 0.0),
420             ('name','=', _('Opening Balance'))
421         ], context=context)
422         if move_id:
423             move = move_obj.browse(cr, uid, move_id[0], context=context)
424             move_obj.write(cr, uid, move_id[0], {
425                 name: diff+getattr(move,name)
426             }, context=context)
427         else:
428             if diff<0.0:
429                 raise osv.except_osv(_('Error!'),_("Unable to adapt the initial balance (negative value)!"))
430             nameinv = (name=='credit' and 'debit') or 'credit'
431             move_id = move_obj.create(cr, uid, {
432                 'name': _('Opening Balance'),
433                 'account_id': account_id,
434                 'journal_id': jids[0],
435                 'period_id': pids[0],
436                 name: diff,
437                 nameinv: 0.0
438             }, context=context)
439         return True
440
441     _columns = {
442         'name': fields.char('Name', size=256, required=True, select=True),
443         'currency_id': fields.many2one('res.currency', 'Secondary Currency', help="Forces all moves for this account to have this secondary currency."),
444         'code': fields.char('Code', size=64, required=True, select=1),
445         'type': fields.selection([
446             ('view', 'View'),
447             ('other', 'Regular'),
448             ('receivable', 'Receivable'),
449             ('payable', 'Payable'),
450             ('liquidity','Liquidity'),
451             ('consolidation', 'Consolidation'),
452             ('closed', 'Closed'),
453         ], 'Internal Type', required=True, help="The 'Internal Type' is used for features available on "\
454             "different types of accounts: view can not have journal items, consolidation are accounts that "\
455             "can have children accounts for multi-company consolidations, payable/receivable are for "\
456             "partners accounts (for debit/credit computations), closed for depreciated accounts."),
457         'user_type': fields.many2one('account.account.type', 'Account Type', required=True,
458             help="Account Type is used for information purpose, to generate "
459               "country-specific legal reports, and set the rules to close a fiscal year and generate opening entries."),
460         'financial_report_ids': fields.many2many('account.financial.report', 'account_account_financial_report', 'account_id', 'report_line_id', 'Financial Reports'),
461         'parent_id': fields.many2one('account.account', 'Parent', ondelete='cascade', domain=[('type','=','view')]),
462         'child_parent_ids': fields.one2many('account.account','parent_id','Children'),
463         'child_consol_ids': fields.many2many('account.account', 'account_account_consol_rel', 'child_id', 'parent_id', 'Consolidated Children'),
464         'child_id': fields.function(_get_child_ids, type='many2many', relation="account.account", string="Child Accounts"),
465         'balance': fields.function(__compute, digits_compute=dp.get_precision('Account'), string='Balance', multi='balance'),
466         'credit': fields.function(__compute, fnct_inv=_set_credit_debit, digits_compute=dp.get_precision('Account'), string='Credit', multi='balance'),
467         'debit': fields.function(__compute, fnct_inv=_set_credit_debit, digits_compute=dp.get_precision('Account'), string='Debit', multi='balance'),
468         'foreign_balance': fields.function(__compute, digits_compute=dp.get_precision('Account'), string='Foreign Balance', multi='balance',
469                                            help="Total amount (in Secondary currency) for transactions held in secondary currency for this account."),
470         'adjusted_balance': fields.function(__compute, digits_compute=dp.get_precision('Account'), string='Adjusted Balance', multi='balance',
471                                             help="Total amount (in Company currency) for transactions held in secondary currency for this account."),
472         'unrealized_gain_loss': fields.function(__compute, digits_compute=dp.get_precision('Account'), string='Unrealized Gain or Loss', multi='balance',
473                                                 help="Value of Loss or Gain due to changes in exchange rate when doing multi-currency transactions."),
474         'reconcile': fields.boolean('Allow Reconciliation', help="Check this box if this account allows reconciliation of journal items."),
475         'exchange_rate': fields.related('currency_id', 'rate', type='float', string='Exchange Rate', digits=(12,6)),
476         'shortcut': fields.char('Shortcut', size=12),
477         'tax_ids': fields.many2many('account.tax', 'account_account_tax_default_rel',
478             'account_id', 'tax_id', 'Default Taxes'),
479         'note': fields.text('Note'),
480         'company_currency_id': fields.function(_get_company_currency, type='many2one', relation='res.currency', string='Company Currency'),
481         'company_id': fields.many2one('res.company', 'Company', required=True),
482         'active': fields.boolean('Active', select=2, help="If the active field is set to False, it will allow you to hide the account without removing it."),
483
484         'parent_left': fields.integer('Parent Left', select=1),
485         'parent_right': fields.integer('Parent Right', select=1),
486         'currency_mode': fields.selection([('current', 'At Date'), ('average', 'Average Rate')], 'Outgoing Currencies Rate',
487             help=
488             'This will select how the current currency rate for outgoing transactions is computed. '\
489             'In most countries the legal method is "average" but only a few software systems are able to '\
490             'manage this. So if you import from another software system you may have to use the rate at date. ' \
491             'Incoming transactions always use the rate at date.', \
492             required=True),
493         'level': fields.function(_get_level, string='Level', method=True, type='integer',
494              store={
495                     'account.account': (_get_children_and_consol, ['level', 'parent_id'], 10),
496                    }),
497     }
498
499     _defaults = {
500         'type': 'other',
501         'reconcile': False,
502         'active': True,
503         'currency_mode': 'current',
504         'company_id': lambda s, cr, uid, c: s.pool.get('res.company')._company_default_get(cr, uid, 'account.account', context=c),
505     }
506
507     def _check_recursion(self, cr, uid, ids, context=None):
508         obj_self = self.browse(cr, uid, ids[0], context=context)
509         p_id = obj_self.parent_id and obj_self.parent_id.id
510         if (obj_self in obj_self.child_consol_ids) or (p_id and (p_id is obj_self.id)):
511             return False
512         while(ids):
513             cr.execute('SELECT DISTINCT child_id '\
514                        'FROM account_account_consol_rel '\
515                        'WHERE parent_id IN %s', (tuple(ids),))
516             child_ids = map(itemgetter(0), cr.fetchall())
517             c_ids = child_ids
518             if (p_id and (p_id in c_ids)) or (obj_self.id in c_ids):
519                 return False
520             while len(c_ids):
521                 s_ids = self.search(cr, uid, [('parent_id', 'in', c_ids)])
522                 if p_id and (p_id in s_ids):
523                     return False
524                 c_ids = s_ids
525             ids = child_ids
526         return True
527
528     def _check_type(self, cr, uid, ids, context=None):
529         if context is None:
530             context = {}
531         accounts = self.browse(cr, uid, ids, context=context)
532         for account in accounts:
533             if account.child_id and account.type not in ('view', 'consolidation'):
534                 return False
535         return True
536
537     def _check_account_type(self, cr, uid, ids, context=None):
538         for account in self.browse(cr, uid, ids, context=context):
539             if account.type in ('receivable', 'payable') and account.user_type.close_method != 'unreconciled':
540                 return False
541         return True
542
543     _constraints = [
544         (_check_recursion, 'Error ! You can not create recursive accounts.', ['parent_id']),
545         (_check_type, 'Configuration Error! \nYou can not define children to an account with internal type different of "View"! ', ['type']),
546         (_check_account_type, 'Configuration Error! \nYou can not select an account type with a deferral method different of "Unreconciled" for accounts with internal type "Payable/Receivable"! ', ['user_type','type']),
547     ]
548     _sql_constraints = [
549         ('code_company_uniq', 'unique (code,company_id)', 'The code of the account must be unique per company !')
550     ]
551     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=100):
552         if not args:
553             args = []
554         args = args[:]
555         ids = []
556         try:
557             if name and str(name).startswith('partner:'):
558                 part_id = int(name.split(':')[1])
559                 part = self.pool.get('res.partner').browse(cr, user, part_id, context=context)
560                 args += [('id', 'in', (part.property_account_payable.id, part.property_account_receivable.id))]
561                 name = False
562             if name and str(name).startswith('type:'):
563                 type = name.split(':')[1]
564                 args += [('type', '=', type)]
565                 name = False
566         except:
567             pass
568         if name:
569             ids = self.search(cr, user, [('code', '=like', name+"%")]+args, limit=limit)
570             if not ids:
571                 ids = self.search(cr, user, [('shortcut', '=', name)]+ args, limit=limit)
572             if not ids:
573                 ids = self.search(cr, user, [('name', operator, name)]+ args, limit=limit)
574             if not ids and len(name.split()) >= 2:
575                 #Separating code and name of account for searching
576                 operand1,operand2 = name.split(' ',1) #name can contain spaces e.g. OpenERP S.A.
577                 ids = self.search(cr, user, [('code', operator, operand1), ('name', operator, operand2)]+ args, limit=limit)
578         else:
579             ids = self.search(cr, user, args, context=context, limit=limit)
580         return self.name_get(cr, user, ids, context=context)
581
582     def name_get(self, cr, uid, ids, context=None):
583         if not ids:
584             return []
585         reads = self.read(cr, uid, ids, ['name', 'code'], context=context)
586         res = []
587         for record in reads:
588             name = record['name']
589             if record['code']:
590                 name = record['code'] + ' ' + name
591             res.append((record['id'], name))
592         return res
593
594     def copy(self, cr, uid, id, default={}, context=None, done_list=[], local=False):
595         account = self.browse(cr, uid, id, context=context)
596         new_child_ids = []
597         if not default:
598             default = {}
599         default = default.copy()
600         default['code'] = (account['code'] or '') + '(copy)'
601         if not local:
602             done_list = []
603         if account.id in done_list:
604             return False
605         done_list.append(account.id)
606         if account:
607             for child in account.child_id:
608                 child_ids = self.copy(cr, uid, child.id, default, context=context, done_list=done_list, local=True)
609                 if child_ids:
610                     new_child_ids.append(child_ids)
611             default['child_parent_ids'] = [(6, 0, new_child_ids)]
612         else:
613             default['child_parent_ids'] = False
614         return super(account_account, self).copy(cr, uid, id, default, context=context)
615
616     def _check_moves(self, cr, uid, ids, method, context=None):
617         line_obj = self.pool.get('account.move.line')
618         account_ids = self.search(cr, uid, [('id', 'child_of', ids)])
619
620         if line_obj.search(cr, uid, [('account_id', 'in', account_ids)]):
621             if method == 'write':
622                 raise osv.except_osv(_('Error !'), _('You can not desactivate an account that contains some journal items.'))
623             elif method == 'unlink':
624                 raise osv.except_osv(_('Error !'), _('You can not remove an account containing journal items.'))
625         #Checking whether the account is set as a property to any Partner or not
626         value = 'account.account,' + str(ids[0])
627         partner_prop_acc = self.pool.get('ir.property').search(cr, uid, [('value_reference','=',value)], context=context)
628         if partner_prop_acc:
629             raise osv.except_osv(_('Warning !'), _('You can not remove/desactivate an account which is set on a customer or supplier.'))
630         return True
631
632     def _check_allow_type_change(self, cr, uid, ids, new_type, context=None):
633         group1 = ['payable', 'receivable', 'other']
634         group2 = ['consolidation','view']
635         line_obj = self.pool.get('account.move.line')
636         for account in self.browse(cr, uid, ids, context=context):
637             old_type = account.type
638             account_ids = self.search(cr, uid, [('id', 'child_of', [account.id])])
639             if line_obj.search(cr, uid, [('account_id', 'in', account_ids)]):
640                 #Check for 'Closed' type
641                 if old_type == 'closed' and new_type !='closed':
642                     raise osv.except_osv(_('Warning !'), _("You cannot change the type of account from 'Closed' to any other type which contains journal items!"))
643                 #Check for change From group1 to group2 and vice versa
644                 if (old_type in group1 and new_type in group2) or (old_type in group2 and new_type in group1):
645                     raise osv.except_osv(_('Warning !'), _("You cannot change the type of account from '%s' to '%s' type as it contains journal items!") % (old_type,new_type,))
646         return True
647
648     def write(self, cr, uid, ids, vals, context=None):
649
650         if context is None:
651             context = {}
652         if not ids:
653             return True
654         if isinstance(ids, (int, long)):
655             ids = [ids]
656
657         # Dont allow changing the company_id when account_move_line already exist
658         if 'company_id' in vals:
659             move_lines = self.pool.get('account.move.line').search(cr, uid, [('account_id', 'in', ids)])
660             if move_lines:
661                 # Allow the write if the value is the same
662                 for i in [i['company_id'][0] for i in self.read(cr,uid,ids,['company_id'])]:
663                     if vals['company_id']!=i:
664                         raise osv.except_osv(_('Warning !'), _('You cannot change the owner company of an account that already contains journal items.'))
665         if 'active' in vals and not vals['active']:
666             self._check_moves(cr, uid, ids, "write", context=context)
667         if 'type' in vals.keys():
668             self._check_allow_type_change(cr, uid, ids, vals['type'], context=context)
669         return super(account_account, self).write(cr, uid, ids, vals, context=context)
670
671     def unlink(self, cr, uid, ids, context=None):
672         self._check_moves(cr, uid, ids, "unlink", context=context)
673         return super(account_account, self).unlink(cr, uid, ids, context=context)
674
675 account_account()
676
677 class account_journal_view(osv.osv):
678     _name = "account.journal.view"
679     _description = "Journal View"
680     _columns = {
681         'name': fields.char('Journal View', size=64, required=True),
682         'columns_id': fields.one2many('account.journal.column', 'view_id', 'Columns')
683     }
684     _order = "name"
685
686 account_journal_view()
687
688
689 class account_journal_column(osv.osv):
690
691     def _col_get(self, cr, user, context=None):
692         result = []
693         cols = self.pool.get('account.move.line')._columns
694         for col in cols:
695             if col in ('period_id', 'journal_id'):
696                 continue
697             result.append( (col, cols[col].string) )
698         result.sort()
699         return result
700
701     _name = "account.journal.column"
702     _description = "Journal Column"
703     _columns = {
704         'name': fields.char('Column Name', size=64, required=True),
705         'field': fields.selection(_col_get, 'Field Name', required=True, size=32),
706         'view_id': fields.many2one('account.journal.view', 'Journal View', select=True),
707         'sequence': fields.integer('Sequence', help="Gives the sequence order to journal column.", readonly=True),
708         'required': fields.boolean('Required'),
709         'readonly': fields.boolean('Readonly'),
710     }
711     _order = "view_id, sequence"
712
713 account_journal_column()
714
715 class account_journal(osv.osv):
716     _name = "account.journal"
717     _description = "Journal"
718     _columns = {
719         'name': fields.char('Journal Name', size=64, required=True),
720         'code': fields.char('Code', size=5, required=True, help="The code will be displayed on reports."),
721         'type': fields.selection([('sale', 'Sale'),('sale_refund','Sale Refund'), ('purchase', 'Purchase'), ('purchase_refund','Purchase Refund'), ('cash', 'Cash'), ('bank', 'Bank and Cheques'), ('general', 'General'), ('situation', 'Opening/Closing Situation')], 'Type', size=32, required=True,
722                                  help="Select 'Sale' for customer invoices journals."\
723                                  " Select 'Purchase' for supplier invoices journals."\
724                                  " Select 'Cash' or 'Bank' for journals that are used in customer or supplier payments."\
725                                  " Select 'General' for miscellaneous operations journals."\
726                                  " Select 'Opening/Closing Situation' for entries generated for new fiscal years."),
727         'type_control_ids': fields.many2many('account.account.type', 'account_journal_type_rel', 'journal_id','type_id', 'Type Controls', domain=[('code','<>','view'), ('code', '<>', 'closed')]),
728         'account_control_ids': fields.many2many('account.account', 'account_account_type_rel', 'journal_id','account_id', 'Account', domain=[('type','<>','view'), ('type', '<>', 'closed')]),
729         'view_id': fields.many2one('account.journal.view', 'Display Mode', required=True, help="Gives the view used when writing or browsing entries in this journal. The view tells OpenERP which fields should be visible, required or readonly and in which order. You can create your own view for a faster encoding in each journal."),
730         'default_credit_account_id': fields.many2one('account.account', 'Default Credit Account', domain="[('type','!=','view')]", help="It acts as a default account for credit amount"),
731         'default_debit_account_id': fields.many2one('account.account', 'Default Debit Account', domain="[('type','!=','view')]", help="It acts as a default account for debit amount"),
732         'centralisation': fields.boolean('Centralised counterpart', help="Check this box to determine that each entry of this journal won't create a new counterpart but will share the same counterpart. This is used in fiscal year closing."),
733         'update_posted': fields.boolean('Allow Cancelling Entries', help="Check this box if you want to allow the cancellation the entries related to this journal or of the invoice related to this journal"),
734         'group_invoice_lines': fields.boolean('Group Invoice Lines', help="If this box is checked, the system will try to group the accounting lines when generating them from invoices."),
735         'sequence_id': fields.many2one('ir.sequence', 'Entry Sequence', help="This field contains the informatin related to the numbering of the journal entries of this journal.", required=True),
736         'user_id': fields.many2one('res.users', 'User', help="The user responsible for this journal"),
737         'groups_id': fields.many2many('res.groups', 'account_journal_group_rel', 'journal_id', 'group_id', 'Groups'),
738         'currency': fields.many2one('res.currency', 'Currency', help='The currency used to enter statement'),
739         'entry_posted': fields.boolean('Skip \'Draft\' State for Manual Entries', help='Check this box if you don\'t want new journal entries to pass through the \'draft\' state and instead goes directly to the \'posted state\' without any manual validation. \nNote that journal entries that are automatically created by the system are always skipping that state.'),
740         'company_id': fields.many2one('res.company', 'Company', required=True, select=1, help="Company related to this journal"),
741         'allow_date':fields.boolean('Check Date in Period', help= 'If set to True then do not accept the entry if the entry date is not into the period dates'),
742     }
743
744     _defaults = {
745         'user_id': lambda self, cr, uid, context: uid,
746         'company_id': lambda self, cr, uid, c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
747     }
748     _sql_constraints = [
749         ('code_company_uniq', 'unique (code, company_id)', 'The code of the journal must be unique per company !'),
750         ('name_company_uniq', 'unique (name, company_id)', 'The name of the journal must be unique per company !'),
751     ]
752
753     _order = 'code'
754
755     def _check_currency(self, cr, uid, ids, context=None):
756         for journal in self.browse(cr, uid, ids, context=context):
757             if journal.currency:
758                 if journal.default_credit_account_id and not journal.default_credit_account_id.currency_id.id == journal.currency.id:
759                     return False
760                 if journal.default_debit_account_id and not journal.default_debit_account_id.currency_id.id == journal.currency.id:
761                     return False
762         return True
763
764     _constraints = [
765         (_check_currency, 'Configuration error! The currency chosen should be shared by the default accounts too.', ['currency','default_debit_account_id','default_credit_account_id']),
766     ]
767
768     def copy(self, cr, uid, id, default={}, context=None, done_list=[], local=False):
769         journal = self.browse(cr, uid, id, context=context)
770         if not default:
771             default = {}
772         default = default.copy()
773         default['code'] = (journal['code'] or '') + '(copy)'
774         default['name'] = (journal['name'] or '') + '(copy)'
775         default['sequence_id'] = False
776         return super(account_journal, self).copy(cr, uid, id, default, context=context)
777
778     def write(self, cr, uid, ids, vals, context=None):
779         if context is None:
780             context = {}
781         if isinstance(ids, (int, long)):
782             ids = [ids]
783         for journal in self.browse(cr, uid, ids, context=context):
784             if 'company_id' in vals and journal.company_id.id != vals['company_id']:
785                 move_lines = self.pool.get('account.move.line').search(cr, uid, [('journal_id', 'in', ids)])
786                 if move_lines:
787                     raise osv.except_osv(_('Warning !'), _('You can not modify the company of this journal as its related record exist in journal items'))
788         return super(account_journal, self).write(cr, uid, ids, vals, context=context)
789
790     def create_sequence(self, cr, uid, vals, context=None):
791         """ Create new no_gap entry sequence for every new Joural
792         """
793         # in account.journal code is actually the prefix of the sequence
794         # whereas ir.sequence code is a key to lookup global sequences.
795         prefix = vals['code'].upper()
796
797         seq = {
798             'name': vals['name'],
799             'implementation':'no_gap',
800             'prefix': prefix + "/%(year)s/",
801             'padding': 4,
802             'number_increment': 1
803         }
804         if 'company_id' in vals:
805             seq['company_id'] = vals['company_id']
806         return self.pool.get('ir.sequence').create(cr, uid, seq)
807
808     def create(self, cr, uid, vals, context=None):
809         if not 'sequence_id' in vals or not vals['sequence_id']:
810             # if we have the right to create a journal, we should be able to
811             # create it's sequence.
812             vals.update({'sequence_id': self.create_sequence(cr, 1, vals, context)})
813         return super(account_journal, self).create(cr, uid, vals, context)
814
815     def name_get(self, cr, user, ids, context=None):
816         """
817         Returns a list of tupples containing id, name.
818         result format: {[(id, name), (id, name), ...]}
819
820         @param cr: A database cursor
821         @param user: ID of the user currently logged in
822         @param ids: list of ids for which name should be read
823         @param context: context arguments, like lang, time zone
824
825         @return: Returns a list of tupples containing id, name
826         """
827         result = self.browse(cr, user, ids, context=context)
828         res = []
829         for rs in result:
830             if rs.currency:
831                 currency = rs.currency
832             else:
833                 currency = rs.company_id.currency_id
834             name = "%s (%s)" % (rs.name, currency.name)
835             res += [(rs.id, name)]
836         return res
837
838     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=100):
839         if not args:
840             args = []
841         if context is None:
842             context = {}
843         ids = []
844         if context.get('journal_type', False):
845             args += [('type','=',context.get('journal_type'))]
846         if name:
847             ids = self.search(cr, user, [('code', 'ilike', name)]+ args, limit=limit, context=context)
848         if not ids:
849             ids = self.search(cr, user, [('name', 'ilike', name)]+ args, limit=limit, context=context)#fix it ilike should be replace with operator
850
851         return self.name_get(cr, user, ids, context=context)
852
853     def onchange_type(self, cr, uid, ids, type, currency, context=None):
854         obj_data = self.pool.get('ir.model.data')
855         user_pool = self.pool.get('res.users')
856
857         type_map = {
858             'sale':'account_sp_journal_view',
859             'sale_refund':'account_sp_refund_journal_view',
860             'purchase':'account_sp_journal_view',
861             'purchase_refund':'account_sp_refund_journal_view',
862             'cash':'account_journal_bank_view',
863             'bank':'account_journal_bank_view',
864             'general':'account_journal_view',
865             'situation':'account_journal_view'
866         }
867
868         res = {}
869         view_id = type_map.get(type, 'account_journal_view')
870         user = user_pool.browse(cr, uid, uid)
871         if type in ('cash', 'bank') and currency and user.company_id.currency_id.id != currency:
872             view_id = 'account_journal_bank_view_multi'
873         data_id = obj_data.search(cr, uid, [('model','=','account.journal.view'), ('name','=',view_id)])
874         data = obj_data.browse(cr, uid, data_id[0], context=context)
875
876         res.update({
877             'centralisation':type == 'situation',
878             'view_id':data.res_id,
879         })
880         return {
881             'value':res
882         }
883
884 account_journal()
885
886 class account_fiscalyear(osv.osv):
887     _name = "account.fiscalyear"
888     _description = "Fiscal Year"
889     _columns = {
890         'name': fields.char('Fiscal Year', size=64, required=True),
891         'code': fields.char('Code', size=6, required=True),
892         'company_id': fields.many2one('res.company', 'Company', required=True),
893         'date_start': fields.date('Start Date', required=True),
894         'date_stop': fields.date('End Date', required=True),
895         'period_ids': fields.one2many('account.period', 'fiscalyear_id', 'Periods'),
896         'state': fields.selection([('draft','Open'), ('done','Closed')], 'State', readonly=True),
897     }
898     _defaults = {
899         'state': 'draft',
900         'company_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
901     }
902     _order = "date_start, id"
903
904
905     def _check_duration(self, cr, uid, ids, context=None):
906         obj_fy = self.browse(cr, uid, ids[0], context=context)
907         if obj_fy.date_stop < obj_fy.date_start:
908             return False
909         return True
910
911     _constraints = [
912         (_check_duration, 'Error! The start date of the fiscal year must be before his end date.', ['date_start','date_stop'])
913     ]
914
915     def create_period3(self, cr, uid, ids, context=None):
916         return self.create_period(cr, uid, ids, context, 3)
917
918     def create_period(self, cr, uid, ids, context=None, interval=1):
919         period_obj = self.pool.get('account.period')
920         for fy in self.browse(cr, uid, ids, context=context):
921             ds = datetime.strptime(fy.date_start, '%Y-%m-%d')
922             period_obj.create(cr, uid, {
923                     'name':  "%s %s" % (_('Opening Period'), ds.strftime('%Y')),
924                     'code': ds.strftime('00/%Y'),
925                     'date_start': ds,
926                     'date_stop': ds,
927                     'special': True,
928                     'fiscalyear_id': fy.id,
929                 })
930             while ds.strftime('%Y-%m-%d') < fy.date_stop:
931                 de = ds + relativedelta(months=interval, days=-1)
932
933                 if de.strftime('%Y-%m-%d') > fy.date_stop:
934                     de = datetime.strptime(fy.date_stop, '%Y-%m-%d')
935
936                 period_obj.create(cr, uid, {
937                     'name': ds.strftime('%m/%Y'),
938                     'code': ds.strftime('%m/%Y'),
939                     'date_start': ds.strftime('%Y-%m-%d'),
940                     'date_stop': de.strftime('%Y-%m-%d'),
941                     'fiscalyear_id': fy.id,
942                 })
943                 ds = ds + relativedelta(months=interval)
944         return True
945
946     def find(self, cr, uid, dt=None, exception=True, context=None):
947         res = self.finds(cr, uid, dt, exception, context=context)
948         return res and res[0] or False
949
950     def finds(self, cr, uid, dt=None, exception=True, context=None):
951         if context is None: context = {}
952         if not dt:
953             dt = fields.date.context_today(self,cr,uid,context=context)
954         args = [('date_start', '<=' ,dt), ('date_stop', '>=', dt)]
955         if context.get('company_id', False):
956             company_id = context['company_id']
957         else:
958             company_id = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.id
959         args.append(('company_id', '=', company_id))
960         ids = self.search(cr, uid, args, context=context)
961         if not ids:
962             if exception:
963                 raise osv.except_osv(_('Error !'), _('No fiscal year defined for this date !\nPlease create one from the configuration of the accounting menu.'))
964             else:
965                 return []
966         return ids
967
968     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=80):
969         if args is None:
970             args = []
971         if context is None:
972             context = {}
973         ids = []
974         if name:
975             ids = self.search(cr, user, [('code', 'ilike', name)]+ args, limit=limit)
976         if not ids:
977             ids = self.search(cr, user, [('name', operator, name)]+ args, limit=limit)
978         return self.name_get(cr, user, ids, context=context)
979
980 account_fiscalyear()
981
982 class account_period(osv.osv):
983     _name = "account.period"
984     _description = "Account period"
985     _columns = {
986         'name': fields.char('Period Name', size=64, required=True),
987         'code': fields.char('Code', size=12),
988         'special': fields.boolean('Opening/Closing Period', size=12,
989             help="These periods can overlap."),
990         'date_start': fields.date('Start of Period', required=True, states={'done':[('readonly',True)]}),
991         'date_stop': fields.date('End of Period', required=True, states={'done':[('readonly',True)]}),
992         'fiscalyear_id': fields.many2one('account.fiscalyear', 'Fiscal Year', required=True, states={'done':[('readonly',True)]}, select=True),
993         'state': fields.selection([('draft','Open'), ('done','Closed')], 'State', readonly=True,
994                                   help='When monthly periods are created. The state is \'Draft\'. At the end of monthly period it is in \'Done\' state.'),
995         'company_id': fields.related('fiscalyear_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True)
996     }
997     _defaults = {
998         'state': 'draft',
999     }
1000     _order = "date_start, special desc"
1001     _sql_constraints = [
1002         ('name_company_uniq', 'unique(name, company_id)', 'The name of the period must be unique per company!'),
1003     ]
1004
1005     def _check_duration(self,cr,uid,ids,context=None):
1006         obj_period = self.browse(cr, uid, ids[0], context=context)
1007         if obj_period.date_stop < obj_period.date_start:
1008             return False
1009         return True
1010
1011     def _check_year_limit(self,cr,uid,ids,context=None):
1012         for obj_period in self.browse(cr, uid, ids, context=context):
1013             if obj_period.special:
1014                 continue
1015
1016             if obj_period.fiscalyear_id.date_stop < obj_period.date_stop or \
1017                obj_period.fiscalyear_id.date_stop < obj_period.date_start or \
1018                obj_period.fiscalyear_id.date_start > obj_period.date_start or \
1019                obj_period.fiscalyear_id.date_start > obj_period.date_stop:
1020                 return False
1021
1022             pids = self.search(cr, uid, [('date_stop','>=',obj_period.date_start),('date_start','<=',obj_period.date_stop),('special','=',False),('id','<>',obj_period.id)])
1023             for period in self.browse(cr, uid, pids):
1024                 if period.fiscalyear_id.company_id.id==obj_period.fiscalyear_id.company_id.id:
1025                     return False
1026         return True
1027
1028     _constraints = [
1029         (_check_duration, 'Error ! The duration of the Period(s) is/are invalid. ', ['date_stop']),
1030         (_check_year_limit, 'Invalid period ! Some periods overlap or the date period is not in the scope of the fiscal year. ', ['date_stop'])
1031     ]
1032
1033     def next(self, cr, uid, period, step, context=None):
1034         ids = self.search(cr, uid, [('date_start','>',period.date_start)])
1035         if len(ids)>=step:
1036             return ids[step-1]
1037         return False
1038
1039     def find(self, cr, uid, dt=None, context=None):
1040         if context is None: context = {}
1041         if not dt:
1042             dt = fields.date.context_today(self,cr,uid,context=context)
1043 #CHECKME: shouldn't we check the state of the period?
1044         args = [('date_start', '<=' ,dt), ('date_stop', '>=', dt)]
1045         if context.get('company_id', False):
1046             args.append(('company_id', '=', context['company_id']))
1047         else:
1048             company_id = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.id
1049             args.append(('company_id', '=', company_id))
1050         result = []
1051         if context.get('account_period_prefer_normal'):
1052             # look for non-special periods first, and fallback to all if no result is found
1053             result = self.search(cr, uid, args + [('special', '=', False)], context=context)
1054         if not result:
1055             result = self.search(cr, uid, args, context=context)
1056         if not result:
1057             raise osv.except_osv(_('Error !'), _('No period defined for this date: %s !\nPlease create one.')%dt)
1058         return result
1059
1060     def action_draft(self, cr, uid, ids, *args):
1061         mode = 'draft'
1062         cr.execute('update account_journal_period set state=%s where period_id in %s', (mode, tuple(ids),))
1063         cr.execute('update account_period set state=%s where id in %s', (mode, tuple(ids),))
1064         return True
1065
1066     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=100):
1067         if args is None:
1068             args = []
1069         if context is None:
1070             context = {}
1071         ids = []
1072         if name:
1073             ids = self.search(cr, user, [('code','ilike',name)]+ args, limit=limit)
1074         if not ids:
1075             ids = self.search(cr, user, [('name',operator,name)]+ args, limit=limit)
1076         return self.name_get(cr, user, ids, context=context)
1077
1078     def write(self, cr, uid, ids, vals, context=None):
1079         if 'company_id' in vals:
1080             move_lines = self.pool.get('account.move.line').search(cr, uid, [('period_id', 'in', ids)])
1081             if move_lines:
1082                 raise osv.except_osv(_('Warning !'), _('You can not modify company of this period as some journal items exists.'))
1083         return super(account_period, self).write(cr, uid, ids, vals, context=context)
1084
1085     def build_ctx_periods(self, cr, uid, period_from_id, period_to_id):
1086         if period_from_id == period_to_id:
1087             return [period_from_id]
1088         period_from = self.browse(cr, uid, period_from_id)
1089         period_date_start = period_from.date_start
1090         company1_id = period_from.company_id.id
1091         period_to = self.browse(cr, uid, period_to_id)
1092         period_date_stop = period_to.date_stop
1093         company2_id = period_to.company_id.id
1094         if company1_id != company2_id:
1095             raise osv.except_osv(_('Error'), _('You should have chosen periods that belongs to the same company'))
1096         if period_date_start > period_date_stop:
1097             raise osv.except_osv(_('Error'), _('Start period should be smaller then End period'))
1098         #for period from = january, we want to exclude the opening period (but it has same date_from, so we have to check if period_from is special or not to include that clause or not in the search).
1099         if period_from.special:
1100             return self.search(cr, uid, [('date_start', '>=', period_date_start), ('date_stop', '<=', period_date_stop), ('company_id', '=', company1_id)])
1101         return self.search(cr, uid, [('date_start', '>=', period_date_start), ('date_stop', '<=', period_date_stop), ('company_id', '=', company1_id), ('special', '=', False)])
1102
1103 account_period()
1104
1105 class account_journal_period(osv.osv):
1106     _name = "account.journal.period"
1107     _description = "Journal Period"
1108
1109     def _icon_get(self, cr, uid, ids, field_name, arg=None, context=None):
1110         result = {}.fromkeys(ids, 'STOCK_NEW')
1111         for r in self.read(cr, uid, ids, ['state']):
1112             result[r['id']] = {
1113                 'draft': 'STOCK_NEW',
1114                 'printed': 'STOCK_PRINT_PREVIEW',
1115                 'done': 'STOCK_DIALOG_AUTHENTICATION',
1116             }.get(r['state'], 'STOCK_NEW')
1117         return result
1118
1119     _columns = {
1120         'name': fields.char('Journal-Period Name', size=64, required=True),
1121         'journal_id': fields.many2one('account.journal', 'Journal', required=True, ondelete="cascade"),
1122         'period_id': fields.many2one('account.period', 'Period', required=True, ondelete="cascade"),
1123         'icon': fields.function(_icon_get, string='Icon', type='char', size=32),
1124         'active': fields.boolean('Active', required=True, help="If the active field is set to False, it will allow you to hide the journal period without removing it."),
1125         'state': fields.selection([('draft','Draft'), ('printed','Printed'), ('done','Done')], 'State', required=True, readonly=True,
1126                                   help='When journal period is created. The state is \'Draft\'. If a report is printed it comes to \'Printed\' state. When all transactions are done, it comes in \'Done\' state.'),
1127         'fiscalyear_id': fields.related('period_id', 'fiscalyear_id', string='Fiscal Year', type='many2one', relation='account.fiscalyear'),
1128         'company_id': fields.related('journal_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True)
1129     }
1130
1131     def _check(self, cr, uid, ids, context=None):
1132         for obj in self.browse(cr, uid, ids, context=context):
1133             cr.execute('select * from account_move_line where journal_id=%s and period_id=%s limit 1', (obj.journal_id.id, obj.period_id.id))
1134             res = cr.fetchall()
1135             if res:
1136                 raise osv.except_osv(_('Error !'), _('You can not modify/delete a journal with entries for this period !'))
1137         return True
1138
1139     def write(self, cr, uid, ids, vals, context=None):
1140         self._check(cr, uid, ids, context=context)
1141         return super(account_journal_period, self).write(cr, uid, ids, vals, context=context)
1142
1143     def create(self, cr, uid, vals, context=None):
1144         period_id = vals.get('period_id',False)
1145         if period_id:
1146             period = self.pool.get('account.period').browse(cr, uid, period_id, context=context)
1147             vals['state']=period.state
1148         return super(account_journal_period, self).create(cr, uid, vals, context)
1149
1150     def unlink(self, cr, uid, ids, context=None):
1151         self._check(cr, uid, ids, context=context)
1152         return super(account_journal_period, self).unlink(cr, uid, ids, context=context)
1153
1154     _defaults = {
1155         'state': 'draft',
1156         'active': True,
1157     }
1158     _order = "period_id"
1159
1160 account_journal_period()
1161
1162 class account_fiscalyear(osv.osv):
1163     _inherit = "account.fiscalyear"
1164     _description = "Fiscal Year"
1165     _columns = {
1166         'end_journal_period_id':fields.many2one('account.journal.period','End of Year Entries Journal', readonly=True),
1167     }
1168
1169     def copy(self, cr, uid, id, default={}, context=None):
1170         default.update({
1171             'period_ids': [],
1172             'end_journal_period_id': False
1173         })
1174         return super(account_fiscalyear, self).copy(cr, uid, id, default=default, context=context)
1175
1176 account_fiscalyear()
1177 #----------------------------------------------------------
1178 # Entries
1179 #----------------------------------------------------------
1180 class account_move(osv.osv):
1181     _name = "account.move"
1182     _description = "Account Entry"
1183     _order = 'id desc'
1184
1185     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=80):
1186         """
1187         Returns a list of tupples containing id, name, as internally it is called {def name_get}
1188         result format: {[(id, name), (id, name), ...]}
1189
1190         @param cr: A database cursor
1191         @param user: ID of the user currently logged in
1192         @param name: name to search
1193         @param args: other arguments
1194         @param operator: default operator is 'ilike', it can be changed
1195         @param context: context arguments, like lang, time zone
1196         @param limit: Returns first 'n' ids of complete result, default is 80.
1197
1198         @return: Returns a list of tuples containing id and name
1199         """
1200
1201         if not args:
1202           args = []
1203         ids = []
1204         if name:
1205             ids += self.search(cr, user, [('name','ilike',name)]+args, limit=limit, context=context)
1206
1207         if not ids and name and type(name) == int:
1208             ids += self.search(cr, user, [('id','=',name)]+args, limit=limit, context=context)
1209
1210         if not ids:
1211             ids += self.search(cr, user, args, limit=limit, context=context)
1212
1213         return self.name_get(cr, user, ids, context=context)
1214
1215     def name_get(self, cursor, user, ids, context=None):
1216         if isinstance(ids, (int, long)):
1217             ids = [ids]
1218         if not ids:
1219             return []
1220         res = []
1221         data_move = self.pool.get('account.move').browse(cursor, user, ids, context=context)
1222         for move in data_move:
1223             if move.state=='draft':
1224                 name = '*' + str(move.id)
1225             else:
1226                 name = move.name
1227             res.append((move.id, name))
1228         return res
1229
1230     def _get_period(self, cr, uid, context=None):
1231         ctx = dict(context or {}, account_period_prefer_normal=True)
1232         period_ids = self.pool.get('account.period').find(cr, uid, context=ctx)
1233         return period_ids[0]
1234
1235     def _amount_compute(self, cr, uid, ids, name, args, context, where =''):
1236         if not ids: return {}
1237         cr.execute( 'SELECT move_id, SUM(debit) '\
1238                     'FROM account_move_line '\
1239                     'WHERE move_id IN %s '\
1240                     'GROUP BY move_id', (tuple(ids),))
1241         result = dict(cr.fetchall())
1242         for id in ids:
1243             result.setdefault(id, 0.0)
1244         return result
1245
1246     def _search_amount(self, cr, uid, obj, name, args, context):
1247         ids = set()
1248         for cond in args:
1249             amount = cond[2]
1250             if isinstance(cond[2],(list,tuple)):
1251                 if cond[1] in ['in','not in']:
1252                     amount = tuple(cond[2])
1253                 else:
1254                     continue
1255             else:
1256                 if cond[1] in ['=like', 'like', 'not like', 'ilike', 'not ilike', 'in', 'not in', 'child_of']:
1257                     continue
1258
1259             cr.execute("select move_id from account_move_line group by move_id having sum(debit) %s %%s" % (cond[1]),(amount,))
1260             res_ids = set(id[0] for id in cr.fetchall())
1261             ids = ids and (ids & res_ids) or res_ids
1262         if ids:
1263             return [('id', 'in', tuple(ids))]
1264         return [('id', '=', '0')]
1265
1266     _columns = {
1267         'name': fields.char('Number', size=64, required=True),
1268         'ref': fields.char('Reference', size=64),
1269         'period_id': fields.many2one('account.period', 'Period', required=True, states={'posted':[('readonly',True)]}),
1270         'journal_id': fields.many2one('account.journal', 'Journal', required=True, states={'posted':[('readonly',True)]}),
1271         'state': fields.selection([('draft','Unposted'), ('posted','Posted')], 'State', required=True, readonly=True,
1272             help='All manually created new journal entries are usually in the state \'Unposted\', but you can set the option to skip that state on the related journal. In that case, they will be behave as journal entries automatically created by the system on document validation (invoices, bank statements...) and will be created in \'Posted\' state.'),
1273         'line_id': fields.one2many('account.move.line', 'move_id', 'Entries', states={'posted':[('readonly',True)]}),
1274         'to_check': fields.boolean('To Review', help='Check this box if you are unsure of that journal entry and if you want to note it as \'to be reviewed\' by an accounting expert.'),
1275         'partner_id': fields.related('line_id', 'partner_id', type="many2one", relation="res.partner", string="Partner", store=True),
1276         'amount': fields.function(_amount_compute, string='Amount', digits_compute=dp.get_precision('Account'), type='float', fnct_search=_search_amount),
1277         'date': fields.date('Date', required=True, states={'posted':[('readonly',True)]}, select=True),
1278         'narration':fields.text('Internal Note'),
1279         'company_id': fields.related('journal_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
1280         'balance': fields.float('balance', digits_compute=dp.get_precision('Account'), help="This is a field only used for internal purpose and shouldn't be displayed"),
1281     }
1282
1283     _defaults = {
1284         'name': '/',
1285         'state': 'draft',
1286         'period_id': _get_period,
1287         'date': fields.date.context_today,
1288         'company_id': lambda self, cr, uid, c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
1289     }
1290
1291     def _check_centralisation(self, cursor, user, ids, context=None):
1292         for move in self.browse(cursor, user, ids, context=context):
1293             if move.journal_id.centralisation:
1294                 move_ids = self.search(cursor, user, [
1295                     ('period_id', '=', move.period_id.id),
1296                     ('journal_id', '=', move.journal_id.id),
1297                     ])
1298                 if len(move_ids) > 1:
1299                     return False
1300         return True
1301
1302     _constraints = [
1303         (_check_centralisation,
1304             'You can not create more than one move per period on centralized journal',
1305             ['journal_id']),
1306     ]
1307
1308     def post(self, cr, uid, ids, context=None):
1309         if context is None:
1310             context = {}
1311         invoice = context.get('invoice', False)
1312         valid_moves = self.validate(cr, uid, ids, context)
1313
1314         if not valid_moves:
1315             raise osv.except_osv(_('Integrity Error !'), _('You can not validate a non-balanced entry !\nMake sure you have configured payment terms properly !\nThe latest payment term line should be of the type "Balance" !'))
1316         obj_sequence = self.pool.get('ir.sequence')
1317         for move in self.browse(cr, uid, valid_moves, context=context):
1318             if move.name =='/':
1319                 new_name = False
1320                 journal = move.journal_id
1321
1322                 if invoice and invoice.internal_number:
1323                     new_name = invoice.internal_number
1324                 else:
1325                     if journal.sequence_id:
1326                         c = {'fiscalyear_id': move.period_id.fiscalyear_id.id}
1327                         new_name = obj_sequence.next_by_id(cr, uid, journal.sequence_id.id, c)
1328                     else:
1329                         raise osv.except_osv(_('Error'), _('No sequence defined on the journal !'))
1330
1331                 if new_name:
1332                     self.write(cr, uid, [move.id], {'name':new_name})
1333
1334         cr.execute('UPDATE account_move '\
1335                    'SET state=%s '\
1336                    'WHERE id IN %s',
1337                    ('posted', tuple(valid_moves),))
1338         return True
1339
1340     def button_validate(self, cursor, user, ids, context=None):
1341         for move in self.browse(cursor, user, ids, context=context):
1342             # check that all accounts have the same topmost ancestor
1343             top_common = None
1344             for line in move.line_id:
1345                 account = line.account_id
1346                 top_account = account
1347                 while top_account.parent_id:
1348                     top_account = top_account.parent_id
1349                 if not top_common:
1350                     top_common = top_account
1351                 elif top_account.id != top_common.id:
1352                     raise osv.except_osv(_('Error !'),
1353                                          _('You cannot validate this journal entry because account "%s" does not belong to chart of accounts "%s"!') % (account.name, top_common.name))
1354         return self.post(cursor, user, ids, context=context)
1355
1356     def button_cancel(self, cr, uid, ids, context=None):
1357         for line in self.browse(cr, uid, ids, context=context):
1358             if line.period_id.state == 'done':
1359                 raise osv.except_osv(_('Error !'), _('You can not modify a posted entry of closed periods'))
1360             elif not line.journal_id.update_posted:
1361                 raise osv.except_osv(_('Error !'), _('You can not modify a posted entry of this journal !\nYou should set the journal to allow cancelling entries if you want to do that.'))
1362         if ids:
1363             cr.execute('UPDATE account_move '\
1364                        'SET state=%s '\
1365                        'WHERE id IN %s', ('draft', tuple(ids),))
1366         return True
1367
1368     def onchange_line_id(self, cr, uid, ids, line_ids, context=None):
1369         balance = 0.0
1370         line_ids = [ line for line in line_ids if not (isinstance(line, (tuple, list)) and line and line[0] == 2) ]
1371         line_ids = self.resolve_o2m_commands_to_record_dicts(cr, uid, 'line_id', line_ids, context=context)
1372         for line in line_ids:
1373             balance += (line['debit'] or 0.00)- (line['credit'] or 0.00)
1374         return {'value': {'balance': balance}}
1375
1376     def write(self, cr, uid, ids, vals, context=None):
1377         if context is None:
1378             context = {}
1379         c = context.copy()
1380         c['novalidate'] = True
1381         result = super(account_move, self).write(cr, uid, ids, vals, c)
1382         self.validate(cr, uid, ids, context=context)
1383         return result
1384
1385     #
1386     # TODO: Check if period is closed !
1387     #
1388     def create(self, cr, uid, vals, context=None):
1389         if context is None:
1390             context = {}
1391         if 'line_id' in vals and context.get('copy'):
1392             for l in vals['line_id']:
1393                 if not l[0]:
1394                     l[2].update({
1395                         'reconcile_id':False,
1396                         'reconcil_partial_id':False,
1397                         'analytic_lines':False,
1398                         'invoice':False,
1399                         'ref':False,
1400                         'balance':False,
1401                         'account_tax_id':False,
1402                     })
1403
1404             if 'journal_id' in vals and vals.get('journal_id', False):
1405                 for l in vals['line_id']:
1406                     if not l[0]:
1407                         l[2]['journal_id'] = vals['journal_id']
1408                 context['journal_id'] = vals['journal_id']
1409             if 'period_id' in vals:
1410                 for l in vals['line_id']:
1411                     if not l[0]:
1412                         l[2]['period_id'] = vals['period_id']
1413                 context['period_id'] = vals['period_id']
1414             else:
1415                 default_period = self._get_period(cr, uid, context)
1416                 for l in vals['line_id']:
1417                     if not l[0]:
1418                         l[2]['period_id'] = default_period
1419                 context['period_id'] = default_period
1420
1421         if 'line_id' in vals:
1422             c = context.copy()
1423             c['novalidate'] = True
1424             result = super(account_move, self).create(cr, uid, vals, c)
1425             self.validate(cr, uid, [result], context)
1426         else:
1427             result = super(account_move, self).create(cr, uid, vals, context)
1428         return result
1429
1430     def copy(self, cr, uid, id, default={}, context=None):
1431         if context is None:
1432             context = {}
1433         default.update({
1434             'state':'draft',
1435             'name':'/',
1436         })
1437         context.update({
1438             'copy':True
1439         })
1440         return super(account_move, self).copy(cr, uid, id, default, context)
1441
1442     def unlink(self, cr, uid, ids, context=None, check=True):
1443         if context is None:
1444             context = {}
1445         toremove = []
1446         obj_move_line = self.pool.get('account.move.line')
1447         for move in self.browse(cr, uid, ids, context=context):
1448             if move['state'] != 'draft':
1449                 raise osv.except_osv(_('UserError'),
1450                         _('You can not delete a posted journal entry "%s"!') % \
1451                                 move['name'])
1452             line_ids = map(lambda x: x.id, move.line_id)
1453             context['journal_id'] = move.journal_id.id
1454             context['period_id'] = move.period_id.id
1455             obj_move_line._update_check(cr, uid, line_ids, context)
1456             obj_move_line.unlink(cr, uid, line_ids, context=context)
1457             toremove.append(move.id)
1458         result = super(account_move, self).unlink(cr, uid, toremove, context)
1459         return result
1460
1461     def _compute_balance(self, cr, uid, id, context=None):
1462         move = self.browse(cr, uid, id, context=context)
1463         amount = 0
1464         for line in move.line_id:
1465             amount+= (line.debit - line.credit)
1466         return amount
1467
1468     def _centralise(self, cr, uid, move, mode, context=None):
1469         assert mode in ('debit', 'credit'), 'Invalid Mode' #to prevent sql injection
1470         currency_obj = self.pool.get('res.currency')
1471         if context is None:
1472             context = {}
1473
1474         if mode=='credit':
1475             account_id = move.journal_id.default_debit_account_id.id
1476             mode2 = 'debit'
1477             if not account_id:
1478                 raise osv.except_osv(_('UserError'),
1479                         _('There is no default default debit account defined \n' \
1480                                 'on journal "%s"') % move.journal_id.name)
1481         else:
1482             account_id = move.journal_id.default_credit_account_id.id
1483             mode2 = 'credit'
1484             if not account_id:
1485                 raise osv.except_osv(_('UserError'),
1486                         _('There is no default default credit account defined \n' \
1487                                 'on journal "%s"') % move.journal_id.name)
1488
1489         # find the first line of this move with the current mode
1490         # or create it if it doesn't exist
1491         cr.execute('select id from account_move_line where move_id=%s and centralisation=%s limit 1', (move.id, mode))
1492         res = cr.fetchone()
1493         if res:
1494             line_id = res[0]
1495         else:
1496             context.update({'journal_id': move.journal_id.id, 'period_id': move.period_id.id})
1497             line_id = self.pool.get('account.move.line').create(cr, uid, {
1498                 'name': _(mode.capitalize()+' Centralisation'),
1499                 'centralisation': mode,
1500                 'account_id': account_id,
1501                 'move_id': move.id,
1502                 'journal_id': move.journal_id.id,
1503                 'period_id': move.period_id.id,
1504                 'date': move.period_id.date_stop,
1505                 'debit': 0.0,
1506                 'credit': 0.0,
1507             }, context)
1508
1509         # find the first line of this move with the other mode
1510         # so that we can exclude it from our calculation
1511         cr.execute('select id from account_move_line where move_id=%s and centralisation=%s limit 1', (move.id, mode2))
1512         res = cr.fetchone()
1513         if res:
1514             line_id2 = res[0]
1515         else:
1516             line_id2 = 0
1517
1518         cr.execute('SELECT SUM(%s) FROM account_move_line WHERE move_id=%%s AND id!=%%s' % (mode,), (move.id, line_id2))
1519         result = cr.fetchone()[0] or 0.0
1520         cr.execute('update account_move_line set '+mode2+'=%s where id=%s', (result, line_id))
1521
1522         #adjust also the amount in currency if needed
1523         cr.execute("select currency_id, sum(amount_currency) as amount_currency from account_move_line where move_id = %s and currency_id is not null group by currency_id", (move.id,))
1524         for row in cr.dictfetchall():
1525             currency_id = currency_obj.browse(cr, uid, row['currency_id'], context=context)
1526             if not currency_obj.is_zero(cr, uid, currency_id, row['amount_currency']):
1527                 amount_currency = row['amount_currency'] * -1
1528                 account_id = amount_currency > 0 and move.journal_id.default_debit_account_id.id or move.journal_id.default_credit_account_id.id
1529                 cr.execute('select id from account_move_line where move_id=%s and centralisation=\'currency\' and currency_id = %slimit 1', (move.id, row['currency_id']))
1530                 res = cr.fetchone()
1531                 if res:
1532                     cr.execute('update account_move_line set amount_currency=%s , account_id=%s where id=%s', (amount_currency, account_id, res[0]))
1533                 else:
1534                     context.update({'journal_id': move.journal_id.id, 'period_id': move.period_id.id})
1535                     line_id = self.pool.get('account.move.line').create(cr, uid, {
1536                         'name': _('Currency Adjustment'),
1537                         'centralisation': 'currency',
1538                         'account_id': account_id,
1539                         'move_id': move.id,
1540                         'journal_id': move.journal_id.id,
1541                         'period_id': move.period_id.id,
1542                         'date': move.period_id.date_stop,
1543                         'debit': 0.0,
1544                         'credit': 0.0,
1545                         'currency_id': row['currency_id'],
1546                         'amount_currency': amount_currency,
1547                     }, context)
1548
1549         return True
1550
1551     #
1552     # Validate a balanced move. If it is a centralised journal, create a move.
1553     #
1554     def validate(self, cr, uid, ids, context=None):
1555         if context and ('__last_update' in context):
1556             del context['__last_update']
1557
1558         valid_moves = [] #Maintains a list of moves which can be responsible to create analytic entries
1559         obj_analytic_line = self.pool.get('account.analytic.line')
1560         obj_move_line = self.pool.get('account.move.line')
1561         for move in self.browse(cr, uid, ids, context):
1562             # Unlink old analytic lines on move_lines
1563             for obj_line in move.line_id:
1564                 for obj in obj_line.analytic_lines:
1565                     obj_analytic_line.unlink(cr,uid,obj.id)
1566
1567             journal = move.journal_id
1568             amount = 0
1569             line_ids = []
1570             line_draft_ids = []
1571             company_id = None
1572             for line in move.line_id:
1573                 amount += line.debit - line.credit
1574                 line_ids.append(line.id)
1575                 if line.state=='draft':
1576                     line_draft_ids.append(line.id)
1577
1578                 if not company_id:
1579                     company_id = line.account_id.company_id.id
1580                 if not company_id == line.account_id.company_id.id:
1581                     raise osv.except_osv(_('Error'), _("Couldn't create move between different companies"))
1582
1583                 if line.account_id.currency_id and line.currency_id:
1584                     if line.account_id.currency_id.id != line.currency_id.id and (line.account_id.currency_id.id != line.account_id.company_id.currency_id.id):
1585                         raise osv.except_osv(_('Error'), _("""Couldn't create move with currency different from the secondary currency of the account "%s - %s". Clear the secondary currency field of the account definition if you want to accept all currencies.""") % (line.account_id.code, line.account_id.name))
1586
1587             if abs(amount) < 10 ** -4:
1588                 # If the move is balanced
1589                 # Add to the list of valid moves
1590                 # (analytic lines will be created later for valid moves)
1591                 valid_moves.append(move)
1592
1593                 # Check whether the move lines are confirmed
1594
1595                 if not line_draft_ids:
1596                     continue
1597                 # Update the move lines (set them as valid)
1598
1599                 obj_move_line.write(cr, uid, line_draft_ids, {
1600                     'state': 'valid'
1601                 }, context, check=False)
1602
1603                 account = {}
1604                 account2 = {}
1605
1606                 if journal.type in ('purchase','sale'):
1607                     for line in move.line_id:
1608                         code = amount = 0
1609                         key = (line.account_id.id, line.tax_code_id.id)
1610                         if key in account2:
1611                             code = account2[key][0]
1612                             amount = account2[key][1] * (line.debit + line.credit)
1613                         elif line.account_id.id in account:
1614                             code = account[line.account_id.id][0]
1615                             amount = account[line.account_id.id][1] * (line.debit + line.credit)
1616                         if (code or amount) and not (line.tax_code_id or line.tax_amount):
1617                             obj_move_line.write(cr, uid, [line.id], {
1618                                 'tax_code_id': code,
1619                                 'tax_amount': amount
1620                             }, context, check=False)
1621             elif journal.centralisation:
1622                 # If the move is not balanced, it must be centralised...
1623
1624                 # Add to the list of valid moves
1625                 # (analytic lines will be created later for valid moves)
1626                 valid_moves.append(move)
1627
1628                 #
1629                 # Update the move lines (set them as valid)
1630                 #
1631                 self._centralise(cr, uid, move, 'debit', context=context)
1632                 self._centralise(cr, uid, move, 'credit', context=context)
1633                 obj_move_line.write(cr, uid, line_draft_ids, {
1634                     'state': 'valid'
1635                 }, context, check=False)
1636             else:
1637                 # We can't validate it (it's unbalanced)
1638                 # Setting the lines as draft
1639                 obj_move_line.write(cr, uid, line_ids, {
1640                     'state': 'draft'
1641                 }, context, check=False)
1642         # Create analytic lines for the valid moves
1643         for record in valid_moves:
1644             obj_move_line.create_analytic_lines(cr, uid, [line.id for line in record.line_id], context)
1645
1646         valid_moves = [move.id for move in valid_moves]
1647         return len(valid_moves) > 0 and valid_moves or False
1648
1649 account_move()
1650
1651 class account_move_reconcile(osv.osv):
1652     _name = "account.move.reconcile"
1653     _description = "Account Reconciliation"
1654     _columns = {
1655         'name': fields.char('Name', size=64, required=True),
1656         'type': fields.char('Type', size=16, required=True),
1657         'line_id': fields.one2many('account.move.line', 'reconcile_id', 'Entry Lines'),
1658         'line_partial_ids': fields.one2many('account.move.line', 'reconcile_partial_id', 'Partial Entry lines'),
1659         'create_date': fields.date('Creation date', readonly=True),
1660     }
1661     _defaults = {
1662         'name': lambda self,cr,uid,ctx=None: self.pool.get('ir.sequence').get(cr, uid, 'account.reconcile', context=ctx) or '/',
1663     }
1664
1665     def reconcile_partial_check(self, cr, uid, ids, type='auto', context=None):
1666         total = 0.0
1667         for rec in self.browse(cr, uid, ids, context=context):
1668             for line in rec.line_partial_ids:
1669                 if line.account_id.currency_id:
1670                     total += line.amount_currency
1671                 else:
1672                     total += (line.debit or 0.0) - (line.credit or 0.0)
1673         if not total:
1674             self.pool.get('account.move.line').write(cr, uid,
1675                 map(lambda x: x.id, rec.line_partial_ids),
1676                 {'reconcile_id': rec.id }
1677             )
1678         return True
1679
1680     def name_get(self, cr, uid, ids, context=None):
1681         if not ids:
1682             return []
1683         result = []
1684         for r in self.browse(cr, uid, ids, context=context):
1685             total = reduce(lambda y,t: (t.debit or 0.0) - (t.credit or 0.0) + y, r.line_partial_ids, 0.0)
1686             if total:
1687                 name = '%s (%.2f)' % (r.name, total)
1688                 result.append((r.id,name))
1689             else:
1690                 result.append((r.id,r.name))
1691         return result
1692
1693 account_move_reconcile()
1694
1695 #----------------------------------------------------------
1696 # Tax
1697 #----------------------------------------------------------
1698 """
1699 a documenter
1700 child_depend: la taxe depend des taxes filles
1701 """
1702 class account_tax_code(osv.osv):
1703     """
1704     A code for the tax object.
1705
1706     This code is used for some tax declarations.
1707     """
1708     def _sum(self, cr, uid, ids, name, args, context, where ='', where_params=()):
1709         parent_ids = tuple(self.search(cr, uid, [('parent_id', 'child_of', ids)]))
1710         if context.get('based_on', 'invoices') == 'payments':
1711             cr.execute('SELECT line.tax_code_id, sum(line.tax_amount) \
1712                     FROM account_move_line AS line, \
1713                         account_move AS move \
1714                         LEFT JOIN account_invoice invoice ON \
1715                             (invoice.move_id = move.id) \
1716                     WHERE line.tax_code_id IN %s '+where+' \
1717                         AND move.id = line.move_id \
1718                         AND ((invoice.state = \'paid\') \
1719                             OR (invoice.id IS NULL)) \
1720                             GROUP BY line.tax_code_id',
1721                                 (parent_ids,) + where_params)
1722         else:
1723             cr.execute('SELECT line.tax_code_id, sum(line.tax_amount) \
1724                     FROM account_move_line AS line, \
1725                     account_move AS move \
1726                     WHERE line.tax_code_id IN %s '+where+' \
1727                     AND move.id = line.move_id \
1728                     GROUP BY line.tax_code_id',
1729                        (parent_ids,) + where_params)
1730         res=dict(cr.fetchall())
1731         obj_precision = self.pool.get('decimal.precision')
1732         res2 = {}
1733         for record in self.browse(cr, uid, ids, context=context):
1734             def _rec_get(record):
1735                 amount = res.get(record.id, 0.0)
1736                 for rec in record.child_ids:
1737                     amount += _rec_get(rec) * rec.sign
1738                 return amount
1739             res2[record.id] = round(_rec_get(record), obj_precision.precision_get(cr, uid, 'Account'))
1740         return res2
1741
1742     def _sum_year(self, cr, uid, ids, name, args, context=None):
1743         if context is None:
1744             context = {}
1745         move_state = ('posted', )
1746         if context.get('state', 'all') == 'all':
1747             move_state = ('draft', 'posted', )
1748         if context.get('fiscalyear_id', False):
1749             fiscalyear_id = [context['fiscalyear_id']]
1750         else:
1751             fiscalyear_id = self.pool.get('account.fiscalyear').finds(cr, uid, exception=False)
1752         where = ''
1753         where_params = ()
1754         if fiscalyear_id:
1755             pids = []
1756             for fy in fiscalyear_id:
1757                 pids += map(lambda x: str(x.id), self.pool.get('account.fiscalyear').browse(cr, uid, fy).period_ids)
1758             if pids:
1759                 where = ' AND line.period_id IN %s AND move.state IN %s '
1760                 where_params = (tuple(pids), move_state)
1761         return self._sum(cr, uid, ids, name, args, context,
1762                 where=where, where_params=where_params)
1763
1764     def _sum_period(self, cr, uid, ids, name, args, context):
1765         if context is None:
1766             context = {}
1767         move_state = ('posted', )
1768         if context.get('state', False) == 'all':
1769             move_state = ('draft', 'posted', )
1770         if context.get('period_id', False):
1771             period_id = context['period_id']
1772         else:
1773             period_id = self.pool.get('account.period').find(cr, uid)
1774             if not period_id:
1775                 return dict.fromkeys(ids, 0.0)
1776             period_id = period_id[0]
1777         return self._sum(cr, uid, ids, name, args, context,
1778                 where=' AND line.period_id=%s AND move.state IN %s', where_params=(period_id, move_state))
1779
1780     _name = 'account.tax.code'
1781     _description = 'Tax Code'
1782     _rec_name = 'code'
1783     _columns = {
1784         'name': fields.char('Tax Case Name', size=64, required=True, translate=True),
1785         'code': fields.char('Case Code', size=64),
1786         'info': fields.text('Description'),
1787         'sum': fields.function(_sum_year, string="Year Sum"),
1788         'sum_period': fields.function(_sum_period, string="Period Sum"),
1789         'parent_id': fields.many2one('account.tax.code', 'Parent Code', select=True),
1790         'child_ids': fields.one2many('account.tax.code', 'parent_id', 'Child Codes'),
1791         'line_ids': fields.one2many('account.move.line', 'tax_code_id', 'Lines'),
1792         'company_id': fields.many2one('res.company', 'Company', required=True),
1793         'sign': fields.float('Coefficent for parent', required=True, help='You can specify here the coefficient that will be used when consolidating the amount of this case into its parent. For example, set 1/-1 if you want to add/substract it.'),
1794         'notprintable':fields.boolean("Not Printable in Invoice", help="Check this box if you don't want any VAT related to this Tax Code to appear on invoices"),
1795         'sequence': fields.integer('Sequence', help="Determine the display order in the report 'Accounting \ Reporting \ Generic Reporting \ Taxes \ Taxes Report'"),
1796     }
1797
1798     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=80):
1799         if not args:
1800             args = []
1801         if context is None:
1802             context = {}
1803         ids = self.search(cr, user, ['|',('name',operator,name),('code',operator,name)] + args, limit=limit, context=context)
1804         return self.name_get(cr, user, ids, context)
1805
1806     def name_get(self, cr, uid, ids, context=None):
1807         if isinstance(ids, (int, long)):
1808             ids = [ids]
1809         if not ids:
1810             return []
1811         if isinstance(ids, (int, long)):
1812             ids = [ids]
1813         reads = self.read(cr, uid, ids, ['name','code'], context, load='_classic_write')
1814         return [(x['id'], (x['code'] and (x['code'] + ' - ') or '') + x['name']) \
1815                 for x in reads]
1816
1817     def _default_company(self, cr, uid, context=None):
1818         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1819         if user.company_id:
1820             return user.company_id.id
1821         return self.pool.get('res.company').search(cr, uid, [('parent_id', '=', False)])[0]
1822     _defaults = {
1823         'company_id': _default_company,
1824         'sign': 1.0,
1825         'notprintable': False,
1826     }
1827
1828     def copy(self, cr, uid, id, default=None, context=None):
1829         if default is None:
1830             default = {}
1831         default = default.copy()
1832         default.update({'line_ids': []})
1833         return super(account_tax_code, self).copy(cr, uid, id, default, context)
1834
1835     _check_recursion = check_cycle
1836     _constraints = [
1837         (_check_recursion, 'Error ! You can not create recursive accounts.', ['parent_id'])
1838     ]
1839     _order = 'code'
1840
1841 account_tax_code()
1842
1843 class account_tax(osv.osv):
1844     """
1845     A tax object.
1846
1847     Type: percent, fixed, none, code
1848         PERCENT: tax = price * amount
1849         FIXED: tax = price + amount
1850         NONE: no tax line
1851         CODE: execute python code. localcontext = {'price_unit':pu, 'address':address_object}
1852             return result in the context
1853             Ex: result=round(price_unit*0.21,4)
1854     """
1855
1856     def get_precision_tax():
1857         def change_digit_tax(cr):
1858             res = pooler.get_pool(cr.dbname).get('decimal.precision').precision_get(cr, 1, 'Account')
1859             return (16, res+2)
1860         return change_digit_tax
1861
1862     _name = 'account.tax'
1863     _description = 'Tax'
1864     _columns = {
1865         'name': fields.char('Tax Name', size=64, required=True, translate=True, help="This name will be displayed on reports"),
1866         'sequence': fields.integer('Sequence', required=True, help="The sequence field is used to order the tax lines from the lowest sequences to the higher ones. The order is important if you have a tax with several tax children. In this case, the evaluation order is important."),
1867         'amount': fields.float('Amount', required=True, digits_compute=get_precision_tax(), help="For taxes of type percentage, enter % ratio between 0-1."),
1868         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the tax without removing it."),
1869         'type': fields.selection( [('percent','Percentage'), ('fixed','Fixed Amount'), ('none','None'), ('code','Python Code'), ('balance','Balance')], 'Tax Type', required=True,
1870             help="The computation method for the tax amount."),
1871         'applicable_type': fields.selection( [('true','Always'), ('code','Given by Python Code')], 'Applicability', required=True,
1872             help="If not applicable (computed through a Python code), the tax won't appear on the invoice."),
1873         'domain':fields.char('Domain', size=32, help="This field is only used if you develop your own module allowing developers to create specific taxes in a custom domain."),
1874         'account_collected_id':fields.many2one('account.account', 'Invoice Tax Account'),
1875         'account_paid_id':fields.many2one('account.account', 'Refund Tax Account'),
1876         'parent_id':fields.many2one('account.tax', 'Parent Tax Account', select=True),
1877         'child_ids':fields.one2many('account.tax', 'parent_id', 'Child Tax Accounts'),
1878         'child_depend':fields.boolean('Tax on Children', help="Set if the tax computation is based on the computation of child taxes rather than on the total amount."),
1879         'python_compute':fields.text('Python Code'),
1880         'python_compute_inv':fields.text('Python Code (reverse)'),
1881         'python_applicable':fields.text('Python Code'),
1882
1883         #
1884         # Fields used for the VAT declaration
1885         #
1886         'base_code_id': fields.many2one('account.tax.code', 'Account Base Code', help="Use this code for the VAT declaration."),
1887         'tax_code_id': fields.many2one('account.tax.code', 'Account Tax Code', help="Use this code for the VAT declaration."),
1888         'base_sign': fields.float('Base Code Sign', help="Usually 1 or -1."),
1889         'tax_sign': fields.float('Tax Code Sign', help="Usually 1 or -1."),
1890
1891         # Same fields for refund invoices
1892
1893         'ref_base_code_id': fields.many2one('account.tax.code', 'Refund Base Code', help="Use this code for the VAT declaration."),
1894         'ref_tax_code_id': fields.many2one('account.tax.code', 'Refund Tax Code', help="Use this code for the VAT declaration."),
1895         'ref_base_sign': fields.float('Base Code Sign', help="Usually 1 or -1."),
1896         'ref_tax_sign': fields.float('Tax Code Sign', help="Usually 1 or -1."),
1897         'include_base_amount': fields.boolean('Included in base amount', help="Indicates if the amount of tax must be included in the base amount for the computation of the next taxes"),
1898         'company_id': fields.many2one('res.company', 'Company', required=True),
1899         'description': fields.char('Tax Code',size=32),
1900         'price_include': fields.boolean('Tax Included in Price', help="Check this if the price you use on the product and invoices includes this tax."),
1901         'type_tax_use': fields.selection([('sale','Sale'),('purchase','Purchase'),('all','All')], 'Tax Application', required=True)
1902
1903     }
1904     _sql_constraints = [
1905         ('name_company_uniq', 'unique(name, company_id)', 'Tax Name must be unique per company!'),
1906     ]
1907
1908     def name_search(self, cr, user, name, args=None, operator='ilike', context=None, limit=80):
1909         """
1910         Returns a list of tupples containing id, name, as internally it is called {def name_get}
1911         result format: {[(id, name), (id, name), ...]}
1912
1913         @param cr: A database cursor
1914         @param user: ID of the user currently logged in
1915         @param name: name to search
1916         @param args: other arguments
1917         @param operator: default operator is 'ilike', it can be changed
1918         @param context: context arguments, like lang, time zone
1919         @param limit: Returns first 'n' ids of complete result, default is 80.
1920
1921         @return: Returns a list of tupples containing id and name
1922         """
1923         if not args:
1924             args = []
1925         if context is None:
1926             context = {}
1927         ids = []
1928         if name:
1929             ids = self.search(cr, user, [('description', '=', name)] + args, limit=limit, context=context)
1930             if not ids:
1931                 ids = self.search(cr, user, [('name', operator, name)] + args, limit=limit, context=context)
1932         else:
1933             ids = self.search(cr, user, args, limit=limit, context=context or {})
1934         return self.name_get(cr, user, ids, context=context)
1935
1936     def write(self, cr, uid, ids, vals, context=None):
1937         if vals.get('type', False) and vals['type'] in ('none', 'code'):
1938             vals.update({'amount': 0.0})
1939         return super(account_tax, self).write(cr, uid, ids, vals, context=context)
1940
1941     def search(self, cr, uid, args, offset=0, limit=None, order=None, context=None, count=False):
1942         journal_pool = self.pool.get('account.journal')
1943
1944         if context and context.has_key('type'):
1945             if context.get('type') in ('out_invoice','out_refund'):
1946                 args += [('type_tax_use','in',['sale','all'])]
1947             elif context.get('type') in ('in_invoice','in_refund'):
1948                 args += [('type_tax_use','in',['purchase','all'])]
1949
1950         if context and context.has_key('journal_id'):
1951             journal = journal_pool.browse(cr, uid, context.get('journal_id'))
1952             if journal.type in ('sale', 'purchase'):
1953                 args += [('type_tax_use','in',[journal.type,'all'])]
1954
1955         return super(account_tax, self).search(cr, uid, args, offset, limit, order, context, count)
1956
1957     def name_get(self, cr, uid, ids, context=None):
1958         if not ids:
1959             return []
1960         res = []
1961         for record in self.read(cr, uid, ids, ['description','name'], context=context):
1962             name = record['description'] and record['description'] or record['name']
1963             res.append((record['id'],name ))
1964         return res
1965
1966     def _default_company(self, cr, uid, context=None):
1967         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1968         if user.company_id:
1969             return user.company_id.id
1970         return self.pool.get('res.company').search(cr, uid, [('parent_id', '=', False)])[0]
1971
1972     _defaults = {
1973         'python_compute': '''# price_unit\n# address: res.partner.address object or False\n# product: product.product object or None\n# partner: res.partner object or None\n\nresult = price_unit * 0.10''',
1974         'python_compute_inv': '''# price_unit\n# address: res.partner.address object or False\n# product: product.product object or False\n\nresult = price_unit * 0.10''',
1975         'applicable_type': 'true',
1976         'type': 'percent',
1977         'amount': 0,
1978         'price_include': 0,
1979         'active': 1,
1980         'type_tax_use': 'all',
1981         'sequence': 1,
1982         'ref_tax_sign': 1,
1983         'ref_base_sign': 1,
1984         'tax_sign': 1,
1985         'base_sign': 1,
1986         'include_base_amount': False,
1987         'company_id': _default_company,
1988     }
1989     _order = 'sequence'
1990
1991     def _applicable(self, cr, uid, taxes, price_unit, address_id=None, product=None, partner=None):
1992         res = []
1993         obj_partener_address = self.pool.get('res.partner.address')
1994         for tax in taxes:
1995             if tax.applicable_type=='code':
1996                 localdict = {'price_unit':price_unit, 'address':obj_partener_address.browse(cr, uid, address_id), 'product':product, 'partner':partner}
1997                 exec tax.python_applicable in localdict
1998                 if localdict.get('result', False):
1999                     res.append(tax)
2000             else:
2001                 res.append(tax)
2002         return res
2003
2004     def _unit_compute(self, cr, uid, taxes, price_unit, address_id=None, product=None, partner=None, quantity=0):
2005         taxes = self._applicable(cr, uid, taxes, price_unit, address_id, product, partner)
2006         res = []
2007         cur_price_unit=price_unit
2008         obj_partener_address = self.pool.get('res.partner.address')
2009         for tax in taxes:
2010             # we compute the amount for the current tax object and append it to the result
2011             data = {'id':tax.id,
2012                     'name':tax.description and tax.description + " - " + tax.name or tax.name,
2013                     'account_collected_id':tax.account_collected_id.id,
2014                     'account_paid_id':tax.account_paid_id.id,
2015                     'base_code_id': tax.base_code_id.id,
2016                     'ref_base_code_id': tax.ref_base_code_id.id,
2017                     'sequence': tax.sequence,
2018                     'base_sign': tax.base_sign,
2019                     'tax_sign': tax.tax_sign,
2020                     'ref_base_sign': tax.ref_base_sign,
2021                     'ref_tax_sign': tax.ref_tax_sign,
2022                     'price_unit': cur_price_unit,
2023                     'tax_code_id': tax.tax_code_id.id,
2024                     'ref_tax_code_id': tax.ref_tax_code_id.id,
2025             }
2026             res.append(data)
2027             if tax.type=='percent':
2028                 amount = cur_price_unit * tax.amount
2029                 data['amount'] = amount
2030
2031             elif tax.type=='fixed':
2032                 data['amount'] = tax.amount
2033                 data['tax_amount']=quantity
2034                # data['amount'] = quantity
2035             elif tax.type=='code':
2036                 address = address_id and obj_partener_address.browse(cr, uid, address_id) or None
2037                 localdict = {'price_unit':cur_price_unit, 'address':address, 'product':product, 'partner':partner}
2038                 exec tax.python_compute in localdict
2039                 amount = localdict['result']
2040                 data['amount'] = amount
2041             elif tax.type=='balance':
2042                 data['amount'] = cur_price_unit - reduce(lambda x,y: y.get('amount',0.0)+x, res, 0.0)
2043                 data['balance'] = cur_price_unit
2044
2045             amount2 = data.get('amount', 0.0)
2046             if tax.child_ids:
2047                 if tax.child_depend:
2048                     latest = res.pop()
2049                 amount = amount2
2050                 child_tax = self._unit_compute(cr, uid, tax.child_ids, amount, address_id, product, partner, quantity)
2051                 res.extend(child_tax)
2052                 if tax.child_depend:
2053                     for r in res:
2054                         for name in ('base','ref_base'):
2055                             if latest[name+'_code_id'] and latest[name+'_sign'] and not r[name+'_code_id']:
2056                                 r[name+'_code_id'] = latest[name+'_code_id']
2057                                 r[name+'_sign'] = latest[name+'_sign']
2058                                 r['price_unit'] = latest['price_unit']
2059                                 latest[name+'_code_id'] = False
2060                         for name in ('tax','ref_tax'):
2061                             if latest[name+'_code_id'] and latest[name+'_sign'] and not r[name+'_code_id']:
2062                                 r[name+'_code_id'] = latest[name+'_code_id']
2063                                 r[name+'_sign'] = latest[name+'_sign']
2064                                 r['amount'] = data['amount']
2065                                 latest[name+'_code_id'] = False
2066             if tax.include_base_amount:
2067                 cur_price_unit+=amount2
2068         return res
2069
2070     def compute_all(self, cr, uid, taxes, price_unit, quantity, address_id=None, product=None, partner=None, force_excluded=False):
2071         """
2072         :param force_excluded: boolean used to say that we don't want to consider the value of field price_include of
2073             tax. It's used in encoding by line where you don't matter if you encoded a tax with that boolean to True or
2074             False
2075         RETURN: {
2076                 'total': 0.0,                # Total without taxes
2077                 'total_included: 0.0,        # Total with taxes
2078                 'taxes': []                  # List of taxes, see compute for the format
2079             }
2080         """
2081         precision = self.pool.get('decimal.precision').precision_get(cr, uid, 'Account')
2082         totalin = totalex = round(price_unit * quantity, precision)
2083         tin = []
2084         tex = []
2085         for tax in taxes:
2086             if not tax.price_include or force_excluded:
2087                 tex.append(tax)
2088             else:
2089                 tin.append(tax)
2090         tin = self.compute_inv(cr, uid, tin, price_unit, quantity, address_id=address_id, product=product, partner=partner)
2091         for r in tin:
2092             totalex -= r.get('amount', 0.0)
2093         totlex_qty = 0.0
2094         try:
2095             totlex_qty = totalex/quantity
2096         except:
2097             pass
2098         tex = self._compute(cr, uid, tex, totlex_qty, quantity, address_id=address_id, product=product, partner=partner)
2099         for r in tex:
2100             totalin += r.get('amount', 0.0)
2101         return {
2102             'total': totalex,
2103             'total_included': totalin,
2104             'taxes': tin + tex
2105         }
2106
2107     def compute(self, cr, uid, taxes, price_unit, quantity, address_id=None, product=None, partner=None):
2108         logger = netsvc.Logger()
2109         logger.notifyChannel("warning", netsvc.LOG_WARNING,
2110             "Deprecated, use compute_all(...)['taxes'] instead of compute(...) to manage prices with tax included")
2111         return self._compute(cr, uid, taxes, price_unit, quantity, address_id, product, partner)
2112
2113     def _compute(self, cr, uid, taxes, price_unit, quantity, address_id=None, product=None, partner=None):
2114         """
2115         Compute tax values for given PRICE_UNIT, QUANTITY and a buyer/seller ADDRESS_ID.
2116
2117         RETURN:
2118             [ tax ]
2119             tax = {'name':'', 'amount':0.0, 'account_collected_id':1, 'account_paid_id':2}
2120             one tax for each tax id in IDS and their children
2121         """
2122         res = self._unit_compute(cr, uid, taxes, price_unit, address_id, product, partner, quantity)
2123         total = 0.0
2124         precision_pool = self.pool.get('decimal.precision')
2125         for r in res:
2126             if r.get('balance',False):
2127                 r['amount'] = round(r.get('balance', 0.0) * quantity, precision_pool.precision_get(cr, uid, 'Account')) - total
2128             else:
2129                 r['amount'] = round(r.get('amount', 0.0) * quantity, precision_pool.precision_get(cr, uid, 'Account'))
2130                 total += r['amount']
2131         return res
2132
2133     def _unit_compute_inv(self, cr, uid, taxes, price_unit, address_id=None, product=None, partner=None):
2134         taxes = self._applicable(cr, uid, taxes, price_unit, address_id, product, partner)
2135         obj_partener_address = self.pool.get('res.partner.address')
2136         res = []
2137         taxes.reverse()
2138         cur_price_unit = price_unit
2139
2140         tax_parent_tot = 0.0
2141         for tax in taxes:
2142             if (tax.type=='percent') and not tax.include_base_amount:
2143                 tax_parent_tot += tax.amount
2144
2145         for tax in taxes:
2146             if (tax.type=='fixed') and not tax.include_base_amount:
2147                 cur_price_unit -= tax.amount
2148
2149         for tax in taxes:
2150             if tax.type=='percent':
2151                 if tax.include_base_amount:
2152                     amount = cur_price_unit - (cur_price_unit / (1 + tax.amount))
2153                 else:
2154                     amount = (cur_price_unit / (1 + tax_parent_tot)) * tax.amount
2155
2156             elif tax.type=='fixed':
2157                 amount = tax.amount
2158
2159             elif tax.type=='code':
2160                 address = address_id and obj_partener_address.browse(cr, uid, address_id) or None
2161                 localdict = {'price_unit':cur_price_unit, 'address':address, 'product':product, 'partner':partner}
2162                 exec tax.python_compute_inv in localdict
2163                 amount = localdict['result']
2164             elif tax.type=='balance':
2165                 amount = cur_price_unit - reduce(lambda x,y: y.get('amount',0.0)+x, res, 0.0)
2166
2167             if tax.include_base_amount:
2168                 cur_price_unit -= amount
2169                 todo = 0
2170             else:
2171                 todo = 1
2172             res.append({
2173                 'id': tax.id,
2174                 'todo': todo,
2175                 'name': tax.name,
2176                 'amount': amount,
2177                 'account_collected_id': tax.account_collected_id.id,
2178                 'account_paid_id': tax.account_paid_id.id,
2179                 'base_code_id': tax.base_code_id.id,
2180                 'ref_base_code_id': tax.ref_base_code_id.id,
2181                 'sequence': tax.sequence,
2182                 'base_sign': tax.base_sign,
2183                 'tax_sign': tax.tax_sign,
2184                 'ref_base_sign': tax.ref_base_sign,
2185                 'ref_tax_sign': tax.ref_tax_sign,
2186                 'price_unit': cur_price_unit,
2187                 'tax_code_id': tax.tax_code_id.id,
2188                 'ref_tax_code_id': tax.ref_tax_code_id.id,
2189             })
2190             if tax.child_ids:
2191                 if tax.child_depend:
2192                     del res[-1]
2193                     amount = price_unit
2194
2195             parent_tax = self._unit_compute_inv(cr, uid, tax.child_ids, amount, address_id, product, partner)
2196             res.extend(parent_tax)
2197
2198         total = 0.0
2199         for r in res:
2200             if r['todo']:
2201                 total += r['amount']
2202         for r in res:
2203             r['price_unit'] -= total
2204             r['todo'] = 0
2205         return res
2206
2207     def compute_inv(self, cr, uid, taxes, price_unit, quantity, address_id=None, product=None, partner=None):
2208         """
2209         Compute tax values for given PRICE_UNIT, QUANTITY and a buyer/seller ADDRESS_ID.
2210         Price Unit is a VAT included price
2211
2212         RETURN:
2213             [ tax ]
2214             tax = {'name':'', 'amount':0.0, 'account_collected_id':1, 'account_paid_id':2}
2215             one tax for each tax id in IDS and their children
2216         """
2217         res = self._unit_compute_inv(cr, uid, taxes, price_unit, address_id, product, partner=None)
2218         total = 0.0
2219         obj_precision = self.pool.get('decimal.precision')
2220         for r in res:
2221             prec = obj_precision.precision_get(cr, uid, 'Account')
2222             if r.get('balance',False):
2223                 r['amount'] = round(r['balance'] * quantity, prec) - total
2224             else:
2225                 r['amount'] = round(r['amount'] * quantity, prec)
2226                 total += r['amount']
2227         return res
2228
2229 account_tax()
2230
2231 # ---------------------------------------------------------
2232 # Account Entries Models
2233 # ---------------------------------------------------------
2234
2235 class account_model(osv.osv):
2236     _name = "account.model"
2237     _description = "Account Model"
2238     _columns = {
2239         'name': fields.char('Model Name', size=64, required=True, help="This is a model for recurring accounting entries"),
2240         'journal_id': fields.many2one('account.journal', 'Journal', required=True),
2241         'company_id': fields.related('journal_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True),
2242         'lines_id': fields.one2many('account.model.line', 'model_id', 'Model Entries'),
2243         'legend': fields.text('Legend', readonly=True, size=100),
2244     }
2245
2246     _defaults = {
2247         'legend': lambda self, cr, uid, context:_('You can specify year, month and date in the name of the model using the following labels:\n\n%(year)s: To Specify Year \n%(month)s: To Specify Month \n%(date)s: Current Date\n\ne.g. My model on %(date)s'),
2248     }
2249     def generate(self, cr, uid, ids, datas={}, context=None):
2250         move_ids = []
2251         entry = {}
2252         account_move_obj = self.pool.get('account.move')
2253         account_move_line_obj = self.pool.get('account.move.line')
2254         pt_obj = self.pool.get('account.payment.term')
2255         period_obj = self.pool.get('account.period')
2256
2257         if context is None:
2258             context = {}
2259
2260         if datas.get('date', False):
2261             context.update({'date': datas['date']})
2262
2263         move_date = context.get('date', time.strftime('%Y-%m-%d'))
2264         move_date = datetime.strptime(move_date,"%Y-%m-%d")
2265         for model in self.browse(cr, uid, ids, context=context):
2266             ctx = context.copy()
2267             ctx.update({'company_id': model.company_id.id})
2268             period_ids = period_obj.find(cr, uid, dt=context.get('date', False), context=ctx)
2269             period_id = period_ids and period_ids[0] or False
2270             ctx.update({'journal_id': model.journal_id.id,'period_id': period_id})
2271             try:
2272                 entry['name'] = model.name%{'year': move_date.strftime('%Y'), 'month': move_date.strftime('%m'), 'date': move_date.strftime('%Y-%m')}
2273             except:
2274                 raise osv.except_osv(_('Wrong model !'), _('You have a wrong expression "%(...)s" in your model !'))
2275             move_id = account_move_obj.create(cr, uid, {
2276                 'ref': entry['name'],
2277                 'period_id': period_id,
2278                 'journal_id': model.journal_id.id,
2279                 'date': context.get('date', fields.date.context_today(self,cr,uid,context=context))
2280             })
2281             move_ids.append(move_id)
2282             for line in model.lines_id:
2283                 analytic_account_id = False
2284                 if line.analytic_account_id:
2285                     if not model.journal_id.analytic_journal_id:
2286                         raise osv.except_osv(_('No Analytic Journal !'),_("You have to define an analytic journal on the '%s' journal!") % (model.journal_id.name,))
2287                     analytic_account_id = line.analytic_account_id.id
2288                 val = {
2289                     'move_id': move_id,
2290                     'journal_id': model.journal_id.id,
2291                     'period_id': period_id,
2292                     'analytic_account_id': analytic_account_id
2293                 }
2294
2295                 date_maturity = context.get('date',time.strftime('%Y-%m-%d'))
2296                 if line.date_maturity == 'partner':
2297                     if not line.partner_id:
2298                         raise osv.except_osv(_('Error !'), _("Maturity date of entry line generated by model line '%s' of model '%s' is based on partner payment term!" \
2299                                                                 "\nPlease define partner on it!")%(line.name, model.name))
2300                     if line.partner_id.property_payment_term:
2301                         payment_term_id = line.partner_id.property_payment_term.id
2302                         pterm_list = pt_obj.compute(cr, uid, payment_term_id, value=1, date_ref=date_maturity)
2303                         if pterm_list:
2304                             pterm_list = [l[0] for l in pterm_list]
2305                             pterm_list.sort()
2306                             date_maturity = pterm_list[-1]
2307
2308                 val.update({
2309                     'name': line.name,
2310                     'quantity': line.quantity,
2311                     'debit': line.debit,
2312                     'credit': line.credit,
2313                     'account_id': line.account_id.id,
2314                     'move_id': move_id,
2315                     'partner_id': line.partner_id.id,
2316                     'date': context.get('date', fields.date.context_today(self,cr,uid,context=context)),
2317                     'date_maturity': date_maturity
2318                 })
2319                 account_move_line_obj.create(cr, uid, val, context=ctx)
2320
2321         return move_ids
2322
2323 account_model()
2324
2325 class account_model_line(osv.osv):
2326     _name = "account.model.line"
2327     _description = "Account Model Entries"
2328     _columns = {
2329         'name': fields.char('Name', size=64, required=True),
2330         'sequence': fields.integer('Sequence', required=True, help="The sequence field is used to order the resources from lower sequences to higher ones."),
2331         'quantity': fields.float('Quantity', digits_compute=dp.get_precision('Account'), help="The optional quantity on entries."),
2332         'debit': fields.float('Debit', digits_compute=dp.get_precision('Account')),
2333         'credit': fields.float('Credit', digits_compute=dp.get_precision('Account')),
2334         'account_id': fields.many2one('account.account', 'Account', required=True, ondelete="cascade"),
2335         'analytic_account_id': fields.many2one('account.analytic.account', 'Analytic Account', ondelete="cascade"),
2336         'model_id': fields.many2one('account.model', 'Model', required=True, ondelete="cascade", select=True),
2337         'amount_currency': fields.float('Amount Currency', help="The amount expressed in an optional other currency."),
2338         'currency_id': fields.many2one('res.currency', 'Currency'),
2339         'partner_id': fields.many2one('res.partner', 'Partner'),
2340         'date_maturity': fields.selection([('today','Date of the day'), ('partner','Partner Payment Term')], 'Maturity Date', help="The maturity date of the generated entries for this model. You can choose between the creation date or the creation date of the entries plus the partner payment terms."),
2341     }
2342     _order = 'sequence'
2343     _sql_constraints = [
2344         ('credit_debit1', 'CHECK (credit*debit=0)',  'Wrong credit or debit value in model, they must be positive!'),
2345         ('credit_debit2', 'CHECK (credit+debit>=0)', 'Wrong credit or debit value in model, they must be positive!'),
2346     ]
2347 account_model_line()
2348
2349 # ---------------------------------------------------------
2350 # Account Subscription
2351 # ---------------------------------------------------------
2352
2353
2354 class account_subscription(osv.osv):
2355     _name = "account.subscription"
2356     _description = "Account Subscription"
2357     _columns = {
2358         'name': fields.char('Name', size=64, required=True),
2359         'ref': fields.char('Reference', size=16),
2360         'model_id': fields.many2one('account.model', 'Model', required=True),
2361         'date_start': fields.date('Start Date', required=True),
2362         'period_total': fields.integer('Number of Periods', required=True),
2363         'period_nbr': fields.integer('Period', required=True),
2364         'period_type': fields.selection([('day','days'),('month','month'),('year','year')], 'Period Type', required=True),
2365         'state': fields.selection([('draft','Draft'),('running','Running'),('done','Done')], 'State', required=True, readonly=True),
2366         'lines_id': fields.one2many('account.subscription.line', 'subscription_id', 'Subscription Lines')
2367     }
2368     _defaults = {
2369         'date_start': fields.date.context_today,
2370         'period_type': 'month',
2371         'period_total': 12,
2372         'period_nbr': 1,
2373         'state': 'draft',
2374     }
2375     def state_draft(self, cr, uid, ids, context=None):
2376         self.write(cr, uid, ids, {'state':'draft'})
2377         return False
2378
2379     def check(self, cr, uid, ids, context=None):
2380         todone = []
2381         for sub in self.browse(cr, uid, ids, context=context):
2382             ok = True
2383             for line in sub.lines_id:
2384                 if not line.move_id.id:
2385                     ok = False
2386                     break
2387             if ok:
2388                 todone.append(sub.id)
2389         if todone:
2390             self.write(cr, uid, todone, {'state':'done'})
2391         return False
2392
2393     def remove_line(self, cr, uid, ids, context=None):
2394         toremove = []
2395         for sub in self.browse(cr, uid, ids, context=context):
2396             for line in sub.lines_id:
2397                 if not line.move_id.id:
2398                     toremove.append(line.id)
2399         if toremove:
2400             self.pool.get('account.subscription.line').unlink(cr, uid, toremove)
2401         self.write(cr, uid, ids, {'state':'draft'})
2402         return False
2403
2404     def compute(self, cr, uid, ids, context=None):
2405         for sub in self.browse(cr, uid, ids, context=context):
2406             ds = sub.date_start
2407             for i in range(sub.period_total):
2408                 self.pool.get('account.subscription.line').create(cr, uid, {
2409                     'date': ds,
2410                     'subscription_id': sub.id,
2411                 })
2412                 if sub.period_type=='day':
2413                     ds = (datetime.strptime(ds, '%Y-%m-%d') + relativedelta(days=sub.period_nbr)).strftime('%Y-%m-%d')
2414                 if sub.period_type=='month':
2415                     ds = (datetime.strptime(ds, '%Y-%m-%d') + relativedelta(months=sub.period_nbr)).strftime('%Y-%m-%d')
2416                 if sub.period_type=='year':
2417                     ds = (datetime.strptime(ds, '%Y-%m-%d') + relativedelta(years=sub.period_nbr)).strftime('%Y-%m-%d')
2418         self.write(cr, uid, ids, {'state':'running'})
2419         return True
2420
2421 account_subscription()
2422
2423 class account_subscription_line(osv.osv):
2424     _name = "account.subscription.line"
2425     _description = "Account Subscription Line"
2426     _columns = {
2427         'subscription_id': fields.many2one('account.subscription', 'Subscription', required=True, select=True),
2428         'date': fields.date('Date', required=True),
2429         'move_id': fields.many2one('account.move', 'Entry'),
2430     }
2431
2432     def move_create(self, cr, uid, ids, context=None):
2433         tocheck = {}
2434         all_moves = []
2435         obj_model = self.pool.get('account.model')
2436         for line in self.browse(cr, uid, ids, context=context):
2437             datas = {
2438                 'date': line.date,
2439             }
2440             move_ids = obj_model.generate(cr, uid, [line.subscription_id.model_id.id], datas, context)
2441             tocheck[line.subscription_id.id] = True
2442             self.write(cr, uid, [line.id], {'move_id':move_ids[0]})
2443             all_moves.extend(move_ids)
2444         if tocheck:
2445             self.pool.get('account.subscription').check(cr, uid, tocheck.keys(), context)
2446         return all_moves
2447
2448     _rec_name = 'date'
2449
2450 account_subscription_line()
2451
2452 #  ---------------------------------------------------------------
2453 #   Account Templates: Account, Tax, Tax Code and chart. + Wizard
2454 #  ---------------------------------------------------------------
2455
2456 class account_tax_template(osv.osv):
2457     _name = 'account.tax.template'
2458 account_tax_template()
2459
2460 class account_account_template(osv.osv):
2461     _order = "code"
2462     _name = "account.account.template"
2463     _description ='Templates for Accounts'
2464
2465     _columns = {
2466         'name': fields.char('Name', size=256, required=True, select=True),
2467         'currency_id': fields.many2one('res.currency', 'Secondary Currency', help="Forces all moves for this account to have this secondary currency."),
2468         'code': fields.char('Code', size=64, required=True, select=1),
2469         'type': fields.selection([
2470             ('receivable','Receivable'),
2471             ('payable','Payable'),
2472             ('view','View'),
2473             ('consolidation','Consolidation'),
2474             ('liquidity','Liquidity'),
2475             ('other','Regular'),
2476             ('closed','Closed'),
2477             ], 'Internal Type', required=True,help="This type is used to differentiate types with "\
2478             "special effects in OpenERP: view can not have entries, consolidation are accounts that "\
2479             "can have children accounts for multi-company consolidations, payable/receivable are for "\
2480             "partners accounts (for debit/credit computations), closed for depreciated accounts."),
2481         'user_type': fields.many2one('account.account.type', 'Account Type', required=True,
2482             help="These types are defined according to your country. The type contains more information "\
2483             "about the account and its specificities."),
2484         'financial_report_ids': fields.many2many('account.financial.report', 'account_template_financial_report', 'account_template_id', 'report_line_id', 'Financial Reports'),
2485         'reconcile': fields.boolean('Allow Reconciliation', help="Check this option if you want the user to reconcile entries in this account."),
2486         'shortcut': fields.char('Shortcut', size=12),
2487         'note': fields.text('Note'),
2488         'parent_id': fields.many2one('account.account.template', 'Parent Account Template', ondelete='cascade'),
2489         'child_parent_ids':fields.one2many('account.account.template', 'parent_id', 'Children'),
2490         'tax_ids': fields.many2many('account.tax.template', 'account_account_template_tax_rel', 'account_id', 'tax_id', 'Default Taxes'),
2491         'nocreate': fields.boolean('Optional create', help="If checked, the new chart of accounts will not contain this by default."),
2492         'chart_template_id': fields.many2one('account.chart.template', 'Chart Template', help="This optional field allow you to link an account template to a specific chart template that may differ from the one its root parent belongs to. This allow you to define chart templates that extend another and complete it with few new accounts (You don't need to define the whole structure that is common to both several times)."),
2493     }
2494
2495     _defaults = {
2496         'reconcile': False,
2497         'type': 'view',
2498         'nocreate': False,
2499     }
2500
2501     def _check_type(self, cr, uid, ids, context=None):
2502         if context is None:
2503             context = {}
2504         accounts = self.browse(cr, uid, ids, context=context)
2505         for account in accounts:
2506             if account.parent_id and account.parent_id.type != 'view':
2507                 return False
2508         return True
2509
2510     _check_recursion = check_cycle
2511     _constraints = [
2512         (_check_recursion, 'Error ! You can not create recursive account templates.', ['parent_id']),
2513         (_check_type, 'Configuration Error!\nYou can not define children to an account with internal type different of "View"! ', ['type']),
2514
2515     ]
2516
2517     def name_get(self, cr, uid, ids, context=None):
2518         if not ids:
2519             return []
2520         reads = self.read(cr, uid, ids, ['name','code'], context=context)
2521         res = []
2522         for record in reads:
2523             name = record['name']
2524             if record['code']:
2525                 name = record['code']+' '+name
2526             res.append((record['id'],name ))
2527         return res
2528
2529     def generate_account(self, cr, uid, chart_template_id, tax_template_ref, acc_template_ref, code_digits, company_id, context=None):
2530         """
2531         This method for generating accounts from templates.
2532
2533         :param chart_template_id: id of the chart template chosen in the wizard
2534         :param tax_template_ref: Taxes templates reference for write taxes_id in account_account.
2535         :paramacc_template_ref: dictionary with the mappping between the account templates and the real accounts.
2536         :param code_digits: number of digits got from wizard.multi.charts.accounts, this is use for account code.
2537         :param company_id: company_id selected from wizard.multi.charts.accounts.
2538         :returns: return acc_template_ref for reference purpose.
2539         :rtype: dict
2540         """
2541         if context is None:
2542             context = {}
2543         obj_acc = self.pool.get('account.account')
2544         company_name = self.pool.get('res.company').browse(cr, uid, company_id, context=context).name
2545         template = self.pool.get('account.chart.template').browse(cr, uid, chart_template_id, context=context)
2546         #deactivate the parent_store functionnality on account_account for rapidity purpose
2547         ctx = context.copy()
2548         ctx.update({'defer_parent_store_computation': True})
2549         level_ref = {}
2550         children_acc_criteria = [('chart_template_id','=', chart_template_id)]
2551         if template.account_root_id.id:
2552             children_acc_criteria = ['|'] + children_acc_criteria + ['&',('parent_id','child_of', [template.account_root_id.id]),('chart_template_id','=', False)]
2553         children_acc_template = self.search(cr, uid, [('nocreate','!=',True)] + children_acc_criteria, order='id')
2554         for account_template in self.browse(cr, uid, children_acc_template, context=context):
2555             # skip the root of COA if it's not the main one
2556             if (template.account_root_id.id == account_template.id) and template.parent_id:
2557                 continue
2558             tax_ids = []
2559             for tax in account_template.tax_ids:
2560                 tax_ids.append(tax_template_ref[tax.id])
2561
2562             code_main = account_template.code and len(account_template.code) or 0
2563             code_acc = account_template.code or ''
2564             if code_main > 0 and code_main <= code_digits and account_template.type != 'view':
2565                 code_acc = str(code_acc) + (str('0'*(code_digits-code_main)))
2566             parent_id = account_template.parent_id and ((account_template.parent_id.id in acc_template_ref) and acc_template_ref[account_template.parent_id.id]) or False
2567             #the level as to be given as well at the creation time, because of the defer_parent_store_computation in
2568             #context. Indeed because of this, the parent_left and parent_right are not computed and thus the child_of
2569             #operator does not return the expected values, with result of having the level field not computed at all.
2570             if parent_id:
2571                 level = parent_id in level_ref and level_ref[parent_id] + 1 or obj_acc._get_level(cr, uid, [parent_id], 'level', None, context=context)[parent_id] + 1
2572             else:
2573                 level = 0
2574             vals={
2575                 'name': (template.account_root_id.id == account_template.id) and company_name or account_template.name,
2576                 'currency_id': account_template.currency_id and account_template.currency_id.id or False,
2577                 'code': code_acc,
2578                 'type': account_template.type,
2579                 'user_type': account_template.user_type and account_template.user_type.id or False,
2580                 'reconcile': account_template.reconcile,
2581                 'shortcut': account_template.shortcut,
2582                 'note': account_template.note,
2583                 'financial_report_ids': account_template.financial_report_ids and [(6,0,[x.id for x in account_template.financial_report_ids])] or False,
2584                 'parent_id': parent_id,
2585                 'tax_ids': [(6,0,tax_ids)],
2586                 'company_id': company_id,
2587                 'level': level,
2588             }
2589             new_account = obj_acc.create(cr, uid, vals, context=ctx)
2590             acc_template_ref[account_template.id] = new_account
2591             level_ref[new_account] = level
2592
2593         #reactivate the parent_store functionnality on account_account
2594         obj_acc._parent_store_compute(cr)
2595         return acc_template_ref
2596
2597 account_account_template()
2598
2599 class account_add_tmpl_wizard(osv.osv_memory):
2600     """Add one more account from the template.
2601
2602     With the 'nocreate' option, some accounts may not be created. Use this to add them later."""
2603     _name = 'account.addtmpl.wizard'
2604
2605     def _get_def_cparent(self, cr, uid, context=None):
2606         acc_obj = self.pool.get('account.account')
2607         tmpl_obj = self.pool.get('account.account.template')
2608         tids = tmpl_obj.read(cr, uid, [context['tmpl_ids']], ['parent_id'])
2609         if not tids or not tids[0]['parent_id']:
2610             return False
2611         ptids = tmpl_obj.read(cr, uid, [tids[0]['parent_id'][0]], ['code'])
2612         res = None
2613         if not ptids or not ptids[0]['code']:
2614             raise osv.except_osv(_('Error !'), _('I can not locate a parent code for the template account!'))
2615             res = acc_obj.search(cr, uid, [('code','=',ptids[0]['code'])])
2616         return res and res[0] or False
2617
2618     _columns = {
2619         'cparent_id':fields.many2one('account.account', 'Parent target', help="Creates an account with the selected template under this existing parent.", required=True),
2620     }
2621     _defaults = {
2622         'cparent_id': _get_def_cparent,
2623     }
2624
2625     def action_create(self,cr,uid,ids,context=None):
2626         if context is None:
2627             context = {}
2628         acc_obj = self.pool.get('account.account')
2629         tmpl_obj = self.pool.get('account.account.template')
2630         data = self.read(cr, uid, ids)[0]
2631         company_id = acc_obj.read(cr, uid, [data['cparent_id'][0]], ['company_id'])[0]['company_id'][0]
2632         account_template = tmpl_obj.browse(cr, uid, context['tmpl_ids'])
2633         vals = {
2634             'name': account_template.name,
2635             'currency_id': account_template.currency_id and account_template.currency_id.id or False,
2636             'code': account_template.code,
2637             'type': account_template.type,
2638             'user_type': account_template.user_type and account_template.user_type.id or False,
2639             'reconcile': account_template.reconcile,
2640             'shortcut': account_template.shortcut,
2641             'note': account_template.note,
2642             'parent_id': data['cparent_id'][0],
2643             'company_id': company_id,
2644             }
2645         acc_obj.create(cr, uid, vals)
2646         return {'type':'state', 'state': 'end' }
2647
2648     def action_cancel(self, cr, uid, ids, context=None):
2649         return { 'type': 'state', 'state': 'end' }
2650
2651 account_add_tmpl_wizard()
2652
2653 class account_tax_code_template(osv.osv):
2654
2655     _name = 'account.tax.code.template'
2656     _description = 'Tax Code Template'
2657     _order = 'code'
2658     _rec_name = 'code'
2659     _columns = {
2660         'name': fields.char('Tax Case Name', size=64, required=True),
2661         'code': fields.char('Case Code', size=64),
2662         'info': fields.text('Description'),
2663         'parent_id': fields.many2one('account.tax.code.template', 'Parent Code', select=True),
2664         'child_ids': fields.one2many('account.tax.code.template', 'parent_id', 'Child Codes'),
2665         'sign': fields.float('Sign For Parent', required=True),
2666         'notprintable':fields.boolean("Not Printable in Invoice", help="Check this box if you don't want any VAT related to this Tax Code to appear on invoices"),
2667     }
2668
2669     _defaults = {
2670         'sign': 1.0,
2671         'notprintable': False,
2672     }
2673
2674     def generate_tax_code(self, cr, uid, tax_code_root_id, company_id, context=None):
2675         '''
2676         This function generates the tax codes from the templates of tax code that are children of the given one passed
2677         in argument. Then it returns a dictionary with the mappping between the templates and the real objects.
2678
2679         :param tax_code_root_id: id of the root of all the tax code templates to process
2680         :param company_id: id of the company the wizard is running for
2681         :returns: dictionary with the mappping between the templates and the real objects.
2682         :rtype: dict
2683         '''
2684         obj_tax_code_template = self.pool.get('account.tax.code.template')
2685         obj_tax_code = self.pool.get('account.tax.code')
2686         tax_code_template_ref = {}
2687         company = self.pool.get('res.company').browse(cr, uid, company_id, context=context)
2688
2689         #find all the children of the tax_code_root_id
2690         children_tax_code_template = tax_code_root_id and obj_tax_code_template.search(cr, uid, [('parent_id','child_of',[tax_code_root_id])], order='id') or []
2691         for tax_code_template in obj_tax_code_template.browse(cr, uid, children_tax_code_template, context=context):
2692             vals = {
2693                 'name': (tax_code_root_id == tax_code_template.id) and company.name or tax_code_template.name,
2694                 'code': tax_code_template.code,
2695                 'info': tax_code_template.info,
2696                 'parent_id': tax_code_template.parent_id and ((tax_code_template.parent_id.id in tax_code_template_ref) and tax_code_template_ref[tax_code_template.parent_id.id]) or False,
2697                 'company_id': company_id,
2698                 'sign': tax_code_template.sign,
2699             }
2700             #check if this tax code already exists
2701             rec_list = obj_tax_code.search(cr, uid, [('name', '=', vals['name']),('code', '=', vals['code']),('company_id', '=', vals['company_id'])], context=context)
2702             if not rec_list:
2703                 #if not yet, create it
2704                 new_tax_code = obj_tax_code.create(cr, uid, vals)
2705                 #recording the new tax code to do the mapping
2706                 tax_code_template_ref[tax_code_template.id] = new_tax_code
2707         return tax_code_template_ref
2708
2709     def name_get(self, cr, uid, ids, context=None):
2710         if not ids:
2711             return []
2712         if isinstance(ids, (int, long)):
2713             ids = [ids]
2714         reads = self.read(cr, uid, ids, ['name','code'], context, load='_classic_write')
2715         return [(x['id'], (x['code'] and x['code'] + ' - ' or '') + x['name']) \
2716                 for x in reads]
2717
2718     _check_recursion = check_cycle
2719     _constraints = [
2720         (_check_recursion, 'Error ! You can not create recursive Tax Codes.', ['parent_id'])
2721     ]
2722     _order = 'code,name'
2723 account_tax_code_template()
2724
2725
2726 class account_chart_template(osv.osv):
2727     _name="account.chart.template"
2728     _description= "Templates for Account Chart"
2729
2730     _columns={
2731         'name': fields.char('Name', size=64, required=True),
2732         'parent_id': fields.many2one('account.chart.template', 'Parent Chart Template'),
2733         'code_digits': fields.integer('# of Digits', required=True, help="No. of Digits to use for account code"), 
2734         'visible': fields.boolean('Can be Visible?', help="Set this to False if you don't want this template to be used actively in the wizard that generate Chart of Accounts from templates, this is useful when you want to generate accounts of this template only when loading its child template."),
2735         'complete_tax_set': fields.boolean('Complete Set of Taxes', help='This boolean helps you to choose if you want to propose to the user to encode the sale and purchase rates or choose from list of taxes. This last choice assumes that the set of tax defined on this template is complete'),
2736         'account_root_id': fields.many2one('account.account.template', 'Root Account', domain=[('parent_id','=',False)]),
2737         'tax_code_root_id': fields.many2one('account.tax.code.template', 'Root Tax Code', domain=[('parent_id','=',False)]),
2738         'tax_template_ids': fields.one2many('account.tax.template', 'chart_template_id', 'Tax Template List', help='List of all the taxes that have to be installed by the wizard'),
2739         'bank_account_view_id': fields.many2one('account.account.template', 'Bank Account'),
2740         'property_account_receivable': fields.many2one('account.account.template', 'Receivable Account'),
2741         'property_account_payable': fields.many2one('account.account.template', 'Payable Account'),
2742         'property_account_expense_categ': fields.many2one('account.account.template', 'Expense Category Account'),
2743         'property_account_income_categ': fields.many2one('account.account.template', 'Income Category Account'),
2744         'property_account_expense': fields.many2one('account.account.template', 'Expense Account on Product Template'),
2745         'property_account_income': fields.many2one('account.account.template', 'Income Account on Product Template'),
2746         'property_reserve_and_surplus_account': fields.many2one('account.account.template', 'Reserve and Profit/Loss Account', domain=[('type', '=', 'payable')], help='This Account is used for transferring Profit/Loss(If It is Profit: Amount will be added, Loss: Amount will be deducted.), Which is calculated from Profilt & Loss Report'),
2747         'property_account_income_opening': fields.many2one('account.account.template', 'Opening Entries Income Account'),
2748         'property_account_expense_opening': fields.many2one('account.account.template', 'Opening Entries Expense Account'),
2749     }
2750
2751     _defaults = {
2752         'visible': True,
2753         'code_digits': 6,
2754         'complete_tax_set': True,
2755     }
2756
2757 account_chart_template()
2758
2759 class account_tax_template(osv.osv):
2760
2761     _name = 'account.tax.template'
2762     _description = 'Templates for Taxes'
2763
2764     _columns = {
2765         'chart_template_id': fields.many2one('account.chart.template', 'Chart Template', required=True),
2766         'name': fields.char('Tax Name', size=64, required=True),
2767         'sequence': fields.integer('Sequence', required=True, help="The sequence field is used to order the taxes lines from lower sequences to higher ones. The order is important if you have a tax that has several tax children. In this case, the evaluation order is important."),
2768         'amount': fields.float('Amount', required=True, digits=(14,4), help="For Tax Type percent enter % ratio between 0-1."),
2769         'type': fields.selection( [('percent','Percent'), ('fixed','Fixed'), ('none','None'), ('code','Python Code'), ('balance','Balance')], 'Tax Type', required=True),
2770         'applicable_type': fields.selection( [('true','True'), ('code','Python Code')], 'Applicable Type', required=True, help="If not applicable (computed through a Python code), the tax won't appear on the invoice."),
2771         'domain':fields.char('Domain', size=32, help="This field is only used if you develop your own module allowing developers to create specific taxes in a custom domain."),
2772         'account_collected_id':fields.many2one('account.account.template', 'Invoice Tax Account'),
2773         'account_paid_id':fields.many2one('account.account.template', 'Refund Tax Account'),
2774         'parent_id':fields.many2one('account.tax.template', 'Parent Tax Account', select=True),
2775         'child_depend':fields.boolean('Tax on Children', help="Set if the tax computation is based on the computation of child taxes rather than on the total amount."),
2776         'python_compute':fields.text('Python Code'),
2777         'python_compute_inv':fields.text('Python Code (reverse)'),
2778         'python_applicable':fields.text('Python Code'),
2779
2780         #
2781         # Fields used for the VAT declaration
2782         #
2783         'base_code_id': fields.many2one('account.tax.code.template', 'Base Code', help="Use this code for the VAT declaration."),
2784         'tax_code_id': fields.many2one('account.tax.code.template', 'Tax Code', help="Use this code for the VAT declaration."),
2785         'base_sign': fields.float('Base Code Sign', help="Usually 1 or -1."),
2786         'tax_sign': fields.float('Tax Code Sign', help="Usually 1 or -1."),
2787
2788         # Same fields for refund invoices
2789
2790         'ref_base_code_id': fields.many2one('account.tax.code.template', 'Refund Base Code', help="Use this code for the VAT declaration."),
2791         'ref_tax_code_id': fields.many2one('account.tax.code.template', 'Refund Tax Code', help="Use this code for the VAT declaration."),
2792         'ref_base_sign': fields.float('Base Code Sign', help="Usually 1 or -1."),
2793         'ref_tax_sign': fields.float('Tax Code Sign', help="Usually 1 or -1."),
2794         'include_base_amount': fields.boolean('Include in Base Amount', help="Set if the amount of tax must be included in the base amount before computing the next taxes."),
2795         'description': fields.char('Internal Name', size=32),
2796         'type_tax_use': fields.selection([('sale','Sale'),('purchase','Purchase'),('all','All')], 'Tax Use In', required=True,),
2797         'price_include': fields.boolean('Tax Included in Price', help="Check this if the price you use on the product and invoices includes this tax."),
2798     }
2799
2800     def name_get(self, cr, uid, ids, context=None):
2801         if not ids:
2802             return []
2803         res = []
2804         for record in self.read(cr, uid, ids, ['description','name'], context=context):
2805             name = record['description'] and record['description'] or record['name']
2806             res.append((record['id'],name ))
2807         return res
2808
2809     def _default_company(self, cr, uid, context=None):
2810         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
2811         if user.company_id:
2812             return user.company_id.id
2813         return self.pool.get('res.company').search(cr, uid, [('parent_id', '=', False)])[0]
2814
2815     _defaults = {
2816         'python_compute': lambda *a: '''# price_unit\n# address: res.partner.address object or False\n# product: product.product object or None\n# partner: res.partner object or None\n\nresult = price_unit * 0.10''',
2817         'python_compute_inv': lambda *a: '''# price_unit\n# address: res.partner.address object or False\n# product: product.product object or False\n\nresult = price_unit * 0.10''',
2818         'applicable_type': 'true',
2819         'type': 'percent',
2820         'amount': 0,
2821         'sequence': 1,
2822         'ref_tax_sign': 1,
2823         'ref_base_sign': 1,
2824         'tax_sign': 1,
2825         'base_sign': 1,
2826         'include_base_amount': False,
2827         'type_tax_use': 'all',
2828         'price_include': 0,
2829     }
2830     _order = 'sequence'
2831
2832     def _generate_tax(self, cr, uid, tax_templates, tax_code_template_ref, company_id, context=None):
2833         """
2834         This method generate taxes from templates.
2835
2836         :param tax_templates: list of browse record of the tax templates to process
2837         :param tax_code_template_ref: Taxcode templates reference.
2838         :param company_id: id of the company the wizard is running for
2839         :returns:
2840             {
2841             'tax_template_to_tax': mapping between tax template and the newly generated taxes corresponding,
2842             'account_dict': dictionary containing a to-do list with all the accounts to assign on new taxes
2843             }
2844         """
2845         if context is None:
2846             context = {}
2847         res = {}
2848         todo_dict = {}
2849         tax_template_to_tax = {}
2850         for tax in tax_templates:
2851             vals_tax = {
2852                 'name':tax.name,
2853                 'sequence': tax.sequence,
2854                 'amount': tax.amount,
2855                 'type': tax.type,
2856                 'applicable_type': tax.applicable_type,
2857                 'domain': tax.domain,
2858                 'parent_id': tax.parent_id and ((tax.parent_id.id in tax_template_to_tax) and tax_template_to_tax[tax.parent_id.id]) or False,
2859                 'child_depend': tax.child_depend,
2860                 'python_compute': tax.python_compute,
2861                 'python_compute_inv': tax.python_compute_inv,
2862                 'python_applicable': tax.python_applicable,
2863                 'base_code_id': tax.base_code_id and ((tax.base_code_id.id in tax_code_template_ref) and tax_code_template_ref[tax.base_code_id.id]) or False,
2864                 'tax_code_id': tax.tax_code_id and ((tax.tax_code_id.id in tax_code_template_ref) and tax_code_template_ref[tax.tax_code_id.id]) or False,
2865                 'base_sign': tax.base_sign,
2866                 'tax_sign': tax.tax_sign,
2867                 'ref_base_code_id': tax.ref_base_code_id and ((tax.ref_base_code_id.id in tax_code_template_ref) and tax_code_template_ref[tax.ref_base_code_id.id]) or False,
2868                 'ref_tax_code_id': tax.ref_tax_code_id and ((tax.ref_tax_code_id.id in tax_code_template_ref) and tax_code_template_ref[tax.ref_tax_code_id.id]) or False,
2869                 'ref_base_sign': tax.ref_base_sign,
2870                 'ref_tax_sign': tax.ref_tax_sign,
2871                 'include_base_amount': tax.include_base_amount,
2872                 'description': tax.description,
2873                 'company_id': company_id,
2874                 'type_tax_use': tax.type_tax_use,
2875                 'price_include': tax.price_include
2876             }
2877             new_tax = self.pool.get('account.tax').create(cr, uid, vals_tax)
2878             tax_template_to_tax[tax.id] = new_tax
2879             #as the accounts have not been created yet, we have to wait before filling these fields
2880             todo_dict[new_tax] = {
2881                 'account_collected_id': tax.account_collected_id and tax.account_collected_id.id or False,
2882                 'account_paid_id': tax.account_paid_id and tax.account_paid_id.id or False,
2883             }
2884         res.update({'tax_template_to_tax': tax_template_to_tax, 'account_dict': todo_dict})
2885         return res
2886
2887 account_tax_template()
2888
2889 # Fiscal Position Templates
2890
2891 class account_fiscal_position_template(osv.osv):
2892     _name = 'account.fiscal.position.template'
2893     _description = 'Template for Fiscal Position'
2894
2895     _columns = {
2896         'name': fields.char('Fiscal Position Template', size=64, required=True),
2897         'chart_template_id': fields.many2one('account.chart.template', 'Chart Template', required=True),
2898         'account_ids': fields.one2many('account.fiscal.position.account.template', 'position_id', 'Account Mapping'),
2899         'tax_ids': fields.one2many('account.fiscal.position.tax.template', 'position_id', 'Tax Mapping'),
2900         'note': fields.text('Notes', translate=True),
2901     }
2902
2903     def generate_fiscal_position(self, cr, uid, chart_temp_id, tax_template_ref, acc_template_ref, company_id, context=None):
2904         """
2905         This method generate Fiscal Position, Fiscal Position Accounts and Fiscal Position Taxes from templates.
2906
2907         :param chart_temp_id: Chart Template Id.
2908         :param taxes_ids: Taxes templates reference for generating account.fiscal.position.tax.
2909         :param acc_template_ref: Account templates reference for generating account.fiscal.position.account.
2910         :param company_id: company_id selected from wizard.multi.charts.accounts.
2911         :returns: True
2912         """
2913         if context is None:
2914             context = {}
2915         obj_tax_fp = self.pool.get('account.fiscal.position.tax')
2916         obj_ac_fp = self.pool.get('account.fiscal.position.account')
2917         obj_fiscal_position = self.pool.get('account.fiscal.position')
2918         fp_ids = self.search(cr, uid, [('chart_template_id', '=', chart_temp_id)])
2919         for position in self.browse(cr, uid, fp_ids, context=context):
2920             new_fp = obj_fiscal_position.create(cr, uid, {'company_id': company_id, 'name': position.name, 'note': position.note})
2921             for tax in position.tax_ids:
2922                 obj_tax_fp.create(cr, uid, {
2923                     'tax_src_id': tax_template_ref[tax.tax_src_id.id],
2924                     'tax_dest_id': tax.tax_dest_id and tax_template_ref[tax.tax_dest_id.id] or False,
2925                     'position_id': new_fp
2926                 })
2927             for acc in position.account_ids:
2928                 obj_ac_fp.create(cr, uid, {
2929                     'account_src_id': acc_template_ref[acc.account_src_id.id],
2930                     'account_dest_id': acc_template_ref[acc.account_dest_id.id],
2931                     'position_id': new_fp
2932                 })
2933         return True
2934
2935 account_fiscal_position_template()
2936
2937 class account_fiscal_position_tax_template(osv.osv):
2938     _name = 'account.fiscal.position.tax.template'
2939     _description = 'Template Tax Fiscal Position'
2940     _rec_name = 'position_id'
2941
2942     _columns = {
2943         'position_id': fields.many2one('account.fiscal.position.template', 'Fiscal Position', required=True, ondelete='cascade'),
2944         'tax_src_id': fields.many2one('account.tax.template', 'Tax Source', required=True),
2945         'tax_dest_id': fields.many2one('account.tax.template', 'Replacement Tax')
2946     }
2947
2948 account_fiscal_position_tax_template()
2949
2950 class account_fiscal_position_account_template(osv.osv):
2951     _name = 'account.fiscal.position.account.template'
2952     _description = 'Template Account Fiscal Mapping'
2953     _rec_name = 'position_id'
2954     _columns = {
2955         'position_id': fields.many2one('account.fiscal.position.template', 'Fiscal Mapping', required=True, ondelete='cascade'),
2956         'account_src_id': fields.many2one('account.account.template', 'Account Source', domain=[('type','<>','view')], required=True),
2957         'account_dest_id': fields.many2one('account.account.template', 'Account Destination', domain=[('type','<>','view')], required=True)
2958     }
2959
2960 account_fiscal_position_account_template()
2961
2962 # ---------------------------------------------------------
2963 # Account generation from template wizards
2964 # ---------------------------------------------------------
2965
2966 class wizard_multi_charts_accounts(osv.osv_memory):
2967     """
2968     Create a new account chart for a company.
2969     Wizards ask for:
2970         * a company
2971         * an account chart template
2972         * a number of digits for formatting code of non-view accounts
2973         * a list of bank accounts owned by the company
2974     Then, the wizard:
2975         * generates all accounts from the template and assigns them to the right company
2976         * generates all taxes and tax codes, changing account assignations
2977         * generates all accounting properties and assigns them correctly
2978     """
2979     _name='wizard.multi.charts.accounts'
2980     _inherit = 'res.config'
2981
2982     _columns = {
2983         'company_id':fields.many2one('res.company', 'Company', required=True),
2984         'chart_template_id': fields.many2one('account.chart.template', 'Chart Template', required=True),
2985         'bank_accounts_id': fields.one2many('account.bank.accounts.wizard', 'bank_account_id', 'Cash and Banks', required=True),
2986         'code_digits':fields.integer('# of Digits', required=True, help="No. of Digits to use for account code"),
2987         'seq_journal':fields.boolean('Separated Journal Sequences', help="Check this box if you want to use a different sequence for each created journal. Otherwise, all will use the same sequence."),
2988         "sale_tax": fields.many2one("account.tax.template", "Default Sale Tax"),
2989         "purchase_tax": fields.many2one("account.tax.template", "Default Purchase Tax"),
2990         'sale_tax_rate': fields.float('Sales Tax(%)'),
2991         'purchase_tax_rate': fields.float('Purchase Tax(%)'),
2992         'complete_tax_set': fields.boolean('Complete Set of Taxes', help='This boolean helps you to choose if you want to propose to the user to encode the sales and purchase rates or use the usual m2o fields. This last choice assumes that the set of tax defined for the chosen template is complete'),
2993     }
2994     def onchange_tax_rate(self, cr, uid, ids, rate=False, context=None):
2995         return {'value': {'purchase_tax_rate': rate or False}}
2996
2997     def onchange_chart_template_id(self, cr, uid, ids, chart_template_id=False, context=None):
2998         res = {}
2999         tax_templ_obj = self.pool.get('account.tax.template')
3000         res['value'] = {'complete_tax_set': False, 'sale_tax': False, 'purchase_tax': False}
3001         if chart_template_id:
3002             data = self.pool.get('account.chart.template').browse(cr, uid, chart_template_id, context=context)
3003             res['value'].update({'complete_tax_set': data.complete_tax_set})
3004             if data.complete_tax_set:
3005             # default tax is given by the lowest sequence. For same sequence we will take the latest created as it will be the case for tax created while isntalling the generic chart of account
3006                 sale_tax_ids = tax_templ_obj.search(cr, uid, [("chart_template_id"
3007                                               , "=", chart_template_id), ('type_tax_use', 'in', ('sale','all'))], order="sequence, id desc")
3008                 purchase_tax_ids = tax_templ_obj.search(cr, uid, [("chart_template_id"
3009                                               , "=", chart_template_id), ('type_tax_use', 'in', ('purchase','all'))], order="sequence, id desc")
3010                 res['value'].update({'sale_tax': sale_tax_ids and sale_tax_ids[0] or False, 'purchase_tax': purchase_tax_ids and purchase_tax_ids[0] or False})
3011
3012             if data.code_digits:
3013                res['value'].update({'code_digits': data.code_digits})
3014         return res
3015
3016     def default_get(self, cr, uid, fields, context=None):
3017         res = super(wizard_multi_charts_accounts, self).default_get(cr, uid, fields, context=context) 
3018         tax_templ_obj = self.pool.get('account.tax.template')
3019
3020         if 'bank_accounts_id' in fields:
3021             res.update({'bank_accounts_id': [{'acc_name': _('Cash'), 'account_type': 'cash'},{'acc_name': _('Bank'), 'account_type': 'bank'}]})
3022         if 'company_id' in fields:
3023             res.update({'company_id': self.pool.get('res.users').browse(cr, uid, [uid], context=context)[0].company_id.id})
3024         if 'seq_journal' in fields:
3025             res.update({'seq_journal': True})
3026
3027         ids = self.pool.get('account.chart.template').search(cr, uid, [('visible', '=', True)], context=context)
3028         if ids:
3029             if 'chart_template_id' in fields:
3030                 res.update({'chart_template_id': ids[0]})
3031             if 'sale_tax' in fields:
3032                 sale_tax_ids = tax_templ_obj.search(cr, uid, [("chart_template_id"
3033                                               , "=", ids[0]), ('type_tax_use', 'in', ('sale','all'))], order="sequence")
3034                 res.update({'sale_tax': sale_tax_ids and sale_tax_ids[0] or False})
3035             if 'purchase_tax' in fields:
3036                 purchase_tax_ids = tax_templ_obj.search(cr, uid, [("chart_template_id"
3037                                           , "=", ids[0]), ('type_tax_use', 'in', ('purchase','all'))], order="sequence")
3038                 res.update({'purchase_tax': purchase_tax_ids and purchase_tax_ids[0] or False})
3039         res.update({
3040             'purchase_tax_rate': 15.0,
3041             'sale_tax_rate': 15.0,
3042         })
3043         return res
3044
3045     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
3046         res = super(wizard_multi_charts_accounts, self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar,submenu=False)
3047         cmp_select = []
3048         acc_template_obj = self.pool.get('account.chart.template')
3049         company_obj = self.pool.get('res.company')
3050
3051         company_ids = company_obj.search(cr, uid, [], context=context)
3052         #display in the widget selection of companies, only the companies that haven't been configured yet (but don't care about the demo chart of accounts)
3053         cr.execute("SELECT company_id FROM account_account WHERE active = 't' AND account_account.parent_id IS NULL AND name != %s", ("Chart For Automated Tests",))
3054         configured_cmp = [r[0] for r in cr.fetchall()]
3055         unconfigured_cmp = list(set(company_ids)-set(configured_cmp))
3056         for field in res['fields']:
3057             if field == 'company_id':
3058                 res['fields'][field]['domain'] = [('id','in',unconfigured_cmp)]
3059                 res['fields'][field]['selection'] = [('', '')]
3060                 if unconfigured_cmp:
3061                     cmp_select = [(line.id, line.name) for line in company_obj.browse(cr, uid, unconfigured_cmp)]
3062                     res['fields'][field]['selection'] = cmp_select
3063         return res
3064
3065     def check_created_journals(self, cr, uid, vals_journal, company_id, context=None):
3066         """
3067         This method used for checking journals already created or not. If not then create new journal.
3068         """
3069         obj_journal = self.pool.get('account.journal')
3070         rec_list = obj_journal.search(cr, uid, [('name','=', vals_journal['name']),('company_id', '=', company_id)], context=context)
3071         if not rec_list:
3072             obj_journal.create(cr, uid, vals_journal, context=context)
3073         return True
3074
3075     def generate_journals(self, cr, uid, chart_template_id, acc_template_ref, company_id, context=None):
3076         """
3077         This method is used for creating journals.
3078
3079         :param chart_temp_id: Chart Template Id.
3080         :param acc_template_ref: Account templates reference.
3081         :param company_id: company_id selected from wizard.multi.charts.accounts.
3082         :returns: True
3083         """
3084         journal_data = self._prepare_all_journals(cr, uid, chart_template_id, acc_template_ref, company_id, context=context)
3085         for vals_journal in journal_data:
3086             self.check_created_journals(cr, uid, vals_journal, company_id, context=context)
3087         return True
3088
3089     def _prepare_all_journals(self, cr, uid, chart_template_id, acc_template_ref, company_id, context=None):
3090         def _get_analytic_journal(journal_type):
3091             # Get the analytic journal
3092             data = False
3093             if journal_type in ('sale', 'sale_refund'):
3094                 data = obj_data.get_object_reference(cr, uid, 'account', 'analytic_journal_sale') 
3095             elif journal_type in ('purchase', 'purchase_refund'):
3096                 pass
3097             elif journal_type == 'general':
3098                 pass
3099             return data and data[1] or False
3100
3101         def _get_default_account(journal_type, type='debit'):
3102             # Get the default accounts
3103             default_account = False
3104             if journal_type in ('sale', 'sale_refund'):
3105                 default_account = acc_template_ref.get(template.property_account_income_categ.id)
3106             elif journal_type in ('purchase', 'purchase_refund'):
3107                 default_account = acc_template_ref.get(template.property_account_expense_categ.id)
3108             elif journal_type == 'situation':
3109                 if type == 'debit':
3110                     default_account = acc_template_ref.get(template.property_account_expense_opening.id)
3111                 else:
3112                     default_account = acc_template_ref.get(template.property_account_income_opening.id)
3113             return default_account
3114
3115         def _get_view_id(journal_type):
3116             # Get the journal views
3117             if journal_type in ('general', 'situation'):
3118                 data = obj_data.get_object_reference(cr, uid, 'account', 'account_journal_view')
3119             elif journal_type in ('sale_refund', 'purchase_refund'):
3120                 data = obj_data.get_object_reference(cr, uid, 'account', 'account_sp_refund_journal_view') 
3121             else:
3122                 data = obj_data.get_object_reference(cr, uid, 'account', 'account_sp_journal_view')
3123             return data and data[1] or False
3124
3125         journal_names = {
3126             'sale': _('Sales Journal'),
3127             'purchase': _('Purchase Journal'),
3128             'sale_refund': _('Sales Refund Journal'),
3129             'purchase_refund': _('Purchase Refund Journal'),
3130             'general': _('Miscellaneous Journal'),
3131             'situation': _('Opening Entries Journal'),
3132         }
3133         journal_codes = {
3134             'sale': _('SAJ'),
3135             'purchase': _('EXJ'),
3136             'sale_refund': _('SCNJ'),
3137             'purchase_refund': _('ECNJ'),
3138             'general': _('MISC'),
3139             'situation': _('OPEJ'),
3140         }
3141
3142         obj_data = self.pool.get('ir.model.data')
3143         analytic_journal_obj = self.pool.get('account.analytic.journal')
3144         template = self.pool.get('account.chart.template').browse(cr, uid, chart_template_id, context=context)
3145
3146         journal_data = []
3147         for journal_type in ['sale', 'purchase', 'sale_refund', 'purchase_refund', 'general', 'situation']:
3148             vals = {
3149                 'type': journal_type,
3150                 'name': journal_names[journal_type],
3151                 'code': journal_codes[journal_type],
3152                 'company_id': company_id,
3153                 'centralisation': journal_type == 'situation',
3154                 'view_id': _get_view_id(journal_type),
3155                 'analytic_journal_id': _get_analytic_journal(journal_type),
3156                 'default_credit_account_id': _get_default_account(journal_type, 'credit'),
3157                 'default_debit_account_id': _get_default_account(journal_type, 'debit'),
3158             }
3159             journal_data.append(vals)
3160         return journal_data
3161
3162     def generate_properties(self, cr, uid, chart_template_id, acc_template_ref, company_id, context=None):
3163         """
3164         This method used for creating properties.
3165
3166         :param chart_template_id: id of the current chart template for which we need to create properties
3167         :param acc_template_ref: Mapping between ids of account templates and real accounts created from them
3168         :param company_id: company_id selected from wizard.multi.charts.accounts.
3169         :returns: True
3170         """
3171         property_obj = self.pool.get('ir.property')
3172         field_obj = self.pool.get('ir.model.fields')
3173         todo_list = [
3174             ('property_account_receivable','res.partner','account.account'),
3175             ('property_account_payable','res.partner','account.account'),
3176             ('property_account_expense_categ','product.category','account.account'),
3177             ('property_account_income_categ','product.category','account.account'),
3178             ('property_account_expense','product.template','account.account'),
3179             ('property_account_income','product.template','account.account'),
3180             ('property_reserve_and_surplus_account','res.company','account.account')
3181         ]
3182         template = self.pool.get('account.chart.template').browse(cr, uid, chart_template_id, context=context)
3183         for record in todo_list:
3184             account = getattr(template, record[0])
3185             value = account and 'account.account,' + str(acc_template_ref[account.id]) or False
3186             if value:
3187                 field = field_obj.search(cr, uid, [('name', '=', record[0]),('model', '=', record[1]),('relation', '=', record[2])], context=context)
3188                 vals = {
3189                     'name': record[0],
3190                     'company_id': company_id,
3191                     'fields_id': field[0],
3192                     'value': value,
3193                 }
3194                 property_ids = property_obj.search(cr, uid, [('name','=', record[0]),('company_id', '=', company_id)], context=context)
3195                 if property_ids:
3196                     #the property exist: modify it
3197                     property_obj.write(cr, uid, property_ids, vals, context=context)
3198                 else:
3199                     #create the property
3200                     property_obj.create(cr, uid, vals, context=context)
3201         return True
3202
3203     def _install_template(self, cr, uid, template_id, company_id, code_digits=None, obj_wizard=None, acc_ref={}, taxes_ref={}, tax_code_ref={}, context=None):
3204         '''
3205         This function recursively loads the template objects and create the real objects from them.
3206
3207         :param template_id: id of the chart template to load
3208         :param company_id: id of the company the wizard is running for
3209         :param code_digits: integer that depicts the number of digits the accounts code should have in the COA
3210         :param obj_wizard: the current wizard for generating the COA from the templates
3211         :param acc_ref: Mapping between ids of account templates and real accounts created from them
3212         :param taxes_ref: Mapping between ids of tax templates and real taxes created from them
3213         :param tax_code_ref: Mapping between ids of tax code templates and real tax codes created from them
3214         :returns: return a tuple with a dictionary containing
3215             * the mapping between the account template ids and the ids of the real accounts that have been generated
3216               from them, as first item,
3217             * a similar dictionary for mapping the tax templates and taxes, as second item,
3218             * a last identical containing the mapping of tax code templates and tax codes
3219         :rtype: tuple(dict, dict, dict)
3220         '''
3221         template = self.pool.get('account.chart.template').browse(cr, uid, template_id, context=context)
3222         if template.parent_id:
3223             tmp1, tmp2, tmp3 = self._install_template(cr, uid, template.parent_id.id, company_id, code_digits=code_digits, acc_ref=acc_ref, taxes_ref=taxes_ref, tax_code_ref=tax_code_ref, context=context)
3224             acc_ref.update(tmp1)
3225             taxes_ref.update(tmp2)
3226             tax_code_ref.update(tmp3)
3227         tmp1, tmp2, tmp3 = self._load_template(cr, uid, template_id, company_id, code_digits=code_digits, obj_wizard=obj_wizard, account_ref=acc_ref, taxes_ref=taxes_ref, tax_code_ref=tax_code_ref, context=context)
3228         acc_ref.update(tmp1)
3229         taxes_ref.update(tmp2)
3230         tax_code_ref.update(tmp3)
3231         return acc_ref, taxes_ref, tax_code_ref
3232
3233     def _load_template(self, cr, uid, template_id, company_id, code_digits=None, obj_wizard=None, account_ref={}, taxes_ref={}, tax_code_ref={}, context=None):
3234         '''
3235         This function generates all the objects from the templates
3236
3237         :param template_id: id of the chart template to load
3238         :param company_id: id of the company the wizard is running for
3239         :param code_digits: integer that depicts the number of digits the accounts code should have in the COA
3240         :param obj_wizard: the current wizard for generating the COA from the templates
3241         :param acc_ref: Mapping between ids of account templates and real accounts created from them
3242         :param taxes_ref: Mapping between ids of tax templates and real taxes created from them
3243         :param tax_code_ref: Mapping between ids of tax code templates and real tax codes created from them
3244         :returns: return a tuple with a dictionary containing
3245             * the mapping between the account template ids and the ids of the real accounts that have been generated
3246               from them, as first item,
3247             * a similar dictionary for mapping the tax templates and taxes, as second item,
3248             * a last identical containing the mapping of tax code templates and tax codes
3249         :rtype: tuple(dict, dict, dict)
3250         '''
3251         template = self.pool.get('account.chart.template').browse(cr, uid, template_id, context=context)
3252         obj_tax_code_template = self.pool.get('account.tax.code.template')
3253         obj_acc_tax = self.pool.get('account.tax')
3254         obj_tax_temp = self.pool.get('account.tax.template')
3255         obj_acc_template = self.pool.get('account.account.template')
3256         obj_fiscal_position_template = self.pool.get('account.fiscal.position.template')
3257
3258         # create all the tax code.
3259         tax_code_ref.update(obj_tax_code_template.generate_tax_code(cr, uid, template.tax_code_root_id.id, company_id, context=context))
3260
3261         # Generate taxes from templates.
3262         tax_templates = [x for x in template.tax_template_ids]
3263         generated_tax_res = obj_tax_temp._generate_tax(cr, uid, tax_templates, tax_code_ref, company_id, context=context)
3264         taxes_ref.update(generated_tax_res['tax_template_to_tax'])
3265
3266         # Generating Accounts from templates.
3267         account_template_ref = obj_acc_template.generate_account(cr, uid, template_id, taxes_ref, account_ref, code_digits, company_id, context=context)
3268         account_ref.update(account_template_ref)
3269
3270         # writing account values on tax after creation of accounts
3271         for key,value in generated_tax_res['account_dict'].items():
3272             if value['account_collected_id'] or value['account_paid_id']:
3273                 obj_acc_tax.write(cr, uid, [key], {
3274                     'account_collected_id': account_ref.get(value['account_collected_id'], False),
3275                     'account_paid_id': account_ref.get(value['account_paid_id'], False),
3276                 })
3277
3278         # Create Journals
3279         self.generate_journals(cr, uid, template_id, account_ref, company_id, context=context)
3280
3281         # generate properties function
3282         self.generate_properties(cr, uid, template_id, account_ref, company_id, context=context)
3283
3284         # Generate Fiscal Position , Fiscal Position Accounts and Fiscal Position Taxes from templates
3285         obj_fiscal_position_template.generate_fiscal_position(cr, uid, template_id, taxes_ref, account_ref, company_id, context=context)
3286
3287         return account_ref, taxes_ref, tax_code_ref
3288
3289     def _create_tax_templates_from_rates(self, cr, uid, obj_wizard, company_id, context=None):
3290         '''
3291         This function checks if the chosen chart template is configured as containing a full set of taxes, and if
3292         it's not the case, it creates the templates for account.tax.code and for account.account.tax objects accordingly
3293         to the provided sale/purchase rates. Then it saves the new tax templates as default taxes to use for this chart
3294         template.
3295
3296         :param obj_wizard: browse record of wizard to generate COA from templates
3297         :param company_id: id of the company for wich the wizard is running
3298         :return: True
3299         '''
3300         obj_tax_code_template = self.pool.get('account.tax.code.template')
3301         obj_tax_temp = self.pool.get('account.tax.template')
3302         chart_template = obj_wizard.chart_template_id
3303         vals = {}
3304         # get the ids of all the parents of the selected account chart template
3305         current_chart_template = chart_template
3306         all_parents = [current_chart_template.id]
3307         while current_chart_template.parent_id:
3308             current_chart_template = current_chart_template.parent_id
3309             all_parents.append(current_chart_template.id)
3310         # create tax templates and tax code templates from purchase_tax_rate and sale_tax_rate fields
3311         if not chart_template.complete_tax_set:
3312             value = obj_wizard.sale_tax_rate
3313             ref_tax_ids = obj_tax_temp.search(cr, uid, [('type_tax_use','in', ('sale','all')), ('chart_template_id', 'in', all_parents)], context=context, order="sequence, id desc", limit=1)
3314             obj_tax_temp.write(cr, uid, ref_tax_ids, {'amount': value/100.0, 'name': _('Tax %.2f%%') % value})
3315             value = obj_wizard.purchase_tax_rate
3316             ref_tax_ids = obj_tax_temp.search(cr, uid, [('type_tax_use','in', ('purchase','all')), ('chart_template_id', 'in', all_parents)], context=context, order="sequence, id desc", limit=1)
3317             obj_tax_temp.write(cr, uid, ref_tax_ids, {'amount': value/100.0, 'name': _('Purchase Tax %.2f%%') % value})
3318         return True
3319
3320     def execute(self, cr, uid, ids, context=None):
3321         '''
3322         This function is called at the confirmation of the wizard to generate the COA from the templates. It will read
3323         all the provided information to create the accounts, the banks, the journals, the taxes, the tax codes, the
3324         accounting properties... accordingly for the chosen company.
3325         '''
3326         ir_values_obj = self.pool.get('ir.values')
3327         obj_wizard = self.browse(cr, uid, ids[0])
3328         company_id = obj_wizard.company_id.id
3329         # If the floats for sale/purchase rates have been filled, create templates from them
3330         self._create_tax_templates_from_rates(cr, uid, obj_wizard, company_id, context=context)
3331
3332         # Install all the templates objects and generate the real objects
3333         acc_template_ref, taxes_ref, tax_code_ref = self._install_template(cr, uid, obj_wizard.chart_template_id.id, company_id, code_digits=obj_wizard.code_digits, obj_wizard=obj_wizard, context=context)
3334
3335         # write values of default taxes for product
3336         if obj_wizard.sale_tax and taxes_ref:
3337             ir_values_obj.set(cr, 1, key='default', key2=False, name="taxes_id", company=company_id,
3338                                 models =[('product.product',False)], value=[taxes_ref[obj_wizard.sale_tax.id]])
3339         if obj_wizard.purchase_tax and taxes_ref:
3340             ir_values_obj.set(cr, 1, key='default', key2=False, name="supplier_taxes_id", company=company_id,
3341                                 models =[('product.product',False)], value=[taxes_ref[obj_wizard.purchase_tax.id]])
3342
3343         # Create Bank journals
3344         self._create_bank_journals_from_o2m(cr, uid, obj_wizard, company_id, acc_template_ref, context=context)
3345         action = {
3346             'type': 'ir.actions.act_window',
3347             'view_type': 'form',
3348             'view_mode': 'form',
3349             'res_model': 'board.board',
3350             'view_id': self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account', 'board_account_form')[1],
3351             'menu_id': self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account', 'menu_finance')[1]
3352         }
3353         return action
3354
3355     def _prepare_bank_journal(self, cr, uid, line, current_num, default_account_id, company_id, context=None):
3356         '''
3357         This function prepares the value to use for the creation of a bank journal created through the wizard of 
3358         generating COA from templates.
3359
3360         :param line: dictionary containing the values encoded by the user related to his bank account
3361         :param current_num: integer corresponding to a counter of the already created bank journals through this wizard.
3362         :param default_account_id: id of the default debit.credit account created before for this journal.
3363         :param company_id: id of the company for which the wizard is running
3364         :return: mapping of field names and values
3365         :rtype: dict
3366         '''
3367         obj_data = self.pool.get('ir.model.data')
3368         obj_journal = self.pool.get('account.journal')
3369         # Get the id of journal views
3370         tmp = obj_data.get_object_reference(cr, uid, 'account', 'account_journal_bank_view_multi')
3371         view_id_cur = tmp and tmp[1] or False
3372         tmp = obj_data.get_object_reference(cr, uid, 'account', 'account_journal_bank_view')
3373         view_id_cash = tmp and tmp[1] or False
3374
3375         # we need to loop again to find next number for journal code 
3376         # because we can't rely on the value current_num as,
3377         # its possible that we already have bank journals created (e.g. by the creation of res.partner.bank) 
3378         # and the next number for account code might have been already used before for journal
3379         for num in xrange(current_num, 100):
3380             # journal_code has a maximal size of 5, hence we can enforce the boundary num < 100
3381             journal_code = _('BNK')[:3] + str(num)
3382             ids = obj_journal.search(cr, uid, [('code', '=', journal_code), ('company_id', '=', company_id)], context=context)
3383             if not ids:
3384                 break
3385         else:
3386             raise osv.except_osv(_('Error'), _('Cannot generate an unused journal code.'))
3387
3388         vals = {
3389                 'name': line['acc_name'],
3390                 'code': journal_code,
3391                 'type': line['account_type'] == 'cash' and 'cash' or 'bank',
3392                 'company_id': company_id,
3393                 'analytic_journal_id': False,
3394                 'currency': False,
3395                 'default_credit_account_id': default_account_id,
3396                 'default_debit_account_id': default_account_id,
3397         }
3398         if line['currency_id']:
3399             vals['view_id'] = view_id_cur
3400             vals['currency'] = line['currency_id']
3401         else:
3402             vals['view_id'] = view_id_cash
3403         return vals
3404
3405     def _prepare_bank_account(self, cr, uid, line, new_code, acc_template_ref, ref_acc_bank, company_id, context=None):
3406         '''
3407         This function prepares the value to use for the creation of the default debit and credit accounts of a
3408         bank journal created through the wizard of generating COA from templates.
3409
3410         :param line: dictionary containing the values encoded by the user related to his bank account
3411         :param new_code: integer corresponding to the next available number to use as account code
3412         :param acc_template_ref: the dictionary containing the mapping between the ids of account templates and the ids
3413             of the accounts that have been generated from them.
3414         :param ref_acc_bank: browse record of the account template set as root of all bank accounts for the chosen
3415             template
3416         :param company_id: id of the company for which the wizard is running
3417         :return: mapping of field names and values
3418         :rtype: dict
3419         '''
3420         obj_data = self.pool.get('ir.model.data')
3421
3422         # Get the id of the user types fr-or cash and bank
3423         tmp = obj_data.get_object_reference(cr, uid, 'account', 'data_account_type_cash')
3424         cash_type = tmp and tmp[1] or False
3425         tmp = obj_data.get_object_reference(cr, uid, 'account', 'data_account_type_bank')
3426         bank_type = tmp and tmp[1] or False
3427         return {
3428                 'name': line['acc_name'],
3429                 'currency_id': line['currency_id'],
3430                 'code': new_code,
3431                 'type': 'liquidity',
3432                 'user_type': line['account_type'] == 'cash' and cash_type or bank_type,
3433                 'parent_id': acc_template_ref[ref_acc_bank.id] or False,
3434                 'company_id': company_id,
3435         }
3436
3437     def _create_bank_journals_from_o2m(self, cr, uid, obj_wizard, company_id, acc_template_ref, context=None):
3438         '''
3439         This function creates bank journals and its accounts for each line encoded in the field bank_accounts_id of the
3440         wizard.
3441
3442         :param obj_wizard: the current wizard that generates the COA from the templates.
3443         :param company_id: the id of the company for which the wizard is running.
3444         :param acc_template_ref: the dictionary containing the mapping between the ids of account templates and the ids
3445             of the accounts that have been generated from them.
3446         :return: True
3447         '''
3448         obj_acc = self.pool.get('account.account')
3449         obj_journal = self.pool.get('account.journal')
3450         code_digits = obj_wizard.code_digits
3451
3452         # Build a list with all the data to process
3453         journal_data = []
3454         if obj_wizard.bank_accounts_id:
3455             for acc in obj_wizard.bank_accounts_id:
3456                 vals = {
3457                     'acc_name': acc.acc_name,
3458                     'account_type': acc.account_type,
3459                     'currency_id': acc.currency_id.id,
3460                 }
3461                 journal_data.append(vals)
3462         ref_acc_bank = obj_wizard.chart_template_id.bank_account_view_id
3463         if journal_data and not ref_acc_bank.code:
3464             raise osv.except_osv(_('Configuration Error !'), _('The bank account defined on the selected chart of accounts hasn\'t a code.'))
3465
3466         current_num = 1
3467         for line in journal_data:
3468             # Seek the next available number for the account code
3469             while True:
3470                 new_code = str(ref_acc_bank.code.ljust(code_digits-len(str(current_num)), '0')) + str(current_num)
3471                 ids = obj_acc.search(cr, uid, [('code', '=', new_code), ('company_id', '=', company_id)])
3472                 if not ids:
3473                     break
3474                 else:
3475                     current_num += 1
3476             # Create the default debit/credit accounts for this bank journal
3477             vals = self._prepare_bank_account(cr, uid, line, new_code, acc_template_ref, ref_acc_bank, company_id, context=context)
3478             default_account_id  = obj_acc.create(cr, uid, vals, context=context)
3479
3480             #create the bank journal
3481             vals_journal = self._prepare_bank_journal(cr, uid, line, current_num, default_account_id, company_id, context=context)
3482             obj_journal.create(cr, uid, vals_journal)
3483             current_num += 1
3484         return True
3485
3486 wizard_multi_charts_accounts()
3487
3488 class account_bank_accounts_wizard(osv.osv_memory):
3489     _name='account.bank.accounts.wizard'
3490
3491     _columns = {
3492         'acc_name': fields.char('Account Name.', size=64, required=True),
3493         'bank_account_id': fields.many2one('wizard.multi.charts.accounts', 'Bank Account', required=True),
3494         'currency_id': fields.many2one('res.currency', 'Secondary Currency', help="Forces all moves for this account to have this secondary currency."),
3495         'account_type': fields.selection([('cash','Cash'), ('check','Check'), ('bank','Bank')], 'Account Type', size=32),
3496     }
3497
3498 account_bank_accounts_wizard()
3499
3500 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: