本文深入探讨了在marshmallow中如何将模型实例的简单字符串字段序列化为嵌套字典结构。通过详细分析`fields.method`的使用，结合代码示例，教程展示了如何灵活地转换数据，以满足api响应的特定格式要求，从而提升数据结构化和可读性。
引言：Marshmallow中模型实例字段的包装需求在构建RESTful API时，我们经常需要将数据库模型中的数据序列化为特定的JSON格式。一个常见的需求是，模型实例中某个字段（例如，一个关联对象的ID，如parent_id）存储为简单的字符串或整数，但在API响应中，我们希望将其包装成一个嵌套的字典结构，例如将”123-345″序列化为{“id”: “123-345”}。这种包装有助于提高API响应的清晰度、可读性，并使其与其他嵌套资源保持一致。
问题分析与传统方法局限对于上述需求，初学者可能会尝试使用fields.Nested结合一个辅助Schema。例如，定义一个IdSchema来表示{“id”: “…”}结构，然后尝试在主Schema中使用fields.Nested(IdSchema)。然而，如果模型实例的parent属性直接就是一个字符串ID（如”123-345″），fields.Nested(IdSchema)将无法直接处理，因为它期望接收一个字典或对象，而不是一个简单的字符串。
虽然可以在IdSchema中使用@pre_dump来预处理数据，但这种方法通常用于在序列化整个IdSchema的输入之前进行转换。如果主Schema直接将一个字符串赋给fields.Nested字段，Marshmallow在类型检查阶段就会报错，因为字符串不符合嵌套Schema的预期输入类型。因此，我们需要一种更直接的方式来在主Schema中将原始字符串转换为目标嵌套字典。
推荐解决方案：使用 fields.MethodMarshmallow提供了fields.Method，这是一个非常强大且灵活的字段类型，允许我们通过定义一个方法来动态地生成字段的输出值。这完美契合了将模型实例中的简单字段转换为复杂嵌套结构的需求。
核心思想fields.Method接收一个字符串参数，该字符串是Schema类中一个方法的名称。这个方法将接收当前正在被序列化的模型实例作为参数，并返回该字段的最终值。
示例代码
首先，我们定义一个简单的模型类来模拟我们的数据：
from marshmallow import Schema, fieldsclass 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""
接下来，我们创建UserSchema并使用fields.Method来处理parent_id：
豆包大模型
字节跳动自主研发的一系列大型语言模型
834 查看详情 
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}
代码解析
parent = fields.Method(“get_parent_wrapped_id”, dump_only=True):我们定义了一个名为parent的字段，它将是最终输出JSON中的键。fields.Method指示Marshmallow在序列化时调用UserSchema类中的get_parent_wrapped_id方法来获取parent字段的值。dump_only=True是一个重要的参数，它表示这个字段只用于序列化（dump），而不用于反序列化（load）。这通常适用于由多个源数据生成或经过复杂转换的字段。def get_parent_wrapped_id(self, obj: UserModel)::这个方法是fields.Method引用的实际逻辑。self指向当前的UserSchema实例，允许你访问Schema的配置或其他方法。obj是当前正在被序列化的模型实例（在这里是UserModel的实例）。通过obj.parent_id，我们可以访问到原始的字符串ID。方法内部的逻辑很简单：如果obj.parent_id存在，就将其包装成{“id”: obj.parent_id}的字典形式返回；否则返回None，这样在JSON中parent字段将为null。
fields.Function 替代方案除了fields.Method，Marshmallow还提供了fields.Function。它的用法与fields.Method类似，但它接受一个可调用对象（如一个独立的函数）而不是一个方法名字符串。当转换逻辑不依赖于Schema实例的self上下文时，fields.Function是一个很好的选择。
# ... (UserModel 定义同上)def wrap_parent_id_function(obj: UserModel): """ 一个独立的函数，用于包装 parent_id。 """ if obj.parent_id: return {"id": obj.parent_id} return Noneclass 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'}}
fields.Function的优势在于可以将转换逻辑从Schema类中分离出来，提高代码的模块化和复用性。
注意事项与最佳实践
字段命名灵活性： 使用fields.Method或fields.Function时，Schema中的字段名（如parent）可以与模型实例中的原始属性名（如`parent_
以上就是Marshmallow 教程：将模型实例字段包装为嵌套字典结构的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/574472.html
微信扫一扫 
支付宝扫一扫 
