From 6ec648bd1c6acc6e4cac6c17a76e64080cf8cec4 Mon Sep 17 00:00:00 2001
From: Hironsan <light.tree.1.13@gmail.com>
Date: Fri, 9 Apr 2021 18:06:43 +0900
Subject: [PATCH] Handle UnicodeDecodeError

---
 app/api/views/upload/dataset.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/app/api/views/upload/dataset.py b/app/api/views/upload/dataset.py
index 45870e3e..a8804f18 100644
--- a/app/api/views/upload/dataset.py
+++ b/app/api/views/upload/dataset.py
@@ -59,7 +59,11 @@ class Dataset:
 
     def __iter__(self) -> Iterator[Record]:
         for filename in self.filenames:
-            yield from self.load(filename)
+            try:
+                yield from self.load(filename)
+            except UnicodeDecodeError as err:
+                message = str(err)
+                raise FileParseException(filename, line_num=-1, message=message)
 
     def load(self, filename: str) -> Iterator[Record]:
         """Loads a file content."""