3 回答
TA贡献1784条经验 获得超2个赞
这基本上是这样做的方式,AFAIK没有随时可用的快捷方式。
你可以概括它的当然:
def get_or_create(session, model, defaults=None, **kwargs):
instance = session.query(model).filter_by(**kwargs).first()
if instance:
return instance, False
else:
params = dict((k, v) for k, v in kwargs.iteritems() if not isinstance(v, ClauseElement))
params.update(defaults or {})
instance = model(**params)
session.add(instance)
return instance, True
TA贡献1777条经验 获得超3个赞
根据@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
有了这个,我就可以get_or_create我的模型的任何对象。
假设我的模型对象是:
class Country(Base):
__tablename__ = 'countries'
id = Column(Integer, primary_key=True)
name = Column(String, unique=True)
为了获取或创建我的对象,我写道:
myCountry = get_or_create(session, Country, name=countryName)
添加回答
举报