from rest_framework_filters import FilterSet, AllLookupsFilter, RelatedFilter
from .models import Location


class LocationFilterSet(FilterSet):
    children = RelatedFilter('common.filters.LocationFilterSet')
    id = AllLookupsFilter()
    level = AllLookupsFilter()
    lft = AllLookupsFilter()
    name = AllLookupsFilter()
    parent = RelatedFilter('common.filters.LocationFilterSet')
    rght = AllLookupsFilter()
    tree_id = AllLookupsFilter()
    type = AllLookupsFilter()

    class Meta:
        model = Location