SQLAlchemy Obtener la clave principal con autoincrement Antes de confirmar

Question

Cuando he creado una tabla con una clave principal de incremento automático, ¿hay alguna manera de obtener la clave primaria (es decir, hacer algo así como reservar la clave principal) sin en realidad cometiendo?

Me gustaría ubicar dos operaciones dentro de una transacción, sin embargo, una de las operaciones dependerá de qué clave primaria se asignó en la operación anterior.

Answer 1

Puede usar múltiples transacciones y administrarlas dentro del alcance.

Answer 2

No necesita commit, solo necesita flush. Aquí hay un código de muestra. Después de la llamada al flush, puede acceder a la clave principal que se le asignó. Tenga en cuenta que esto es con SA 0.4.8.

from sqlalchemy import * 
from sqlalchemy.databases.mysql import * 
import sqlalchemy.ext.declarative 

Base = sqlalchemy.ext.declarative.declarative_base() 

class User(Base): 
    __tablename__ = 'user' 
    user_id = Column('user_id', Integer, primary_key=True) 
    name = Column('name', String) 

if __name__ == '__main__': 
    import unittest 
    from sqlalchemy.orm import * 
    import datetime 

    class Blah(unittest.TestCase): 
     def setUp(self): 
      self.engine = create_engine('sqlite:///:memory:', echo=True) 
      self.sessionmaker = scoped_session(sessionmaker(bind=self.engine)) 
      Base.metadata.bind = self.engine 
      Base.metadata.create_all() 
      self.now = datetime.datetime.now() 

     def test_pkid(self): 
      user = User(name="Joe") 
      session = self.sessionmaker() 
      session.save(user) 
      session.flush() 
      print 'user_id', user.user_id 
      session.commit() 
      session.close() 

    unittest.main()