Django：测试获取查询

Question

好吧，我厌倦了写这个......

res = Something.objects.filter(asdf=something)

if res:
  single = res[0]
else:
  single = None

if single:
  # do some stuff

我更愿意能够做这样的事情：

single = Something.objects.filter(asdf=something)
if single:
  #do some stuff

我希望能够在不测试过滤结果的情况下抓取一个单个对象。

换句话说，当我知道将有 1 个或 0 个匹配条目时，我想直接跳到该条目，否则只会得到“无”。当尝试将这些查询压缩到一行中时，与 .get 一起出现的DoesNotExist 错误并不总是那么有效。

有什么办法可以做到我所描述的吗？

Answer 1

django-annoying 项目包含一个 get_object_or_None 快捷方式来执行此操作，尽管自己写起来很简单。

Answer 2

创建自定义管理器它将您厌倦重复的部分封装为一种方法（具有比下面的更好的名称），或者只编写一个实用程序函数，它可以在不影响模型定义的情况下执行相同的操作：

class MyManager(models.Manager):
    def get_or_none(self, **kwargs):
        try:
            return self.get(**kwargs)
        except self.model.DoesNotExist:
            return None

class MyModel(models.Model):
    objects = MyManager()

用法：

MyModel.objects.get_or_none(asdf=something)