l10n_se_ocr: Adding support of OCR Number for Invoice and Vendor Bill. Invoice supports OCR level 1 to 4 on both Parter ref and Invoice. Vendor Bill supports default vendor OCR number and OCR checksum.

closes odoo/odoo#53132

X-original-commit: 739e7c0b11c0cda705a2e7db3674863436e346ea
Signed-off-by: Josse Colpaert <jco@openerp.com>

4 files changed, 117 insertions, 0 deletions

diff --git a/addons/l10n_se_ocr/models/__init__.py b/addons/l10n_se_ocr/models/__init__.py
new file mode 100644
index 00000000000..f78304565c8
--- /dev/null
+++ b/addons/l10n_se_ocr/models/__init__.py
@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from . import account_move
+from . import account_journal
+from . import res_partner
\ No newline at end of file
diff --git a/addons/l10n_se_ocr/models/account_journal.py b/addons/l10n_se_ocr/models/account_journal.py
new file mode 100644
index 00000000000..671ac4d2497
--- /dev/null
+++ b/addons/l10n_se_ocr/models/account_journal.py
@@ -0,0 +1,17 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models, api, _
+from odoo.exceptions import ValidationError
+
+
+class AccountJournal(models.Model):
+    _inherit = 'account.journal'
+
+    invoice_reference_model = fields.Selection(selection_add=[('se_ocr2', 'Sweden OCR Level 1 & 2'), ('se_ocr3', 'Sweden OCR Level 3'), ('se_ocr4', 'Sweden OCR Level 4')])
+    l10n_se_invoice_ocr_length = fields.Integer(string='OCR Number Length', help="Total length of OCR Reference Number including checksum.", default=6)
+
+    @api.constrains('l10n_se_invoice_ocr_length')
+    def _check_l10n_se_invoice_ocr_length(self):
+        if self.l10n_se_invoice_ocr_length < 6:
+            return ValidationError(_('OCR Reference Number length need to be greater than 5. Please correct settings under invoice journal settings.'))
diff --git a/addons/l10n_se_ocr/models/account_move.py b/addons/l10n_se_ocr/models/account_move.py
new file mode 100644
index 00000000000..4a2d94bb71d
--- /dev/null
+++ b/addons/l10n_se_ocr/models/account_move.py
@@ -0,0 +1,73 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import api, fields, models, _
+from odoo.exceptions import UserError
+from stdnum import luhn
+
+
+class AccountMove(models.Model):
+    _inherit = 'account.move'
+
+    def _get_invoice_reference_se_ocr2(self, reference):
+        self.ensure_one()
+        return reference + luhn.calc_check_digit(reference)
+
+    def _get_invoice_reference_se_ocr3(self, reference):
+        self.ensure_one()
+        reference = reference + str(len(reference) + 2)[:1]
+        return reference + luhn.calc_check_digit(reference)
+
+    def _get_invoice_reference_se_ocr4(self, reference):
+        self.ensure_one()
+
+        ocr_length = self.journal_id.l10n_se_invoice_ocr_length
+
+        if len(reference) + 1 > ocr_length:
+            raise UserError(_("OCR Reference Number length is greater than allowed. Allowed length in invoice journal setting is %s.") % str(ocr_length))
+
+        reference = reference.rjust(ocr_length - 1, '0')
+        return reference + luhn.calc_check_digit(reference)
+
+
+    def _get_invoice_reference_se_ocr2_invoice(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr2(str(self.id))
+
+    def _get_invoice_reference_se_ocr3_invoice(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr3(str(self.id))
+
+    def _get_invoice_reference_se_ocr4_invoice(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr4(str(self.id))
+
+    def _get_invoice_reference_se_ocr2_partner(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr2(self.partner_id.ref if str(self.partner_id.ref).isdecimal() else str(self.partner_id.id))
+
+    def _get_invoice_reference_se_ocr3_partner(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr3(self.partner_id.ref if str(self.partner_id.ref).isdecimal() else str(self.partner_id.id))
+
+    def _get_invoice_reference_se_ocr4_partner(self):
+        self.ensure_one()
+        return self._get_invoice_reference_se_ocr4(self.partner_id.ref if str(self.partner_id.ref).isdecimal() else str(self.partner_id.id))
+
+    @api.onchange('partner_id')
+    def _onchange_partner_id(self):
+        """ If Vendor Bill and Vendor OCR is set, add it. """
+        if self.partner_id and self.type == 'in_invoice' and self.partner_id.l10n_se_default_vendor_payment_ref:
+            self.invoice_payment_ref = self.partner_id.l10n_se_default_vendor_payment_ref
+        return super(AccountMove, self)._onchange_partner_id()
+
+    @api.onchange('invoice_payment_ref')
+    def _onchange_invoice_payment_ref(self):
+        """ If Vendor Bill and Payment Reference is changed check validation. """
+        if self.partner_id and self.type == 'in_invoice' and self.partner_id.l10n_se_check_vendor_ocr:
+            reference = self.invoice_payment_ref
+            try:
+                luhn.validate(reference)
+            except: 
+                return {'warning': {'title': _('Warning'), 'message': _('Vendor require OCR Number as payment reference. Payment reference isn\'t a valid OCR Number.')}}
+        return super(AccountMove, self)._onchange_invoice_payment_ref()
diff --git a/addons/l10n_se_ocr/models/res_partner.py b/addons/l10n_se_ocr/models/res_partner.py
new file mode 100644
index 00000000000..e42b3f1e4bd
--- /dev/null
+++ b/addons/l10n_se_ocr/models/res_partner.py
@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# Part of Odoo. See LICENSE file for full copyright and licensing details.
+
+from odoo import fields, models, api, _
+from stdnum import luhn
+
+
+class ResPartner(models.Model):
+    _inherit = 'res.partner'
+
+    l10n_se_check_vendor_ocr = fields.Boolean(string='Check Vendor OCR', help='This Vendor uses OCR Number on their Vendor Bills.')
+    l10n_se_default_vendor_payment_ref = fields.Char(string='Default Vendor Payment Ref', help='If set, the vendor uses the same Default Payment Reference or OCR Number on all their Vendor Bills.')
+
+    @api.onchange('l10n_se_default_vendor_payment_ref')
+    def onchange_l10n_se_default_vendor_payment_ref(self):
+        if not self.l10n_se_default_vendor_payment_ref == "" and self.l10n_se_check_vendor_ocr:
+            reference = self.l10n_se_default_vendor_payment_ref
+            try:
+                luhn.validate(reference)
+            except: 
+                return {'warning': {'title': _('Warning'), 'message': _('Default vendor OCR number isn\'t a valid OCR number.')}}