from django.utils.translation import ugettext_lazy as _
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',)

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