phzhik/gault-millau
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix positions by establishment type/subtype

Browse Source
This commit is contained in:
Kuroshini 2020-02-04 15:42:15 +03:00
parent 9982c2ca77
commit 5aa74ff7c9
2 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
apps/establishment/filters.py
View File

@ -173,5 +173,5 @@ class PositionsByEstablishmentFilter(filters.FilterSet):
def by_subtype(self, queryset, name, value): def by_subtype(self, queryset, name, value):
"""filter by establishment subtype""" """filter by establishment subtype"""
if value not in EMPTY_VALUES: if value not in EMPTY_VALUES:
return queryset.by_establishment_subtype(value) return queryset.by_establishment_subtypes(value.split('__'))
return queryset return queryset

8
apps/establishment/models.py
View File

@ -970,10 +970,12 @@ class EstablishmentGallery(IntermediateGalleryModelMixin):
class PositionQuerySet(models.QuerySet): class PositionQuerySet(models.QuerySet):
def by_establishment_type(self, value: str): def by_establishment_type(self, value: str):
return self.filter(establishment_type__index_name=value) return self.filter(Q(establishment_type__index_name=value) |
Q(establishment_type__isnull=True, establishment_subtype__isnull=True))
def by_establishment_subtype(self, value: str): def by_establishment_subtypes(self, value: List[str]):
return self.filter(establishment_subtype__index_name=value) return self.filter(Q(establishment_subtype__index_name__in=value) |
Q(establishment_type__isnull=True, establishment_subtype__isnull=True))
class Position(BaseAttributes, TranslatedFieldsMixin): class Position(BaseAttributes, TranslatedFieldsMixin):