import django_filters
from .models import School, SchoolResponsibility

class SchoolFilter(django_filters.FilterSet):
    name = django_filters.CharFilter(lookup_expr="icontains")
    address = django_filters.CharFilter(lookup_expr="icontains")
    institution_type_name = django_filters.CharFilter(
        field_name="institution_type__name", lookup_expr="icontains"
    )

    class Meta:
        model = School
        fields = ["name", "address", "institution_type_name"]


class SchoolResponsibilityFilter(django_filters.FilterSet):
    academic_year = django_filters.CharFilter(field_name="academic_year__year", lookup_expr="exact")
    school_name = django_filters.CharFilter(field_name="school__name", lookup_expr="icontains")
    responsible_name = django_filters.CharFilter(
        field_name="responsible__account__first_name", lookup_expr="icontains"
    )

    class Meta:
        model = SchoolResponsibility
        fields = ["academic_year", "school_name", "responsible_name"]