import django_filters
from .models import AddFunctionModel
from .models import MenuByFunctionModel

class ListingFilter(django_filters.FilterSet):
    function_name = django_filters.CharFilter(lookup_expr='exact')


    class Meta:
        model = AddFunctionModel
        fields= ['function_name']


class ListingFilterMenu(django_filters.FilterSet):
    title = django_filters.CharFilter(lookup_expr='exact')


    class Meta:
        model = MenuByFunctionModel
        fields= ['title']