使用marshmallow序列化两个嵌套模式

时间:2016-09-04 14:59:28

标签: python flask-sqlalchemy marshmallow

我对python很新。我有两个SQLAlchemy模型如下:

class listing(db.Model):
 id = db.Integer(primary_key=True)
 title = db.String()
 location_id = db.Column(db.Integer, db.ForeignKey('location.id'))
 location = db.relationship('Location', lazy='joined')

class location(db.Model):
 id = db.Integer(primary_key=True)
 title = db.String()

我有两个Marshmallow架构类:

class ListingSchema(Schema):
 id = fields.Int()
 title = fields.Str()
 location_id = fields.Int()

class LocationSchema(Schema):
 id = fields.Int()
 title = fields.Str()

我创建了一个嵌套的架构类,如:

class NestedSchema(Schema):
 listing = fields.Nested(ListingSchema)
 location fields.Nested(LocationSchema)

我正在进行连接查询,如:

listing,location = db.session.query(Listing,Location)\
                            .join(Location, and_(Listing.location_id == Location.id))\
                            .filter(Listing.id == listing_id).first()

数据在我检查过的对象中加载。如何解析这个架构? 我试过了

result,errors = nested_listing_Schema(listing,location)

这给出了错误:“列表对象不可迭代。”

1 个答案:

答案 0 :(得分:8)

右边是使用您创建的NestedSchema而不是nested_schema的类,请执行以下操作:

result,errors = NestedSchema().dump({'listing':listing,'location':location})

结果将是:

dict: {
       u'listing': {u'id': 8, u'title': u'foo'},
       u'location': {u'id': 30, u'title': u'bar'}
      }

但我不明白为什么你想制作一个" NestedSchema",我认为你可以用另一种方式来做。

首先,忘记课程" NestedSchema"。

之后,更改您的" ListingSchema",如下所示:

class ListingSchema(Schema):
    id = fields.Int()
    title = fields.Str()
    location_id = fields.Int()
    location = fields.Nested("LocationSchema") #The diff is here

现在你可以这样做:

listing = db.session.query(Listing).get(listing_id) # Suppose listing_id = 8
result,errors = ListingSchema().dump(listing)
print result

结果将是:

dict: {
        u'id': 8, u'title': u'foo', u'location_id': 30, u'location': {u'id': 30, u'title': u'bar'}
      }

请注意,现在," location"是"列出"。

的财产

你仍然可以创建一个Two-Way Nesting,只需在Listing(模型)中添加一个backref,并将ListingSchema添加为嵌套在LocationSchema中

class Listing(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(45), nullable=False)
    location_id = db.Column(db.Integer, db.ForeignKey('location.id'))
    location = db.relationship('Location', lazy='joined', backref="listings") #the diff is here

class LocationSchema(Schema):
    id = fields.Int()
    title = fields.Str()
    listings = fields.Nested("ListingSchema", many=True, exclude=("location",)) #The property name is the same as in bakcref

many=True是因为我们有一对多的关系。 exclude=("location")是为了避免递归异常。

现在我们也可以按位置搜索。

location = db.session.query(Location).get(location_id) # Suppose location_id = 30
result,errors = LocationSchema().dump(location)
print result

dict: {u'id': 30, u'title': u'bar', 
       u'listings': [
             {u'location_id': 30, u'id': 8, u'title': u'foo'},
             {u'location_id': 30, u'id': 9, u'title': u'foo bar baz'},
             ...
             ]
       }

您可以查看有关它的文档here