Source code for kawaz.apps.events.views

import io
from django.views.generic.detail import DetailView
from django.views.generic.list import ListView, MultipleObjectMixin
from django.views.generic.edit import CreateView, UpdateView, DeleteView
from django.views.generic.dates import YearArchiveView, MonthArchiveView, BaseArchiveIndexView
from django.contrib import messages
from django.contrib.messages.views import SuccessMessageMixin
from django.http.response import HttpResponseNotFound
from django.core.urlresolvers import reverse_lazy
from django.core.exceptions import PermissionDenied
from django.http.response import HttpResponseRedirect, HttpResponseForbidden, HttpResponseNotAllowed
from django_filters.views import FilterView
from django.utils.translation import ugettext as _
from .filters import EventFilter
from django.http.response import StreamingHttpResponse

from wsgiref.util import FileWrapper

from permission.decorators import permission_required
from kawaz.core.views.delete import DeleteSuccessMessageMixin

from kawaz.core.views.preview import SingleObjectPreviewViewMixin

from .models import Event
from .forms import EventForm, EventUpdateForm, EventCreationForm
from .utils.ical import generate_ical

class EventPublishedQuerySetMixin(MultipleObjectMixin):
    def get_queryset(self):
        return Event.objects.published(self.request.user)


class EventActiveQuerySetMixin(MultipleObjectMixin):
    def get_queryset(self):
        return Event.objects.active(self.request.user)


class EventDateArchiveMixin(BaseArchiveIndexView):
    def get(self, request, *args, **kwargs):
        self.date_list, self.object_list, extra_context = self.get_dated_items()
        self.object_list = self.object_list.order_by('period_start') # 日付昇順にならない
        context = self.get_context_data(object_list=self.object_list,
                                        date_list=self.date_list)
        context.update(extra_context)
        return self.render_to_response(context)


class EventListView(FilterView, EventActiveQuerySetMixin):
    model = Event
    filterset_class = EventFilter
    template_name_suffix = '_list'


@permission_required('events.view_event')
class EventDetailView(DetailView):
    model = Event


@permission_required('events.add_event')
class EventCreateView(SuccessMessageMixin, CreateView):
    model = Event
    form_class = EventCreationForm

    def get_success_message(self, cleaned_data):
        return _("""A event '%(title)s' was successfully created.""") % {
            'title': cleaned_data['title']
        }


    def form_valid(self, form):
        form.instance.organizer = self.request.user
        return super().form_valid(form)

@permission_required('events.change_event')
class EventUpdateView(SuccessMessageMixin, UpdateView):
    model = Event
    form_class = EventUpdateForm

    def get_success_message(self, cleaned_data):
        return _("""The event '%(title)s' was successfully updated.""") % {
            'title': cleaned_data['title']
        }


@permission_required('events.delete_event')
class EventDeleteView(DeleteSuccessMessageMixin, DeleteView):
    model = Event
    success_url = reverse_lazy('events_event_list')

    def get_success_message(self):
        return _("The event was successfully deleted.")

@permission_required('events.attend_event')
class EventAttendView(UpdateView):
    model = Event

    def get_success_url(self):
        return self.object.get_absolute_url()

    def attend(self, request, *args, **kwargs):
        """
        Calls the attend() method on the fetched object and then
        redirects to the success URL.
        """
        self.object = self.get_object()
        success_url = self.get_success_url()
        messages.add_message(request, messages.SUCCESS,
                             _('You have just attended the event.'))
        try:
            self.object.attend(request.user)
            return HttpResponseRedirect(success_url)
        except PermissionDenied:
            return HttpResponseForbidden

    def get(self, request, *args, **kwargs):
        return HttpResponseNotAllowed(['POST',])

    def post(self, request, *args, **kwargs):
        return self.attend(request, *args, **kwargs)


@permission_required('events.quit_event')
class EventQuitView(UpdateView):
    model = Event

    def get_success_url(self):
        return self.object.get_absolute_url()

    def quit(self, request, *args, **kwargs):
        """
        Calls the quit() method on the fetched object and then
        redirects to the success URL.
        """
        self.object = self.get_object()
        success_url = self.get_success_url()
        messages.add_message(request, messages.SUCCESS,
                             _('You have just withdraw from the event.'))
        try:
            self.object.quit(request.user)
            return HttpResponseRedirect(success_url)
        except PermissionDenied:
            return HttpResponseForbidden

    def get(self, request, *args, **kwargs):
        return HttpResponseNotAllowed(['POST',])

    def post(self, request, *args, **kwargs):
        return self.quit(request, *args, **kwargs)


class EventYearListView(YearArchiveView, EventPublishedQuerySetMixin, EventDateArchiveMixin):
    model = Event
    date_field = 'period_start'
    allow_empty = True
    allow_future = True
    make_object_list = True
    paginate_by = 10


class EventMonthListView(MonthArchiveView, EventPublishedQuerySetMixin, EventDateArchiveMixin):
    model = Event
    date_field = 'period_start'
    allow_empty = True
    allow_future = True
    month_format = '%m'


class EventPreviewView(SingleObjectPreviewViewMixin, DetailView):
    model = Event
    template_name = "events/components/event_detail.html"


@permission_required('events.view_event')
class EventCalendarView(DetailView):
    """
    EventをiCal形式でダウンロードするView
    """
    model = Event
    MIMETYPE = 'text/calendar'

    def render_to_response(self, context, **response_kwargs):
        object = context['object']

        if not object.period_start or object.pub_state == 'draft':
            return HttpResponseNotFound(_(
                'The event is private or does not have start time.'
            ))
        cal = generate_ical(object)
        file = io.BytesIO(cal.to_ical())
        # テストの際に、closeされてしまうため
        # HttpResponseの代わりにStreamingHttpResponseを使っている
        # http://stackoverflow.com/questions/19359451/django-test-file-download-valueerror-i-o-operation-on-closed-file
        response = StreamingHttpResponse(FileWrapper(file), content_type=self.MIMETYPE)
        response['Content-Disposition'] = 'attachment; filename={}.ics'.format(object.pk)
        return response