Django: pobierz obiekt z bazy danych lub 'None' jeśli nic nie pasuje

Czy jest jakaś funkcja Django, która pozwoli mi pobrać obiekt z bazy danych lub Brak, jeśli nic nie pasuje?

W tej chwili używam czegoś takiego:

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

Ale to nie jest zbyt jasne, a mieć wszędzie bałagan.

W Django 1.6 możesz użyć first()metody Queryset. Zwraca pierwszy obiekt dopasowany przez zestaw zapytań lub None, jeśli nie ma pasującego obiektu.

Stosowanie:

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

Można to zrobić na dwa sposoby;

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

Lub możesz użyć opakowania:

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

Nazwij to tak

foo = get_or_none(Foo, baz=bar)

Aby dodać przykładowy kod do odpowiedzi sorki (dodałbym to jako komentarz, ale to jest mój pierwszy post i nie mam wystarczającej reputacji, aby dodawać komentarze), zaimplementowałem niestandardowego menedżera get_or_none w następujący sposób:

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

A teraz mogę to zrobić:

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

Możesz także spróbować użyć django irytującego (ma inne przydatne funkcje!)

zainstaluj za pomocą:

pip install django-annoying

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

Daj Foo jego niestandardowego menedżera . To całkiem proste - po prostu umieść kod w funkcji w menedżerze niestandardowym, ustaw menedżera niestandardowego w modelu i wywołaj go za pomocą Foo.objects.your_new_func(...).

Jeśli potrzebujesz funkcji ogólnej (aby używać jej w dowolnym modelu, nie tylko z niestandardowym menedżerem), napisz własną i umieść ją w dowolnym miejscu na ścieżce Pythona i zaimportuj, nie brudząc się już.

Niezależnie od tego, czy robisz to za pośrednictwem menedżera, czy funkcji ogólnej, możesz również chcieć złapać `` MultipleObjectsReturned '' w instrukcji TRY, ponieważ funkcja get () podniesie to, jeśli twoje kwargs pobiorą więcej niż jeden obiekt.

Opierając się na funkcji ogólnej:

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

aw menedżerze:

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

Oto odmiana funkcji pomocniczej, która pozwala opcjonalnie przekazać QuerySetinstancję, na wypadek gdybyś chciał pobrać unikalny obiekt (jeśli jest obecny) z zestawu zapytań innego niż allzestaw zapytań obiektów modelu (np. Z podzbioru elementów potomnych należących do instancja nadrzędna):

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

Można to wykorzystać na dwa sposoby, np .:

1. obj = get_unique_or_none(Model, *args, **kwargs) jak omówiono wcześniej
2. obj = get_unique_or_none(Model, parent.children, *args, **kwargs)

Myślę, że w większości przypadków możesz po prostu użyć:

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

Tylko jeśli nie jest krytyczne, że nowy wpis zostanie dodany w tabeli Foo (inne kolumny będą miały wartości Brak / domyślne)