X-Git-Url: https://git.mdrn.pl/edumed.git/blobdiff_plain/d12ee4e0ec425e7f508c93ad1295a21a67ae1a30..836e87743276449d019848fb53b520b4c4851908:/contact/views.py?ds=inline diff --git a/contact/views.py b/contact/views.py index d18598c..5a2f2ae 100644 --- a/contact/views.py +++ b/contact/views.py @@ -1,40 +1,62 @@ +# -*- coding: utf-8 -*- +from urllib import unquote + from django.contrib.auth.decorators import permission_required -from django.http import Http404, HttpResponse +from django.http import Http404 from django.shortcuts import get_object_or_404, redirect, render -from django.utils.translation import ugettext_lazy as _ from fnpdjango.utils.views import serve_file +from honeypot.decorators import check_honeypot + from .forms import contact_forms from .models import Attachment -def form(request, form_tag): +@check_honeypot +def form(request, form_tag, force_enabled=False): try: form_class = contact_forms[form_tag] except KeyError: raise Http404 + if (getattr(form_class, 'disabled', False) and + not (force_enabled and request.user.is_superuser)): + template = getattr(form_class, 'disabled_template', None) + if template: + return render(request, template, {'title': form_class.form_title}) + raise Http404 if request.method == 'POST': form = form_class(request.POST, request.FILES) - if form.is_valid(): - form.save(request) - return redirect('contact_thanks', form_tag) else: form = form_class(initial=request.GET) - return render(request, - ['contact/%s/form.html' % form_tag, 'contact/form.html'], - {'form': form} - ) + formset_classes = getattr(form, 'form_formsets', {}) + if request.method == 'POST': + formsets = { + prefix: formset_class(request.POST, request.FILES, prefix=prefix) + for prefix, formset_class in formset_classes.iteritems()} + if form.is_valid() and all(formset.is_valid() for formset in formsets.itervalues()): + form.save(request, formsets.values()) + return redirect('contact_thanks', form_tag) + else: + formsets = {prefix: formset_class(prefix=prefix) for prefix, formset_class in formset_classes.iteritems()} + + return render( + request, ['contact/%s/form.html' % form_tag, 'contact/form.html'], + {'form': form, 'formsets': formsets} + ) def thanks(request, form_tag): - if form_tag not in contact_forms: + try: + form_class = contact_forms[form_tag] + except KeyError: raise Http404 - return render(request, - ['contact/%s/thanks.html' % form_tag, 'contact/thanks.html'] - ) + return render( + request, ['contact/%s/thanks.html' % form_tag, 'contact/thanks.html'], + {'base_template': getattr(form_class, 'base_template', None)}) @permission_required('contact.change_attachment') def attachment(request, contact_id, tag): attachment = get_object_or_404(Attachment, contact_id=contact_id, tag=tag) - return serve_file(attachment.file.url) + attachment_url = unquote(attachment.file.url) + return serve_file(attachment_url)