import django_filters

from apps.calls.models import Notification


class NumberInFilter(django_filters.BaseInFilter, django_filters.NumberFilter):
    pass

class NotificationFilter(django_filters.FilterSet):
    is_read = django_filters.BooleanFilter(field_name='is_read')
    notification_type = NumberInFilter(field_name='notification_type')

    class Meta:
        model = Notification
        fields = ['is_read', 'notification_type']