Flask Marshmallow / SqlAlchemy:序列化多对多关系

发布于 2021-01-29 15:58:44

我正在使用Flask,flask-sqlalchemy和flask-marshmallow构建一个小的REST
api。对于某些请求,我想返回一个由我的sqlalchemy对象组成的json序列化响应。但是,当使用多对多关系/辅助表时,我无法使序列化与急切加载的sqlalchemy对象一起使用。

这是一个简单的示例,或多或少地从flask-marshmallow docs复制/粘贴:

from flask import Flask
from flask_sqlalchemy import SQLAlchemy
from flask_marshmallow import Marshmallow
from sqlalchemy.orm import joinedload

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'

# Order matters: Initialize SQLAlchemy before Marshmallow
db = SQLAlchemy(app)
ma = Marshmallow(app)

secondary_foo = db.Table('secondary_foo',
                            db.Column('author_id', db.Integer, db.ForeignKey('author.id')),
                            db.Column('book_id', db.Integer, db.ForeignKey('book.id')))

class Author(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255))

class Book(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    title = db.Column(db.String(255))
    authors = db.relationship('Author', secondary="secondary_foo", backref='books', lazy="joined")

class AuthorSchema(ma.ModelSchema):
    class Meta:
        model = Author

class BookSchema(ma.ModelSchema):
    #authors = ma.Nested(AuthorSchema) <-- Doesn't work, authors will be serialized to empty json object, instead of list of ids
    class Meta:
        model = Book


db.drop_all()
db.create_all()
author_schema = AuthorSchema()
book_schema = BookSchema()
author = Author(name='Chuck Paluhniuk')
book = Book(title='Fight Club')
book.authors.append(author)
db.session.add(author)
db.session.add(book)
db.session.commit()

s = BookSchema(many=True)

根据上述代码,我可以热切地加载书籍并获得authors对象。但是当序列化深层对象时,序列化成一个ID列表:

print(Book.query.filter(1==1).options(joinedload('authors')).all()[0].authors)
//--> [<__main__.Author object at 0x1043a0dd8>]

print(s.dump(Book.query.filter(1==1).options(joinedload('authors')).all()).data)
//--> [{'authors': [1], 'title': 'Fight Club', 'id': 1}]

这是我想要的结果:

print(s.dump(Book.query.filter(1==1).options(joinedload('authors')).all()).data)
//--> [{'authors': [{'name':'Chuck Paluhniuk', 'id':'1'}], 'title': 'Fight Club', 'id': 1}]

我怎么做?

关注者
0
被浏览
145
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

    在中BookSchema,您将希望Nested
    authors像以前一样添加一个字段(但已将其注释掉),但是您将要使用many关键字arguments来指定它将是一个列表。

    class BookSchema(ma.ModelSchema):
    
        # A list of author objects
        authors = ma.Nested(AuthorSchema, many=True)
    
        class Meta:
            model = Book
    


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看