shanghui 3 月之前
父节点
当前提交
cacfebd3bb
共有 1 个文件被更改,包括 6 次插入1 次删除
  1. 6 1
      src/core/callcenter/exception.py

+ 6 - 1
src/core/callcenter/exception.py

@@ -5,7 +5,12 @@ from src.core.callcenter.enumeration import BizErrorCode
 
 class BizException(Exception):
     def __init__(self, message, status_code=400):
-        if isinstance(message, tuple):
+        if isinstance(message, BizErrorCode):
+            # 处理 BizErrorCode 对象
+            status_code = message.code
+            message = message.message
+        elif isinstance(message, tuple):
+            # 处理直接传入的元组
             status_code, message = message
         super().__init__(message)
         self.message = message