Обновление SQLAlchemy, если существует уникальный ключ

У меня есть класс:

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

При добавлении тега:

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

Я хочу, чтобы он обновлял существующий тег .

Конечно, я мог бы сделать это:

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

но сохранение такой логики в классе Tagкажется более понятным — возможно, уберегает меня от операции дробовиком.

​​Как мне туда добраться? Я новичок в Pythonи SQLAlchemy, поэтому буду признателен за любые дополнительные мысли о моем коде.

Спасибо.

П.С. SQLAlchemyНАСТОЛЬКО ГИГАНТСКАЯ, и они не предоставляют удобного способа сделать INSERT ... ON DUPLICATE KEY UPDATE, да? ВАУ!

18
задан vishes_shell 11 July 2017 в 21:21
поделиться