diff options
| author | Tom Christie | 2015-02-04 23:32:48 +0000 | 
|---|---|---|
| committer | Tom Christie | 2015-02-04 23:32:48 +0000 | 
| commit | 83673e8f74abc6dfd77e9b87eca6da9ec319da2b (patch) | |
| tree | 417c9ef574d96d15c2771960211f0498c57ce3c8 | |
| parent | 41b213414df57d7e39f1bbf3aaa35a1b033e89a3 (diff) | |
| parent | 030f01afdbcd4018a288250ef1f4c12de28e63bb (diff) | |
| download | django-rest-framework-83673e8f74abc6dfd77e9b87eca6da9ec319da2b.tar.bz2 | |
Merge branch 'version-3.1-2489' of git://github.com/brandoncazander/django-rest-framework into brandoncazander-version-3.1-2489
| -rw-r--r-- | rest_framework/relations.py | 7 | ||||
| -rw-r--r-- | rest_framework/reverse.py | 15 | ||||
| -rw-r--r-- | rest_framework/versioning.py | 16 | ||||
| -rw-r--r-- | tests/test_versioning.py | 41 | 
4 files changed, 73 insertions, 6 deletions
| diff --git a/rest_framework/relations.py b/rest_framework/relations.py index 66857a41..809d3db9 100644 --- a/rest_framework/relations.py +++ b/rest_framework/relations.py @@ -1,7 +1,7 @@  # coding: utf-8  from __future__ import unicode_literals  from django.core.exceptions import ObjectDoesNotExist, ImproperlyConfigured -from django.core.urlresolvers import resolve, get_script_prefix, NoReverseMatch, Resolver404 +from django.core.urlresolvers import get_script_prefix, NoReverseMatch, Resolver404  from django.db.models.query import QuerySet  from django.utils import six  from django.utils.encoding import smart_text @@ -9,7 +9,7 @@ from django.utils.six.moves.urllib import parse as urlparse  from django.utils.translation import ugettext_lazy as _  from rest_framework.compat import OrderedDict  from rest_framework.fields import get_attribute, empty, Field -from rest_framework.reverse import reverse +from rest_framework.reverse import reverse, resolve  from rest_framework.utils import html @@ -205,6 +205,7 @@ class HyperlinkedRelatedField(RelatedField):          return self.reverse(view_name, kwargs=kwargs, request=request, format=format)      def to_internal_value(self, data): +        request = self.context.get('request', None)          try:              http_prefix = data.startswith(('http:', 'https:'))          except AttributeError: @@ -218,7 +219,7 @@ class HyperlinkedRelatedField(RelatedField):                  data = '/' + data[len(prefix):]          try: -            match = self.resolve(data) +            match = self.resolve(data, request=request)          except Resolver404:              self.fail('no_match') diff --git a/rest_framework/reverse.py b/rest_framework/reverse.py index 8fcca55b..0d1d94a7 100644 --- a/rest_framework/reverse.py +++ b/rest_framework/reverse.py @@ -1,12 +1,25 @@  """ -Provide reverse functions that return fully qualified URLs +Provide urlresolver functions that return fully qualified URLs or view names  """  from __future__ import unicode_literals  from django.core.urlresolvers import reverse as django_reverse +from django.core.urlresolvers import resolve as django_resolve  from django.utils import six  from django.utils.functional import lazy +def resolve(path, urlconf=None, request=None): +    """ +    If versioning is being used then we pass any `resolve` calls through +    to the versioning scheme instance, so that the resulting view name +    can be modified if needed. +    """ +    scheme = getattr(request, 'versioning_scheme', None) +    if scheme is not None: +        return scheme.resolve(path, urlconf, request) +    return django_resolve(path, urlconf) + +  def reverse(viewname, args=None, kwargs=None, request=None, format=None, **extra):      """      If versioning is being used then we pass any `reverse` calls through diff --git a/rest_framework/versioning.py b/rest_framework/versioning.py index a07b629f..a76da17a 100644 --- a/rest_framework/versioning.py +++ b/rest_framework/versioning.py @@ -1,6 +1,8 @@  # coding: utf-8  from __future__ import unicode_literals  from django.utils.translation import ugettext_lazy as _ +from django.core.urlresolvers import resolve as django_resolve +from django.core.urlresolvers import ResolverMatch  from rest_framework import exceptions  from rest_framework.compat import unicode_http_header  from rest_framework.reverse import _reverse @@ -24,6 +26,9 @@ class BaseVersioning(object):      def reverse(self, viewname, args=None, kwargs=None, request=None, format=None, **extra):          return _reverse(viewname, args, kwargs, request, format, **extra) +    def resolve(self, path, urlconf=None): +        return django_resolve(path, urlconf) +      def is_allowed_version(self, version):          if not self.allowed_versions:              return True @@ -127,6 +132,17 @@ class NamespaceVersioning(BaseVersioning):              viewname, args, kwargs, request, format, **extra          ) +    def resolve(self, path, urlconf=None, request=None): +        match = django_resolve(path, urlconf) +        if match.namespace: +            _, view_name = match.view_name.split(':') +            return ResolverMatch(func=match.func, +                                 args=match.args, +                                 kwargs=match.kwargs, +                                 url_name=view_name, +                                 app_name=match.app_name) +        return match +  class HostNameVersioning(BaseVersioning):      """ diff --git a/tests/test_versioning.py b/tests/test_versioning.py index c44f727d..e7c8485e 100644 --- a/tests/test_versioning.py +++ b/tests/test_versioning.py @@ -1,9 +1,13 @@ +from .utils import MockObject, MockQueryset  from django.conf.urls import include, url +from django.core.exceptions import ObjectDoesNotExist +from rest_framework import serializers  from rest_framework import status, versioning  from rest_framework.decorators import APIView  from rest_framework.response import Response  from rest_framework.reverse import reverse -from rest_framework.test import APIRequestFactory, APITestCase +from rest_framework.test import APIRequestFactory, APITestCase, APISimpleTestCase +from rest_framework.versioning import NamespaceVersioning  class RequestVersionView(APIView): @@ -29,15 +33,18 @@ class RequestInvalidVersionView(APIView):  factory = APIRequestFactory()  mock_view = lambda request: None +dummy_view = lambda request, pk: None  included_patterns = [      url(r'^namespaced/$', mock_view, name='another'), +    url(r'^example/(?P<pk>\d+)/$', dummy_view, name='example-detail')  ]  urlpatterns = [      url(r'^v1/', include(included_patterns, namespace='v1')),      url(r'^another/$', mock_view, name='another'), -    url(r'^(?P<version>[^/]+)/another/$', mock_view, name='another') +    url(r'^(?P<version>[^/]+)/another/$', mock_view, name='another'), +    url(r'^example/(?P<pk>\d+)/$', dummy_view, name='example-detail')  ] @@ -221,3 +228,33 @@ class TestInvalidVersion:          request.resolver_match = FakeResolverMatch          response = view(request, version='v3')          assert response.status_code == status.HTTP_404_NOT_FOUND + + +class TestHyperlinkedRelatedField(APISimpleTestCase): +    urls = 'tests.test_versioning' + +    def setUp(self): + +        class HyperlinkedMockQueryset(MockQueryset): +            def get(self, **lookup): +                for item in self.items: +                    if item.pk == int(lookup.get('pk', -1)): +                        return item +                raise ObjectDoesNotExist() + +        self.queryset = HyperlinkedMockQueryset([ +            MockObject(pk=1, name='foo'), +            MockObject(pk=2, name='bar'), +            MockObject(pk=3, name='baz') +        ]) +        self.field = serializers.HyperlinkedRelatedField( +            view_name='example-detail', +            queryset=self.queryset +        ) +        request = factory.post('/', urlconf='tests.test_versioning') +        request.versioning_scheme = NamespaceVersioning() +        self.field._context = {'request': request} + +    def test_bug_2489(self): +        self.field.to_internal_value('/example/3/') +        self.field.to_internal_value('/v1/example/3/') | 
