Add a RIB-specific 'format_layout' on the res.partner.bank.type
[odoo/odoo.git] / addons / l10n_fr_rib / bank.py
1 # -*- encoding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2011 Numérigraphe SARL.
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 netsvc
23 from osv import fields, osv
24 from tools.translate import _
25
26 class res_partner_bank(osv.osv):
27     """Add fields and behavior for French RIB"""
28     _inherit = "res.partner.bank"
29
30     def _check_key(self, cr, uid, ids):
31         """Check the RIB key"""
32         for bank_acc in self.browse(cr, uid, ids):
33             # Ignore the accounts of type other than rib
34             if bank_acc.state <> 'rib':
35                 continue
36             # Fail if the needed values are empty of too short 
37             if (not bank_acc.bank_code
38             or len(bank_acc.bank_code) != 5
39             or not bank_acc.office or len(bank_acc.office) != 5
40             or not bank_acc.rib_acc_number or len(bank_acc.rib_acc_number) != 11
41             or not bank_acc.key or len(bank_acc.key) != 2):
42                 return False
43             # Get the rib data (without the key)
44             rib = "%s%s%s" % (bank_acc.bank_code, bank_acc.office,
45                               bank_acc.rib_acc_number)
46             # Translate letters into numbers according to a specific table
47             #    (notice how s -> 2)
48             table = dict((ord(a), b) for a, b in zip(
49                 u'abcdefghijklmnopqrstuvwxyz', u'12345678912345678923456789'))
50             rib = rib.lower().translate(table)
51             # compute the key   
52             key = 97 - (100 * int(rib)) % 97
53             if int(bank_acc.key) != key:
54                 raise osv.except_osv(_('Error'),
55                      _("The RIB key %s does not correspond to the other "
56                             "codes: %s %s %s.") %
57                             (bank_acc.key, bank_acc.bank_code,
58                             bank_acc.office, bank_acc.rib_acc_number) )
59             if bank_acc.acc_number:
60                 if not self.is_iban_valid(cr, uid, bank_acc.acc_number):
61                     raise osv.except_osv(_('Error'),
62                         _("The IBAN %s is not valid.") % bank_acc.acc_number)
63         return True
64
65     def onchange_bank_id(self, cr, uid, ids, bank_id, context=None):
66         """Change the bank code"""
67         result = super(res_partner_bank, self).onchange_bank_id(cr, uid, ids, bank_id,
68                                                         context=context)
69         if bank_id:
70             value = result.setdefault('value', {})
71             bank = self.pool.get('res.bank').browse(cr, uid, bank_id, 
72                                                     context=context)
73             value['bank_code'] = bank.rib_code
74         return result
75
76     _columns = {
77         'acc_number': fields.char('Account Number', size=64, required=False),
78         'rib_acc_number': fields.char('RIB account number', size=11, readonly=True,),
79         'bank_code': fields.char('Bank Code', size=64, readonly=True,),
80         'office': fields.char('Office Code', size=5, readonly=True,),
81         'key': fields.char('Key', size=2, readonly=True,
82                            help="The key is a number allowing to check the "
83                                 "correctness of the other codes."),
84     }
85
86     _constraints = [(_check_key, 'Error message in raise',
87          ['rib_acc_number', 'bank_code', 'office', 'key'])]
88
89 res_partner_bank()
90
91 class res_bank(osv.osv):
92     """Add the bank code to make it easier to enter RIB data"""
93     _inherit = 'res.bank'
94
95     def name_search(self, cr, user, name, args=None, operator='ilike',
96                     context=None, limit=80):
97         """Search by bank code in addition to the standard search"""
98         # Get the standard results
99         results = super(res_bank, self).name_search(cr, user,
100              name, args=args ,operator=operator, context=context, limit=limit)
101         # Get additional results using the RIB code
102         ids = self.search(cr, user, [('rib_code', operator, name)],
103                               limit=limit, context=context)
104         # Merge the results
105         results = list(set(results + self.name_get(cr, user, ids, context)))
106         return results
107         
108     _columns = {
109         'rib_code': fields.char('RIB Bank Code', size=64),
110     }
111 res_bank()
112 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
113