import django_filters

from apps.content.models import Banner


class BannerFilter(django_filters.FilterSet):
    placement = django_filters.ChoiceFilter(choices=Banner.PLACEMENT_CHOICES)
    is_active = django_filters.BooleanFilter()
    created_at = django_filters.DateFromToRangeFilter()

    class Meta:
        model = Banner
        fields = ["placement", "is_active", "created_at"]