diff options
Diffstat (limited to 'scipy/views.py')
| -rw-r--r-- | scipy/views.py | 24 |
1 files changed, 19 insertions, 5 deletions
diff --git a/scipy/views.py b/scipy/views.py index 25932ff..f77c467 100644 --- a/scipy/views.py +++ b/scipy/views.py @@ -49,15 +49,19 @@ def user_logout(request): # User Register View def user_register(request): + context = {} if request.user.is_anonymous(): if request.method == 'POST': form = UserRegisterForm(request.POST) - if form.is_valid: + if form.is_valid(): form.save() - return HttpResponseRedirect('/2013/accounts/upload-document') + return HttpResponseRedirect('/2013/accounts/upload-document') + else: + context.update(csrf(request)) + context['form'] = form + return render_to_response('register.html', context) else: form = UserRegisterForm() - context = {} context.update(csrf(request)) context['form'] = form return render_to_response('register.html', context) @@ -76,9 +80,20 @@ def user_profile(request): # Document Upload View def upload_document(request): + allowed_files = ['doc', 'docx', 'txt', 'pdf'] + invalid_file_msg = None + context = {} if request.user.is_authenticated(): if request.method == 'POST': form = DocumentUploadForm(request.POST, request.FILES) + content_type = request.FILES['attachments'].content_type.split('/')[1] + if not content_type in allowed_files: + invalid_file_msg = "Only PDF, DOC, DOCX & TXT files are allowed" + context['current_user'] = request.user + context['invalid_file'] = invalid_file_msg + context.update(csrf(request)) + context['form'] = form + return render_to_response('upload-document.html', context) if form.is_valid(): data = form.save(commit=False) data.user = request.user @@ -93,11 +108,10 @@ def upload_document(request): return render_to_response('upload-document.html', context) else: form = DocumentUploadForm() - - context = {} context.update(csrf(request)) context['form'] = DocumentUploadForm() context['current_user'] = request.user + context['invalid_file'] = invalid_file_msg return render_to_response('upload-document.html', context) else: return HttpResponseRedirect('/2013/accounts/login') |