使用 SQLAlchemy 从 MySQL 获取最后插入的值

Question

我想返回 MySQL 中 auto_increment 字段的最后插入值。

我见过提到使用 session.flush() 添加记录然后检索 id 的示例。然而，这似乎总是返回 0。

我也看到过提到使用 session.refresh() 的示例，但这会引发以下错误：

InvalidRequestError: Could not refresh instance '<MyModel....>'

我的代码看起来像这样：

class Foo(Base):
    __tablename__ = 'tblfoo'
    __table_args__ = {'mysql_engine':'InnoDB'}

    ModelID = Column(INTEGER(unsigned=True), default=0, primary_key=True, autoincrement=True)
    ModelName = Column(Unicode(255), nullable=True, index=True)
    ModelMemo = Column(Unicode(255), nullable=True)

f = Foo(ModelName='Bar', ModelMemo='Foo')
session.add(f)
session.flush()

此时，对象 f 已被推送到数据库，并已自动分配一个唯一的主键 id。但是，我似乎找不到一种方法来获取在某些附加操作中使用的值。我想执行以下操作：

my_new_id = f.ModelID

我知道我可以简单地执行另一个查询来根据其他参数查找 ModelID，但如果可能的话，我宁愿不这样做。

Answer 1

问题是您正在为自动增量设置 default 。因此，当它运行插入查询时，服务器的日志是

2011-12-21 13:44:26,561 INFO sqlalchemy.engine.base.Engine.0x...1150 INSERT INTO tblfoo (`ModelID`, `ModelName`, `ModelMemo`) VALUES (%s, %s, %s)
2011-12-21 13:44:26,561 INFO sqlalchemy.engine.base.Engine.0x...1150 (0, 'Bar', 'Foo')
ID : 0

所以输出是 0 这是默认值并且被传递，因为您正在为 autoincrement 列设置默认值。

如果我在没有 default 的情况下运行相同的代码，那么它会给出正确的输出。

请尝试这个代码

from sqlalchemy import create_engine
engine = create_engine('mysql://test:test@localhost/test1', echo=True)

from sqlalchemy.ext.declarative import declarative_base
Base = declarative_base()

from sqlalchemy.orm import sessionmaker
Session = sessionmaker(bind=engine)

session = Session()

from sqlalchemy import Column, Integer, Unicode

class Foo(Base):
    __tablename__ = 'tblfoo'
    __table_args__ = {'mysql_engine':'InnoDB'}

    ModelID = Column(Integer, primary_key=True, autoincrement=True)
    ModelName = Column(Unicode(255), nullable=True, index=True)
    ModelMemo = Column(Unicode(255), nullable=True)

Base.metadata.create_all(engine)

f = Foo(ModelName='Bar', ModelMemo='Foo')
session.add(f)
session.flush()

print "ID :", f.ModelID

Answer 2

尝试使用 session.commit() 而不是 session.flush()。然后您可以使用f.ModelID。

Answer 3

不确定为什么标记的答案对您有用。但就我而言，这实际上并没有将行插入表中。我最终需要调用 commit() 。

所以最后几行代码是：

f = Foo(ModelName='Bar', ModelMemo='Foo')
session.add(f)
session.flush()

print "ID:", f.ModelID

session.commit()