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.
27 lines
783 B
Python
27 lines
783 B
Python
from django.contrib.auth.mixins import LoginRequiredMixin, UserPassesTestMixin
|
|
from django.views.generic import ListView
|
|
|
|
from booking.models import Booking
|
|
|
|
|
|
class BookingListView(LoginRequiredMixin, UserPassesTestMixin, ListView):
|
|
template_name = 'employee/booking-list.html'
|
|
model = Booking
|
|
context_object_name = 'objects'
|
|
paginate_by = 20
|
|
|
|
def test_func(self):
|
|
user = self.request.user
|
|
return self.request.user.user_type == 'EMP' or 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(status='active')
|
|
|
|
queryset = queryset.order_by('-created_on')
|
|
|
|
return queryset
|