addons/account/account_bank.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 from tools.translate import _
  23 from osv import fields, osv
  24
  25 class bank(osv.osv):
  26     _inherit = "res.partner.bank"
  27     _columns = {
  28         'journal_id': fields.many2one('account.journal', 'Account Journal', help="This journal will be created automatically for this bank account when you save the record"),
  29     }
  30     def create(self, cr, uid, data, context={}):
  31         result = super(bank, self).create(cr, uid, data, context=context)
  32         self.post_write(cr, uid, [result], context=context)
  33         return result
  34
  35     def write(self, cr, uid, ids, data, context={}):
  36         result = super(bank, self).write(cr, uid, ids, data, context=context)
  37         self.post_write(cr, uid, ids, context=context)
  38         return result
  39
  40     def _prepare_name(self, bank):
  41         "Return the name to use when creating a bank journal"
  42         return (bank.bank_name or '') + ' ' + bank.acc_number
  43
  44     def post_write(self, cr, uid, ids, context={}):
  45         if isinstance(ids, (int, long)):
  46           ids = [ids]
  47
  48         obj_acc = self.pool.get('account.account')
  49         obj_data = self.pool.get('ir.model.data')
  50
  51         for bank in self.browse(cr, uid, ids, context):
  52             if bank.company_id and not bank.journal_id:
  53                 # Find the code and parent of the bank account to create
  54                 dig = 6
  55                 current_num = 1
  56                 ids = obj_acc.search(cr, uid, [('type','=','liquidity')], context=context)
  57                 # No liquidity account exists, no template available
  58                 if not ids: continue
  59
  60                 ref_acc_bank_temp = obj_acc.browse(cr, uid, ids[0], context=context)
  61                 ref_acc_bank = ref_acc_bank_temp.parent_id
  62                 while True:
  63                     new_code = str(ref_acc_bank.code.ljust(dig-len(str(current_num)), '0')) + str(current_num)
  64                     ids = obj_acc.search(cr, uid, [('code', '=', new_code), ('company_id', '=', bank.company_id.id)])
  65                     if not ids:
  66                         break
  67                     current_num += 1
  68                 name = self._prepare_name(bank)
  69                 acc = {
  70                     'name': name,
  71                     'currency_id': bank.company_id.currency_id.id,
  72                     'code': new_code,
  73                     'type': 'liquidity',
  74                     'user_type': ref_acc_bank_temp.user_type.id,
  75                     'reconcile': False,
  76                     'parent_id': ref_acc_bank.id,
  77                     'company_id': bank.company_id.id,
  78                 }
  79                 acc_bank_id  = obj_acc.create(cr,uid,acc,context=context)
  80
  81                 # Get the journal view id
  82                 data_id = obj_data.search(cr, uid, [('model','=','account.journal.view'), ('name','=','account_journal_bank_view')])
  83                 data = obj_data.browse(cr, uid, data_id[0], context=context)
  84                 view_id_cash = data.res_id
  85
  86                 jour_obj = self.pool.get('account.journal')
  87                 new_code = 1
  88                 while True:
  89                     code = _('BNK')+str(new_code)
  90                     ids = jour_obj.search(cr, uid, [('code','=',code)], context=context)
  91                     if not ids:
  92                         break
  93                     new_code += 1
  94
  95                 #create the bank journal
  96                 vals_journal = {
  97                     'name': name,
  98                     'code': code,
  99                     'type': 'bank',
 100                     'company_id': bank.company_id.id,
 101                     'analytic_journal_id': False,
 102                     'currency_id': False,
 103                     'default_credit_account_id': acc_bank_id,
 104                     'default_debit_account_id': acc_bank_id,
 105                     'view_id': view_id_cash
 106                 }
 107                 journal_id = jour_obj.create(cr, uid, vals_journal, context=context)
 108
 109                 self.write(cr, uid, [bank.id], {'journal_id': journal_id}, context=context)
 110         return True
 111
 112 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: