filters.ModelChoiceFilter not looks like filters.ChoiceFilter about drf-yasg

[deveshbedmutha1406]

You can achieve this by creating a custom Filter subclass that overrides the field_class attribute.

from django_filters import rest_framework as filters
from django.db import models
from drf_yasg import openapi

class ModelChoiceField(filters.ModelChoiceFilter):
    field_class = filters.ModelChoiceFilter.field_class

    def get_schema_fields(self, view):
        field = super().get_schema_fields(view)[0]
        field['type'] = 'integer'
        field['enum'] = list(self.model.objects.values_list('id', flat=True))
        return [field]

class ClinicFilter(filters.FilterSet):
    test = ModelChoiceField(queryset=Clinic.objects.all())

[Gamafoot]

thank you!