diff options
| author | Tom Christie | 2013-05-28 11:57:11 +0100 |
|---|---|---|
| committer | Tom Christie | 2013-05-28 11:57:11 +0100 |
| commit | 138f0cacdb0f093a4dc65ef98bf48ddbc27e18b2 (patch) | |
| tree | d478a9d3df0e658e5096e1d5d5691fe5cb9ad815 /rest_framework/generics.py | |
| parent | 7123f0b1e6b29778c41476341bd2370f60c279fa (diff) | |
| download | django-rest-framework-138f0cacdb0f093a4dc65ef98bf48ddbc27e18b2.tar.bz2 | |
Raise 404 on incorrect lookup type in URL, not 500. Closes #890.
Diffstat (limited to 'rest_framework/generics.py')
| -rw-r--r-- | rest_framework/generics.py | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/rest_framework/generics.py b/rest_framework/generics.py index afcb8a9f..9ccc7898 100644 --- a/rest_framework/generics.py +++ b/rest_framework/generics.py @@ -6,7 +6,7 @@ from __future__ import unicode_literals from django.core.exceptions import ImproperlyConfigured, PermissionDenied from django.core.paginator import Paginator, InvalidPage from django.http import Http404 -from django.shortcuts import get_object_or_404 +from django.shortcuts import get_object_or_404 as _get_object_or_404 from django.utils.translation import ugettext as _ from rest_framework import views, mixins, exceptions from rest_framework.request import clone_request @@ -14,6 +14,17 @@ from rest_framework.settings import api_settings import warnings +def get_object_or_404(queryset, **filter_kwargs): + """ + Same as Django's standard shortcut, but make sure to raise 404 + if the filter_kwargs don't match the required types. + """ + try: + return _get_object_or_404(queryset, **filter_kwargs) + except (TypeError, ValueError): + raise Http404 + + class GenericAPIView(views.APIView): """ Base class for all other generic views. |
