Какие существуют варианты фильтрации значений Enums из документации Swagger?

Я использую Django с пакетом drf-spectacular для создания документации Swagger. Мне интересно, есть ли лучший подход для фильтрации некоторых значений из секции Enum внутри Schema. Сейчас я сделал это с помощью пользовательского хука preprocess_schema_enums

Энумы задаются как поле в файле модели

hook.py

def preprocess_schema_enums(result, generator, request, public):
  excluded = ['value1', 'value2']
  enums_response = result['components']['schemas']['CustomEnum']['enum']
  filtered = [res for res in enums_resoinse if res not in excluded]
  result['components']['schemas']['CustomEnum']['enum'] = filtered
  return result

Вернуться на верх