Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Django; is it possible to have a default value for queryset

Tags:

django

employee = Employee.objects.filter('age' = 99)

We assume this queryset is empty.

If I use employee[0], that will return index out of range error, so is it possible to have a None as default value here?

`employee[0] or None`? # This won't work, but this is what I mean.
like image 280
user469652 Avatar asked Jan 18 '11 10:01

user469652


4 Answers

If you just want to get a single instance, use get, not filter:

employee = Employee.objects.get(age = 99)

If that doesn't exist, you'll get a Employee.DoesNotExist exception, which you'll need to catch. If there's more than one 99 year-old employee, you'll get a Employee.MultipleObjectsReturned exception, which you may want to catch.

There's always django-annoying's get_object_or_None if you're feeling lazy!

from annoying.functions import get_object_or_None

obj = get_object_or_None(Employee, age=99)

If you don't want to use all of django-annoying, you can always add get_object_or_None somewhere, it just looks like this:

def get_object_or_None(klass, *args, **kwargs):
    """
    Uses get() to return an object or None if the object does not exist.

    klass may be a Model, Manager, or QuerySet object. All other passed
    arguments and keyword arguments are used in the get() query.

    Note: Like with get(), a MultipleObjectsReturned will be raised if
    more than one object is found.
    """
    queryset = _get_queryset(klass)
    try:
        return queryset.get(*args, **kwargs)
    except queryset.model.DoesNotExist:
        return None
like image 135
Dominic Rodger Avatar answered Nov 10 '22 13:11

Dominic Rodger


On Django >= 1.6 there's the first method, so you can just do:

employee = Employee.objects.filter(age=99).first()

Note, however, that this only makes sense if you know the QuerySet is guaranteed to return either 1 or 0 results.

like image 23
Ariel Avatar answered Nov 10 '22 11:11

Ariel


This should work:

employee[0] if employee else None
like image 27
ikostia Avatar answered Nov 10 '22 12:11

ikostia


first_employee = employee[0] if employee.count() > 0 else None
like image 1
Bernhard Vallant Avatar answered Nov 10 '22 13:11

Bernhard Vallant