openerp/addons/base/res/res_bank.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 from openerp.osv import fields, osv
  23 from openerp.tools.translate import _
  24
  25 class Bank(osv.osv):
  26     _description='Bank'
  27     _name = 'res.bank'
  28     _order = 'name'
  29     _columns = {
  30         'name': fields.char('Name', required=True),
  31         'street': fields.char('Street'),
  32         'street2': fields.char('Street2'),
  33         'zip': fields.char('Zip', change_default=True, size=24),
  34         'city': fields.char('City'),
  35         'state': fields.many2one("res.country.state", 'Fed. State',
  36             domain="[('country_id', '=', country)]"),
  37         'country': fields.many2one('res.country', 'Country'),
  38         'email': fields.char('Email'),
  39         'phone': fields.char('Phone'),
  40         'fax': fields.char('Fax'),
  41         'active': fields.boolean('Active'),
  42         'bic': fields.char('Bank Identifier Code', size=64,
  43             help="Sometimes called BIC or Swift."),
  44     }
  45     _defaults = {
  46         'active': lambda *a: 1,
  47     }
  48     def name_get(self, cr, uid, ids, context=None):
  49         result = []
  50         for bank in self.browse(cr, uid, ids, context):
  51             result.append((bank.id, (bank.bic and (bank.bic + ' - ') or '') + bank.name))
  52         return result
  53
  54 class res_partner_bank_type(osv.osv):
  55     _description='Bank Account Type'
  56     _name = 'res.partner.bank.type'
  57     _order = 'name'
  58     _columns = {
  59         'name': fields.char('Name', required=True, translate=True),
  60         'code': fields.char('Code', size=64, required=True),
  61         'field_ids': fields.one2many('res.partner.bank.type.field', 'bank_type_id', 'Type Fields'),
  62         'format_layout': fields.text('Format Layout', translate=True)
  63     }
  64     _defaults = {
  65         'format_layout': lambda *args: "%(bank_name)s: %(acc_number)s"
  66     }
  67
  68 class res_partner_bank_type_fields(osv.osv):
  69     _description='Bank type fields'
  70     _name = 'res.partner.bank.type.field'
  71     _order = 'name'
  72     _columns = {
  73         'name': fields.char('Field Name', required=True, translate=True),
  74         'bank_type_id': fields.many2one('res.partner.bank.type', 'Bank Type', required=True, ondelete='cascade'),
  75         'required': fields.boolean('Required'),
  76         'readonly': fields.boolean('Readonly'),
  77         'size': fields.integer('Max. Size'),
  78     }
  79
  80 class res_partner_bank(osv.osv):
  81     '''Bank Accounts'''
  82     _name = "res.partner.bank"
  83     _rec_name = "acc_number"
  84     _description = __doc__
  85     _order = 'sequence'
  86
  87     def _bank_type_get(self, cr, uid, context=None):
  88         bank_type_obj = self.pool.get('res.partner.bank.type')
  89
  90         result = []
  91         type_ids = bank_type_obj.search(cr, uid, [])
  92         bank_types = bank_type_obj.browse(cr, uid, type_ids, context=context)
  93         for bank_type in bank_types:
  94             result.append((bank_type.code, bank_type.name))
  95         return result
  96
  97     def _default_value(self, cursor, user, field, context=None):
  98         if context is None: context = {}
  99         if field in ('country_id', 'state_id'):
 100             value = False
 101         else:
 102             value = ''
 103         if not context.get('address'):
 104             return value
 105
 106         for address in self.pool.get('res.partner').resolve_2many_commands(
 107             cursor, user, 'address', context['address'], ['type', field], context=context):
 108
 109             if address.get('type') == 'default':
 110                 return address.get(field, value)
 111             elif not address.get('type'):
 112                 value = address.get(field, value)
 113         return value
 114
 115     _columns = {
 116         'name': fields.char('Bank Account'), # to be removed in v6.2 ?
 117         'acc_number': fields.char('Account Number', size=64, required=True),
 118         'bank': fields.many2one('res.bank', 'Bank'),
 119         'bank_bic': fields.char('Bank Identifier Code', size=16),
 120         'bank_name': fields.char('Bank Name'),
 121         'owner_name': fields.char('Account Owner Name'),
 122         'street': fields.char('Street'),
 123         'zip': fields.char('Zip', change_default=True, size=24),
 124         'city': fields.char('City'),
 125         'country_id': fields.many2one('res.country', 'Country',
 126             change_default=True),
 127         'state_id': fields.many2one("res.country.state", 'Fed. State',
 128             change_default=True, domain="[('country_id','=',country_id)]"),
 129         'company_id': fields.many2one('res.company', 'Company',
 130             ondelete='cascade', help="Only if this bank account belong to your company"),
 131         'partner_id': fields.many2one('res.partner', 'Account Owner', ondelete='cascade', select=True, domain=['|',('is_company','=',True),('parent_id','=',False)]),
 132         'state': fields.selection(_bank_type_get, 'Bank Account Type', required=True,
 133             change_default=True),
 134         'sequence': fields.integer('Sequence'),
 135         'footer': fields.boolean("Display on Reports", help="Display this bank account on the footer of printed documents like invoices and sales orders.")
 136     }
 137
 138     _defaults = {
 139         'owner_name': lambda obj, cursor, user, context: obj._default_value(
 140             cursor, user, 'name', context=context),
 141         'street': lambda obj, cursor, user, context: obj._default_value(
 142             cursor, user, 'street', context=context),
 143         'city': lambda obj, cursor, user, context: obj._default_value(
 144             cursor, user, 'city', context=context),
 145         'zip': lambda obj, cursor, user, context: obj._default_value(
 146             cursor, user, 'zip', context=context),
 147         'country_id': lambda obj, cursor, user, context: obj._default_value(
 148             cursor, user, 'country_id', context=context),
 149         'state_id': lambda obj, cursor, user, context: obj._default_value(
 150             cursor, user, 'state_id', context=context),
 151         'name': '/'
 152     }
 153
 154     def fields_get(self, cr, uid, allfields=None, context=None):
 155         res = super(res_partner_bank, self).fields_get(cr, uid, allfields=allfields, context=context)
 156         bank_type_obj = self.pool.get('res.partner.bank.type')
 157         type_ids = bank_type_obj.search(cr, uid, [])
 158         types = bank_type_obj.browse(cr, uid, type_ids)
 159         for type in types:
 160             for field in type.field_ids:
 161                 if field.name in res:
 162                     res[field.name].setdefault('states', {})
 163                     res[field.name]['states'][type.code] = [
 164                             ('readonly', field.readonly),
 165                             ('required', field.required)]
 166         return res
 167
 168     def _prepare_name_get(self, cr, uid, bank_dicts, context=None):
 169         """ Format the name of a res.partner.bank.
 170             This function is designed to be inherited to add replacement fields.
 171             :param bank_dicts: a list of res.partner.bank dicts, as returned by the method read()
 172             :return: [(id, name), ...], as returned by the method name_get()
 173         """
 174         # prepare a mapping {code: format_layout} for all bank types
 175         bank_type_obj = self.pool.get('res.partner.bank.type')
 176         bank_types = bank_type_obj.browse(cr, uid, bank_type_obj.search(cr, uid, []), context=context)
 177         bank_code_format = dict((bt.code, bt.format_layout) for bt in bank_types)
 178
 179         res = []
 180         for data in bank_dicts:
 181             name = data['acc_number']
 182             if data['state'] and bank_code_format.get(data['state']):
 183                 try:
 184                     if not data.get('bank_name'):
 185                         data['bank_name'] = _('BANK')
 186                     name = bank_code_format[data['state']] % data
 187                 except Exception:
 188                     raise osv.except_osv(_("Formating Error"), _("Invalid Bank Account Type Name format."))
 189             res.append((data.get('id', False), name))
 190         return res
 191
 192     def name_get(self, cr, uid, ids, context=None):
 193         if not len(ids):
 194             return []
 195         bank_dicts = self.read(cr, uid, ids, self.fields_get_keys(cr, uid, context=context), context=context)
 196         return self._prepare_name_get(cr, uid, bank_dicts, context=context)
 197
 198     def onchange_company_id(self, cr, uid, ids, company_id, context=None):
 199         result = {}
 200         if company_id:
 201             c = self.pool.get('res.company').browse(cr, uid, company_id, context=context)
 202             if c.partner_id:
 203                 r = self.onchange_partner_id(cr, uid, ids, c.partner_id.id, context=context)
 204                 r['value']['partner_id'] = c.partner_id.id
 205                 r['value']['footer'] = 1
 206                 result = r
 207         return result
 208
 209     def onchange_bank_id(self, cr, uid, ids, bank_id, context=None):
 210         result = {}
 211         if bank_id:
 212             bank = self.pool.get('res.bank').browse(cr, uid, bank_id, context=context)
 213             result['bank_name'] = bank.name
 214             result['bank_bic'] = bank.bic
 215         return {'value': result}
 216
 217
 218     def onchange_partner_id(self, cr, uid, id, partner_id, context=None):
 219         result = {}
 220         if partner_id:
 221             part = self.pool.get('res.partner').browse(cr, uid, partner_id, context=context)
 222             result['owner_name'] = part.name
 223             result['street'] = part.street or False
 224             result['city'] = part.city or False
 225             result['zip'] =  part.zip or False
 226             result['country_id'] =  part.country_id.id
 227             result['state_id'] = part.state_id.id
 228         return {'value': result}
 229
 230 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: