Show message when a uploaded file is not a .poll_result

1 files changed, 14 insertions, 8 deletions

diff --git a/webapp/polls/views.py b/webapp/polls/views.py
index e195242..6ca16ee 100644
--- a/webapp/polls/views.py
+++ b/webapp/polls/views.py
@@ -360,17 +360,23 @@ class UnploadPollResultFormView(TemplateView):
 
         files = dict(request.FILES.lists()).get('result', [])
 
-        # si se subió algún archivo:
-        if len(files):
-            # Esto hace todo
+        to_json = []
 
-            to_json = []
-
-            # genera una lista de dict python en to_json
-            for file_ in files:
-                path = file_.temporary_file_path()
+        # genera una lista de dict python en to_json
+        for index, file_ in enumerate(files):
+            path = file_.temporary_file_path()
+            try:
                 prf = PollResultFile(path)
                 to_json.append(prf.get_data())
+            except ValueError:
+                del files[index]
+                fname = file_.name
+                msg = u'{0}: No es un .poll_result válido.'.format(fname)
+                messages.add_message(self.request, messages.ERROR, msg)
+
+        # si se subió algún archivo:
+        if len(files):
+            # Esto hace todo
 
             # comprueba si los archivos ya existían y los excluye
             poll_id = to_json[0].get("poll_id", None)