update meal_type filter on MealPlan to be a list

This commit is contained in:
smilerz 2023-12-19 12:43:16 -06:00
parent 8bbccad7a9
commit 99cd9bfb5b
No known key found for this signature in database
GPG Key ID: 39444C7606D47126

View File

@ -708,7 +708,7 @@ class MealPlanViewSet(viewsets.ModelViewSet):
query_params = [ query_params = [
QueryParam(name='from_date', description=_('Filter meal plans from date (inclusive) in the format of YYYY-MM-DD.'), qtype='string'), QueryParam(name='from_date', description=_('Filter meal plans from date (inclusive) in the format of YYYY-MM-DD.'), qtype='string'),
QueryParam(name='to_date', description=_('Filter meal plans to date (inclusive) in the format of YYYY-MM-DD.'), qtype='string'), QueryParam(name='to_date', description=_('Filter meal plans to date (inclusive) in the format of YYYY-MM-DD.'), qtype='string'),
QueryParam(name='meal_type', description=_('Filter meal plans with MealType ID.'), qtype='int'), QueryParam(name='meal_type', description=_('Filter meal plans with MealType ID. For multiple repeat parameter.'), qtype='int'),
] ]
schema = QueryParamAutoSchema() schema = QueryParamAutoSchema()
@ -726,9 +726,9 @@ class MealPlanViewSet(viewsets.ModelViewSet):
if to_date is not None: if to_date is not None:
queryset = queryset.filter(to_date__lte=to_date) queryset = queryset.filter(to_date__lte=to_date)
meal_type = self.request.query_params.get('meal_type', None) meal_type = self.request.query_params.getlist('meal_type', [])
if meal_type is not None: if meal_type:
queryset = queryset.filter(meal_type__id=meal_type) queryset = queryset.filter(meal_type__in=meal_type)
return queryset return queryset