mirror of
https://gitlab.com/flectra-community/server-ux.git
synced 2024-11-22 14:12:07 +00:00
70 lines
2.3 KiB
Python
70 lines
2.3 KiB
Python
|
# 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))
|