Django REST framework 異常處理
更新時間：2021年07月05日 09:39:58 作者：西紅柿蛋炒飯
本文將結合實例代碼，介紹Django REST framework 異常處理，文中通過示例代碼介紹的非常詳細，需要的朋友們下面隨著小編來一起學習學習吧
寫在前面
這兩天一直在思索關于 DRF 還有哪些是項目必備的而且還沒有說到的基礎性的知識。這不昨天寫到日志相關的功能就直接想到還有異常處理相關的功能，其實在之前項目中初期是沒有統(tǒng)一的異常捕獲手段?？赡苁?DRF 自帶的異常 能滿足大多數(shù)功能，也可能是比較懶，就使用比較粗暴的方式，以狀態(tài)碼 500 的方式去拋出異常，然后在日志中可以看到所有的異常信息。這么做呢，代碼其實是不夠健壯的，前端在調用的時候莫名的 500 也是不夠友好的，所以今天就補充一下異常相關的知識。
DRF異常處理
1. DRF 常見的異常
- AuthenticationFailed/ NotAuthenticated 一般該異常狀態(tài)碼為"401 Unauthenticated"，主要是沒有登錄鑒權的時候會返回，可以用在自定義登錄的時候。
- PermissionDenied 一般用在鑒權時候使用，一般狀態(tài)碼為"403 Forbidden"。
- ValidationError 一般狀態(tài)碼為"400 Bad Request"，主要是 serializers 中對字段的校驗，比如對字段類型的校驗、字段長度的校驗以及自定義字段格式的校驗。
2. 自定義異常
這里對異常的定義主要的想法來自 ValidationError，統(tǒng)一異常返回的格式，方便前端統(tǒng)一處理類似異常。
自定義異常
# 新建 utils/custom_exception.py
class CustomException(Exception):
_default_code = 400
def __init__(
self,
message: str = "",
status_code=status.HTTP_400_BAD_REQUEST,
data=None,
code: int = _default_code,
):
self.code = code
self.status = status_code
self.message = message
if data is None:
self.data = {"detail": message}
else:
self.data = data
def __str__(self):
return self.message
自定義異常處理
# utils/custom_exception.py
from rest_framework.views import exception_handler
def custom_exception_handler(exc, context):
# Call REST framework's default exception handler first,
# to get the standard error response.
# 這里對自定義的 CustomException 直接返回，保證系統(tǒng)其他異常不受影響
if isinstance(exc, CustomException):
return Response(data=exc.data, status=exc.status)
response = exception_handler(exc, context)
return response
配置自定義異常處理類
REST_FRAMEWORK = {
# ...
"EXCEPTION_HANDLER": "utils.custom_exception.custom_exception_handler",
}
3. 使用自定義異常
使用之前文章的接口用來測試自定義異常的處理
class ArticleViewSet(viewsets.ModelViewSet):
"""
允許用戶查看或編輯的API路徑。
"""
queryset = Article.objects.all()
serializer_class = ArticleSerializer
@action(detail=False, methods=["get"], url_name="exception", url_path="exception")
def exception(self, request, *args, **kwargs):
# 日志使用 demo
logger.error("自定義異常")
raise CustomException(data={"detail": "自定義異常"})
4. 驗證結果
$ curl -H 'Accept: application/json; indent=4' -u admin:admin http://127.0.0.1:8000/api/article/exception/
{
"detail": "自定義異常"
}
異常處理進階
上面的代碼雖說是可以滿足90%的需求，但是錯誤的定義太泛泛。難以集中定義管理錯誤，與常見項目中自定義的異常比較優(yōu)點就是靈活，但是隨著代碼中拋出的異常越來越多加之散落在各個角落，不利于更新維護。所以下面對修改一下代碼，對異常有統(tǒng)一的定義，同時也支持自定義返回HTTP狀態(tài)碼。
1. 修改自定義異常
# utils/custom_exception.py
class CustomException(Exception):
# 自定義code
default_code = 400
# 自定義 message
default_message = None
def __init__(
self,
status_code=status.HTTP_400_BAD_REQUEST,
code: int = None,
message: str = None,
data=None,
):
self.status = status_code
self.code = self.default_code if code is None else code
self.message = self.default_message if message is None else message
if data is None:
self.data = {"detail": self.message, "code": self.code}
else:
self.data = data
def __str__(self):
return str(self.code) + self.message
2. 自定義更多異常
class ExecuteError(CustomException):
"""執(zhí)行出錯"""
default_code = 500
default_message = "執(zhí)行出錯"
class UnKnowError(CustomException):
"""執(zhí)行出錯"""
default_code = 500
default_message = "未知出錯"
3. 新增測試接口
class ArticleViewSet(viewsets.ModelViewSet):
"""
允許用戶查看或編輯的API路徑。
"""
queryset = Article.objects.all()
serializer_class = ArticleSerializer
@action(detail=False, methods=["get"], url_name="exception", url_path="exception")
def exception(self, request, *args, **kwargs):
# 日志使用 demo
logger.error("自定義異常")
raise CustomException(data={"detail": "自定義異常"})
@action(detail=False, methods=["get"], url_name="unknown", url_path="unknown")
def unknown(self, request, *args, **kwargs):
# 日志使用 demo
logger.error("未知錯誤")
raise UnknownError()
@action(detail=False, methods=["get"], url_name="execute", url_path="execute")
def execute(self, request, *args, **kwargs):
# 日志使用 demo
logger.error("執(zhí)行錯誤")
raise ExecuteError()
4. 驗證結果
curl -H 'Accept: application/json; indent=4' -u admin:admin http://127.0.0.1:8000/api/article/unknown/
{
"detail": "未知出錯",
"code": 500
}
$ curl -H 'Accept: application/json; indent=4' -u admin:admin http://127.0.0.1:8000/api/article/execute/
{
"detail": "執(zhí)行出錯",
"code": 500
}
總結
需要注意自定義的異常處理函數(shù)需要在處理完成自定義異常后繼續(xù)執(zhí)行 rest_framework.views.exception_handler，因為這里的執(zhí)行仍然需要兼容已有的異常處理；下面貼一下 DRF 有關的異常處理邏輯。
該處理函數(shù)默認處理 APIException以及 Django 內部的 Http404 PermissionDenied，其他的異常會返回 None ，會觸發(fā) DRF 500 的錯誤。
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
參考資料
Django REST framework 異常文檔
Django 異常文檔
到此這篇關于Django REST framework 異常處理的文章就介紹到這了,更多相關Django REST framework 異常內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！
相關文章
這篇文章主要介紹了python對指定目錄下文件進行批量重命名的方法,涉及Python中replace及join方法的使用技巧,非常具有實用價值,需要的朋友可以參考下2015-04-04
django中上傳圖片分頁三級聯(lián)動效果的實現(xiàn)代碼
這篇文章主要介紹了django中上傳圖片分頁三級聯(lián)動效果的實現(xiàn)代碼，非常不錯，具有一定的參考借鑒價值，需要的朋友參考下吧2019-08-08
這篇文章主要介紹了Python BautifulSoup 節(jié)點信息，文章圍繞主題展開詳細的內容介紹，具有一定的參考價值，需要的小伙伴可以參考一下2022-08-08
這篇文章主要介紹了python基礎詳解之if循環(huán)語句,文中有非常詳細的代碼示例,對正在學習python的小伙伴們有很好的幫助需要的朋友可以參考下2021-04-04
ansible-playbook實現(xiàn)自動部署KVM及安裝python3的詳細教程
這篇文章主要介紹了ansible-playbook實現(xiàn)自動部署KVM及安裝python3的詳細教程，本文給大家介紹的非常詳細，對大家的學習或工作具有一定的參考借鑒價值,需要的朋友可以參考下2020-05-05
當你寫好一個python應用以后（有可能是命令行，有可能是GUI），你或許希望分享給他人使用，而別人可能并沒有python環(huán)境，那么我們需要尋找一種方法生成可執(zhí)行文件（比如Windows上的exe或macOs上的app）2021-10-10
這篇文章主要介紹了Python中反轉二維數(shù)組的行和列問題，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教2023-01-01
這篇文章主要介紹了Python中統(tǒng)計函數(shù)運行耗時的方法,涉及Python時間操作的相關技巧,非常簡單實用,需要的朋友可以參考下2015-05-05
根據(jù)一些我收到的反饋，大家似乎有點混淆——Cython是用來生成 C 擴展到而不是獨立的程序的。所有的加速都是針對一個已經(jīng)存在的 Python 應用的一個函數(shù)進行的。2009-09-09
隨機數(shù)廣泛應用在科學研究,但是計算機無法產(chǎn)生真正的隨機數(shù),一般成為偽隨機數(shù),下面這篇文章主要給大家介紹了關于Pytorch隨機數(shù)生成常用的4種方法,需要的朋友可以參考下2023-05-05