from rest_framework_filters import FilterSet, RelatedFilter, AllLookupsFilter

from .models import Message, Notes


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