python – 如何使用Marshmallow序列化MongoDB ObjectId?

前端之家收集整理的这篇文章主要介绍了python – 如何使用Marshmallow序列化MongoDB ObjectId?前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

我正在使用棉花糖和mongoengine在Flask上构建和API.当我打电话并且ID应该被序列化时,我收到以下错误

TypeError: ObjectId('54c117322053049ba3ef31f3') is not JSON serializable

我看到了其他库的一些方法来覆盖ObjectId的处理方式.我还没有想到Marshmallow,有谁知道怎么做?

我的模型是:

class Process(db.Document):
    name = db.StringField(max_length=255,required=True,unique=True)
    created_at = db.DateTimeField(default=datetime.datetime.now,required=True)

我的序列化器:

class ProcessSerializer(Serializer):
    class Meta:
        fields = ("id","created_at","name")

并且观点:

class ProcessView(Resource):
    def get(self,id):
        process = Process.objects.get_or_404(id)
        return ProcessSerializer(process).data
最佳答案
当您将Meta.fields传递给模式时,Marshmallow会尝试为每个属性选择一个字段类型.由于它不知道ObjectId是什么,它只是将它传递给序列化的dict.当您尝试将其转储到JSON时,它不知道ObjectId是什么并引发错误.要解决这个问题,您需要告诉Marshmallow要用于id的字段.可以将BSON ObjectId转换为字符串,因此请使用String字段.

from marshmallow import Schema,fields

class ProcessSchema(Schema):
    id = fields.String()

    class Meta:
        additional =  ('created_at','name')

您还可以告诉Marshmallow对ObjectId类型使用哪个字段,这样您就不必每次都添加字段.

from bson import ObjectId
from marshmallow import Schema,fields

Schema.TYPE_MAPPING[ObjectId] = fields.String

猜你在找的Python相关文章