mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-11-15 02:32:06 +00:00
68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
# Copyright (C) 2017 Creu Blanca
|
|
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
|
|
|
|
from flectra import fields, models, api, _
|
|
from flectra.exceptions import ValidationError
|
|
import logging
|
|
_logger = logging.getLogger(__name__)
|
|
try:
|
|
from stdnum.iso7064 import mod_97_10
|
|
from stdnum.iso7064 import mod_37_2, mod_37_36
|
|
from stdnum.iso7064 import mod_11_2, mod_11_10
|
|
from stdnum import luhn, damm, verhoeff
|
|
except(ImportError, IOError) as err:
|
|
_logger.debug(err)
|
|
|
|
|
|
class IrSequence(models.Model):
|
|
_inherit = "ir.sequence"
|
|
|
|
check_digit_formula = fields.Selection(
|
|
selection=[
|
|
('none', 'None'),
|
|
('luhn', 'Luhn'),
|
|
('damm', 'Damm'),
|
|
('verhoeff', 'Verhoeff'),
|
|
('ISO7064_11_2', 'ISO 7064 Mod 11, 2'),
|
|
('ISO7064_11_10', 'ISO 7064 Mod 11, 10'),
|
|
('ISO7064_37_2', 'ISO 7064 Mod 37, 2'),
|
|
('ISO7064_37_36', 'ISO 7064 Mod 37, 36'),
|
|
('ISO7064_97_10', 'ISO 7064 Mod 97, 10'),
|
|
], default='none'
|
|
)
|
|
|
|
@api.constrains('check_digit_formula', 'prefix', 'suffix')
|
|
def check_check_digit_formula(self):
|
|
try:
|
|
self.get_next_char(0)
|
|
except Exception:
|
|
raise ValidationError(_('Format is not accepted'))
|
|
|
|
def get_check_digit(self, code):
|
|
try:
|
|
return self.get_formula_map()[self.check_digit_formula](code)
|
|
except KeyError:
|
|
raise ValidationError(_('%s is not an implemented function'
|
|
% self.check_digit_formula))
|
|
except Exception:
|
|
raise ValidationError(_('Format is not accepted'))
|
|
|
|
def get_formula_map(self):
|
|
return {
|
|
'none': lambda _: '',
|
|
'luhn': luhn.calc_check_digit,
|
|
'damm': damm.calc_check_digit,
|
|
'verhoeff': verhoeff.calc_check_digit,
|
|
'ISO7064_11_2': mod_11_2.calc_check_digit,
|
|
'ISO7064_11_10': mod_11_10.calc_check_digit,
|
|
'ISO7064_37_2': mod_37_2.calc_check_digit,
|
|
'ISO7064_37_36': mod_37_36.calc_check_digit,
|
|
'ISO7064_97_10': mod_97_10.calc_check_digits
|
|
}
|
|
|
|
def get_next_char(self, number_next):
|
|
code = super(IrSequence, self).get_next_char(number_next)
|
|
if not self.check_digit_formula:
|
|
return code
|
|
return '%s%s' % (code, self.get_check_digit(code))
|