python - SQLAlchemy 是否与 Django 的 get_or_create 等效？

Question

如果它已经存在(基于提供的参数)，我想从数据库中获取一个对象，如果不存在则创建它。

Django 的 get_or_create (或 source )这样做。 SQLAlchemy 中是否有等效的快捷方式？

我目前正在这样明确地写出来:

def get_or_create_instrument(session, serial_number):
    instrument = session.query(Instrument).filter_by(serial_number=serial_number).first()
    if instrument:
        return instrument
    else:
        instrument = Instrument(serial_number)
        session.add(instrument)
        return instrument

Answer 1

按照@WoLpH的解决方案，这是对我有用的代码(简单版):

def get_or_create(session, model, **kwargs):
    instance = session.query(model).filter_by(**kwargs).first()
    if instance:
        return instance
    else:
        instance = model(**kwargs)
        session.add(instance)
        session.commit()
        return instance

有了这个，我可以获取或创建我模型的任何对象。

假设我的模型对象是:

class Country(Base):
    __tablename__ = 'countries'
    id = Column(Integer, primary_key=True)
    name = Column(String, unique=True)

要获取或创建我的对象，我会写:

myCountry = get_or_create(session, Country, name=countryName)