diff --git a/program/filters.py b/program/filters.py index ca507c827a6d70580ecc078e5e68545a06cc9ffb..823ed52e38ecd1435828b37db744f1816774815a 100644 --- a/program/filters.py +++ b/program/filters.py @@ -174,6 +174,16 @@ class TimeSlotFilterSet(filters.FilterSet): ), ) + scheduleIds = IntegerInFilter( + field_name="schedule", + help_text="Return only timeslots that belong to the specified schedule(s).", + ) + + showIds = IntegerInFilter( + field_name="schedule__show", + help_text="Return only timeslots that belong to the specified show(s).", + ) + def filter_surrounding(self, queryset: QuerySet, name: str, value: datetime.datetime): nearest_timeslots_in_future = ( models.TimeSlot.objects.filter(start__gte=value)