如何使用 SQLAlchemy 在一次提交中切换唯一行的两个字段？

Question

假设有一个具有唯一名称的对象。现在要切换两个对象的名称：

这是布局：

import sqlalchemy as sa
import sqlalchemy.orm as orm
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class MyObject(Base):
  __tablename__ = 'my_objects'
  id = sa.Column(sa.Integer, primary_key=True)
  name = sa.Column(sa.Text, unique=True)

if __name__ == "__main__":
  engine = sa.create_engine('sqlite:///:memory:', echo=True)
  Session = orm.sessionmaker(bind=engine)
  Base.metadata.create_all(engine)
  session = Session()

我想这样做：

a = MyObject(name="Max")
b = MyObject(name="Moritz")
session.add_all([a, b])
session.commit()

# Now: switch names!
tmp = a.name
a.name = b.name
b.name = tmp
session.commit()

这会引发IntegrityError。有没有办法在一次提交中切换这些字段而不会出现此错误？

Answer 1

Python 允许这种语法（使用元组）：

a.name, b.name = b.name, a.name

这样切换两个普通参数绝对没问题，但没有在你的情况下测试过，也许你可以试一试？

Answer 2

您在名称字段中提供了unique=True，因此当您尝试提交时，它将运行更新查询，它会引发错误。

这种情况是当您更改名称时它将在内存中设置。但是当它尝试运行更新查询时，旧记录已经存在同名，所以它会给出IntegrityError。

改名的方法是

a = MyObject(name="Max")
b = MyObject(name="Moritz")
session.add_all([a, b])
session.commit()

# Now: switch names!
atmp = a.name
btemp = b.name

a.name = a.name+btemp # Temp set the any random name
session.commit()

b.name = atemp
a.name = btemp
session.commit() # Run the update query for update the record.

Answer 3

更纯粹的选择是删除 a，重命名 b，然后重新添加重命名的：

session.delete(a)
sqlalchemy.orm.session.make_transient(a)
a.name, b.name = b.name, a.name
session.flush()
session.add(a)
session.commit()

Answer 4

感谢Philipp der Rautenberg's comment，我找到了更好的解决方案，它适用于 PostgreSQL，但不适用于 SQLite。

有关其工作原理的更多信息，请参见 this answer、PostgreSQL docs 和 SQLAlchemy docs。

这个想法是首先将唯一约束声明为deferrable，然后通过在initially中使用它来将其设置为DEFERRED：

import sqlalchemy as sa
import sqlalchemy.orm as orm
from sqlalchemy.ext.declarative import declarative_base

Base = declarative_base()

class MyObject(Base):
    __tablename__ = 'my_object'
    id = sa.Column(sa.Integer, primary_key=True)
    name = sa.Column(sa.Text)
    name_unique_constraint = sa.UniqueConstraint(name, deferrable=True, initially='DEFERRED')


if __name__ == "__main__":
    engine = sa.create_engine('postgresql+psycopg2://user:password@hostname/database_name', echo=True)
    Session = orm.sessionmaker(bind=engine)
    Base.metadata.create_all(engine)
    session = Session()

    a = MyObject(name="Max")
    b = MyObject(name="Moritz")
    session.add_all([a, b])
    session.commit()

    # Now: switch names!
    a.name, b.name = b.name, a.name

    session.add_all([a, b]) # this line seems to be necessary to raise exception if swap woudn't be possible
    session.commit()

或者用明确的sql语句：

db.execute('SET CONSTRAINTS ALL DEFERRED')

或

db.execute('SET CONSTRAINTS uq_my_object_name DEFERRED')

但请记住在使用SET CONSTRAINTS 时：

此命令仅更改当前事务中约束的行为。