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 = RelatedFilter('projects.filters.OrderFilterSet')
#     # order = 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()
    team = RelatedFilter('users.filters.TeamFilterSet')
    order = RelatedFilter('projects.filters.OrderFilterSet')

    class Meta:
        model = Message




class DocumentFilterSet(FilterSet):
    id = AllLookupsFilter()
    # order = RelatedFilter('projects.filters.OrderFilterSet')

    class Meta:
        model = Documents



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

    class Meta:
        model = Notes