輕鬆將sqlAlchemy中的對象轉換爲dict

在用sqlAlchemy寫web應用的時候,常常會用json進行通訊,跟json最接近的對象就是dict,有時候操做dict也會比操做ORM對象更爲方便,畢竟不用管數據庫session的狀態了。web

假設數據庫裏有一張post表,其中一種方法就是sql

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

但因爲p是sqlAlchemy的對象,因此p.__dict__中會有一些其餘的屬性好比_sa_instance這種咱們不須要關注的數據庫

那麼咱們能夠給model的基類加一個方法,假設models.py中原來是這樣json

Base = sqlalchemy.ext.declarative.declarative_base()

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

那麼咱們能夠加一個to_dict()方法到Base類中session

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

Base.to_dict = to_dict

這樣就能夠post

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

固然,若是model沒有和table綁定的話model裏是沒有__table__的信息的,可能也會出問題,不過我目前以爲這樣最方便了code

相關文章
相關標籤/搜索