flask_marshmallow_sqlalchemy_model_json_serialization.py

1from flask import Flask, jsonify
2from flask_sqlalchemy import SQLAlchemy
3from flask_marshmallow import Marshmallow
4
5app = Flask(__name__)
6app.config["SQLALCHEMY_DATABASE_URI"] = "sqlite:///:memory:"
7
8db = SQLAlchemy(app)
9ma = Marshmallow(app)
10
11class User(db.Model):
12    id = db.Column(db.Integer, primary_key=True)
13    username = db.Column(db.String(80), unique=True)
14    email = db.Column(db.String(120), unique=True)
15
16    def __init__(self, username, email):
17        self.username = username
18        self.email = email
19
20class UserSchema(ma.SQLAlchemyAutoSchema):
21    class Meta:
22        model = User
23
24user_schema = UserSchema()
25users_schema = UserSchema(many=True)
26
27@app.route("/api/users/")
28def get_users():
29    all_users = User.query.all()
30    return jsonify(users_schema.dump(all_users))
31
32@app.route("/api/users/<id>")
33def user_detail(id):
34    user = User.query.get(id)
35    return user_schema.jsonify(user)
36
37if __name__ == "__main__":
38    with app.app_context():
39        db.create_all()
40    app.run(debug=True)