fix: 修复响应模型中msg=None时的Pydantic验证错误

Author: JiayuXu

src/schemas/base.py

@@ -11,6 +11,9 @@ def __init__(
         data: Any | None = None,
         **kwargs,
     ):
+        # 确保msg不为None
+        if msg is None:
+            msg = "OK"
         content = {"code": code, "msg": msg, "data": data}
         content.update(kwargs)
         super().__init__(content=content, status_code=code)
@@ -24,6 +27,9 @@ def __init__(
         data: Any | None = None,
         **kwargs,
     ):
+        # 确保msg不为None
+        if msg is None:
+            msg = "Error"
         content = {"code": code, "msg": msg, "data": data}
         content.update(kwargs)
         super().__init__(content=content, status_code=code)
@@ -40,6 +46,9 @@ def __init__(
         page_size: int = 20,
         **kwargs,
     ):
+        # 确保msg不为None
+        if msg is None:
+            msg = "OK"
         content = {
             "code": code,
             "msg": msg,

src/schemas/response.py

@@ -4,7 +4,7 @@
 """
 from typing import Any, Generic, TypeVar
 from datetime import datetime
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, field_validator
 
 T = TypeVar("T")
 
@@ -15,6 +15,12 @@ class ResponseBase(BaseModel, Generic[T]):
     msg: str = Field(default="OK", description="响应消息")
     data: T | None = Field(default=None, description="响应数据")
 
+    @field_validator("msg", mode="before")
+    @classmethod
+    def set_default_msg(cls, v):
+        """当msg为None时，设置默认值"""
+        return "OK" if v is None else v
+
     class Config:
         json_schema_extra = {
             "example": {
@@ -34,6 +40,12 @@ class PageResponse(BaseModel, Generic[T]):
     page: int = Field(default=1, description="当前页码")
     page_size: int = Field(default=20, description="每页数量")
 
+    @field_validator("msg", mode="before")
+    @classmethod
+    def set_default_msg(cls, v):
+        """当msg为None时，设置默认值"""
+        return "OK" if v is None else v
+
     class Config:
         json_schema_extra = {
             "example": {