From b6f5261620d6518a0e5a9da1e638ab97be716609 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Thu, 11 Oct 2018 16:33:12 +0200 Subject: Import: catch error in the CSV file --- ishtar_common/models_imports.py | 18 ++++++++++++++---- 1 file changed, 14 insertions(+), 4 deletions(-) (limited to 'ishtar_common/models_imports.py') diff --git a/ishtar_common/models_imports.py b/ishtar_common/models_imports.py index d1ffb3656..2dec39c84 100644 --- a/ishtar_common/models_imports.py +++ b/ishtar_common/models_imports.py @@ -44,7 +44,8 @@ from ishtar_common.utils import create_slug, \ put_session_var, get_session_var, num2col from ishtar_common.data_importer import Importer, ImportFormater, \ IntegerFormater, FloatFormater, UnicodeFormater, DateFormater, \ - TypeFormater, YearFormater, StrToBoolean, FileFormater, InseeFormater + TypeFormater, YearFormater, StrToBoolean, FileFormater, InseeFormater, \ + ImporterError if settings.USE_BACKGROUND_TASK: from background_task import background @@ -1028,16 +1029,25 @@ class Import(models.Model): return vals except UnicodeDecodeError: pass # try the next encoding + except unicodecsv.Error: + raise ImporterError(_(u"Error in the CSV file.")) if tmpdir: shutil.rmtree(tmpdir) return [] - def initialize(self, user=None): + def initialize(self, user=None, session_key=None): self.state = 'AP' self.end_date = datetime.datetime.now() self.save() - self.get_importer_instance().initialize(self.data_table, - user=user, output='db') + try: + self.get_importer_instance().initialize( + self.data_table, user=user, output='db') + except ImporterError as e: + if session_key: + put_session_message(session_key, e.msg, "danger") + self.state = 'C' + self.save() + return self.state = 'A' self.end_date = datetime.datetime.now() self.save() -- cgit v1.2.3