|
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
|