Add basic sanitization

OCA · Mar 4, 2023 · a45df7d · a45df7d
1 parent 70bd0ba
commit a45df7d
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/account_invoice_import_invoice2data/wizard/account_invoice_import.py b/account_invoice_import_invoice2data/wizard/account_invoice_import.py
@@ -5,6 +5,7 @@
 import datetime
 import logging
 import os
+import re
 import shutil
 from tempfile import NamedTemporaryFile
 
@@ -153,7 +154,7 @@ def invoice2data_to_parsed_inv(self, invoice2data_res):
 
         parsed_inv = {
             "partner": {
-                "vat": invoice2data_res.get("vat"),
+                "vat": re.sub(r"\W+", "", invoice2data_res.get("vat")).upper(),
                 "name": invoice2data_res.get("partner_name"),
                 "street": invoice2data_res.get("partner_street"),
                 "street2": invoice2data_res.get("partner_street2"),
@@ -164,10 +165,12 @@ def invoice2data_to_parsed_inv(self, invoice2data_res):
                 "website": invoice2data_res.get("partner_website"),
                 "phone": invoice2data_res.get("telephone"),
                 "siren": invoice2data_res.get("siren"),
-                "coc_registration_number": invoice2data_res.get("partner_coc"),
+                "coc_registration_number": re.sub(
+                    r"\D+", "", invoice2data_res.get("partner_coc", "")
+                ),
             },
-            "bic": invoice2data_res.get("bic"),
-            "iban": invoice2data_res.get("iban"),
+            "bic": re.sub(r"\W+", "", invoice2data_res.get("bic", "")).upper(),
+            "iban": re.sub(r"\W+", "", invoice2data_res.get("iban", "")).upper(),
             "currency": {
                 "iso": invoice2data_res.get("currency"),
             },