SqlAlchemy: lấy id của hồ sơ cuối cùng được chèn

Question

Tôi đang sử dụng SQLAlchemy không có ORM, tức là sử dụng các biểu tượng SQL được tạo thủ công để tương tác trực tiếp với db phụ trợ. Tôi đang sử dụng PG như db phụ trợ của tôi (psycopg2 như trình điều khiển DB) trong trường hợp này - Tôi không biết nếu điều đó ảnh hưởng đến câu trả lời.

Tôi có báo cáo như thế này (cho ngắn gọn, giả conn đó là một kết nối hợp lệ đến db):

conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123)") 

cũng Giả sử rằng các bảng người dùng bao gồm các cột (id [SERIAL PRIMARY KEY], tên, country_id)

Làm thế nào tôi có thể lấy id của người dùng mới (lý tưởng, mà không va vào db nữa không?)

Answer 1

Bạn có thể có thể sử dụng RETURNING khoản của INSERT statement như thế này:

result = conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123) 
         RETURNING *") 

Nếu bạn chỉ muốn kết quả id:

rslt_id = conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123) 
         RETURNING id") 

Answer 2

tài lastrowid

result = conn.execute("INSERT INTO user (name, country_id) VALUES ('Homer', 123)") 
result.lastrowid 

Answer 3

hiện SQLAlchemy documentation gợi ý

result.inserted_primary_key nên làm việc!