我正在使用带有drf-spectacular
包的 Django 来生成 Swagger 文档。我想知道是否有更好的方法可以从 Schema 中的 Enum 部分过滤掉一些值。现在我已经使用自定义钩子完成了这个preprocess_schema_enums
枚举被指定为模型文件中的字段
钩子.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