python - query - Actualización de SQLAlchemy si existe una clave única

sqlalchemy relationship

Question

Tengo una clase:

class Tag(Base, TimestampMixin):
"""Tags"""
__tablename__ = ''tags''
__table_args__ = {''mysql_engine'' : ''InnoDB'', ''mysql_charset'' : ''utf8'' }
id = Column(Integer(11), autoincrement = True, primary_key = True)
tag = Column(String(32), nullable = False, unique = True)
cnt = Column(Integer(11), index = True, nullable = False, default = 1)
def __init__(self, tag):
t = session.query(Tag).filter_by(tag=tag).first()
if t:
self.cnt = t.cnt+1
self.tag = t.tag
else:
self.tag = tag
def __repr__(self):
return "<Tag(''%s'')>" % (self.tag, )
def __unicode__(self):
return "%s" % (self.tag, )

Al agregar la etiqueta:

tag = Tag(''tag'')
session.add(tag)
session.commit()

Quiero que actualice la tag existente.

Por supuesto, podría haber hecho esto:

tag = session.query(Tag).filter_by(tag=''tag'').first()
if tag:
tag.cnt++
else:
tag = Tag(''tag'')
session.add(tag)
session.commit()

pero, mantener esa lógica en la clase Tag parece ser más claro, posiblemente me mantenga alejado de la cirugía de disparo.

¿Como llego hasta ahí? Soy bastante nuevo en Python y SQLAlchemy , por lo que cualquier idea adicional sobre mi código será apreciada.

Gracias.

PS SQLAlchemy es TAN GIGANTE y no proporcionan una forma práctica de hacerlo INSERT ... ON DUPLICATE KEY UPDATE , ¿eh? ¡GUAUU!

Answer 1

Puedes intentar esto

def get_or_increase_tag(tag_name):
tag = session.query(Tag).filter_by(tag=tag_name).first()
if not tag:
tag = Tag(tag_name)
else:
tag.cnt += 1
return tag

Puede consultar el enlace https://.com/search?q=Insert+on+duplicate+update+sqlalchemy

Answer 2

Desde la versión 1.2 SQLAlchemy será compatible con on_duplicate_key_update para MySQL

También hay ejemplos de cómo usarlo:

from sqlalchemy.dialects.mysql import insert
insert_stmt = insert(my_table).values(
id=''some_existing_id'',
data=''inserted value'')
on_duplicate_key_stmt = insert_stmt.on_duplicate_key_update(
data=insert_stmt.values.data,
status=''U''
)
conn.execute(on_duplicate_key_stmt)

Desde la versión 1.1 Soporte de SQLAlchemy on_conflict_do_update para PostgreSQL

Ejemplos :

from sqlalchemy.dialects.postgresql import insert
insert_stmt = insert(my_table).values(
id=''some_existing_id'',
data=''inserted value'')
do_update_stmt = insert_stmt.on_conflict_do_update(
constraint=''pk_my_table'',
set_=dict(data=''updated value'')
)
conn.execute(do_update_stmt)