用棉花糖序列化SQLAlchemy

时间:2020-01-13 17:40:19

标签: python python-3.x flask flask-sqlalchemy marshmallow

我正在学习教程,并使用以下代码。我还使用Postman查看http://localhost:5000/planets的服务器状态,但是当我应该看到自己创建的行星的JSON数据时,出现 500 INTERNAL SERVER ERROR 。 在命令行中,我还看到: AttributeError:'list'对象没有属性'data'

我认为这可能与以下行有关:return jsonify(result.data),但我不确定。

from flask import Flask, jsonify, request
from flask_sqlalchemy import SQLAlchemy
from sqlalchemy import Column, Integer, String, Float
import os
from flask_marshmallow import Marshmallow
from marshmallow import Schema

app = Flask(__name__)

basedir = os.path.abspath(os.path.dirname(__file__))
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///C:/Users/krist/Py3/flask2/planets.db'

db = SQLAlchemy(app)
ma = Marshmallow(app)

@app.cli.command('db_create')
def db_create():
    db.create_all()
    print("DB Created")

@app.cli.command('db_seed')
def deb_seed():
    mercury = Planet(planet_name='Mercury',
                     planet_type='Class D',
                     home_star='Sol',
                     mass=3.25e23,
                     radius=1516,
                     distance=35.98e6)

    venus = Planet(planet_name='Venus',
                     planet_type='Class K',
                     home_star='Sol',
                     mass=8.95e24,
                     radius=3516,
                     distance=67.98e6)

    earth = Planet(planet_name='Earth',
                     planet_type='Class M',
                     home_star='Sol',
                     mass=5.97e24,
                     radius=3916,
                     distance=92.96e6)

    db.session.add(mercury)
    db.session.add(venus)
    db.session.add(earth)

    test_user = User(first_name='William',
                     last_name='Hershel',
                     email='test@test.com',
                     password='p@ssw0rd')

    db.session.add(test_user)
    db.session.commit()
    print("DB Seeded")

@app.route('/planets', methods=['GET'])
def planets():
    planets_list = Planet.query.all()
    result = planets_schema.dump(planets_list)
    return jsonify(result.data)


class User(db.Model):
    __tablename__ = 'users'
    id = Column(Integer, primary_key=True)
    first_name = Column(String)
    last_name = Column(String)
    email = Column(String, unique=True)
    password = Column(String)

class Planet(db.Model):
    __tablename__ = 'planets'
    planet_id = Column(Integer, primary_key=True)
    planet_name = Column(String)
    planet_type = Column(String)
    home_star = Column(String)
    mass = Column(Float)
    radius = Column(Float)
    distance = Column(Float)


class UserSchema(ma.Schema):
    class Meta:
        fields = ('id', 'first_name', 'last_name', 'email', 'password')

class PlanetSchema(ma.Schema):
    class Meta:
        fields = ('planet_id', 'planet_name', 'planet_type', 'home_star', 'mass', 'radius', 'distance')

user_schema = UserSchema()
users_schema = UserSchema(many=True)

planet_schema = PlanetSchema()
planets_schema = PlanetSchema(many=True)


if __name__ == '__main__':
    app.run()

1 个答案:

答案 0 :(得分:3)

代替

result = planets_schema.dump(planets_list)
return jsonify(result.data)

尝试

result = planets_schema.dump(planets_list)
return jsonify(result)

为什么有效:

这里您正在查询Planet Mapper,以返回Planet ORM对象的列表

planets_list = Planet.query.all()

然后,棉花糖模式用于封送或将ORM对象转换为python字典对象。这是封送处理的基本原理-当数据将要传输或存储时,将数据从一种格式转换为另一种格式。因此,在这种情况下,您可以将数据从SQLAlchemy ORM对象列表转换为Python字典对象列表。

result = planets_schema.dump(planets_list)

现在您有了result(可以更恰当地是包含字典对象列表的名称results。 然后,您尝试访问此列表对象上的data变量。但是,Python列表没有data变量,因此会出现错误。

return jsonify(result.data)

flask的jsonify方法接受字典列表作为输入,因此只需将此行修改为以下内容即可:

return jsonify(result)