from django_filters import FilterSet
from django_filters import rest_framework as filters

from establishment import models


class EstablishmentFilter(FilterSet):
    tag_id = filters.NumberFilter(field_name='tags__metadata__id',)
    award_id = filters.NumberFilter(field_name='awards__id',)

    class Meta:
        model = models.Establishment
        fields = (
            'tag_id',
            'award_id'
        )