48 lines
1.4 KiB
Python
48 lines
1.4 KiB
Python
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
|