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