在Django中,queryset提供了一个名为get_or_create的方法,该方法可以返回一个对象或创建一个对象。
但是,与get方法一样,如果查询返回多个对象,get_or_create可能会抛出异常。
有没有一种优雅的方法:
objects = Model.manager.filter(params)
if len(objects) == 0:
obj = Model.objects.create(params)
else:
obj = objects[0]
答案 0 :(得分:30)
get_or_create()只是一个便利功能,所以编写自己的内容没有任何问题,就像pavid所示或
result = Model.objects.filter(field__lookup=value)[0]
if not result:
result = Model.objects.create(...)
return result
修改强> 如建议的那样,将过滤器之后的[:1]切片(返回单项列表)更改为[0](返回实际对象)。这样做的问题是,如果与查询不匹配,它将引发异常。
这也会引起一个类似的例外:
Model.objects.filter(field_lookup=value).latest()
再次查看这个问题,我不确定原始海报是否希望返回多个对象/行,或者只是在检索单个对象/行时绕过引发异常的方法。
这是另一种选择吗?
results = Model.objects.filter(...)
if results.exists():
return results
else:
return Model.objects.create(...)
和另一个:
result = None
try:
result = Model.objects.get(...)
except Model.DoesNotExist:
result = Model.objects.create(...)
抚养和抚养没有错抓住例外!
答案 1 :(得分:21)
从django 1.6开始,有一个方便的方法first(),它返回过滤查询中的第一个结果,或者是None。
obj = Model.manager.filter(params).first()
if obj is None:
obj = Model.objects.create(params)
答案 2 :(得分:5)
我刚刚遇到了这个问题并且想要贡献,因为没有人建议实际捕获IndexError。
try:
obj = Model.objects.filter(params)[0]
except IndexError:
obj = Model.objects.create(params)
答案 3 :(得分:5)
这是一个管理器方法,可以让你优雅地扩展这个功能
class FilterOrCreateManager(models.Manager):
"""Adds filter_or_create method to objects
"""
def filter_or_create(self, **kwargs):
try:
created = False
obj = self.filter(**kwargs).first()
if obj is None:
obj = self.create(**kwargs)
created = True
return (obj,created)
except Exception as e:
print(e)
然后确保将管理器添加到您要使用它的任何模型:
class MyObj(models.Model):
objects = FilterOrCreateManager()
之后,您将能够像get_or_create
一样使用它:
obj_instance, created = MyObj.filter_or_create(somearg='some value')
答案 4 :(得分:1)
你可以试试这个:
result = Model.objects.filter(field__lookup=value)[0]
if not result:
result = Model.objects.create(...)
return result
答案 5 :(得分:0)
这对我有用:
在您看来,请拨打以下内容:
obj = Category.objects.get_or_create_category(form.cleaned_data.get('name'))
在模型管理器中,创建一个如下函数:
class CategoryManager(models.Manager):
def get_or_create_category(self, query):
try:
result = Category.objects.filter(name = query)[0]
except:
result = Category.objects.create(name = query)
return result
逻辑很简单。首先,尝试检索第一个Category对象,该对象的名称与查询字符串(由表单提供)匹配。如果检索失败(因为它不存在),请创建一个以字符串作为名称的新类别。返回结果以在视图中使用。
答案 6 :(得分:0)
在一行中完成:
obj = Model.objects.filter(params).first() or Model.objects.create(params)