24 lines
649 B
Python
24 lines
649 B
Python
"""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',
|
|
)
|