mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-11-10 19:56:59 +03:00
Fix @api_view decorator tests
This commit is contained in:
parent
622e001e0b
commit
0cc7030aab
|
@ -2,4 +2,23 @@
|
||||||
|
|
||||||
# Parsers
|
# Parsers
|
||||||
|
|
||||||
## .parse(request)
|
> Machine interacting web services tend to use more
|
||||||
|
structured formats for sending data than form-encoded, since they're
|
||||||
|
sending more complex data than simple forms
|
||||||
|
>
|
||||||
|
> — Malcom Tredinnick, [Django developers group][cite]
|
||||||
|
|
||||||
|
|
||||||
|
## JSONParser
|
||||||
|
|
||||||
|
## YAMLParser
|
||||||
|
|
||||||
|
## XMLParser
|
||||||
|
|
||||||
|
## FormParser
|
||||||
|
|
||||||
|
## MultiPartParser
|
||||||
|
|
||||||
|
## Custom parsers
|
||||||
|
|
||||||
|
[cite]: https://groups.google.com/d/topic/django-developers/dxI4qVzrBY4/discussion
|
||||||
|
|
|
@ -2,5 +2,22 @@
|
||||||
|
|
||||||
# Renderers
|
# Renderers
|
||||||
|
|
||||||
## .render(response)
|
> Before a TemplateResponse instance can be returned to the client, it must be rendered. The rendering process takes the intermediate representation of template and context, and turns it into the final byte stream that can be served to the client.
|
||||||
|
>
|
||||||
|
> — [Django documentation][cite]
|
||||||
|
|
||||||
|
## JSONRenderer
|
||||||
|
|
||||||
|
## JSONPRenderer
|
||||||
|
|
||||||
|
## YAMLRenderer
|
||||||
|
|
||||||
|
## XMLRenderer
|
||||||
|
|
||||||
|
## DocumentingHTMLRenderer
|
||||||
|
|
||||||
|
## TemplatedHTMLRenderer
|
||||||
|
|
||||||
|
## Custom renderers
|
||||||
|
|
||||||
|
[cite]: https://docs.djangoproject.com/en/dev/ref/template-response/#the-rendering-process
|
|
@ -4,7 +4,7 @@
|
||||||
|
|
||||||
> If you're doing REST-based web service stuff ... you should ignore request.POST.
|
> If you're doing REST-based web service stuff ... you should ignore request.POST.
|
||||||
>
|
>
|
||||||
> — Malcom Tredinnick, [Django developers group][cite]
|
> — Malcom Tredinnick, [Django developers group][cite]
|
||||||
|
|
||||||
REST framework's `Request` class extends the standard `HttpRequest`, adding support for parsing multiple content types, allowing browser-based `PUT`, `DELETE` and other methods, and adding flexible per-request authentication.
|
REST framework's `Request` class extends the standard `HttpRequest`, adding support for parsing multiple content types, allowing browser-based `PUT`, `DELETE` and other methods, and adding flexible per-request authentication.
|
||||||
|
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
<a class="github" href="views.py"></a>
|
<a class="github" href="decorators.py"></a> <a class="github" href="views.py"></a>
|
||||||
|
|
||||||
# Views
|
# Views
|
||||||
|
|
||||||
|
@ -6,36 +6,108 @@
|
||||||
>
|
>
|
||||||
> — [Reinout van Rees][cite]
|
> — [Reinout van Rees][cite]
|
||||||
|
|
||||||
REST framework provides a simple `APIView` class, built on Django's `django.generics.views.View`. The `APIView` class ensures five main things:
|
REST framework provides an `APIView` class, which subclasses Django's `View` class.
|
||||||
|
|
||||||
1. Any requests inside the view will become `Request` instances.
|
`APIView` classes are different from regular `View` classes in the following ways:
|
||||||
2. `Request` instances will have their `renderers` and `authentication` attributes automatically set.
|
|
||||||
3. `Response` instances will have their `parsers` and `serializer` attributes automatically set.
|
|
||||||
4. `APIException` exceptions will be caught and return appropriate responses.
|
|
||||||
5. Any permissions provided will be checked prior to passing the request to a handler method.
|
|
||||||
|
|
||||||
Additionally there are a some minor extras, such as providing a default `options` handler, setting some common headers on the response prior to return, and providing the useful `initial()` and `final()` hooks.
|
* Requests passed to the handler methods will be REST framework's `Request` instances, not Django's `HttpRequest` instances.
|
||||||
|
* Handler methods may return REST framework's `Response`, instead of Django's `HttpResponse`. The view will manage content negotiation and setting the correct renderer on the response.
|
||||||
|
* Any `APIException` exceptions will be caught and mediated into appropriate responses.
|
||||||
|
* Incoming requests will be authenticated and appropriate permission and/or throttle checks will be run before dispatching the request to the handler method.
|
||||||
|
|
||||||
## APIView
|
Using the `APIView` class is pretty much the same as using a regular `View` class, as usual, the incoming request is dispatched to an appropriate handler method such as `.get()` or `.post()`. Additionally, a number of attributes may be set on the class that control various aspects of the API policy.
|
||||||
|
|
||||||
## Method handlers
|
For example:
|
||||||
|
|
||||||
Describe that APIView handles regular .get(), .post(), .put(), .delete() etc...
|
class ListUsers(APIView):
|
||||||
|
"""
|
||||||
|
View to list all users in the system.
|
||||||
|
|
||||||
|
* Requires token authentication.
|
||||||
|
* Only admin users are able to access this view.
|
||||||
|
"""
|
||||||
|
authentication_classes = (authentication.TokenAuthentication,)
|
||||||
|
permission_classes = (permissions.IsAdmin,)
|
||||||
|
|
||||||
## .initial(request, *args, **kwargs)
|
def get(self, request, format=None):
|
||||||
|
"""
|
||||||
|
Return a list of all users.
|
||||||
|
"""
|
||||||
|
users = [user.username for user in User.objects.all()]
|
||||||
|
return Response(users)
|
||||||
|
|
||||||
## .final(request, response, *args, **kwargs)
|
## API policy attributes
|
||||||
|
|
||||||
## .parsers
|
The following attributes control the pluggable aspects of API views.
|
||||||
|
|
||||||
## .renderers
|
### .renderer_classes
|
||||||
|
|
||||||
## .serializer
|
### .parser_classes
|
||||||
|
|
||||||
## .authentication
|
### .authentication_classes
|
||||||
|
|
||||||
## .permissions
|
### .throttle_classes
|
||||||
|
|
||||||
## .headers
|
### .permission_classes
|
||||||
|
|
||||||
|
### .content_negotiation_class
|
||||||
|
|
||||||
|
## API policy instantiation methods
|
||||||
|
|
||||||
|
The following methods are used by REST framework to instantiate the various pluggable API policies. You won't typically need to override these methods.
|
||||||
|
|
||||||
|
### .get_renderers(self)
|
||||||
|
|
||||||
|
### .get_parsers(self)
|
||||||
|
|
||||||
|
### .get_authenticators(self)
|
||||||
|
|
||||||
|
### .get_throttles(self)
|
||||||
|
|
||||||
|
### .get_permissions(self)
|
||||||
|
|
||||||
|
### .get_content_negotiator(self)
|
||||||
|
|
||||||
|
## API policy implementation methods
|
||||||
|
|
||||||
|
The following methods are called before dispatching to the handler method.
|
||||||
|
|
||||||
|
### .check_permissions(...)
|
||||||
|
|
||||||
|
### .check_throttles(...)
|
||||||
|
|
||||||
|
### .perform_content_negotiation(...)
|
||||||
|
|
||||||
|
## Dispatch methods
|
||||||
|
|
||||||
|
The following methods are called directly by the view's `.dispatch()` method.
|
||||||
|
These perform any actions that need to occur before or after calling the handler methods such as `.get()`, `.post()`, `put()` and `.delete()`.
|
||||||
|
|
||||||
|
### .initial(self, request, *args, **kwargs)
|
||||||
|
|
||||||
|
Performs any actions that need to occur before the handler method gets called.
|
||||||
|
This method is used to enforce permissions and throttling, and perform content negotiation.
|
||||||
|
|
||||||
|
You won't typically need to override this method.
|
||||||
|
|
||||||
|
### .handle_exception(self, exc)
|
||||||
|
|
||||||
|
Any exception thrown by the handler method will be passed to this method, which either returns a `Response` instance, or re-raises the exception.
|
||||||
|
|
||||||
|
The default implementation handles any subclass of `rest_framework.exceptions.APIException`, as well as Django's `Http404` and `PermissionDenied` exceptions, and returns an appropriate error response.
|
||||||
|
|
||||||
|
If you need to customize the error responses your API returns you should subclass this method.
|
||||||
|
|
||||||
|
### .initialize_request(self, request, *args, **kwargs)
|
||||||
|
|
||||||
|
Ensures that the request object that is passed to the handler method is an instance of `Request`, rather than the usual Django `HttpRequest`.
|
||||||
|
|
||||||
|
You won't typically need to override this method.
|
||||||
|
|
||||||
|
### .finalize_response(self, request, response, *args, **kwargs)
|
||||||
|
|
||||||
|
Ensures that any `Response` object returned from the handler method will be rendered into the correct content type, as determined by the content negotation.
|
||||||
|
|
||||||
|
You won't typically need to override this method.
|
||||||
|
|
||||||
[cite]: http://reinout.vanrees.org/weblog/2011/08/24/class-based-views-usage.html
|
[cite]: http://reinout.vanrees.org/weblog/2011/08/24/class-based-views-usage.html
|
|
@ -1,11 +1,3 @@
|
||||||
from functools import wraps
|
|
||||||
from django.utils.decorators import available_attrs
|
|
||||||
from django.core.exceptions import PermissionDenied
|
|
||||||
from rest_framework import exceptions
|
|
||||||
from rest_framework import status
|
|
||||||
from rest_framework.response import Response
|
|
||||||
from rest_framework.request import Request
|
|
||||||
from rest_framework.settings import api_settings
|
|
||||||
from rest_framework.views import APIView
|
from rest_framework.views import APIView
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -77,3 +77,10 @@ class Throttled(APIException):
|
||||||
self.detail = format % (self.wait, self.wait != 1 and 's' or '')
|
self.detail = format % (self.wait, self.wait != 1 and 's' or '')
|
||||||
else:
|
else:
|
||||||
self.detail = detail or self.default_detail
|
self.detail = detail or self.default_detail
|
||||||
|
|
||||||
|
|
||||||
|
class ConfigurationError(Exception):
|
||||||
|
"""
|
||||||
|
Indicates an internal server error.
|
||||||
|
"""
|
||||||
|
pass
|
||||||
|
|
|
@ -39,6 +39,10 @@ DEFAULTS = {
|
||||||
'DEFAULT_THROTTLES': (),
|
'DEFAULT_THROTTLES': (),
|
||||||
'DEFAULT_CONTENT_NEGOTIATION':
|
'DEFAULT_CONTENT_NEGOTIATION':
|
||||||
'rest_framework.negotiation.DefaultContentNegotiation',
|
'rest_framework.negotiation.DefaultContentNegotiation',
|
||||||
|
'DEFAULT_THROTTLE_RATES': {
|
||||||
|
'user': None,
|
||||||
|
'anon': None,
|
||||||
|
},
|
||||||
|
|
||||||
'UNAUTHENTICATED_USER': 'django.contrib.auth.models.AnonymousUser',
|
'UNAUTHENTICATED_USER': 'django.contrib.auth.models.AnonymousUser',
|
||||||
'UNAUTHENTICATED_TOKEN': None,
|
'UNAUTHENTICATED_TOKEN': None,
|
||||||
|
|
|
@ -1,10 +1,11 @@
|
||||||
from django.test import TestCase
|
from django.test import TestCase
|
||||||
|
from rest_framework import status
|
||||||
from rest_framework.response import Response
|
from rest_framework.response import Response
|
||||||
from rest_framework.compat import RequestFactory
|
from rest_framework.compat import RequestFactory
|
||||||
from rest_framework.renderers import JSONRenderer
|
from rest_framework.renderers import JSONRenderer
|
||||||
from rest_framework.parsers import JSONParser
|
from rest_framework.parsers import JSONParser
|
||||||
from rest_framework.authentication import BasicAuthentication
|
from rest_framework.authentication import BasicAuthentication
|
||||||
from rest_framework.throttling import SimpleRateThottle
|
from rest_framework.throttling import UserRateThrottle
|
||||||
from rest_framework.permissions import IsAuthenticated
|
from rest_framework.permissions import IsAuthenticated
|
||||||
from rest_framework.views import APIView
|
from rest_framework.views import APIView
|
||||||
from rest_framework.decorators import (
|
from rest_framework.decorators import (
|
||||||
|
@ -23,7 +24,6 @@ class DecoratorTestCase(TestCase):
|
||||||
self.factory = RequestFactory()
|
self.factory = RequestFactory()
|
||||||
|
|
||||||
def _finalize_response(self, request, response, *args, **kwargs):
|
def _finalize_response(self, request, response, *args, **kwargs):
|
||||||
print "HAI"
|
|
||||||
response.request = request
|
response.request = request
|
||||||
return APIView.finalize_response(self, request, response, *args, **kwargs)
|
return APIView.finalize_response(self, request, response, *args, **kwargs)
|
||||||
|
|
||||||
|
@ -87,21 +87,24 @@ class DecoratorTestCase(TestCase):
|
||||||
@api_view(['GET'])
|
@api_view(['GET'])
|
||||||
@permission_classes([IsAuthenticated])
|
@permission_classes([IsAuthenticated])
|
||||||
def view(request):
|
def view(request):
|
||||||
self.assertEqual(request.permission_classes, [IsAuthenticated])
|
|
||||||
return Response({})
|
return Response({})
|
||||||
|
|
||||||
request = self.factory.get('/')
|
request = self.factory.get('/')
|
||||||
view(request)
|
response = view(request)
|
||||||
|
self.assertEquals(response.status_code, status.HTTP_403_FORBIDDEN)
|
||||||
|
|
||||||
# Doesn't look like this bits are working quite yet
|
def test_throttle_classes(self):
|
||||||
|
class OncePerDayUserThrottle(UserRateThrottle):
|
||||||
|
rate = '1/day'
|
||||||
|
|
||||||
# def test_throttle_classes(self):
|
@api_view(['GET'])
|
||||||
|
@throttle_classes([OncePerDayUserThrottle])
|
||||||
|
def view(request):
|
||||||
|
return Response({})
|
||||||
|
|
||||||
# @api_view(['GET'])
|
request = self.factory.get('/')
|
||||||
# @throttle_classes([SimpleRateThottle])
|
response = view(request)
|
||||||
# def view(request):
|
self.assertEquals(response.status_code, status.HTTP_200_OK)
|
||||||
# self.assertEqual(request.throttle_classes, [SimpleRateThottle])
|
|
||||||
# return Response({})
|
|
||||||
|
|
||||||
# request = self.factory.get('/')
|
response = view(request)
|
||||||
# view(request)
|
self.assertEquals(response.status_code, status.HTTP_429_TOO_MANY_REQUESTS)
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
import time
|
import time
|
||||||
from django.core.cache import cache
|
from django.core.cache import cache
|
||||||
|
from rest_framework import exceptions
|
||||||
from rest_framework.settings import api_settings
|
from rest_framework.settings import api_settings
|
||||||
|
|
||||||
|
|
||||||
|
@ -49,8 +50,9 @@ class SimpleRateThottle(BaseThrottle):
|
||||||
|
|
||||||
def __init__(self, view):
|
def __init__(self, view):
|
||||||
super(SimpleRateThottle, self).__init__(view)
|
super(SimpleRateThottle, self).__init__(view)
|
||||||
rate = self.get_rate_description()
|
if not getattr(self, 'rate', None):
|
||||||
self.num_requests, self.duration = self.parse_rate_description(rate)
|
self.rate = self.get_rate()
|
||||||
|
self.num_requests, self.duration = self.parse_rate(self.rate)
|
||||||
|
|
||||||
def get_cache_key(self, request):
|
def get_cache_key(self, request):
|
||||||
"""
|
"""
|
||||||
|
@ -61,21 +63,28 @@ class SimpleRateThottle(BaseThrottle):
|
||||||
"""
|
"""
|
||||||
raise NotImplementedError('.get_cache_key() must be overridden')
|
raise NotImplementedError('.get_cache_key() must be overridden')
|
||||||
|
|
||||||
def get_rate_description(self):
|
def get_rate(self):
|
||||||
"""
|
"""
|
||||||
Determine the string representation of the allowed request rate.
|
Determine the string representation of the allowed request rate.
|
||||||
"""
|
"""
|
||||||
try:
|
if not getattr(self, 'scope', None):
|
||||||
return self.rate
|
msg = ("You must set either `.scope` or `.rate` for '%s' thottle" %
|
||||||
except AttributeError:
|
self.__class__.__name__)
|
||||||
return self.settings.DEFAULT_THROTTLE_RATES.get(self.scope)
|
raise exceptions.ConfigurationError(msg)
|
||||||
|
|
||||||
def parse_rate_description(self, rate):
|
try:
|
||||||
|
return self.settings.DEFAULT_THROTTLE_RATES[self.scope]
|
||||||
|
except KeyError:
|
||||||
|
msg = "No default throttle rate set for '%s' scope" % self.scope
|
||||||
|
raise exceptions.ConfigurationError(msg)
|
||||||
|
|
||||||
|
def parse_rate(self, rate):
|
||||||
"""
|
"""
|
||||||
Given the request rate string, return a two tuple of:
|
Given the request rate string, return a two tuple of:
|
||||||
<allowed number of requests>, <period of time in seconds>
|
<allowed number of requests>, <period of time in seconds>
|
||||||
"""
|
"""
|
||||||
assert rate, "No throttle rate set for '%s'" % self.__class__.__name__
|
if rate is None:
|
||||||
|
return (None, None)
|
||||||
num, period = rate.split('/')
|
num, period = rate.split('/')
|
||||||
num_requests = int(num)
|
num_requests = int(num)
|
||||||
duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
|
duration = {'s': 1, 'm': 60, 'h': 3600, 'd': 86400}[period[0]]
|
||||||
|
@ -88,6 +97,9 @@ class SimpleRateThottle(BaseThrottle):
|
||||||
On success calls `throttle_success`.
|
On success calls `throttle_success`.
|
||||||
On failure calls `throttle_failure`.
|
On failure calls `throttle_failure`.
|
||||||
"""
|
"""
|
||||||
|
if self.rate is None:
|
||||||
|
return True
|
||||||
|
|
||||||
self.key = self.get_cache_key(request)
|
self.key = self.get_cache_key(request)
|
||||||
self.history = cache.get(self.key, [])
|
self.history = cache.get(self.key, [])
|
||||||
self.now = self.timer()
|
self.now = self.timer()
|
||||||
|
@ -188,14 +200,6 @@ class ScopedRateThrottle(SimpleRateThottle):
|
||||||
self.scope = getattr(self.view, self.scope_attr, None)
|
self.scope = getattr(self.view, self.scope_attr, None)
|
||||||
super(ScopedRateThrottle, self).__init__(view)
|
super(ScopedRateThrottle, self).__init__(view)
|
||||||
|
|
||||||
def parse_rate_description(self, rate):
|
|
||||||
"""
|
|
||||||
Subclassed so that we don't fail if `view.throttle_scope` is not set.
|
|
||||||
"""
|
|
||||||
if not rate:
|
|
||||||
return (None, None)
|
|
||||||
return super(ScopedRateThrottle, self).parse_rate_description(rate)
|
|
||||||
|
|
||||||
def get_cache_key(self, request):
|
def get_cache_key(self, request):
|
||||||
"""
|
"""
|
||||||
If `view.throttle_scope` is not set, don't apply this throttle.
|
If `view.throttle_scope` is not set, don't apply this throttle.
|
||||||
|
|
Loading…
Reference in New Issue
Block a user