鍍金池/ 問答/Java  Python  數(shù)據(jù)庫/ 使用 flask-sqlalchemy 如何級聯(lián)創(chuàng)建對象？

使用 flask-sqlalchemy 如何級聯(lián)創(chuàng)建對象？

問題描述

比如我想在 User 對象成功 session.commit() 之后創(chuàng)建一個 Wallet 對象，Wallet 對象創(chuàng)建成功之后可能又需要創(chuàng)建另一個對象。用函數(shù)把創(chuàng)建多個對象的代碼封裝起來這種方式我不想用，因為 User 對象可能不是在代碼中顯式創(chuàng)建的，有時是通過 load fixtures 的方式(我搞了個類似 django 的 load fixtures 功能)創(chuàng)建的。

嘗試過的方法

我試著利用 flask-sqlalchemy 的 models_committed 信號來做觸發(fā)，比如這樣：

    @models_committed.connect_via(app)
    def on_models_committed(sender, changes):
        for obj, change in changes:
            if change == 'insert' and hasattr(obj, '__commit_insert__'):
                obj.__commit_insert__()
            elif change == 'update' and hasattr(obj, '__commit_update__'):
                obj.__commit_update__()
            elif change == 'delete' and hasattr(obj, '__commit_delete__'):
                obj.__commit_delete__()

然后在 User model 中添加 commit_insert 方法，在該方法中創(chuàng)建 Wallet 對象，比如這樣：

def __commit_insert__(self):
    wallet = Wallet(name="my_wallet")
    db.session.add(wallet)
    db.session.commit()

實際看到的錯誤信息又是什么？

然而，在創(chuàng)建 Wallet 對象 commit 的時候，報了這么個錯誤：

This session is in 'committed' state; no further SQL can be emitted within this transaction.

講道理，臨時搞搞 flask 還沒倒出功夫縷源碼看看 flask-sqlalchemy 怎么個邏輯，所以先問一下大家，看看有沒有遇到過這種情況，類似的場景在 flask 中是否有什么通用的處理方式？

另外，
我是有去 stack-overflow 上先看一眼的，結(jié)果這個答案講道理挺尷尬的.

回答
編輯回答
別瞎鬧

試一下SQLAlchemy的event吧（http://docs.sqlalchemy.org/en...）
一個簡單的使用例子

import sqlalchemy
from sqlalchemy import event
from models import Wallet

class User(db.Model):
  name = sqlalchemy.column(s.String)

  @staticmethod
  def after_create(mapper, connection, target):
    wallet = Wallet()
    db.session.add(wallet)
    db.session.commit()
    
event.listen(User, 'after_insert', User.after_create)
2018年9月13日 13:44
編輯回答
乞許

這個問題可以通過直接執(zhí)行 sql 語句來搞

import sqlalchemy
from sqlalchemy import event
from models import Wallet

class User(db.Model):
  name = sqlalchemy.column(s.String)

  @staticmethod
  def after_create(mapper, connection, target):
    wallet_table = Wallet.__table__
    connection.execute(
        wallet_table.insert().values(name='my wallet')
        )
    
event.listen(User, 'after_insert', User.after_create)

參考了 @低默 給的答案，雖然也會報錯，但他的這種 sqlalchemy event 寫法很方便獲取到 connection 對象，而我原問題中寫的 Flask Signaling 方式中如果想獲得 connection 需要這樣寫：

with db.engine.connect() as connection:
    ...

另外還有個問題需要注意就是如果需要執(zhí)行多個 sql 語句，注意使用事務(wù)，例如：

trans = connection.begin()
try:
    ...
except exc.SQLAlchemyError:
    trans.rollback()
        raise
    else:
        trans.commit()

參考鏈接

2018年8月17日 14:42