update token default exception return message (#65)

Author: wu-clan

backend/app/common/exception/errors.py

@@ -4,7 +4,7 @@
 
 from fastapi import HTTPException
 
-from backend.app.common.response.response_code import CodeEnum
+from backend.app.common.response.response_code import CustomCode
 
 
 class BaseExceptionMixin(Exception):
@@ -16,7 +16,14 @@ def __init__(self, *, msg: str = None, data: Any = None):
 
 
 class HTTPError(HTTPException):
-    pass
+    def __init__(self, *, code: int, msg: Any = None, headers: dict[str, Any] | None = None):
+        super().__init__(status_code=code, detail=msg, headers=headers)
+
+
+class CustomError(BaseExceptionMixin):
+    def __init__(self, *, error: CustomCode, data: Any = None):
+        self.code = error.code
+        super().__init__(msg=error.msg, data=data)
 
 
 class RequestError(BaseExceptionMixin):
@@ -54,21 +61,15 @@ def __init__(self, *, msg: str = 'Bad Gateway', data: Any = None):
         super().__init__(msg=msg, data=data)
 
 
-class CodeError(BaseExceptionMixin):
-    def __init__(self, *, error: CodeEnum, data: Any = None):
-        self.code = error.code
-        super().__init__(msg=error.msg, data=data)
-
-
 class AuthorizationError(BaseExceptionMixin):
     code = 401
 
     def __init__(self, *, msg: str = 'Permission denied', data: Any = None):
         super().__init__(msg=msg, data=data)
 
 
-class TokenError(BaseExceptionMixin):
+class TokenError(HTTPError):
     code = 401
 
-    def __init__(self, *, msg: str = 'Token is invalid', data: Any = None):
-        super().__init__(msg=msg, data=data)
+    def __init__(self, *, msg: str = 'Not authenticated', headers: dict[str, Any] | None = None):
+        super().__init__(code=self.code, msg=msg, headers=headers or {'WWW-Authenticate': 'Bearer'})

backend/app/common/response/response_code.py

@@ -3,9 +3,9 @@
 from enum import Enum
 
 
-class CodeEnum(Enum):
+class CustomCode(Enum):
     """
-    错误码
+    自定义错误码
     """
 
     CAPTCHA_ERROR = (40001, '图形验证码错误')