from django.shortcuts import render

from django.views import generic
from website import views

from .models import Sujet

from django.forms import ModelForm

# Create your views here.

class SujetsView(views.WebsiteProtectedMixin):
    title = "Sujets"

    def get_active_app(self):
        return super().get_active_app(app_name='suivi')



class SujetDetailsView(SujetsView, generic.DetailView):
    template_name = "sujets/sujet_details.html"
    model = Sujet



class SujetListView(SujetsView, generic.ListView):
    model = Sujet
    template_name = "sujets/sujet_liste.html"



class SujetUpdateView(SujetsView, generic.edit.UpdateView):
    template_name = "sujets/sujet_update.html"
    model = Sujet
    fields = '__all__'



class SujetCreateForm(ModelForm):
    class Meta:
        model = Sujet
        fields = ['nom', 'surnom', 'prenom', 'genre', 'premiere_rencontre']



class SujetCreateView(SujetsView, generic.edit.CreateView, views.AjaxTemplateMixin):
    template_name = "sujets/sujet_create.html"
    form_class = SujetCreateForm

    title = "Création : Sujet"
    header = "Ajouter un sujet"


    def post(self, request, *args, **kwargs):
        if 'next' in self.request.POST:
            self.success_url = self.request.POST["next"]
        return super().post(self, request, *args, **kwargs)

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        try:
            context['next'] = self.request.GET['next']
        except:
            context['next'] = None
        return context