from django.views.generic import ListView

from booking.models import Booking


class BookingListView(ListView):
    template_name = 'employee/booking-list.html'
    model = Booking
    context_object_name = 'objects'
    paginate_by = 30  # Number of containers per page
    base_template = 'employee-base.html'

    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(status='active')

        queryset = queryset.order_by('-created_on')

        return queryset