103 lines
3.8 KiB
Python
103 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
|
|
import collections
|
|
|
|
from flectra import api, fields, models
|
|
from flectra.osv import expression
|
|
from flectra.tools import pycompat
|
|
|
|
|
|
def sanitize_account_number(acc_number):
|
|
if acc_number:
|
|
return re.sub(r'\W+', '', acc_number).upper()
|
|
return False
|
|
|
|
|
|
class Bank(models.Model):
|
|
_description = 'Bank'
|
|
_name = 'res.bank'
|
|
_order = 'name'
|
|
|
|
name = fields.Char(required=True)
|
|
street = fields.Char()
|
|
street2 = fields.Char()
|
|
zip = fields.Char()
|
|
city = fields.Char()
|
|
state = fields.Many2one('res.country.state', 'Fed. State', domain="[('country_id', '=', country)]")
|
|
country = fields.Many2one('res.country')
|
|
email = fields.Char()
|
|
phone = fields.Char()
|
|
active = fields.Boolean(default=True)
|
|
bic = fields.Char('Bank Identifier Code', index=True, help="Sometimes called BIC or Swift.")
|
|
|
|
@api.multi
|
|
@api.depends('name', 'bic')
|
|
def name_get(self):
|
|
result = []
|
|
for bank in self:
|
|
name = bank.name + (bank.bic and (' - ' + bank.bic) or '')
|
|
result.append((bank.id, name))
|
|
return result
|
|
|
|
@api.model
|
|
def name_search(self, name, args=None, operator='ilike', limit=100):
|
|
args = args or []
|
|
domain = []
|
|
if name:
|
|
domain = ['|', ('bic', '=ilike', name + '%'), ('name', operator, name)]
|
|
if operator in expression.NEGATIVE_TERM_OPERATORS:
|
|
domain = ['&'] + domain
|
|
banks = self.search(domain + args, limit=limit)
|
|
return banks.name_get()
|
|
|
|
|
|
class ResPartnerBank(models.Model):
|
|
_name = 'res.partner.bank'
|
|
_rec_name = 'acc_number'
|
|
_description = 'Bank Accounts'
|
|
_order = 'sequence'
|
|
|
|
acc_type = fields.Char(compute='_compute_acc_type', help='Bank account type, inferred from account number')
|
|
acc_number = fields.Char('Account Number', required=True)
|
|
sanitized_acc_number = fields.Char(compute='_compute_sanitized_acc_number', string='Sanitized Account Number', readonly=True, store=True)
|
|
partner_id = fields.Many2one('res.partner', 'Account Holder', ondelete='cascade', index=True, domain=['|', ('is_company', '=', True), ('parent_id', '=', False)], default=lambda self: self.env.user.company_id.partner_id)
|
|
bank_id = fields.Many2one('res.bank', string='Bank')
|
|
bank_name = fields.Char(related='bank_id.name')
|
|
bank_bic = fields.Char(related='bank_id.bic')
|
|
sequence = fields.Integer()
|
|
currency_id = fields.Many2one('res.currency', string='Currency')
|
|
company_id = fields.Many2one('res.company', 'Company', default=lambda self: self.env.user.company_id, ondelete='cascade')
|
|
|
|
_sql_constraints = [
|
|
('unique_number', 'unique(sanitized_acc_number, company_id)', 'Account Number must be unique'),
|
|
]
|
|
|
|
@api.depends('acc_number')
|
|
def _compute_sanitized_acc_number(self):
|
|
for bank in self:
|
|
bank.sanitized_acc_number = sanitize_account_number(bank.acc_number)
|
|
|
|
@api.multi
|
|
def _compute_acc_type(self):
|
|
for bank in self:
|
|
bank.acc_type = 'bank'
|
|
|
|
@api.model
|
|
def search(self, args, offset=0, limit=None, order=None, count=False):
|
|
pos = 0
|
|
while pos < len(args):
|
|
if args[pos][0] == 'acc_number':
|
|
op = args[pos][1]
|
|
value = args[pos][2]
|
|
if not isinstance(value, pycompat.string_types) and isinstance(value, collections.Iterable):
|
|
value = [sanitize_account_number(i) for i in value]
|
|
else:
|
|
value = sanitize_account_number(value)
|
|
if 'like' in op:
|
|
value = '%' + value + '%'
|
|
args[pos] = ('sanitized_acc_number', op, value)
|
|
pos += 1
|
|
return super(ResPartnerBank, self).search(args, offset, limit, order, count=count)
|