import django_filters
from .models import QuestionPair, Specialty

class QuestionPairFilter(django_filters.FilterSet):
    survey_title = django_filters.CharFilter(field_name="survey__title", lookup_expr="icontains")
    statement_a_text = django_filters.CharFilter(field_name="statement_a__statement", lookup_expr="icontains")
    statement_b_text = django_filters.CharFilter(field_name="statement_b__statement", lookup_expr="icontains")

    class Meta:
        model = QuestionPair
        fields = ["survey_title", "statement_a_text", "statement_b_text"]


class SpecialtyFilter(django_filters.FilterSet):
    institution_name = django_filters.CharFilter(field_name="institution__name", lookup_expr="icontains")

    class Meta:
        model = Specialty
        fields = ["name", "institution_name"]