SQLAlchemy:如何在一个事务中交换唯一值?

时间:2015-04-22 15:34:36

标签: mysql sql sqlite sqlalchemy

我的版本:

$ python
Python 3.4.0 (default, Apr 11 2014, 13:05:11) 
[GCC 4.8.2] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import sqlalchemy
>>> sqlalchemy.__version__
'1.0.0'

让我们看一下sqlite的例子(在生产中我使用mysql,但这里没关系):

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import Column
from sqlalchemy import create_engine
from sqlalchemy import Integer, String

Base=declarative_base()
engine = create_engine('sqlite:///:memory:', echo=True)

class User(Base):
   __tablename__ = 'user'
   id      = Column(Integer, primary_key=True)
   name    = Column(String(16))
   tech_id = Column(Integer, unique=True, nullable=True)

Base.metadata.create_all(engine)

Session = sessionmaker(bind=engine)
s = Session()

现在我们添加两条记录:

u1=User(name="User1", tech_id=None)
u2=User(name="User2", tech_id=10)
s.add(u1)
s.add(u2)
s.commit()

接下来尝试修改它们:

u1=s.query(User).filter(User.name=="User1").first()
u2=s.query(User).filter(User.name=="User2").first()

u2.tech_id=None
u1.tech_id=10
s.commit()

提交后我得到了例外:

<-- cut -->
sqlalchemy.exc.IntegrityError: (sqlite3.IntegrityError) UNIQUE constraint failed: user.tech_id [SQL: 'UPDATE user SET tech_id=? WHERE user.id = ?'] [parameters: ((10, 1), (None, 2))]

如果我喜欢这样:

u2.tech_id=None
s.commit()
u1.tech_id=10
s.commit()

没关系。

是否可以仅通过一次提交(仅通过一次交易)来执行请求?

1 个答案:

答案 0 :(得分:0)

你可以这样做:

#u1=s.query(User).filter(User.name=="User1").first()
#u2=s.query(User).filter(User.name=="User2").first()

#u2.tech_id=None
#u1.tech_id=10
#s.commit()

s.query(User).filter(User.name=="User2").update(
    {User.tech_id: None}
)
s.query(User).filter(User.name=="User1").update(
    {User.tech_id: 10}
)

s.commit()

这种变化的本质是执行正确的行动序列。即使在提交之前,也无法为唯一键(至少在mysql中)创建重复值。但是您可以为唯一列创建多个NULL值。