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

Question

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

Answer 1

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(...)

提出和捕获异常没有错!

Answer 2

从django 1.6开始,有一个方便的方法first(),它返回过滤查询中的第一个结果,或者是None.

obj = Model.manager.filter(params).first()
if obj is None:
    obj = Model.objects.create(params)

Answer 3

这是一个管理器方法，可让您优雅地扩展此功能

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')

Answer 4

我刚刚遇到这个问题,并希望贡献,因为没有人建议实际捕获IndexError.

try:
    obj = Model.objects.filter(params)[0]
except IndexError:
    obj = Model.objects.create(params)