From a45df7dfc629149722efc9986c7ea3029fba4cab Mon Sep 17 00:00:00 2001 From: bosd Date: Sat, 4 Mar 2023 15:47:52 +0100 Subject: [PATCH] Add basic sanitization --- .../wizard/account_invoice_import.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/account_invoice_import_invoice2data/wizard/account_invoice_import.py b/account_invoice_import_invoice2data/wizard/account_invoice_import.py index aa008626c39..41b635235d2 100644 --- 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"), },