mirror of
https://github.com/jazzband/django-avatar.git
synced 2026-05-12 09:33:12 +00:00
27 lines
905 B
Python
27 lines
905 B
Python
from django.shortcuts import _get_queryset
|
|
|
|
|
|
def get_object_or_none(klass, *args, **kwargs):
|
|
"""
|
|
Use get() to return an object, or return 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.
|
|
|
|
Like with QuerySet.get(), MultipleObjectsReturned is raised if more than
|
|
one object is found.
|
|
"""
|
|
queryset = _get_queryset(klass)
|
|
if not hasattr(queryset, "get"):
|
|
klass__name = (
|
|
klass.__name__ if isinstance(klass, type) else klass.__class__.__name__
|
|
)
|
|
raise ValueError(
|
|
"First argument to get_object_or_404() must be a Model, Manager, "
|
|
"or QuerySet, not '%s'." % klass__name
|
|
)
|
|
try:
|
|
return queryset.get(*args, **kwargs)
|
|
except queryset.model.DoesNotExist:
|
|
return None
|