SQLAlchemy zmiana jeśli unikalny klucz istnieje

Question

Mam klasę:

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,) 

Dodając tag:

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

chcę go zaktualizować istniejący tag.

Oczywiście, mogłem zrobiłem to:

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

ale utrzymując taką logikę w Tag klasy wydaje się być bardziej jasne - ewentualnie trzyma mnie z dala od zabiegu shotgun.

Jak się tam dostać? Jestem całkiem nowy dla Python i SQLAlchemy, więc wszelkie dodatkowe uwagi dotyczące mojego kodu zostaną docenione.

Dziękuję.

P.S. SQLAlchemy jest TAK GIGANTYCZNY i nie zapewnia wygodnego sposobu na wykonanie INSERT ... ON DUPLICATE KEY UPDATE, co? ŁAŁ!

Answer 1

Można spróbować tej

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 

Można sprawdzić link https://stackoverflow.com/search?q=Insert+on+duplicate+update+sqlalchemy

Answer 2

od wersji 1,2 SQLAlchemy wesprze on_duplicate_key_update dla MySQL

Istnieje również examples jak używać to:

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) 

---

Od wersji 1,1 SQLAlchemy wsparcia on_conflict_do_update dla PostgreSQL

Examples:

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)