Exemple d'insertion ou de mise à jour dans SQLAlchemy

Question

En Python, en utilisant SQLAlchemy, je veux insérer ou mettre à jour une ligne. J'ai essayé ceci :

existing = db.session.query(Toner)
for row in data:
    new = Toner(row[0], row[1], row[2])

Cela ne fonctionne pas. Comment puis-je INSÉRER ou METTRE À JOUR new en Toner table ? Je soupçonne que c'est fait avec la fusion, mais je ne comprends pas comment faire.

Answer 1

En supposant que certains noms de colonnes...

INSERER un

newToner = Toner(toner_id = 1,
                    toner_color = 'blue',
                    toner_hex = '#0F85FF')

dbsession.add(newToner)   
dbsession.commit()

INSERT multiple

newToner1 = Toner(toner_id = 1,
                    toner_color = 'blue',
                    toner_hex = '#0F85FF')

newToner2 = Toner(toner_id = 2,
                    toner_color = 'red',
                    toner_hex = '#F01731')

dbsession.add_all([newToner1, newToner2])   
dbsession.commit()

UPDATE

q = dbsession.query(Toner)
q = q.filter(Toner.toner_id==1)
record = q.one()
record.toner_color = 'Azure Radiance'

dbsession.commit()

ou en utilisant une phrase unique fantaisiste en utilisant MERGE

record = dbsession.merge(Toner( **kwargs))

Answer 2

J'ai essayé de nombreuses façons et j'ai finalement essayé ceci :

def db_persist(func):
    def persist(*args, **kwargs):
        func(*args, **kwargs)
        try:
            session.commit()
            logger.info("success calling db func: " + func.__name__)
            return True
        except SQLAlchemyError as e:
            logger.error(e.args)
            session.rollback()
            return False

    return persist

et :

@db_persist
def insert_or_update(table_object):
    return session.merge(table_object)

Answer 3

INSERT ON DUPLICATE KEY UPDATE

le sql :

INSERT INTO the_table (id, col1) VALUES (%s, %s) 
   ON DUPLICATE KEY UPDATE col1 = %s

dans le code py :
// test avec sqlalchemy 1.4.x, utiliser mysql

def test_insert_or_update():
    insert_stmt = insert(the_table).values(
        id = 'xxx',
        col1 = 'insert value',
    )
    on_duplicate_key_stmt = insert_stmt.on_duplicate_key_update(
        # col1 = insert_stmt.inserted.data,
        col1 = 'update value',
        col2 = 'mark'
    )
    print(on_duplicate_key_stmt)
    session.execute(on_duplicate_key_stmt)
    session.commit()