如何检查SQLAlchemy会话是脏的还是n

Question

我有一个SQLAlchemySession对象，想知道它是否脏。确切的问题是，我想（隐喻性地）问Session：“如果此时我发布commit()或rollback()，对数据库的影响是否相同？”。

理由是这样的：我想问用户他是否想确认更改。但如果没有变化，我什么都不想问。当然，我可以监视自己在Session上执行的所有操作，并决定是否有修改，但由于程序的结构，这需要一些相当复杂的更改。如果SQLAlchemy已经提供了这个机会，我很乐意利用它。

谢谢大家。

Answer 1

这是我根据@zzzeek的回答和最新评论提出的解决方案。我已经对它进行了单元测试，它似乎可以很好地处理回滚（会话在发出回滚后是干净的）：

from sqlalchemy import event
from sqlalchemy.orm import Session


@event.listens_for(Session, "after_flush")
def log_flush(session, flush_context):
    session.info['flushed'] = True


@event.listens_for(Session, "after_commit")
@event.listens_for(Session, "after_rollback")
def reset_flushed(session):
    if 'flushed' in session.info:
        del session.info['flushed']


def has_uncommitted_changes(session):
    return any(session.new) or any(session.deleted) \
        or any([x for x in session.dirty if session.is_modified(x)]) \
        or session.info.get('flushed', False)

Answer 2

会话具有脏属性

会话.dirty

当前检测到更改的持久对象 （现在，每次调用属性时都会动态创建此集合）

sqlalchemy.orm.session.Session.dirty

Answer 3

您正在寻找在会话事务的整个范围内进行的实际刷新的净计数；虽然有一些线索表明是否发生了这种情况（称为“快照”），但这种结构只是为了帮助回滚，而不是强引用。最直接的方法是跟踪“after_flush”事件，因为此事件仅在调用flush时才会发出，并且flush found状态为flush：

from sqlalchemy import event
import weakref
transactions_with_flushes = weakref.WeakSet()

@event.listens_for(Session, "after_flush")
def log_transaction(session, flush_context):
    for trans in session.transaction._iterate_parents():
        transactions_with_flushes.add(trans)

def session_has_pending_commit(session):
    return session.transaction in transactions_with_flushes

编辑：这里有一个更简单的更新版本：

from sqlalchemy import event

@event.listens_for(Session, "after_flush")
def log_transaction(session, flush_context):
    session.info['has_flushed'] = True

def session_has_pending_commit(session):
    return session.info.get('has_flushed', False)