在SQLAlchemy ORM中将对象标记为干净

时间:2017-01-26 17:43:40

标签: python mysql orm sqlalchemy

有没有办法在SQLAlchemy ORM中将对象明确标记为 clean

这部分与之前关于bulk update strategies的问题有关。

我希望在before_flush事件监听器中标记一堆对象,因为实际上不需要刷新。这是因为他们通过其他方式手动与数据库同步。

我已经尝试了下面的策略,但是它导致对象从会话中删除,然后在延迟加载发生时会导致问题。

@event.listens_for(SignallingSession, 'before_flush')
def before_flush(session, flush_context, instances):
    ledgers = []

    if session.dirty:
        for elem in session.dirty:
            if ( session.is_modified(elem, include_collections=False) ):
                if isinstance(elem, Wallet):
                    session.expunge(elem) # causes problems later
                    ledgers.append(Ledger(id=elem.id, amount=elem.balance))

    if ledgers:
        session.bulk_save_objects(ledgers)
        session.execute('UPDATE wallet w JOIN ledger l on w.id = l.id SET w.balance = l.amount')
        session.execute('TRUNCATE ledger')

我想做类似的事情:

session.dirty.remove(MyObject)

但是这不起作用,因为session.dirty是计算属性,而不是常规属性。我一直在挖掘仪器代码,但无法看到我如何欺骗dirty列表不包含某些内容。我发现在对象状态上还有一个history也需要处理。

有什么想法吗?如果有任何不同,底层数据库就是MySQL。

-Matt

2 个答案:

答案 0 :(得分:1)

在ORM之外修改数据库时,可以使用set_committed_value()让ORM知道当前的数据库状态。

示例:

wallet = session.query(Wallet).filter_by(id=123)
wallet.balance = 0
session.execute("UPDATE wallet SET balance = 0 WHERE id = 123;")
set_committed_value(wallet, "balance", 0)
session.commit()  # won't issue additional SQL to update wallet

如果确实想要将实例标记为不脏,则可以使用SQLAlchemy的内部进行清理:

state = inspect(p)
session.identity_map._modified.discard(state)
state.modified = False
print(p in session.dirty)  # False

答案 1 :(得分:0)

让我总结一下这种精神错乱。

from sqlalchemy.orm import attributes

attributes.instance_state(your_object).committed_state.clear()

容易。 (否)