SQLAlchemy Lấy khóa chính với Autoincrement trước khi Commit

Question

Khi tôi đã tạo một bảng có khóa chính tăng tự động, có cách nào để lấy khóa chính sẽ là gì (tức là, thực hiện điều gì đó như đặt khóa chính) mà không thực sự cam kết?

Tôi muốn đặt hai hoạt động bên trong một giao dịch tuy nhiên một trong các thao tác sẽ phụ thuộc vào khóa chính nào được gán trong thao tác trước đó.

Answer 1

Bạn có thể sử dụng nhiều giao dịch và quản lý nó trong phạm vi.

Answer 2

Bạn không cần phải commit, bạn chỉ cần flush. Đây là một số mã mẫu. Sau cuộc gọi đến flush bạn có thể truy cập khóa chính đã được chỉ định. Lưu ý điều này là với 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()