Django:从过滤器查询中获取第一个对象或创建

时间:2011-05-31 16:42:48

标签: django django-models django-queryset

在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]

7 个答案:

答案 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)