codeforamerica/intake

from django.contrib.auth.decorators import login_required
from django.http.response import HttpResponseForbidden
from django.urls import reverse_lazy
from django.utils.decorators import method_decorator
from django.views.generic.edit import CreateView, UpdateView, DeleteView
from clips.models import Clip


class ClipCreateView(CreateView):
    model = Clip
    fields = ['title', 'query', ]

    @method_decorator(login_required)
    def dispatch(self, *args, **kwargs):
        if not self.request.user.has_perm('clips.add_clip'):
            return HttpResponseForbidden()
        return super(ClipCreateView, self).dispatch(*args, **kwargs)

    def get_context_data(self, **kwargs):
        # Call the base implementation first to get a context
        context = super(ClipCreateView, self).get_context_data(**kwargs)
        # Add in a QuerySet of all the books
        context['clips'] = Clip.objects.all().order_by('title')
        return context


class ClipUpdateView(UpdateView):
    model = Clip
    fields = ['title', 'query', ]
    template_name_suffix = '_update_form'

    @method_decorator(login_required)
    def dispatch(self, *args, **kwargs):
        if not self.request.user.has_perm('clips.change_clip'):
            return HttpResponseForbidden()
        return super(ClipUpdateView, self).dispatch(*args, **kwargs)


class ClipDeleteView(DeleteView):
    model = Clip
    success_url = reverse_lazy('clips-create')

    @method_decorator(login_required)
    def dispatch(self, *args, **kwargs):
        if not self.request.user.has_perm('clips.delete_clip'):
            return HttpResponseForbidden()
        return super(ClipDeleteView, self).dispatch(*args, **kwargs)