python - 在异常时覆盖 django-rest-framework 中接受的渲染器

Question

我正在使用 django-rest-framework 创建一个返回 PDF 的端点。但是，当呈现 PDF 时出错，我想返回一个 JSON 响应。但是 DRF 将异常数据传递给了我的 PDFRenderer 类。只有在出现错误时，我如何才能改用 JSONRenderer？

class PDFRenderer(BaseRenderer):
    """ DRF renderer for PDF binary content. """
    media_type = 'application/pdf'
    format = 'pdf'
    charset = None
    render_style = 'binary'

    def render(self, data, media_type=None, renderer_context=None):
        return bytes(data)

例如，当我的 View 显示为 raise PermissionDenied() 时，因为授权用户没有权限查看请求的 PDF，DRF 将 {'detail': 'You do not have permission to perform this action.'} 作为 data 参数到 PDFRenderer.render。

编辑: 我尝试了 custom exception handler但显然您仍然必须通过 DRF 的异常处理程序运行它，并将其传递给 PDFRenderer。

Answer 1

毕竟我找到了一种使用自定义异常处理程序来执行此操作的方法:

from rest_framework.renderers import JSONRenderer
from rest_framework.views import exception_handler


def custom_exception_handler(exc, context):
    """ Switch from PDFRenderer to JSONRenderer for exceptions """
    if context['request'].accepted_renderer.format == 'pdf':
        context['request'].accepted_renderer = JSONRenderer()
    return exception_handler(exc, context)

感觉很hacky。仍然希望得到更好的答案。