import django_filters

from .models import Event, Status, Prospect

class ProspectFilter(django_filters.FilterSet):
    zip_code = django_filters.CharFilter(lookup_expr='icontains')
    events = django_filters.ModelMultipleChoiceFilter(
        queryset=Event.objects.all(),
        field_name='events',
    )

    class Meta:
        model = Prospect
        fields = ['city', 'events']