import django_filters

from .models import Course


class CourseFilter(django_filters.FilterSet):
    category = django_filters.CharFilter(field_name='category')
    cat = django_filters.CharFilter(field_name='category__title')
    age = django_filters.ChoiceFilter(field_name='age', choices=Course.AGE_CHOICES)

    class Meta:
        model = Course
        fields = ['category', 'age']