欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

Django與DRF結(jié)合的全局異常處理方案詳解

 更新時(shí)間：2022年02月23日 10:22:08   作者：Cao Lilu  
這篇文章主要為大家詳細(xì)介紹了Django與DRF結(jié)合的全局異常處理方案，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下，希望能夠給你帶來(lái)幫助

前言

Django 和 DRF(django rest framawork) 的結(jié)合在 python 后臺(tái)中經(jīng)常出現(xiàn)的組合。對(duì)于異常的全局處理，我們系統(tǒng)能有一個(gè)統(tǒng)一的解決方案，在開(kāi)發(fā)環(huán)境能看到比較全的異常堆棧，而在生產(chǎn)環(huán)境能更好的給用戶(hù)一個(gè)友好的提示，本文旨在提出一個(gè)統(tǒng)一個(gè)全局異常處理方案，僅供參考使用。

實(shí)現(xiàn)的目標(biāo)

如果沒(méi)有 DRF，我們只需要在 Django 中加一個(gè)中間件就可以解決全局異常的處理問(wèn)題，但是 DRF 會(huì)幫我們處理一些異常并自動(dòng)返回到客戶(hù)端，因此我們要協(xié)調(diào)兩者的異常處理策略。

同時(shí)我們希望能使用 Django 的 admin 進(jìn)行一些后臺(tái)的數(shù)據(jù)查看和修改，因此最好要保留 admin 的內(nèi)部異常處理行為。

本文的目標(biāo)如下：

  • 保留 Django 自帶的 admin 的異常處理行為
  • 攔截 DRF 的異常并進(jìn)行全局異常行為處理
  • 攔截除 DRF 的異常之外的其他 Django 異常并進(jìn)行全局異常行為處理

DRF 全局異常攔截的解決思路

首先 DRF 的異常都是繼承自 APIException 這個(gè)類(lèi)的，并且 DRF 跑出的異常會(huì)被 exception_handler 這個(gè)異常處理函數(shù)攔截（這個(gè)函數(shù)的位置在 /python3.7/site-packages/rest_framework/views.py中）。

我們進(jìn)一步查看這個(gè)函數(shù)的源碼：

def exception_handler(exc, context):
    """
    Returns the response that should be used for any given exception.
    By default we handle the REST framework `APIException`, and also
    Django's built-in `Http404` and `PermissionDenied` exceptions.
    Any unhandled exceptions may return `None`, which will cause a 500 error
    to be raised.
    """
    if isinstance(exc, Http404):
        exc = exceptions.NotFound()
    elif isinstance(exc, PermissionDenied):
        exc = exceptions.PermissionDenied()
    if isinstance(exc, exceptions.APIException):
        headers = {}
        if getattr(exc, 'auth_header', None):
            headers['WWW-Authenticate'] = exc.auth_header
        if getattr(exc, 'wait', None):
            headers['Retry-After'] = '%d' % exc.wait
        if isinstance(exc.detail, (list, dict)):
            data = exc.detail
        else:
            data = {'detail': exc.detail}
        set_rollback()
        return Response(data, status=exc.status_code, headers=headers)
    return None

通過(guò)這個(gè)函數(shù)的文檔簽名我們知道，DRF 會(huì)處理所有繼承自 APIException 的異常類(lèi)，并且還會(huì)額外的處理 Django 內(nèi)置的 Http404 和 PermissionDenied 異常，并將這些異常的處理結(jié)果返回到前臺(tái)。

如果不再這些處理范圍之內(nèi)，函數(shù)會(huì)返回 None，這時(shí)候會(huì)給 Django 拋出一個(gè) 500 的服務(wù)器錯(cuò)誤異常。

DRF 支持單獨(dú)配置異常處理函數(shù)，因此第一步現(xiàn)在 setting 中指定自定義的異常處理函數(shù)的位置：

REST_FRAMEWORK = {
    'EXCEPTION_HANDLER': 'server.exception.exception_globe.globe_exception_handler'
}

然后我們定義自己的異常處理程序：

第一步，調(diào)用 DRF 自己的異常處理函數(shù)第二步，對(duì) DRF 攔截的異常進(jìn)行處理第三步，將其他異常拋給 Django 處理

def globe_exception_handler(exc, context):
    """
        Below is the global exception handler of drf
        Http404 / PermissionDenied / APIException
    """
    # Call REST framework's default exception handler
    response = exception_handler(exc, context)
    request = context['request']
    # Exceptions form DRF and Django built-in `Http404` and `PermissionDenied`
    if response is not None:
        if isinstance(response.data, list):
            msg = '; '.join(response.data)
        elif isinstance(response.data, str):
            msg = response.data
        else:
            msg = 'Sorry, we make a mistake (*￣︶￣)!'
        ex_data = {
            "msg": msg,
            "error_code": 1000,
            "request": request.path
        }
        return JsonResponse(data=ex_data, status=response.status_code)
    # Exceptions from others
    # 如果 response 為 None 則直接觸發(fā)上面的 ExceptionGlobeMiddleware
    return response

Django 異常處理方案

從上一步的結(jié)果我們知道，DRF 處理不了的異常我們拋給了 Django，而 Django 支持通過(guò)定義中間件進(jìn)行全局異常處理，因此接下來(lái)我們只需要定一個(gè) Django 全局異常處理的中間件，并將中間件配置到 setting 文件中的 MIDDLEWARE 數(shù)組即可。

try:
    from django.utils.deprecation import MiddlewareMixin  # Django 1.10.x
except ImportError:
    MiddlewareMixin = object  # Django 1.4.x - Django 1.9.x

class ExceptionGlobeMiddleware(MiddlewareMixin):
    """
        Below is the global exception handler of django
    """
    def process_exception(self, request, exception):
        # 直接拋出 django admin 的異常
        if str(request.path).startswith('/admin/'):
            return None
        # 捕獲其他異常，直接返回 500
        ex_data = {
            "msg": "Sorry, we make a mistake (*￣︶￣)!",
            "error_code": 1000,
            "request": request.path
        }
        return JsonResponse(data=ex_data, status=500)

值得注意的是，我們可以在中間件的處理函數(shù)中拿到 request 對(duì)象，因此，我們可以通過(guò)這個(gè)對(duì)象拿到用戶(hù)請(qǐng)求的 url，這樣，我們通過(guò)判斷 url 就可以得到那些請(qǐng)求是來(lái)自 Django 自帶的 admin 的。

參考代碼如下：

# 直接拋出 django admin 的異常
if str(request.path).startswith('/admin/'):
    return None

總結(jié)

通過(guò)上述的配置，我們可以對(duì)完成 Django 結(jié)合 DRF 的全局異常處理，并且保留了 Django 自帶 admin 的異常處理策略。

在實(shí)際的環(huán)境中，我們可以通過(guò)環(huán)境變量進(jìn)行有選擇的日志打印和記錄。

本篇文章就到這里了，希望能夠給你帶來(lái)幫助，也希望您能夠多多關(guān)注腳本之家的更多內(nèi)容!  

相關(guān)文章

最新評(píng)論