185 lines
9.5 KiB
Python
185 lines
9.5 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo, Flectra. See LICENSE file for full copyright and licensing details.
|
|
|
|
# Copyright (c) 2011 Noviat nv/sa (www.noviat.be). All rights reserved.
|
|
|
|
import random
|
|
import re
|
|
|
|
from flectra import api, fields, models, _
|
|
from flectra.exceptions import UserError, ValidationError
|
|
|
|
"""
|
|
account.invoice object:
|
|
- Add support for Belgian structured communication
|
|
- Rename 'reference' field labels to 'Communication'
|
|
"""
|
|
|
|
|
|
class AccountInvoice(models.Model):
|
|
_inherit = 'account.invoice'
|
|
|
|
@api.model
|
|
def _get_reference_type(self):
|
|
"""Add BBA Structured Communication Type and change labels from 'reference' into 'communication' """
|
|
res = super(AccountInvoice, self)._get_reference_type()
|
|
res[[i for i, x in enumerate(res) if x[0] == 'none'][0]] = ('none', _('Free Communication'))
|
|
res.append(('bba', 'BBA Structured Communication'))
|
|
return res
|
|
|
|
reference_type = fields.Selection('_get_reference_type', string='Payment Reference',
|
|
required=True, readonly=True)
|
|
|
|
@api.constrains('reference', 'reference_type')
|
|
def _check_communication(self):
|
|
for inv in self:
|
|
if inv.reference_type == 'bba' and not self.check_bbacomm(inv.reference):
|
|
raise ValidationError(_('Invalid BBA Structured Communication !'))
|
|
|
|
def check_bbacomm(self, val):
|
|
supported_chars = '0-9+*/ '
|
|
pattern = re.compile('[^' + supported_chars + ']')
|
|
if pattern.findall(val or ''):
|
|
return False
|
|
bbacomm = re.sub('\D', '', val or '')
|
|
if len(bbacomm) == 12:
|
|
base = int(bbacomm[:10])
|
|
mod = base % 97 or 97
|
|
if mod == int(bbacomm[-2:]):
|
|
return True
|
|
|
|
@api.onchange('partner_id', 'type', 'reference_type')
|
|
def _onchange_partner_id(self):
|
|
result = super(AccountInvoice, self)._onchange_partner_id()
|
|
reference = False
|
|
reference_type = 'none'
|
|
if self.partner_id:
|
|
if (self.type == 'out_invoice'):
|
|
reference_type = self.partner_id.out_inv_comm_type
|
|
if reference_type:
|
|
reference = self.generate_bbacomm(self.type, reference_type, self.partner_id.id, '')['value']['reference']
|
|
self.reference_type = reference_type or 'none'
|
|
self.reference = reference
|
|
return result
|
|
|
|
def generate_bbacomm(self, type, reference_type, partner_id, reference):
|
|
reference = reference or ''
|
|
algorithm = False
|
|
if partner_id:
|
|
algorithm = self.env['res.partner'].browse(partner_id).out_inv_comm_algorithm
|
|
algorithm = algorithm or 'random'
|
|
if (type == 'out_invoice'):
|
|
if reference_type == 'bba':
|
|
if algorithm == 'date':
|
|
if not self.check_bbacomm(reference):
|
|
date = fields.Date.from_string(fields.Date.today())
|
|
doy = date.strftime('%j')
|
|
year = date.strftime('%Y')
|
|
seq = '001'
|
|
invoices = self.search([('type', '=', 'out_invoice'), ('reference_type', '=', 'bba'),
|
|
('reference', 'like', '+++%s/%s/%%' % (doy, year))], order='reference')
|
|
if invoices:
|
|
prev_seq = int(invoices[-1].reference[12:15])
|
|
if prev_seq < 999:
|
|
seq = '%03d' % (prev_seq + 1)
|
|
else:
|
|
raise UserError(_('The daily maximum of outgoing invoices with an automatically generated BBA Structured Communications has been exceeded!'
|
|
'\nPlease create manually a unique BBA Structured Communication.'))
|
|
bbacomm = doy + year + seq
|
|
base = int(bbacomm)
|
|
mod = base % 97 or 97
|
|
reference = '+++%s/%s/%s%02d+++' % (doy, year, seq, mod)
|
|
elif algorithm == 'partner_ref':
|
|
if not self.check_bbacomm(reference):
|
|
partner_ref = self.env['res.partner'].browse(partner_id).ref
|
|
partner_ref_nr = re.sub('\D', '', partner_ref or '')
|
|
if (len(partner_ref_nr) < 3) or (len(partner_ref_nr) > 7):
|
|
raise UserError(_('The Partner should have a 3-7 digit Reference Number for the generation of BBA Structured Communications!'
|
|
'\nPlease correct the Partner record.'))
|
|
else:
|
|
partner_ref_nr = partner_ref_nr.ljust(7, '0')
|
|
seq = '001'
|
|
invoices = self.search([('type', '=', 'out_invoice'), ('reference_type', '=', 'bba'),
|
|
('reference', 'like', '+++%s/%s/%%' % (partner_ref_nr[:3], partner_ref_nr[3:]))], order='reference')
|
|
if invoices:
|
|
prev_seq = int(invoices[-1].reference[12:15])
|
|
if prev_seq < 999:
|
|
seq = '%03d' % (prev_seq + 1)
|
|
else:
|
|
raise UserError(_('The daily maximum of outgoing invoices with an automatically generated BBA Structured Communications has been exceeded!'
|
|
'\nPlease create manually a unique BBA Structured Communication.'))
|
|
bbacomm = partner_ref_nr + seq
|
|
base = int(bbacomm)
|
|
mod = base % 97 or 97
|
|
reference = '+++%s/%s/%s%02d+++' % (partner_ref_nr[:3], partner_ref_nr[3:], seq, mod)
|
|
elif algorithm == 'random':
|
|
if not self.check_bbacomm(reference):
|
|
base = random.randint(1, 9999999999)
|
|
bbacomm = str(base).rjust(10, '0')
|
|
base = int(bbacomm)
|
|
mod = base % 97 or 97
|
|
mod = str(mod).rjust(2, '0')
|
|
reference = '+++%s/%s/%s%s+++' % (bbacomm[:3], bbacomm[3:7], bbacomm[7:], mod)
|
|
else:
|
|
raise UserError(_("Unsupported Structured Communication Type Algorithm '%s' !"
|
|
"\nPlease contact your Flectra support channel.") % algorithm)
|
|
return {'value': {'reference': reference}}
|
|
|
|
@api.model
|
|
def create(self, vals):
|
|
reference = vals.get('reference', False)
|
|
reference_type = vals.get('reference_type', False)
|
|
if vals.get('type') == 'out_invoice' and not reference_type:
|
|
# fallback on default communication type for partner
|
|
partner = self.env['res.partner'].browse(vals['partner_id'])
|
|
reference_type = partner.out_inv_comm_type
|
|
if reference_type == 'bba':
|
|
reference = self.generate_bbacomm(vals['type'], reference_type, partner.id, '')['value']['reference']
|
|
vals.update({
|
|
'reference_type': reference_type or 'none',
|
|
'reference': reference,
|
|
})
|
|
|
|
if reference_type == 'bba':
|
|
if not reference:
|
|
raise UserError(_('Empty BBA Structured Communication!'
|
|
'\nPlease fill in a unique BBA Structured Communication.'))
|
|
if self.check_bbacomm(reference):
|
|
reference = re.sub('\D', '', reference)
|
|
vals['reference'] = '+++' + reference[0:3] + '/' + reference[3:7] + '/' + reference[7:] + '+++'
|
|
same_ids = self.search([('type', '=', 'out_invoice'), ('reference_type', '=', 'bba'), ('reference', '=', vals['reference'])])
|
|
if same_ids:
|
|
raise UserError(_('The BBA Structured Communication has already been used!'
|
|
'\nPlease create manually a unique BBA Structured Communication.'))
|
|
return super(AccountInvoice, self).create(vals)
|
|
|
|
@api.multi
|
|
def write(self, vals):
|
|
for invoice in self:
|
|
if 'reference_type' in vals:
|
|
reference_type = vals['reference_type']
|
|
else:
|
|
reference_type = invoice.reference_type or ''
|
|
|
|
if reference_type == 'bba' and 'reference' in vals:
|
|
if self.check_bbacomm(vals['reference']):
|
|
reference = re.sub('\D', '', vals['reference'])
|
|
vals['reference'] = '+++' + reference[0:3] + '/' + reference[3:7] + '/' + reference[7:] + '+++'
|
|
same_ids = self.search([('id', '!=', invoice.id), ('type', '=', 'out_invoice'),
|
|
('reference_type', '=', 'bba'), ('reference', '=', vals['reference'])])
|
|
if same_ids:
|
|
raise UserError(_('The BBA Structured Communication has already been used!'
|
|
'\nPlease create manually a unique BBA Structured Communication.'))
|
|
return super(AccountInvoice, self).write(vals)
|
|
|
|
@api.multi
|
|
def copy(self, default=None):
|
|
self.ensure_one()
|
|
default = default or {}
|
|
if self.type in ['out_invoice']:
|
|
reference_type = self.reference_type or 'none'
|
|
default['reference_type'] = reference_type
|
|
if reference_type == 'bba':
|
|
default['reference'] = self.generate_bbacomm(self.type, reference_type, self.partner_id.id, '')['value']['reference']
|
|
return super(AccountInvoice, self).copy(default)
|