如何自定义drf-yasg端点的示例响应？

Question

在我的Django项目中，我使用DRF和drf-yasg。在某些端点，示例响应体显示不正确的示例。如以下示例所示：

​

但是其中一些没有显示正确的响应体示例。

​

这个端点实际上返回access_token和refresh_token，不返回电子邮件和密码。这对前端开发人员来说是错误的信息。有什么办法来改变这个吗？

Answer 1

下面是一个使用带有3个序列化器的@swagger_auto_schema注释的示例。1个请求序列化程序和2个响应序列化程序(成功和错误序列化程序)。

class RequestSerializer(serializers.Serializer):
   param = serializers.CharField()

                  
class SucessSerializer(serializers.Serializer):
   success = serializers.BooleanField()
   message = serializers.CharField()

                     
class ErrorSerializer(serializers.Serializer):
   success = serializers.BooleanField()
   errors = serializers.IntegerField()

class ExampleViewView(APIView):

    @swagger_auto_schema(
        request_body = RequestSerializer,
        responses={
            '200': SucessSerializer,
            '500': ErrorSerializer,
        },
        operation_description = 'Doc description'
    )
    def post(self, request):
        
        return successResponse({
            'sucess': True
        })