Как массово добавить данные с использованием SQLAlchemy ORM с проверкой есть ли они уже?

Question

[AlexBoss]

К примеру, у меня есть 2 таблицы

node = Table(
    "node",
    metadata_obj,
    Column("node_id", Integer, primary_key=True),
    Column("primary_element", Integer, ForeignKey("element.element_id")),
)

element = Table(
    "element",
    metadata_obj,
    Column("element_id", Integer, primary_key=True),
    Column("text", Text),
    Column("parent_node_id", Integer),
    ForeignKeyConstraint(
        ["parent_node_id"], ["node.node_id"], name="fk_element_parent_node_id"
    ),
)

Я хочу вставить в таблицу 'element' данные, если нет записи таким text и node_id из связанной таблицы.
Думал сделать через

objects = [User(name="u1"), User(name="u2"), User(name="u3")]
session.add_all(objects)
session.commit()

Но не совсем понимаю, как сюда добавить ограничение not exists

Comments

[Модератор]

См.п.3.1, 3.4, 3.8

Answer 1

[Elvis]

Например для Postgres on_conflict_do_nothing
Или INSERT…ON CONFLICT
У меня Postgres, я сделал on_conflict_do_update так:

class HeroCard(db.Model):
    id = db.Column(db.Integer, primary_key=True, nullable=False)
    heroid = db.Column(db.Integer, ForeignKey('hero.id'))
    cardid = db.Column(db.Integer, ForeignKey('card.id'))
    count = db.Column(db.Integer)
    db.UniqueConstraint(heroid, cardid, name='hero_card_un')

upd = insert(HeroCard).values(heroid=id, cardid=cid, count=count)
do_upd_upd = upd.on_conflict_do_update(constraint='hero_card_un', set_={'count': HeroCard.count+count})
db.session.execute(do_upd_upd)
db.session.commit()

Comments

[AlexBoss]

Как раз для pg и надо. Просто сказка, обязательно попробую)

[AlexBoss]

По поводу массового добавления, если добавлять 100 записей по производительности все ок будет по сравнению с тем же add_all или bulk_add??

[Elvis]

AlexBoss, по этому поводу не подскажу, не на столько знаю библиотеку.

[Михаил Кростелев]

AlexBoss, если у тебя информация хранится не на магнитных лентах, то 100 записей по производительности будет ок.