"""Partner app filters."""
from django_filters import rest_framework as filters

from partner.models import Partner


class PartnerFilterSet(filters.FilterSet):
    """Establishment filter set."""

    establishment = filters.NumberFilter(
        help_text='Allows to get partner list by establishment ID.')
    type = filters.ChoiceFilter(
        choices=Partner.MODEL_TYPES,
        help_text=f'Allows to filter partner list by partner type. '
                  f'Enum: {dict(Partner.MODEL_TYPES)}')

    class Meta:
        """Meta class."""
        model = Partner
        fields = (
            'establishment',
            'type',
        )