You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
28 lines
966 B
Python
28 lines
966 B
Python
from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
|
|
from django.views.generic import TemplateView, FormView, CreateView, ListView
|
|
from preinfo.forms import PreinfoBaseForm, PreinfoCreateForm, PreinfoEditForm
|
|
from preinfo.models import Preinfo
|
|
|
|
class EmployeePreinfoView(LoginRequiredMixin, UserPassesTestMixin, ListView):
|
|
model = Preinfo
|
|
template_name = 'employee/preinfo-list.html'
|
|
context_object_name = 'objects'
|
|
paginate_by = 20
|
|
form_class = PreinfoEditForm
|
|
|
|
def test_func(self):
|
|
return self.request.user.user_type in ('EM', 'CA')
|
|
|
|
def get_queryset(self):
|
|
queryset = super().get_queryset()
|
|
|
|
data_filter = self.request.GET.get('filter')
|
|
if data_filter == 'all':
|
|
queryset = queryset.filter(deleted=False)
|
|
else:
|
|
queryset = queryset.filter(received=False,deleted=False)
|
|
|
|
queryset = queryset.order_by('-created_on')
|
|
|
|
return queryset
|