import django_filters
from django.db.models import Q
from rest_framework import filters

from files import models


class DocumentFilter(django_filters.rest_framework.FilterSet):
    class Meta:
        model = models.Document
        fields = ('name', 'content_type', 'object_id',)

class DocumentSubTypeFilter(django_filters.rest_framework.FilterSet):
    document_type = django_filters.ModelMultipleChoiceFilter(queryset=models.DocumentType.objects.all(), widget=django_filters.widgets.CSVWidget)

    class Meta:
        model = models.DocumentSubType
        fields = ('document_type',)