使用 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已经被push到DB中，并被自动分配了唯一的主键id。但是，我似乎找不到一种方法来获取在某些附加操作中使用的值。我想做以下事情：

my_new_id = f.ModelID

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

Answer 1

问题是您正在为自动增量设置

defaul

。因此，当它运行插入查询时，服务器的日志是

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()