from django.shortcuts import render
from django.contrib.auth.mixins import LoginRequiredMixin
from django.views.generic import CreateView, ListView, UpdateView, DetailView, DeleteView
from .models import *
from .forms import *
from django.urls import reverse_lazy

# Create your views here.
class RecoverDirManagement(LoginRequiredMixin, ListView):
	model = PersLetter

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)	
		context.update({
			'active_link' : 'recoverdir',
			'persletter' : PersLetter.objects.filter(agency=self.request.user.profile.agency)
		})
		return context
		
class RecoverDirAddPL(CreateView):
	model = PersLetter
	success_url = reverse_lazy('recoverdir')
	form_class = PersLetterForm
	template_name = "recoverdir/rd_pers_add.html"

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)
		context.update({
			'active_link' : 'recoverdir',
		})
		return context

	def form_valid(self, form):
		form.instance.agency = self.request.user.profile.agency
		form.instance.user = self.request.user
		return super().form_valid(form)

class RecoverDirUpdatePL(UpdateView):
	model = PersLetter
	success_url = reverse_lazy('recoverdir')
	form_class = PersLetterForm
	template_name = "recoverdir/rd_pers_update.html"

	def get_context_data(self, **kwargs):
		context = super().get_context_data(**kwargs)
		context.update({'active_link' : 'recoverdir'})
		return context