SQLAlchemy - выполнение массового upsert (если существует, обновление, другое вставка) в postgresql

Я пытаюсь написать массив upsert в python, используя модуль SQLAlchemy (не в SQL!).

У меня есть таблица с именем posts с первичным ключом в столбце id.

В этом примере у меня уже есть строка в db с id=TEST1234. Когда я пытаюсь db.session.add() создать новый объект post с id, установленным на TEST1234, я получаю ошибку выше. У меня создалось впечатление, что если первичный ключ уже существует, запись будет обновляться.

Как я могу выполнить обновление с помощью Flask-SQLAlchemy на основе первичного ключа? Есть ли простое решение?

Если этого не происходит, я всегда могу проверить и удалить любую запись с соответствующим идентификатором, а затем вставить новую запись, но это кажется дорогим для моей ситуации, где я не ожидаю много обновлений.

Ответы

Ответ 1

В SQLAlchemy есть операция upsert-esque:

db.session.merge()

После того, как я нашел эту команду, я смог выполнить upserts, но стоит упомянуть, что эта операция медленна для массового "upsert".

Альтернативой является получение списка первичных ключей, которые вы хотели бы активизировать, и запросить базу данных для любых совпадающих идентификаторов:

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