1. ホーム
  2. python

[解決済み] SQLAlchemy の行オブジェクトを Python の dict に変換するには?

2022-02-27 10:08:52

質問

カラム名と値のペアを反復処理する簡単な方法はありますか?

私のSQLAlchemyのバージョンは0.5.6です。

以下は dict(row) :

import sqlalchemy
from sqlalchemy import *
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker

print "sqlalchemy version:",sqlalchemy.__version__ 

engine = create_engine('sqlite:///:memory:', echo=False)
metadata = MetaData()
users_table = Table('users', metadata,
     Column('id', Integer, primary_key=True),
     Column('name', String),
)
metadata.create_all(engine) 

class User(declarative_base()):
    __tablename__ = 'users'
    
    id = Column(Integer, primary_key=True)
    name = Column(String)
    
    def __init__(self, name):
        self.name = name

Session = sessionmaker(bind=engine)
session = Session()

user1 = User("anurag")
session.add(user1)
session.commit()

# uncommenting next line throws exception 'TypeError: 'User' object is not iterable'
#print dict(user1)
# this one also throws 'TypeError: 'User' object is not iterable'
for u in session.query(User).all():
    print dict(u)

このコードを私のシステムで実行すると、出力されます。

Traceback (most recent call last):
  File "untitled-1.py", line 37, in <module>
    print dict(u)
TypeError: 'User' object is not iterable

解決方法は?

良い回答が得られなかったので、これを使います。

def row2dict(row):
    d = {}
    for column in row.__table__.columns:
        d[column.name] = str(getattr(row, column.name))

    return d

編集:上記の関数が長すぎて好みに合わない場合は、ワンライナー(python 2.7+)をご利用ください。

row2dict = lambda r: {c.name: str(getattr(r, c.name)) for c in r.__table__.columns}