Flask-SQLAlchemy ORM对象序列化问题及解决方案

在使用Python的Flask框架和SQLAlchemy ORM进行数据库操作时，经常会遇到对象序列化难题。本文将分析一个常见问题并提供解决方案。

问题：

使用Flask-SQLAlchemy定义的User模型，尝试通过User.query.all()获取所有用户数据，但直接返回给前端时，出现TypeError: Object of type User is not JSON serializable错误。这是因为Flask无法直接将SQLAlchemy的ORM对象转换为JSON格式。

示例代码片段：

提示词大师-python版

图片提示词生成器？不止如此。 马甲系统 —— 把脑海中的画面，翻译成AI能理解的专业表达。 用得越多，它越懂你：首次需要多问几句确认方向，用久了几乎一说就懂。 用得越多，它越快：缓存机制让后续对话越来越省。 RAG进化：成功案例持续入库，越跑越聪明。 输入「新手指南」查看完整功能介绍

下载

from sqlalchemy import Column, Integer, String
from lib.database import Base  # 假设lib.database包含数据库连接配置

class User(Base):
    __tablename__ = 'users' # 表名应与数据库中表名一致
    id = Column(Integer, primary_key=True)
    app_key = Column(String(50), unique=True)
    secret = Column(String(120), unique=True)

    def __init__(self, app_key=None, secret=None):
        self.app_key = app_key
        self.secret = secret

    def __repr__(self):
        return f'<User {self.app_key}>'

# ...后续代码，例如获取所有用户并返回

解决方案：

为了解决这个问题，需要将SQLAlchemy的ORM对象转换成可序列化的字典格式。在User模型中添加to_dict()方法：

class User(Base):
    # ... (之前的代码) ...

    def to_dict(self):
        return {
            "id": self.id,
            "app_key": self.app_key,
            "secret": self.secret,
        }

然后，在Flask的路由处理函数中，使用to_dict()方法将每个User对象转换为字典，再用jsonify函数转换为JSON响应：

from flask import jsonify, request

@app.route("/users")
def get_users():
    users = User.query.all()
    return jsonify([user.to_dict() for user in users])

这样，就成功地将SQLAlchemy ORM对象转换为JSON可序列化格式，避免了TypeError错误，实现安全可靠的数据传输，方便前端处理。