diff options
Diffstat (limited to 'mediagoblin/submit/views.py')
-rw-r--r-- | mediagoblin/submit/views.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/mediagoblin/submit/views.py b/mediagoblin/submit/views.py index d4ecc75a..e9b5c37e 100644 --- a/mediagoblin/submit/views.py +++ b/mediagoblin/submit/views.py @@ -21,7 +21,7 @@ from werkzeug.utils import secure_filename from mediagoblin.util import render_to_response, redirect from mediagoblin.decorators import require_active_login -from mediagoblin.submit import forms as submit_forms +from mediagoblin.submit import forms as submit_forms, security from mediagoblin.process_media import process_media_initial @@ -38,6 +38,9 @@ def submit_start(request): and request.POST['file'].file): submit_form.file.errors.append( u'You must provide a file.') + elif not security.check_filetype(request.POST['file']): + submit_form.file.errors.append( + u'The file doesn\'t seem to be an image!') else: filename = request.POST['file'].filename @@ -77,11 +80,12 @@ def submit_start(request): return redirect(request, "mediagoblin.submit.success") - return render_to_response(request, + return render_to_response( + request, 'mediagoblin/submit/start.html', {'submit_form': submit_form}) def submit_success(request): - return render_to_response(request, - 'mediagoblin/submit/success.html', {}) + return render_to_response( + request, 'mediagoblin/submit/success.html', {}) |