Я хотел бы получить кучу строк из моей базы данных с помощью набора фильтров.
Мне было интересно, применим ли условный фильтр в django. То есть, "фильтр, если переменная не является None, или не применяет фильтрацию в противном случае".
Что-то вроде этого:
user = User.objects.get(pk=1)
category = Category.objects.get(pk=1)
todays_items = Item.objects.filter(user=user, date=now()).conditional_filter(category=category))
Что бы я хотел сделать, это применить фильтр категории только в том случае, если категория не является None.
Если категория равна None (означает, что она не указана в объекте запроса), этот фильтр вообще не будет применяться. Это избавит меня от кучи ситуаций "if-elif-else".
Есть ли способ сделать это?
Вы можете связать запросы:
user = User.objects.get(pk=1)
category = Category.objects.get(pk=1)
qs = Item.objects.filter(user=user, date=now())
if category:
qs = qs.filter(category=category)
Так как запрос выполняется лениво, результат БД происходит только при отображении элементов.
Ну, это довольно старый вопрос, но для тех, кто хотел бы выполнить условную фильтрацию на одной строке, вот мой подход (Btw, следующий код, вероятно, может быть написан более общим образом):
from django.db.models import Q
def conditional_category_filter(category):
if category != None:
return Q(category=category)
else:
return Q() #Dummy filter
user = User.objects.get(pk=1)
category = Category.objects.get(pk=1)
todays_items = Item.objects.filter(conditional_category_filter(category), user=user, date=now())
Единственное, что вам нужно посмотреть, это использовать вызов conditional_category_filter(category) перед аргументами ключевого слова, например user=user. Например, следующий код вызовет ошибку:
todays_items = Item.objects.filter(user=user, date=now(), conditional_category_filter(category))
Это несколько подходов к вашей проблеме. Один из подходов - играть с Комплексный поиск с объектами Q
from django.db.models import Q
user = User.objects.get(pk=1)
category = Category.objects.get(pk=1)
f1 = Q( user=user, date=now() )
f_cat_is_none = Q( category__isnull = True )
f_cat_is_not_none = Q( category=category )
todays_items = Item.objects.filter( f1 & ( f_cat_is_none | f_cat_is_not_none ) )
Я не понимаю в вашем ответе, если это запрос, который вы ищете, но в этом примере вы можете легко составить свой собственный запрос.
Отредактированный комментарий OP
category__isnull == True означает, что в базе данных элемент не связан с категорией.
Возможно, запрос, который вы ищете, это:
from django.db.models import Q
user_pk = 1
category_pk = 1 #some times None
f = Q( user__pk = user_pk, date=now() )
if category_pk is not None:
f &= Q( category__pk = category_pk )
todays_items = Item.objects.filter( f )
Это всего лишь образец кода, соответствующий его требованиям. Будьте осторожны с одиночными _ и двойными __.