diff --git a/greg/api/filters.py b/greg/api/filters.py index 14ef355eb0bc7935dc090f5cf189055e2df78765..86465b1d1a7290dbd3dc9ed816d827e47ff0b3d4 100644 --- a/greg/api/filters.py +++ b/greg/api/filters.py @@ -1,18 +1,26 @@ -from django_filters import rest_framework as filters +from django_filters.rest_framework import ( + BaseInFilter, + BooleanFilter, + FilterSet, +) -from greg.models import Person, PersonRole, PersonIdentity +from greg.models import ( + Person, + PersonRole, + PersonIdentity, +) -class PersonRoleFilter(filters.FilterSet): - type = filters.BaseInFilter(field_name="role__type", lookup_expr="in") +class PersonRoleFilter(FilterSet): + type = BaseInFilter(field_name="role__type", lookup_expr="in") class Meta: model = PersonRole fields = ["type"] -class PersonFilter(filters.FilterSet): - verified = filters.BooleanFilter( +class PersonFilter(FilterSet): + verified = BooleanFilter( field_name="person__verified_by_id", lookup_expr="isnull", exclude=True ) @@ -21,7 +29,7 @@ class PersonFilter(filters.FilterSet): fields = ["first_name", "last_name", "verified"] -class PersonIdentityFilter(filters.FilterSet): +class PersonIdentityFilter(FilterSet): class Meta: model = PersonIdentity fields = ["type", "verified_by_id"]