SqlAlchemy如何处理表定义中的唯一约束

Question

我有一个具有以下声明性定义的表：

class Type(Base):
    __tablename__ = 'Type'
    id = Column(Integer, primary_key=True)
    name = Column(String, unique = True)
    def __init__(self, name):
        self.name = name

列“name”具有唯一约束，但我能够这样做

type1 = Type('name1')
session.add(type1)
type2 = Type(type1.name)
session.add(type2)

因此，可以看出，根本不检查唯一约束，因为我已添加到会话中2 个同名的对象。

当我执行 session.commit() 时，我收到 mysql 错误，因为约束也在 mysql 表中。

sqlalchemy 是否有可能提前告诉我，我无法制作或识别它，并且不会插入具有相同“名称”列的 2 个条目？ 如果没有，我是否应该保留所有现有名称，以便在创建对象之前检查它们是否存在？

Answer 1

SQLAlechemy 不处理唯一性，因为它不可能做得很好。即使您跟踪创建的对象和/或检查具有该名称的对象是否存在，也存在竞争条件：其他进程中的任何人都可以插入具有您刚刚检查的名称的新对象。唯一的解决方案是在检查之前锁定整个表并在插入后释放锁定（某些数据库支持这种锁定）。

Answer 2

AFAIK，sqlalchemy 不处理 python 行为中的唯一性约束。这些“unique=True”声明仅用于强加数据库级表约束，并且仅当您使用 sqlalchemy 命令（即

Type.__table__.create(engine)

或类似命令）创建表时才使用。如果您针对实际不存在此约束的现有表创建 SA 模型，则就像它不存在一样。

根据您的具体用例，您可能必须使用类似的模式

try:
  existing = session.query(Type).filter_by(name='name1').one()
  # do something with existing
except:
  newobj = Type('name1')
  session.add(newobj)

或变体，或者您只需要捕获 mysql 异常并从那里恢复。

Answer 3

来自文档

class MyClass(Base):
    __tablename__ = 'sometable'
    __table_args__ = (
            ForeignKeyConstraint(['id'], ['remote_table.id']),
            UniqueConstraint('foo'),
            {'autoload':True}
            )

Answer 4

.one() 抛出两种异常：
sqlalchemy.orm.exc.NoResultFound 和 sqlalchemy.orm.exc.MultipleResultsFound

您应该在第一个异常发生时创建该对象，如果发生第二个异常，无论如何您都会被搞砸并且不应该让情况变得更糟。

try:
  existing = session.query(Type).filter_by(name='name1').one()
# do something with existing
except NoResultFound:
  newobj = Type('name1')
  session.add(newobj)