Как лучше всего написать сериализатор для enum и последующего отображения в drf_spectacular?

Я хочу создать конструктор для сериализаторов перечислений, чтобы в дальнейшем использовать их в представлениях и корректно отображать данные в Swagger, особенно в схеме.

Мое типичное перечисление:

class EServicePlatform(models.IntegerChoices):
    ONLINE = 1, 'Online'
    OFFLINE = 2, 'Offline'

Которые напрямую передаются в представления в виде словаря:

django Скопируйте код

class DictionaryPlatforms(APIView):
    @extend_schema(responses={
        200: OpenApiResponse(response=DictionarySerializer(enum_class=EServicePlatform, many=True)),
        400: OpenApiResponse(description='Bad Request')}
    )
    def get(self, request, *args, **kwargs):
        data = [{"value": item.value, "label": item.label} for item in EServicePlatform]
        serializer = DictionarySerializer(data=data, many=True, enum_class=EServicePlatform)

        if serializer.is_valid():
            return Response(serializer.data, status=status.HTTP_200_OK)
        else:
            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)

Сам конструктор сериализатора:

class DictionarySerializer(serializers.Serializer):
    value = serializers.ChoiceField(choices=[])  # Set an empty list by default
    label = serializers.CharField()

    def __init__(self, *args, **kwargs):
        # Expect enum_class to be passed via kwargs
        enum_class = kwargs.pop('enum_class', None)
        if not enum_class:
            raise ValueError("enum_class is required")

        # Dynamically set choices based on the passed enum class
        self.fields['value'].choices = enum_class.choices

        super().__init__(*args, **kwargs)
        self.enum_class = enum_class

    def validate(self, data):
        # Automatically add the label corresponding to the value
        data['label'] = dict(self.enum_class.choices)[data['value']]
        return data

В текущей реализации возникают две проблемы:

1. Мне кажется, что логика слишком сложна и ее можно значительно упростить. Но как?
2. По какой-то причине Swagger берет данные из другого enum, который находится в другом приложении, а не из того, который мы передаем.

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