Django admin custom filter

class NullFilter(admin.SimpleListFilter) :
    parameter_name = 'title'
    title = 'title'

    def lookups(self, request, model_admin) :
        return (
            ('Not Null', 'Not empty only'),
            ('Null', 'Empty only'),
        )

    def queryset(self, request, queryset) :
        title= self.value()
        if title== 'Not Null' :
            return queryset.filter(category__isnull=False)
        elif title== 'Null' :
            return queryset.filter(category__isnull=True)

        return queryset

admin의 list_filter에 위 클래스 추가

    list_filter = ['aaaa','bbbb',NullFilter]

class TrainedFilter(admin.SimpleListFilter) :
    title = 'Trained'
    parameter_name = 'Trained'

    def lookups(self, request, model_admin) :
        return [('True','True'), ('False','False')]

    def queryset(self, request, queryset) :
        if self.value() == 'True' :
            for image in queryset :
                if image.isTrained() == False :
                    queryset = queryset.exclude(id=image.id)
            return queryset
        elif self.value() == 'False' :
            for image in queryset :
                if image.isTrained() == True :
                    queryset = queryset.exclude(id=image.id)
            return queryset
        
        return queryset