Marshmallow 教程：将模型实例字段包装为嵌套字典结构

from marshmallow import Schema, fields

class UserModel:
    """
    模拟用户模型，parent_id存储为字符串。
    """
    def __init__(self, name: str, parent_id: str = None):
        self.name = name
        self.parent_id = parent_id

    def __repr__(self):
        return f"<UserModel(name='{self.name}', parent_id='{self.parent_id}')>"

class UserSchema(Schema):
    """
    用户序列化Schema，将parent_id包装为嵌套字典。
    """
    name = fields.String(required=True)
    # 使用 fields.Method 来调用 Schema 内部的方法 get_parent_wrapped_id
    # dump_only=True 表示此字段仅用于序列化，不用于反序列化
    parent = fields.Method("get_parent_wrapped_id", dump_only=True)

    def get_parent_wrapped_id(self, obj: UserModel):
        """
        根据模型实例的 parent_id 属性生成嵌套的 'parent' 字典。
        obj 是当前被序列化的 UserModel 实例。
        """
        if obj.parent_id:
            return {"id": obj.parent_id}
        return None # 如果没有 parent_id，则返回 None

# 示例用法
user_instance_with_parent = UserModel(name="John Doe", parent_id="123-345")
user_instance_without_parent = UserModel(name="Jane Smith")

schema = UserSchema()

# 序列化带 parent_id 的实例
result_with_parent = schema.dump(user_instance_with_parent)
print("序列化结果 (带 parent_id):", result_with_parent)
# 预期输出: {'name': 'John Doe', 'parent': {'id': '123-345'}}

# 序列化不带 parent_id 的实例
result_without_parent = schema.dump(user_instance_without_parent)
print("序列化结果 (不带 parent_id):", result_without_parent)
# 预期输出: {'name': 'Jane Smith', 'parent': None}

# ... (UserModel 定义同上)

def wrap_parent_id_function(obj: UserModel):
    """
    一个独立的函数，用于包装 parent_id。
    """
    if obj.parent_id:
        return {"id": obj.parent_id}
    return None

class UserSchemaFunction(Schema):
    """
    使用 fields.Function 的 Schema 示例。
    """
    name = fields.String(required=True)
    parent = fields.Function(wrap_parent_id_function, dump_only=True)

# 示例用法
user_instance_func = UserModel(name="Alice Johnson", parent_id="678-901")
schema_func = UserSchemaFunction()
result_func = schema_func.dump(user_instance_func)
print("序列化结果 (使用 fields.Function):", result_func)
# 预期输出: {'name': 'Alice Johnson', 'parent': {'id': '678-901'}}