Validate extension from url

open-contracting-archive · Jul 2, 2020 · 8d5c226 · 8d5c226
1 parent a5363c0
commit 8d5c226
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/default/views.py b/default/views.py
@@ -262,7 +262,10 @@ def upload_url(request):
         if 'input_url' in data:
             url = request.POST.get(data)
             basename = data
-            extension = ".json"
+            if request.POST.get('type') == 'csv xlsx zip':
+                extension = os.path.splitext(urlparse(url).path)[1]
+            else:
+                extension = ".json"
             folder = 'media'
             data_file = DataFile(basename, extension)