from django.db.models import Q
from django.views.generic import ListView

from apps.tasks.models import Task


class TaskListView(ListView):
    model = Task
    template_name = 'tasks/task_list.html'
    context_object_name = 'tasks'
    paginate_by = 10

    def get_queryset(self):
        queryset = Task.objects.select_related('customer', 'assignee').all()
        query = self.request.GET.get('q', '').strip()
        status = self.request.GET.get('status', '').strip()

        if query:
            queryset = queryset.filter(
                Q(title__icontains=query)
                | Q(customer__full_name__icontains=query)
                | Q(customer__company_name__icontains=query)
            )

        if status in Task.Status.values:
            queryset = queryset.filter(status=status)

        return queryset

    def get_context_data(self, **kwargs):
        context = super().get_context_data(**kwargs)
        context['q'] = self.request.GET.get('q', '').strip()
        context['status'] = self.request.GET.get('status', '').strip()
        context['status_options'] = Task.Status.choices
        return context