SQLAlchemy：尝试保存非唯一值后重新保存模型的唯一字段

Question

在我的 SQLAlchemy 应用程序中，我有以下模型：

from sqlalchemy import Column, String
from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import scoped_session, sessionmaker
from zope.sqlalchemy import ZopeTransactionExtension

DBSession = scoped_session(sessionmaker(extension=ZopeTransactionExtension()))

class MyModel(declarative_base()):
    # ...
    label = Column(String(20), unique=True)

    def save(self, force=False):
        DBSession.add(self)
        if force:
            DBSession.flush()

稍后在代码中为每个新的 MyModel 对象随机生成 label，如果生成的值已存在于 DB 中，则重新生成它。
我正在尝试执行以下操作：

# my_model is an object of MyModel
while True:
    my_model.label = generate_label()
    try:
        my_model.save(force=True)
    except IntegrityError:
        # label is not unique - will do one more iteration
        # (*)
        pass
    else:
        # my_model saved successfully - exit the loop
        break

但如果第一次生成 label 不是唯一的并且在第二次（或以后）迭代中调用 save() 时会出现此错误：

 InvalidRequestError: This Session's transaction has been rolled back due to a previous exception during flush. To begin a new transaction with this Session, first issue Session.rollback(). Original exception was: (IntegrityError) column url_label is not unique... 

当我在位置 (*) 中添加 DBSession.rollback() 时，我得到了这个：

 ResourceClosedError: The transaction is closed

我应该怎么做才能正确处理这种情况？
谢谢

Answer 1

如果您的session 对象基本上回滚，您必须创建一个新会话并刷新您的模型，然后才能重新开始。如果您使用zope.sqlalchemy，您应该使用transaction.commit() 和transaction.abort() 来控制事物。所以你的循环看起来像这样：

# you'll also need this import after your zope.sqlalchemy import statement
import transaction

while True:
    my_model.label = generate_label()
    try:
        transaction.commit()
    except IntegrityError:
        # need to use zope.sqlalchemy to clean things up
        transaction.abort()
        # recreate the session and re-add your object
        session = DBSession()
        session.add(my_model)
    else:
        break

我已将会话对象的使用从对象的save 方法中提取出来。我不完全确定ScopedSession 在课堂级别使用时如何刷新自己，就像您所做的那样。就个人而言，我认为在你的模型中嵌入SqlAlchemy 的东西并不能很好地与SqlAlchemy 的unit of work 方法一起工作。

如果您的标签对象确实是生成的唯一值，那么我会同意 TokenMacGuy 并使用 uuid 值。

希望对您有所帮助。

Answer 2

数据库没有一致的方式告诉您为什么事务失败，以自动化可访问的形式。您通常不能尝试事务，然后重试，因为它由于某些特定原因失败。

如果您知道要解决的条件（例如唯一约束），您需要自己检查约束。在 sqlalchemy 中，它看起来像这样：

# Find a unique label
label = generate_label()
while DBsession.query(
        sqlalchemy.exists(sqlalchemy.orm.Query(Model)
                  .filter(Model.lable == label)
                  .statement)).scalar():
    label = generate_label()

# add that label to the model
my_model.label = label
DBSession.add(my_model)
DBSession.flush()

---

edit：回答这个问题的另一种方法是您不应该自动重试事务；您可以改为返回 HTTP 状态代码 307 Temporary Redirect（在重定向的 URL 中添加一些盐），以便事务真正重新开始。

Answer 3

我在用 Pyramid 框架编写的 webapp 中遇到了类似的问题。我为这个问题找到了一些不同的解决方案。

while True:
    try:
        my_model.label = generate_label()
        DBSession.flush()
        break
    except IntegrityError:
        # Rollback will recreate session:
        DBSession.rollback()
        # if my_model was in db it must be merged:
        my_model = DBSession.merge(my_model)

如果之前存储了 my_model，则合并部分至关重要。没有合并会话将是空的，因此刷新不会采取任何行动。