From f3f51a92476e8244e5ee2fef07f84d9629686c0a Mon Sep 17 00:00:00 2001 From: Ben Webb Date: Fri, 24 Sep 2021 18:04:29 +0100 Subject: [PATCH] libcove.lib.converters: Add hook to change warning messages in a cove --- libcove/lib/converters.py | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) diff --git a/libcove/lib/converters.py b/libcove/lib/converters.py index df302d9..7efac02 100644 --- a/libcove/lib/converters.py +++ b/libcove/lib/converters.py @@ -12,16 +12,6 @@ logger = logging.getLogger(__name__) -def filter_conversion_warnings(conversion_warnings): - out = [] - for w in conversion_warnings: - if w.category is flattentool.exceptions.DataErrorWarning: - out.append(str(w.message)) - else: - logger.warning(w) - return out - - @cove_spreadsheet_conversion_error def convert_spreadsheet( upload_dir, @@ -36,7 +26,18 @@ def convert_spreadsheet( xml=False, xml_schemas=None, cache=True, + conversion_warning_hook=lambda w: str(w.message), ): + + def filter_conversion_warnings(conversion_warnings): + out = [] + for w in conversion_warnings: + if w.category is flattentool.exceptions.DataErrorWarning: + out.append(conversion_warning_hook(w)) + else: + logger.warning(w) + return out + context = {} if xml: output_file = "unflattened.xml"