python - Django：get_context_data 之后的 get_queryset

Question

我正在使用 ListView 中的 get_context_data 由经过身份验证的用户过滤我的项目模型对象

class myprojects(LoginRequiredMixin, ListView):
    model = Project
    template_name = 'my_project_list.html'
    ordering = ['project_title']

    def get_context_data(self, *args, **kwargs):
        context = super(myprojects, self).get_context_data(**kwargs)
        context['my_projects'] = Project.objects.filter(engineer=self.request.user)
        return context

在模板中

{% for my_project in my_projects %}
{{ my_project }}
{% endfor %}

我的模板中有一个表单来搜索项目，我在同一个 ListView 上使用了 get_queryset。我收到了搜索，但它没有在模板中过滤

class myprojects(LoginRequiredMixin, ListView):
    model = Project
    template_name = 'my_project_list.html'
    ordering = ['project_title']

    def get_context_data(self, *args, **kwargs):
        context = super(myprojects, self).get_context_data(**kwargs)
        context['my_projects'] = Project.objects.filter(engineer=self.request.user)
        return context

    def get_queryset(self, *args, **kwargs):
        context = super().get_queryset(*args, **kwargs)
        search = self.request.GET.get('buscar', None)
        print(search)
        if search:
            context = context.filter(
                Q(project_title__icontains=search) |
                Q(category__title__icontains=search)
            ).distinct()
        return context

注意：当我使用{% for project in object_list %}而不是{% for project in my_projects %}，它搜索没有问题，但我不再有用户过滤器

标签： pythondjangodjango-queryset

Answer 1

只需在 get_queryset 方法中过滤

def get_queryset(self, *args, **kwargs):
    queryset = super().get_queryset(*args, **kwargs)
    search = self.request.GET.get('buscar', None)
    queryset = queryset.filter(engineer=self.request.user)
    if search:
        queryset = queryset.filter(
            Q(project_title__icontains=search) |
            Q(category__title__icontains=search)
        ).distinct()
    return queryset