У меня есть запись, которую я хочу существовать в базе данных, если ее там нет, и если она уже существует (первичный ключ существует), я хочу, чтобы поля обновлялись до текущего состояния. Это часто называют upsert.
Следующий неполный фрагмент кода демонстрирует, что будет работать, но он кажется чрезмерно неуклюжим (особенно, если было больше столбцов). Каков лучший/лучший способ?
Base = declarative_base()
class Template(Base):
__tablename__ = 'templates'
id = Column(Integer, primary_key = True)
name = Column(String(80), unique = True, index = True)
template = Column(String(80), unique = True)
description = Column(String(200))
def __init__(self, Name, Template, Desc):
self.name = Name
self.template = Template
self.description = Desc
def UpsertDefaultTemplate():
sess = Session()
desired_default = Template("default", "AABBCC", "This is the default template")
try:
q = sess.query(Template).filter_by(name = desiredDefault.name)
existing_default = q.one()
except sqlalchemy.orm.exc.NoResultFound:
#default does not exist yet, so add it...
sess.add(desired_default)
else:
#default already exists. Make sure the values are what we want...
assert isinstance(existing_default, Template)
existing_default.name = desired_default.name
existing_default.template = desired_default.template
existing_default.description = desired_default.description
sess.flush()
Есть ли лучший или менее верный способ сделать это? Что-то вроде этого было бы здорово:
sess.upsert_this(desired_default, unique_key = "name")
хотя unique_key
kwarg, очевидно, не нужен (ORM должен иметь возможность легко понять это). Я добавил его только потому, что SQLAlchemy имеет тенденцию работать только с первичным ключом. например: Я смотрел, применим ли Session.merge, но это работает только на первичный ключ, который в этом случае является автоинкрементным id, что не очень полезно для этой цели.
Пример использования для этого - просто при запуске серверного приложения, которое, возможно, обновило ожидаемые по умолчанию данные. т.е.: no concurrency относится к этой проблеме.