event_rsvp.models << index

event_rsvp.views: 121 total statements, 100.0% covered

Generated: Tue 2013-04-16 10:12 CEST

Source file: /home/tobi/Projects/event-rsvp/src/event_rsvp/views.py

Stats: 113 executed, 0 missed, 8 excluded, 106 ignored

  1. """Views for the ``event_rsvp`` app."""
  2. from django.contrib.auth.decorators import login_required
  3. from django.core.urlresolvers import reverse
  4. from django.http import Http404, HttpResponseRedirect
  5. from django.utils import timezone
  6. from django.utils.decorators import method_decorator
  8. from django.views.generic import (
  9. CreateView,
  10. DeleteView,
  11. DetailView,
  12. ListView,
  13. UpdateView,
  14. )
  16. from event_rsvp.forms import EventForm, GuestForm
  17. from event_rsvp.models import Event, Guest
  20. #--------#
  21. # Mixins #
  22. #--------#
  24. class StaffMixin(object):
  25. """Mixin to let only staff member pass."""
  27. @method_decorator(login_required)
  28. def dispatch(self, request, *args, **kwargs):
  29. if not request.user.is_staff:
  30. raise Http404
  31. return super(StaffMixin, self).dispatch(request, *args, **kwargs)
  34. class EventViewMixin(object):
  35. """Mixin to handle event-specific options."""
  36. model = Event
  37. form_class = EventForm
  39. def get_form_kwargs(self):
  40. kwargs = super(EventViewMixin, self).get_form_kwargs()
  41. kwargs.update({'created_by': self.request.user})
  42. return kwargs
  44. def get_success_url(self):
  45. return reverse('rsvp_event_staff')
  48. class EventSecurityMixin(object):
  49. """Mixin to handle event-specific security options."""
  51. def dispatch(self, request, *args, **kwargs):
  52. self.kwargs = kwargs
  53. self.object = self.get_object()
  54. date = self.object.start
  55. # Check the right starting date within the slug
  56. if (date.year != int(kwargs.get('year'))
  57. or date.month != int(kwargs.get('month'))
  58. or date.day != int(kwargs.get('day'))):
  59. redirect_url = getattr(self.object, 'get_{0}_url'.format(
  60. self.url_mode))
  61. return HttpResponseRedirect(redirect_url())
  62. return super(EventSecurityMixin, self).dispatch(request, *args,
  63. **kwargs)
  66. class GuestViewMixin(object):
  67. """Mixin to handle guest-specific functions."""
  68. model = Guest
  69. form_class = GuestForm
  71. def dispatch(self, request, *args, **kwargs):
  72. try:
  73. self.event = Event.objects.get(slug=kwargs.get('event_slug'))
  74. except Event.DoesNotExist:
  75. raise Http404
  76. return super(GuestViewMixin, self).dispatch(request, *args, **kwargs)
  78. def get_context_data(self, **kwargs):
  79. context = super(GuestViewMixin, self).get_context_data(**kwargs)
  80. context.update({'event': self.event, 'user': self.request.user})
  81. if (self.request.user.is_authenticated()
  82. or self.event.allow_anonymous_rsvp):
  83. context.update({'permission_to_book': True})
  84. return context
  86. def get_form_kwargs(self):
  87. kwargs = super(GuestViewMixin, self).get_form_kwargs()
  88. kwargs.update({'event': self.event, 'user': self.request.user})
  89. return kwargs
  91. def get_success_url(self):
  92. return self.event.get_absolute_url()
  95. class GuestSecurityMixin(object):
  96. """Mixin to handle guest-specific security options."""
  97. def get_object(self, *args, **kwargs):
  98. obj = super(GuestSecurityMixin, self).get_object(*args, **kwargs)
  99. if obj.event != self.event:
  100. raise Http404
  101. return obj
  104. #--------#
  105. # Views #
  106. #--------#
  108. class EventListView(ListView):
  109. """List view to display upcoming events."""
  110. def get_queryset(self):
  111. return Event.objects.filter(start__gt=timezone.now(),
  112. is_published=True)
  114. def get_context_data(self, **kwargs):
  115. context = super(EventListView, self).get_context_data(**kwargs)
  116. if self.request.user.is_authenticated():
  117. context.update({
  118. 'my_participations': self.request.user.guest_set.all()})
  119. return context
  122. class EventDetailView(EventSecurityMixin, EventViewMixin, DetailView):
  123. """Detail view to display information of an event."""
  124. url_mode = 'absolute'
  126. def dispatch(self, request, *args, **kwargs):
  127. self.kwargs = kwargs
  128. self.object = self.get_object()
  129. if not self.object.is_published and not request.user.is_staff:
  130. raise Http404
  131. return super(EventDetailView, self).dispatch(request, *args, **kwargs)
  134. class EventCreateView(StaffMixin, EventViewMixin, CreateView):
  135. """Create view to handle information of an event."""
  136. pass
  139. class EventUpdateView(StaffMixin, EventSecurityMixin, EventViewMixin,
  140. UpdateView):
  141. """Update view to handle information of an event."""
  142. url_mode = 'update'
  145. class EventDeleteView(StaffMixin, EventSecurityMixin, EventViewMixin,
  146. DeleteView):
  147. """Delete view to remove the relevant event."""
  148. url_mode = 'delete'
  151. class EventCreateFromTemplateView(StaffMixin, EventViewMixin, CreateView):
  152. """Create view to create information of an event from a template."""
  153. @method_decorator(login_required)
  154. def dispatch(self, request, *args, **kwargs):
  155. try:
  156. # Check if it's really a template
  157. self.template = Event.objects.get(pk=kwargs.get('pk'),
  158. template_name__gt='')
  159. except Event.DoesNotExist:
  160. raise Http404
  161. return super(EventCreateFromTemplateView, self).dispatch(
  162. request, *args, **kwargs)
  164. def get_form_kwargs(self):
  165. kwargs = super(EventCreateFromTemplateView, self).get_form_kwargs()
  166. kwargs.update({'instance': self.template,
  167. 'create_from_template': True})
  168. return kwargs
  171. class StaffDashboardView(StaffMixin, ListView):
  172. """View to display event related functions and lists."""
  173. model = Event
  174. template_name = 'event_rsvp/staff_dashboard.html'
  176. def get_context_data(self, **kwargs):
  177. context = super(StaffDashboardView, self).get_context_data(**kwargs)
  178. templates = self.object_list.exclude(template_name__exact='')
  179. self.object_list = self.object_list.filter(template_name__exact='')
  180. context.update({
  181. 'upcoming': self.object_list.filter(start__gt=timezone.now()),
  182. 'current': self.object_list.filter(start__lte=timezone.now(),
  183. end__gte=timezone.now()),
  184. 'past': self.object_list.filter(end__lt=timezone.now()),
  185. 'templates': templates,
  186. })
  187. return context
  190. class GuestDetailView(StaffMixin, GuestSecurityMixin, GuestViewMixin,
  191. DetailView):
  192. """View to display guest related functions and lists."""
  193. pass
  196. class GuestCreateView(GuestViewMixin, CreateView):
  197. """Create view to add a guest to an event."""
  198. def get_form_kwargs(self):
  199. kwargs = super(GuestCreateView, self).get_form_kwargs()
  200. if self.request.user.is_authenticated():
  201. kwargs.update({'initial': {
  202. 'name': self.request.user.get_full_name(),
  203. 'email': self.request.user.email}})
  204. return kwargs
  207. class GuestUpdateView(GuestSecurityMixin, GuestViewMixin, UpdateView):
  208. """Update view to handle a guest."""
  210. @method_decorator(login_required)
  211. def dispatch(self, request, *args, **kwargs):
  212. try:
  213. self.event = Event.objects.get(slug=kwargs.get('event_slug'))
  214. except Event.DoesNotExist:
  215. raise Http404
  216. self.kwargs = kwargs
  217. self.object = self.get_object()
  218. if (not request.user.is_staff and not self.object.user
  219. and not self.object.user == request.user):
  220. raise Http404
  221. return super(GuestViewMixin, self).dispatch(request, *args, **kwargs)
  224. class GuestDeleteView(StaffMixin, GuestViewMixin, GuestSecurityMixin,
  225. DeleteView):
  226. """Delete view to remove the relevant guest."""
  227. pass
event_rsvp.models << index