34 lines
1.2 KiB
Python
34 lines
1.2 KiB
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
|
|
# base_template = 'employee-base.html'
|
|
|
|
def test_func(self):
|
|
return True # self.request.user.has_employee_perm('can_view_preinfo') or self.request.user.user_type == 'CA'
|
|
|
|
# def get_context_data(self, **kwargs):
|
|
# context = super().get_context_data(**kwargs)
|
|
# context['base_template'] = self.base_template
|
|
# return context
|
|
|
|
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
|