facturx: check unece codes at tax

edocument.py

@@ -5,50 +5,12 @@
 
 import genshi.template
 import os
-import html
-from trytond.exceptions import UserError
-from trytond.i18n import gettext
-from trytond.modules.edocument_uncefact.edocument import Invoice
 from decimal import Decimal
+from trytond.modules.edocument_uncefact.edocument import Invoice
+from .mixin import EdocumentMixin
 
 
-class FacturX(Invoice):
-    'Factur-X'
-    __name__ = 'edocument.facturxext.invoice'
-
-    def get_list_of_comments(self):
-        """ comment, to export in <ram:IncludedNote/>
-
-        Returns:
-            _type_: _description_
-        """
-        result = []
-        if self.invoice.comment:
-            result.append({
-                'content': self.invoice.comment,
-                'subject_code': '',
-                'content_code': ''})
-        return result
-
-    def _get_template(self, version):
-        """ load our own template if 'version' is ours
-        """
-        loader = genshi.template.TemplateLoader(
-            os.path.join(os.path.dirname(__file__), 'template'),
-            auto_reload=True)
-
-        if version == 'Factur-X-1.07.2-extended':
-            if self.type_code in ['380', '389', '381', '261']:
-                return loader.load(os.path.join(version, 'invoice.xml'))
-            else:
-                raise ValueError('invalid type-code "%s"' % self.type_code)
-        else:
-            return super(Invoice, self)._get_template(version)
-
-# end FacturX
-
-
-class Invoice(Invoice):
+class XRechnung(EdocumentMixin, Invoice):
     'EDocument XRechnung'
     __name__ = 'edocument.xrechnung.invoice'
 
@@ -88,74 +50,6 @@ class Invoice(Invoice):
         if notes:
             return '; '.join(notes)
 
-    def invoice_line_tax(self, line):
-        """ get tax of invoice-line,
-            fire exception if no/multiple taxes exists
-        """
-        if len(line.invoice_taxes) != 1:
-            raise UserError(gettext(
-                'edocument_xrechnung.msg_linetax_invalid_number',
-                linename=line.rec_name,
-                numtax=len(line.invoice_taxes)))
-
-        allowed_cat = ['AE', 'L', 'M', 'E', 'S', 'Z', 'G', 'O', 'K', 'B']
-        unece_category_code = self.get_category_code(line.invoice_taxes[0].tax)
-        if unece_category_code not in allowed_cat:
-            raise UserError(gettext(
-                'edocument_xrechnung.msg_linetax_invalid_catcode',
-                taxname=line.invoice_taxes[0].tax.rec_name,
-                allowed=', '.join(allowed_cat)))
-
-        return line.invoice_taxes[0].tax
-
-    def taxident_data(self, tax_identifier):
-        """ get tax-scheme-id and codes
-        """
-        result = {'code': None, 'id': None}
-
-        if tax_identifier:
-            if tax_identifier.type == 'de_vat':
-                result['code'] = 'DE%s' % tax_identifier.code
-                result['id'] = 'VAT'
-        return result
-
-    def tax_rate(self, tax):
-        """ get tax-rate in procent
-        """
-        return (tax.rate * Decimal('100.0')).quantize(Decimal('0.01'))
-
-    def uom_unece_code(self, line):
-        """ 'line': invoice.line
-        """
-        if len(line.unit.unece_code or '') == 0:
-            raise UserError(gettext(
-                'edocument_xrechnung.msg_uom_code_missing',
-                uomname=line.unit.rec_name))
-        return line.unit.unece_code
-
-    def get_category_code(self, tax):
-        while tax:
-            if tax.unece_category_code:
-                return tax.unece_category_code
-                break
-            tax = tax.parent
-
-    def tax_category_code(self, tax):
-        """ read tax-category, fire exception if missing
-        """
-        unece_category_code = self.get_category_code(tax)
-        if not unece_category_code:
-            raise UserError(gettext(
-                'edocument_xrechnung.mds_tax_category_missing',
-                taxname=tax.rec_name))
-        return unece_category_code
-
-    def quote_text(self, text):
-        """ replace critical chars
-        """
-        if text:
-            return html.quote(text)
-
     def _get_template(self, version):
         """ load our own template if 'version' is ours
         """
@@ -176,6 +70,28 @@ class Invoice(Invoice):
             else:
                 raise ValueError('invalid type-code "%s"' % self.type_code)
         else:
-            return super(Invoice, self)._get_template(version)
+            return super(XRechnung, self)._get_template(version)
 
-# end Invoice
+# end XRechnung
+
+
+class FacturX(EdocumentMixin, Invoice):
+    'Factur-X'
+    __name__ = 'edocument.facturxext.invoice'
+
+    def _get_template(self, version):
+        """ load our own template if 'version' is ours
+        """
+        loader = genshi.template.TemplateLoader(
+            os.path.join(os.path.dirname(__file__), 'template'),
+            auto_reload=True)
+
+        if version == 'Factur-X-1.07.2-extended':
+            if self.type_code in ['380', '389', '381', '261']:
+                return loader.load(os.path.join(version, 'invoice.xml'))
+            else:
+                raise ValueError('invalid type-code "%s"' % self.type_code)
+        else:
+            return super(FacturX, self)._get_template(version)
+
+# end FacturX