from django import forms
import django_filters

from evolus.models import Audience, Document, Site, Structure


class DocumentFilter(django_filters.FilterSet):
    audiences = django_filters.ModelMultipleChoiceFilter(
        queryset=Audience.objects.all(),
        widget=forms.CheckboxSelectMultiple
    )
    sites = django_filters.ModelMultipleChoiceFilter(
        queryset=Site.objects.all(),
        widget=forms.CheckboxSelectMultiple
    )
    structures = django_filters.ModelMultipleChoiceFilter(
        queryset=Structure.objects.all(),
        widget=forms.CheckboxSelectMultiple
    )

    class Meta:
        model = Document
        fields = ('audiences', 'sites', 'structures', 'standards')