SQLAlchemy - 在postgresql中執行批量upsert（如果存在，更新，否則插入）

Question

我想在使用SQLAlchemy模塊（不是SQL！）的python中編寫批量upsert。

我得到一個SQLAlchemy的下面的錯誤添加：

sqlalchemy.exc.IntegrityError: (IntegrityError) duplicate key value violates unique constraint "posts_pkey" 
DETAIL: Key (id)=(TEST1234) already exists. 

我有一個表叫posts與上id列主鍵。

在這個例子中，我已經在db中有一行id=TEST1234。當我試圖db.session.add()一個新的帖子對象與id設置爲TEST1234，我得到上述錯誤。我的印象是，如果主鍵已經存在，記錄會被更新。

如何根據主鍵單獨添加Flask-SQLAlchemy？有一個簡單的解決方案嗎？

如果沒有，我總是可以檢查並刪除任何帶有匹配ID的記錄，然後插入新記錄，但這對我的情況來說似乎很昂貴，我不希望有很多更新。

Answer 1

有一個在SQLAlchemy的一個更新插入式的操作：

db.session.merge()

我發現這個命令後，我能夠執行upserts，但值得一提的是，該操作是批量慢「更新插入」。

另一種方法是讓你想UPSERT主鍵的列表，並查詢任何匹配的IDS數據庫：

# Imagine that post1, post5, and post1000 are posts objects with ids 1, 5 and 1000 respectively 
# The goal is to "upsert" these posts. 
# we initialize a dict which maps id to the post object 

my_new_posts = {1: post1, 5: post5, 1000: post1000} 

for each in posts.query.filter(posts.id.in_(my_new_posts.keys())).all(): 
    # Only merge those posts which already exist in the database 
    db.session.merge(my_new_posts.pop(each.id)) 

# Only add those posts which did not exist in the database 
db.session.add_all(my_new_posts.values()) 

# Now we commit our modifications (merges) and inserts (adds) to the database! 
db.session.commit()