私はクラスを持っています:
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
を更新したい。
もちろん、私はこれを行うことができました:
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
にかなり慣れていないので、コードについてさらに考えていただければ幸いです。
ありがとうございました。
P.S. SQLAlchemy
はSO GIGANTICであり、INSERT ... ON DUPLICATE KEY UPDATE
を実行するための便利な方法を提供していませんね?すごい!
あなたはこれを試すことができます
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
リンクを確認できます https://stackoverflow.com/search?q=Insert+on+duplicate+update+sqlalchemy
バージョン1.2からSQLAlchemyは on_duplicate_key_update
forMySQL
それを使用する方法の 例 もあります:
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)
バージョン1.1からSQLAlchemyサポート on_conflict_do_update
forPostgreSQL
例 :
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)