[utils] Add error handling in csvfile validation

Fixes #251 Remove extra whitespace
openwisp · Nov 19, 2019 · 782122a · 782122a
1 parent 3b745b5
commit 782122a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/django_freeradius/utils.py b/django_freeradius/utils.py
@@ -29,7 +29,10 @@ def find_available_username(username, users_list, prefix=False):
 
 def validate_csvfile(csvfile):
     csv_data = csvfile.read()
-    csv_data = csv_data.decode('utf-8') if isinstance(csv_data, bytes) else csv_data
+    try:
+        csv_data = csv_data.decode('utf-8') if isinstance(csv_data, bytes) else csv_data
+    except UnicodeDecodeError:
+        raise ValidationError("Incorrect file format has been uploaded. Please Try Again!")
     reader = csv.reader(StringIO(csv_data), delimiter=',')
     error_message = "The CSV contains a line with invalid data,\
                     line number {} triggered the following error: {}"