server-ux/sequence_check_digit/models/ir_sequence.py

70 lines
2.3 KiB
Python
Raw Normal View History

2021-05-26 11:33:18 +00:00
# Copyright (C) 2017 Creu Blanca
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl.html).
import logging
from flectra import _, api, fields, models
from flectra.exceptions import ValidationError
_logger = logging.getLogger(__name__)
try:
from stdnum import damm, luhn, verhoeff
from stdnum.iso7064 import mod_11_2, mod_11_10, mod_37_2, mod_37_36, mod_97_10
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 "{}{}".format(code, self.get_check_digit(code))