from rest_framework_filters import FilterSet, RelatedFilter, AllLookupsFilter

from .models import Message, Notes, Documents


class DocumentsFilterSet(FilterSet):
    file = AllLookupsFilter()
    sender = RelatedFilter('users.filters.UserFilterSet')
    recipent = RelatedFilter('users.filters.UserFilterSet')
    # order = AllLookupsFilter()
    # team = AllLookupsFilter()

    class Meta:
        model = Documents


class MessageFilterSet(FilterSet):
    text = AllLookupsFilter()
    created = AllLookupsFilter()
    sender = RelatedFilter('users.filters.UserFilterSet')
    recipent = RelatedFilter('users.filters.UserFilterSet')
    private_type = AllLookupsFilter()

    class Meta:
        model = Message


class NoteFilterSet(FilterSet):
    text = AllLookupsFilter()
    created = AllLookupsFilter()
    sender = RelatedFilter('users.filters.UserFilterSet')
    recipent = RelatedFilter('users.filters.UserFilterSet')

    class Meta:
        model = Notes