如何使用 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

更纯粹的选择是删除 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 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

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

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

以这种方式切换两个普通参数是绝对可以的，但没有在你的情况下进行测试，也许你可以尝试一下？

Answer 4

感谢 Philipp der Rautenberg 的评论 我找到了更好的解决方案，可以与 PostgreSQL 一起使用，但不能SQLite。

有关其工作原理的更多信息，请参阅此答案和PostgreSQL 文档 和 SQLAlchemy 文档。

这个想法是首先将唯一约束声明为 deferrable，然后通过在 initially: 中使用它

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 语句将其设置为 DEFERRED：

db.execute('SET CONSTRAINTS ALL DEFERRED')

或者

db.execute('SET CONSTRAINTS uq_my_object_name DEFERRED')

但请记住在使用SET CONSTRAINTS时：

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