icontains 和 getlist django python

Question

我们正在尝试返回 Django API 的标题列表，其中标题可以包含几个关键字。

例如，如果我们使用__icontains 方法搜索“money”和“world” (api.com/?keyworld=money&keyword=world)，这将返回所有包含 money、world 或两者的记录。

相关的SQL语句为：

select * from news
    where news_source = 1 or news_source = 2
        and news_title like '%money%' or news_title like '%world%'

我们正在尝试使用此代码来允许用户拥有多个__icontains 的关键字以及多个来源，因此最终目标是：

api.com/?keyworld=money&keyword=world&source=1&source=2

我们的代码：

def get_queryset(self):
    queryset = News.objects.all()
    title = self.request.query_params.getlist('title')
    source = self.request.query_params.getlist('source')
    if title:
        queryset = queryset.filter(news_title__icontains=title, news_source__in=source)
    return queryset

问题在于，如果使用第二个关键字，这只会返回第二个关键字，而不是在&keyword= 中键入的内容之前的其他关键字。

Answer 1

您不能使用列表执行__icontains，但您可以设计一个函数，例如，为列表构造这些值的逻辑或。例如：

from django.db.models import Q
from functools import reduce
from operator import or_

def or_fold(list_of_qs):
    if list_of_qs:
        return reduce(or_, list_of_qs)
    else:
        return Q()

def unroll_lists_or(qs, **kwargs):
    return qs.filter([
        or_fold(Q(**{k: vi}) for vi in v)
        for k, v in kwargs.items()
    ])

然后您可以使用查询集调用unroll_lists_or，并且每个项目都应该是可迭代的（例如列表）。然后它将在列表的项目之间执行或-逻辑，并在不同键之间执行和-逻辑。如果一个可迭代对象为空，则忽略它。

所以我们可以把支票写成：

unroll_lists_or(queryset, <b>news_title__icontains=title, news_source=source</b>)

如果title 包含两个项目（所以title == [title1, title2]），而source 包含三个项目（所以source = [source1, source2, source3]），那么这将导致：

qs.filter(
    Q(news_title__icontains=title1) | Q(news_title__icontains=title2),
    Q(news_source=source1) | Q(news_source=source2) | Q(news_source=source3)
)

但是，您可以将其与 .filter(..) 结合使用以进行 __in 检查。例如：

queryset = News.objects.all()
if source:
    queryset = queryset.filter(news_source__in=source)
queryset = unroll_lists_or(queryset, news_title__icontains=title)

Answer 2

我能够通过在 get_querset() 函数中创建 2 个单独的函数来解决这个问题，该函数在发出 GET 请求时被调用。

 def get_queryset(self):
        queryset = News.objects.all()
        source_list = self.request.query_params.getlist('source')
        keyword_list = self.request.query_params.getlist('title')
        if source_list or keyword_list:
            def create_q_source(*args):
                list = [*args]
                source = Q()
                for value in list:
                    source.add(Q(news_source=value), Q.OR)
                return source
            def create_q_keyword(*args):
                list = [*args]
                keyword = Q()
                for value in list:
                    keyword.add(Q(news_title__icontains=value), Q.OR)
                return keyword
            queryset = queryset.filter(create_q_source(*source_list),create_q_keyword(*keyword_list))
        return queryset

编辑： 当你去api链接传入参数时，会根据传入的内容进行过滤：

http://127.0.0.1:8000/api/notes/?keyword=trump&keyword=beyond&keyword=money&source=1

SQL 等效项：

select * from news where news_source = 1 AND news_title like '%beyond%' OR news_title like '%money%'