如何检查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)