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

Ви також можете спробувати використовувати джанго дратівливого (він має ще одну корисну функцію!)

встановіть його за допомогою:

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 / default)