Produits Entr'ouvert » Calebasse

from datetime import datetime, timedelta

from dateutil.relativedelta import relativedelta

from django.views.generic import list as list_cbv, edit, base # ListView

# from django.views.generic.edit import # CreateView, DeleteView, UpdateView

from django.shortcuts import get_object_or_404

from django.http import Http404

from calebasse.ressources.models import Service

class ServiceViewMixin(object):

    service = None

    date = None

    popup = False

    def dispatch(self, request, **kwargs):

        self.popup = request.GET.get('popup')

        if 'service' in kwargs:

            self.service = get_object_or_404(Service, slug=kwargs['service'])

        if 'date' in kwargs:

            try:

                self.date = datetime.strptime(kwargs.get('date'),

                        '%Y-%m-%d').date()

            except (TypeError, ValueError):

                raise Http404

        return super(ServiceViewMixin, self).dispatch(request, **kwargs)

    def get_context_data(self, **kwargs):

        context = super(ServiceViewMixin, self).get_context_data(**kwargs)

        context['popup'] = self.popup

        if self.service is not None:

            context['service'] = self.service.slug

            context['service_name'] = self.service.name

        if self.date is not None:

            context['date'] = self.date

            context['previous_day'] = self.date + relativedelta(days=-1)

            context['next_day'] = self.date + relativedelta(days=1)

            context['previous_month'] = self.date + relativedelta(months=-1)

            context['next_month'] = self.date + relativedelta(months=1)

        return context

class TemplateView(ServiceViewMixin, base.TemplateView):

    pass

class ListView(ServiceViewMixin, list_cbv.ListView):

    pass

class CreateView(ServiceViewMixin, edit.CreateView):

    pass

class DeleteView(ServiceViewMixin, edit.DeleteView):

    pass

class UpdateView(ServiceViewMixin, edit.UpdateView):

    pass