82 lines
3.0 KiB
Python
82 lines
3.0 KiB
Python
import django_filters
|
|
from django.conf import settings
|
|
from django.contrib.postgres.search import TrigramSimilarity
|
|
from django.db.models import Q
|
|
from django.utils.translation import gettext as _
|
|
from django_scopes import scopes_disabled
|
|
|
|
from cookbook.forms import MultiSelectWidget
|
|
from cookbook.models import Food, Keyword, Recipe, ShoppingList
|
|
|
|
with scopes_disabled():
|
|
class RecipeFilter(django_filters.FilterSet):
|
|
name = django_filters.CharFilter(method='filter_name')
|
|
keywords = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Keyword.objects.none(),
|
|
widget=MultiSelectWidget,
|
|
method='filter_keywords'
|
|
)
|
|
foods = django_filters.ModelMultipleChoiceFilter(
|
|
queryset=Food.objects.none(),
|
|
widget=MultiSelectWidget,
|
|
method='filter_foods',
|
|
label=_('Ingredients')
|
|
)
|
|
|
|
def __init__(self, data=None, *args, **kwargs):
|
|
space = kwargs.pop('space')
|
|
super().__init__(data, *args, **kwargs)
|
|
self.filters['foods'].queryset = Food.objects.filter(space=space).all()
|
|
self.filters['keywords'].queryset = Keyword.objects.filter(space=space).all()
|
|
|
|
@staticmethod
|
|
def filter_keywords(queryset, name, value):
|
|
if not name == 'keywords':
|
|
return queryset
|
|
for x in value:
|
|
queryset = queryset.filter(keywords=x)
|
|
return queryset
|
|
|
|
@staticmethod
|
|
def filter_foods(queryset, name, value):
|
|
if not name == 'foods':
|
|
return queryset
|
|
for x in value:
|
|
queryset = queryset.filter(steps__ingredients__food__name=x).distinct()
|
|
return queryset
|
|
|
|
@staticmethod
|
|
def filter_name(queryset, name, value):
|
|
if not name == 'name':
|
|
return queryset
|
|
if settings.DATABASES['default']['ENGINE'] in ['django.db.backends.postgresql_psycopg2',
|
|
'django.db.backends.postgresql']:
|
|
queryset = queryset.annotate(similarity=TrigramSimilarity('name', value), ).filter(
|
|
Q(similarity__gt=0.1) | Q(name__unaccent__icontains=value)).order_by('-similarity')
|
|
else:
|
|
queryset = queryset.filter(name__icontains=value)
|
|
return queryset
|
|
|
|
class Meta:
|
|
model = Recipe
|
|
fields = ['name', 'keywords', 'foods', 'internal']
|
|
|
|
class FoodFilter(django_filters.FilterSet):
|
|
name = django_filters.CharFilter(lookup_expr='icontains')
|
|
|
|
class Meta:
|
|
model = Food
|
|
fields = ['name']
|
|
|
|
class ShoppingListFilter(django_filters.FilterSet):
|
|
|
|
def __init__(self, data=None, *args, **kwargs):
|
|
if data is not None:
|
|
data = data.copy()
|
|
data.setdefault("finished", False)
|
|
super().__init__(data, *args, **kwargs)
|
|
|
|
class Meta:
|
|
model = ShoppingList
|
|
fields = ['finished']
|