contact/views.py

   1 from urllib import unquote
   2 from django.contrib.auth.decorators import permission_required
   3 from django.http import Http404, HttpResponse
   4 from django.shortcuts import get_object_or_404, redirect, render
   5 from django.utils.translation import ugettext_lazy as _
   6 from fnpdjango.utils.views import serve_file
   7 from .forms import contact_forms
   8 from .models import Attachment
   9
  10
  11 def form(request, form_tag):
  12     try:
  13         form_class = contact_forms[form_tag]
  14     except KeyError:
  15         raise Http404
  16     if request.method == 'POST':
  17         form = form_class(request.POST, request.FILES)
  18         if form.is_valid():
  19             form.save(request)
  20             return redirect('contact_thanks', form_tag)
  21     else:
  22         form = form_class()
  23     return render(request,
  24                 ['contact/%s/form.html' % form_tag, 'contact/form.html'],
  25                 {'form': form}
  26             )
  27
  28
  29 def thanks(request, form_tag):
  30     if form_tag not in contact_forms:
  31         raise Http404
  32
  33     return render(request,
  34                 ['contact/%s/thanks.html' % form_tag, 'contact/thanks.html']
  35             )
  36
  37
  38 @permission_required('contact.change_attachment')
  39 def attachment(request, contact_id, tag):
  40     attachment = get_object_or_404(Attachment, contact_id=contact_id, tag=tag)
  41     return serve_file(unquote(attachment.file.url))