将查询结果转换为 dict

首先是示例的设置:

import datetime as dt
from sqlalchemy import Column, Date, Integer, Text, create_engine, inspect
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()
Session = sessionmaker()

class User(Base):
    __tablename__ = 'users'

    id = Column(Integer, primary_key=True)
    name = Column(Text, nullable=False)
    birthday = Column(Date)

engine = create_engine('sqlite://')
Base.metadata.create_all(bind=engine)
Session.configure(bind=engine)

session = Session()
session.add(User(name='Alice', birthday=dt.date(1990, 1, 1)))
session.commit()

如果你单独查询列,则该行是 KeyedTuple,其中包含 _asdict 方法。方法名称以单个下划线开头,以匹配 namedtuple API(它不是私有的!)。

query = session.query(User.name, User.birthday)
for row in query:
    print(row._asdict())

使用 ORM 检索对象时,默认情况下不可用。应该使用 SQLAlchemy 检查系统

def object_as_dict(obj):
    return {c.key: getattr(obj, c.key)
            for c in inspect(obj).mapper.column_attrs}

query = session.query(User)
for user in query:
    print(object_as_dict(user))

在这里,我们创建了一个执行转换的函数,但是一个选项是将方法添加到基类。

你可以从自己的类创建它,而不是使用上面的 declarative_base

from sqlalchemy.ext.declarative import as_declarative

@as_declarative()
class Base:
    def _asdict(self):
        return {c.key: getattr(self, c.key)
                for c in inspect(self).mapper.column_attrs}