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 == 'EM' or self.request.user.is_superuser

    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