问题描述
|
在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]
解决方法
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(...)
引发和捕获异常没有错!
, 从Django 1.6开始,有一个便捷的方法first()可以在过滤器查询中返回第一个结果,即None。
obj = Model.manager.filter(params).first()
if obj is None:
obj = Model.objects.create(params)
, 这是一个管理器方法,可让您优雅地扩展此功能
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\')
, 我只是遇到了这个问题,并想做出贡献,因为没有人建议实际捕获IndexError。
try:
obj = Model.objects.filter(params)[0]
except IndexError:
obj = Model.objects.create(params)
, 这对我有用:
在您看来,可以这样调用:
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对象。如果检索失败(因为它不存在),请使用该字符串作为名称创建一个新的类别。返回结果以供您在视图中使用。
, 您可以尝试以下方法:
result = Model.objects.filter(field__lookup=value)[0]
if not result:
result = Model.objects.create(...)
return result