将查询结果转换为 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}