from rest_framework_filters import FilterSet, RelatedFilter, AllLookupsFilter

from .models import Review


class ReviewFilterSet(FilterSet):
    text = AllLookupsFilter()
    type = AllLookupsFilter()
    created = AllLookupsFilter()
    from_customer = RelatedFilter('users.filters.UserFilterSet')
    from_contractor = RelatedFilter('users.filters.UserFilterSet')
    target_customer = RelatedFilter('users.filters.UserFilterSet')
    target_contractor = RelatedFilter('users.filters.UserFilterSet')

    class Meta:
        model = Review