转载 

sqlAlchemy中的ORM 对象转换为JSON

分类:python    731人阅读    IT小君  2021-07-09 23:02

在用sqlAlchemy写web应用的时候,经常会用json进行通信,跟json最接近的对象就是dict,有时候操作dict也会比操作ORM对象更为方便,毕竟不用管数据库session的状态了。

假设数据库里有一张post表,其中一种方法就是

p = session.query(Post).first()
p.__dict__

但由于p是sqlAlchemy的对象,所以p.__dict__中会有一些其他的属性比如_sa_instance这种我们不需要关注的

那么我们可以给model的基类加一个方法,假设models.py中原来是这样

Base = sqlalchemy.ext.declarative.declarative_base()

class Post(Base):
    __tablename__ = 'post'
    id = Column(Integer, primary_key=True)
    title = Column(String)

那么我们可以加一个to_dict()方法到Base类中

def to_dict(self):
    return {c.name: getattr(self, c.name, None) for c in self.__table__.columns}

Base.to_dict = to_dict

这样就可以

p = session.query(Post).first()
p.to_dict()

当然,如果model没有和table绑定的话model里是没有__table__的信息的,可能也会出问题,不过我目前觉得这样最方便了

DateTime 之类的,可能还需要转换下

def to_dict(self):
    def convert_datetime(value):
        if value:
            return value.strftime("%Y-%m-%d %H:%M:%S")
        else:
            return ""

    for col in self.__table__.columns:
        if isinstance(col.type, DateTime):
            value = convert_datetime(getattr(self, col.name))
        elif isinstance(col.type, Numeric):
            value = float(getattr(self, col.name))
        else:
            value = getattr(self, col.name)
        yield (col.name, value)

def to_json(self):
    d = dict(self.__todict__())
    return json.dumps(d)

以上每个model都要添加方法 可以额外写方法来装换减少重复添加方法:

from json import dumps
from sqlalchemy.orm import class_mapper

def serialize(model):
  """Transforms a model into a dictionary which can be dumped to JSON."""
  # first we get the names of all the columns on your model
  columns = [c.key for c in class_mapper(model.__class__).columns]
  # then we return their values in a dict
  return dict((c, getattr(model, c)) for c in columns)

# we can then use this for your particular example
serialized_labels = [
  serialize(label)
  for label in session.query(LabelsData).filter(LabelsData.deleted == False)
]
your_json = dumps(serialized_labels)

以下本站当前使用模式:

class CommentTemp(db.Model):
    __tablename__ = 'comment_temp'
    CommentID = db.Column(db.Integer, autoincrement=True, primary_key=True)
    ResourcesID = db.Column(db.Integer)
    Content = db.Column(db.Text)
    CreateTime = db.Column(db.DateTime)
    ParentID = db.Column(db.Integer)
    UserID = db.Column(db.Integer)
    TypeID = db.Column(db.Integer)
    TypeClass = db.Column(db.String(200))
    fro = db.Column('from', db.Integer)
    to = db.Column(db.Integer)
    isreply = db.Column(db.BOOLEAN)
    email = db.Column(db.String(200))
    floor = db.Column(db.Integer)

    # 多个对象
    def dobule_to_dict(self):
        result = {}
        for key in self.__mapper__.c.keys():
            if getattr(self, key) is not None:
                result[key] = str(getattr(self, key))
            else:
                result[key] = getattr(self, key)
        return result
支付宝打赏 微信打赏

如果文章对你有帮助,欢迎点击上方按钮打赏作者

 工具推荐 更多»