Django: получить объект из БД или None, если ничего не подходит

Есть ли функция Django, которая позволит мне получить объект из базы данных, или None, если ничего не подходит?

Сейчас я использую что-то вроде:

foo = Foo.objects.filter(bar=baz)
foo = len(foo) > 0 and foo.get() or None

Но это не очень понятно, и иметь везде грязно.

В Django 1.6 вы можете использовать first()метод Queryset. Он возвращает первый объект, соответствующий набору запросов, или None, если подходящего объекта нет.

Использование:

p = Article.objects.order_by('title', 'pub_date').first()

Есть два способа сделать это;

try:
    foo = Foo.objects.get(bar=baz)
except model.DoesNotExist:
    foo = None

Или вы можете использовать обертку:

def get_or_none(model, *args, **kwargs):
    try:
        return model.objects.get(*args, **kwargs)
    except model.DoesNotExist:
        return None

Назовите это так

foo = get_or_none(Foo, baz=bar)

Чтобы добавить пример кода к ответу Сорки (я бы добавил это как комментарий, но это мой первый пост, и у меня недостаточно репутации, чтобы оставлять комментарии), я реализовал настраиваемый менеджер get_or_none следующим образом:

from django.db import models

class GetOrNoneManager(models.Manager):
    """Adds get_or_none method to objects
    """
    def get_or_none(self, **kwargs):
        try:
            return self.get(**kwargs)
        except self.model.DoesNotExist:
            return None

class Person(models.Model):
    name = models.CharField(max_length=255)
    objects = GetOrNoneManager()

И теперь я могу это сделать:

bob_or_none = Person.objects.get_or_none(name='Bob')

Также можно попробовать использовать раздражающий django (у него есть и другие полезные функции!)

установите его с помощью:

pip install django-annoying

from annoying.functions import get_object_or_None
get_object_or_None(Foo, bar=baz)

Дайте Foo его собственный менеджер . Это довольно просто - просто введите свой код в функцию в пользовательском диспетчере, установите пользовательский диспетчер в своей модели и вызовите его с помощью Foo.objects.your_new_func(...).

Если вам нужна универсальная функция (чтобы использовать ее в любой модели, а не только с настраиваемым менеджером), напишите свою собственную и поместите ее где-нибудь на своем пути python и импортируйте, не беспорядочно.

Независимо от того, выполняете ли вы это через менеджер или общую функцию, вы также можете захотеть поймать MultipleObjectsReturned в операторе TRY, поскольку функция get () поднимет это, если ваши kwargs извлекут более одного объекта.

Основываясь на общей функции:

def get_unique_or_none(model, *args, **kwargs):
    try:
        return model.objects.get(*args, **kwargs)
    except (model.DoesNotExist, model.MultipleObjectsReturned), err:
        return None

и в диспетчере:

class GetUniqueOrNoneManager(models.Manager):
    """Adds get_unique_or_none method to objects
    """
    def get_unique_or_none(self, *args, **kwargs):
        try:
            return self.get(*args, **kwargs)
        except (self.model.DoesNotExist, self.model.MultipleObjectsReturned), err:
            return None

Вот вариант вспомогательной функции, которая позволяет при желании передать QuerySetэкземпляр, если вы хотите получить уникальный объект (если он есть) из набора запросов, отличного от набора запросов allобъектов модели (например, из подмножества дочерних элементов, принадлежащих к родительский экземпляр):

def get_unique_or_none(model, queryset=None, *args, **kwargs):
    """
        Performs the query on the specified `queryset`
        (defaulting to the `all` queryset of the `model`'s default manager)
        and returns the unique object matching the given
        keyword arguments.  Returns `None` if no match is found.
        Throws a `model.MultipleObjectsReturned` exception
        if more than one match is found.
    """
    if queryset is None:
        queryset = model.objects.all()
    try:
        return queryset.get(*args, **kwargs)
    except model.DoesNotExist:
        return None

Это можно использовать двумя способами, например:

1. obj = get_unique_or_none(Model, *args, **kwargs) как обсуждалось ранее
2. obj = get_unique_or_none(Model, parent.children, *args, **kwargs)

Я думаю, что в большинстве случаев вы можете просто использовать:

foo, created = Foo.objects.get_or_create(bar=baz)

Только если не критично, что новая запись будет добавлена ​​в таблицу Foo (другие столбцы будут иметь значения None / по умолчанию)