from rest_framework.views import exception_handler
def core_exception_handler(exc, context):
# If an exception is thrown that we don't explicitly handle here, we want
# to delegate to the default exception handler offered by DRF. If we do
# handle it, we will still want access to the response
# generated by DRF, so we get it up front.
response = exception_handler(exc, context)
handlers = {
'ProfileDoesNotExist': _handle_generic_error,
'ValidationError': _handle_generic_error
}
# This is how we identify the type of the current exception. We will use
# this in a moment to see whether we should handle this exception or let
# Django REST Framework do its thing.
exception_class = exc.__class__.__name__
if exception_class in handlers:
# If this exception is one that we can handle, then handle it. Otherwise,
# return the response generated earlier by the default exception
# handler.
return handlers[exception_class](exc, context, response)
return response
def _handle_generic_error(exc, context, response):
# the simplest exception handler
response.data = {'errors': response.data}
return response
def _handle_not_found_error(exc, context, response):
view = context.get('view', None)
if view and hasattr(view, 'queryset') and view.queryset is not None:
error_key = view.queryset.model._meta.verbose_name
response.data = {
'errors': {
error_key: response.data['detail']
}
}
else:
response = _handle_generic_error(exc, context, response)
return response
