mirror of
https://github.com/encode/django-rest-framework.git
synced 2024-12-01 05:54:01 +03:00
made changes requested by Tom. In resources : removed decorators, and 'request' parameter from the crud methods
This commit is contained in:
parent
d1c217523b
commit
3a52c213f9
|
@ -471,12 +471,21 @@ class ResourceMixin(object):
|
||||||
"""
|
"""
|
||||||
return self.deserialize(self.request.GET)
|
return self.deserialize(self.request.GET)
|
||||||
|
|
||||||
@property
|
def get_resource_class(self):
|
||||||
def resource(self):
|
if self.resource_class:
|
||||||
if not hasattr(self, '_resource'):
|
return self.resource_class
|
||||||
self._resource = self.resource_class(view=self)
|
elif getattr(self, 'model', None):
|
||||||
return self._resource
|
return ModelResource
|
||||||
|
elif getattr(self, 'form', None):
|
||||||
|
return FormResource
|
||||||
|
elif hasattr(self, 'request') and getattr(self, '%s_form' % self.method.lower(), None):
|
||||||
|
return FormResource
|
||||||
|
else:
|
||||||
|
return Resource
|
||||||
|
|
||||||
|
def get_resource(self):
|
||||||
|
resource_class = self.get_resource_class()
|
||||||
|
return resource_class(view=self)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -496,6 +505,7 @@ class InstanceMixin(object):
|
||||||
associated with this view.
|
associated with this view.
|
||||||
"""
|
"""
|
||||||
view = super(InstanceMixin, cls).as_view(**initkwargs)
|
view = super(InstanceMixin, cls).as_view(**initkwargs)
|
||||||
|
# TODO: FIX !!! Very bad now, since this is attached on the class (thread-safety)
|
||||||
resource_class = getattr(cls(**initkwargs), 'resource_class', None)
|
resource_class = getattr(cls(**initkwargs), 'resource_class', None)
|
||||||
if resource_class:
|
if resource_class:
|
||||||
# We do a little dance when we store the view callable...
|
# We do a little dance when we store the view callable...
|
||||||
|
@ -512,51 +522,55 @@ class InstanceMixin(object):
|
||||||
class GetResourceMixin(object):
|
class GetResourceMixin(object):
|
||||||
|
|
||||||
def get(self, request, *args, **kwargs):
|
def get(self, request, *args, **kwargs):
|
||||||
|
resource = self.get_resource()
|
||||||
try:
|
try:
|
||||||
self.resource.retrieve(request, *args, **kwargs)
|
resource.retrieve(*args, **kwargs)
|
||||||
except self.resource.DoesNotExist:
|
except resource.DoesNotExist:
|
||||||
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
||||||
return self.resource.instance
|
return resource.instance
|
||||||
|
|
||||||
|
|
||||||
class PostResourceMixin(object):
|
class PostResourceMixin(object):
|
||||||
|
|
||||||
def post(self, request, *args, **kwargs):
|
def post(self, request, *args, **kwargs):
|
||||||
self.resource.create(request, *args, **kwargs)
|
resource = self.get_resource()
|
||||||
self.resource.update(self.CONTENT, request, *args, **kwargs)
|
resource.create(*args, **kwargs)
|
||||||
headers = {'Location': self.resource.get_url()}
|
resource.update(self.CONTENT, *args, **kwargs)
|
||||||
return Response(status.HTTP_201_CREATED, self.resource.instance, headers)
|
headers = {'Location': resource.get_url()}
|
||||||
|
return Response(status.HTTP_201_CREATED, resource.instance, headers)
|
||||||
|
|
||||||
|
|
||||||
class PutResourceMixin(object):
|
class PutResourceMixin(object):
|
||||||
|
|
||||||
def put(self, request, *args, **kwargs):
|
def put(self, request, *args, **kwargs):
|
||||||
headers = {}
|
resource = self.get_resource()
|
||||||
try:
|
try:
|
||||||
self.resource.retrieve(request, *args, **kwargs)
|
resource.retrieve(*args, **kwargs)
|
||||||
status_code = status.HTTP_204_NO_CONTENT
|
status_code = status.HTTP_204_NO_CONTENT
|
||||||
except self.resource.DoesNotExist:
|
except resource.DoesNotExist:
|
||||||
self.resource.create(request, *args, **kwargs)
|
resource.create(*args, **kwargs)
|
||||||
status_code = status.HTTP_201_CREATED
|
status_code = status.HTTP_201_CREATED
|
||||||
self.resource.update(self.CONTENT, request, *args, **kwargs)
|
resource.update(self.CONTENT, *args, **kwargs)
|
||||||
return Response(status_code, self.resource.instance, {})
|
return Response(status_code, resource.instance, {})
|
||||||
|
|
||||||
|
|
||||||
class DeleteResourceMixin(object):
|
class DeleteResourceMixin(object):
|
||||||
|
|
||||||
def delete(self, request, *args, **kwargs):
|
def delete(self, request, *args, **kwargs):
|
||||||
|
resource = self.get_resource()
|
||||||
try:
|
try:
|
||||||
self.resource.retrieve(request, *args, **kwargs)
|
resource.retrieve(*args, **kwargs)
|
||||||
except self.resource.DoesNotExist:
|
except resource.DoesNotExist:
|
||||||
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
||||||
self.resource.delete(request, *args, **kwargs)
|
resource.delete(*args, **kwargs)
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
class ListResourceMixin(object):
|
class ListResourceMixin(object):
|
||||||
|
|
||||||
def get(self, request, *args, **kwargs):
|
def get(self, request, *args, **kwargs):
|
||||||
return self.resource.list(request, *args, **kwargs)
|
resource = self.get_resource()
|
||||||
|
return resource.list(*args, **kwargs)
|
||||||
|
|
||||||
|
|
||||||
########## Pagination Mixins ##########
|
########## Pagination Mixins ##########
|
||||||
|
|
|
@ -216,4 +216,4 @@ class PerResourceThrottling(BaseThrottle):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def get_cache_key(self):
|
def get_cache_key(self):
|
||||||
return 'throttle_resource_%s' % self.view.resource.__class__.__name__
|
return 'throttle_resource_%s' % self.view.get_resource_class().__name__
|
||||||
|
|
|
@ -6,14 +6,6 @@ from djangorestframework.response import ErrorResponse
|
||||||
from djangorestframework.serializer import Serializer, _SkipField
|
from djangorestframework.serializer import Serializer, _SkipField
|
||||||
|
|
||||||
|
|
||||||
def bound_resource_required(meth):
|
|
||||||
def _decorated(self, *args, **kwargs):
|
|
||||||
if not self.is_bound():
|
|
||||||
raise Exception("resource needs to be bound") #TODO: what exception?
|
|
||||||
return meth(self, *args, **kwargs)
|
|
||||||
return _decorated
|
|
||||||
|
|
||||||
|
|
||||||
class BaseResource(Serializer):
|
class BaseResource(Serializer):
|
||||||
"""
|
"""
|
||||||
Base class for all Resource classes, which simply defines the interface
|
Base class for all Resource classes, which simply defines the interface
|
||||||
|
@ -26,7 +18,8 @@ class BaseResource(Serializer):
|
||||||
# TODO: Inheritance, like for models
|
# TODO: Inheritance, like for models
|
||||||
class DoesNotExist(Exception): pass
|
class DoesNotExist(Exception): pass
|
||||||
|
|
||||||
def __init__(self, instance=None, view=None, depth=None, stack=[], **kwargs):
|
# !!! `view` should be first kwarg to avoid backward incompatibilities. (lol)
|
||||||
|
def __init__(self, view=None, instance=None, depth=None, stack=[], **kwargs):
|
||||||
super(BaseResource, self).__init__(depth, stack, **kwargs)
|
super(BaseResource, self).__init__(depth, stack, **kwargs)
|
||||||
self.view = view
|
self.view = view
|
||||||
self.instance = instance
|
self.instance = instance
|
||||||
|
@ -39,21 +32,18 @@ class BaseResource(Serializer):
|
||||||
"""
|
"""
|
||||||
return data
|
return data
|
||||||
|
|
||||||
def retrieve(self, request, *args, **kwargs):
|
def retrieve(self, *args, **kwargs):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
def create(self, request, *args, **kwargs):
|
def create(self, *args, **kwargs):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@bound_resource_required
|
def update(self, data, *args, **kwargs):
|
||||||
def update(self, data, request, *args, **kwargs):
|
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@bound_resource_required
|
def delete(self, *args, **kwargs):
|
||||||
def delete(self, request, *args, **kwargs):
|
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
@bound_resource_required
|
|
||||||
def get_url(self):
|
def get_url(self):
|
||||||
raise NotImplementedError()
|
raise NotImplementedError()
|
||||||
|
|
||||||
|
@ -148,6 +138,7 @@ class FormResource(Resource):
|
||||||
if bound_form is None:
|
if bound_form is None:
|
||||||
return data
|
return data
|
||||||
|
|
||||||
|
if self.view is not None:
|
||||||
self.view.bound_form_instance = bound_form
|
self.view.bound_form_instance = bound_form
|
||||||
|
|
||||||
data = data and data or {}
|
data = data and data or {}
|
||||||
|
@ -314,17 +305,17 @@ class ModelResource(FormResource):
|
||||||
is not set.
|
is not set.
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def __init__(self, instance=None, view=None, depth=None, stack=[], **kwargs):
|
def __init__(self, view=None, instance=None, depth=None, stack=[], **kwargs):
|
||||||
"""
|
"""
|
||||||
Allow :attr:`form` and :attr:`model` attributes set on the
|
Allow :attr:`form` and :attr:`model` attributes set on the
|
||||||
:class:`View` to override the :attr:`form` and :attr:`model`
|
:class:`View` to override the :attr:`form` and :attr:`model`
|
||||||
attributes set on the :class:`Resource`.
|
attributes set on the :class:`Resource`.
|
||||||
"""
|
"""
|
||||||
super(ModelResource, self).__init__(instance=instance, view=view, depth=depth, stack=stack, **kwargs)
|
super(ModelResource, self).__init__(view=None, instance=instance, depth=depth, stack=stack, **kwargs)
|
||||||
|
|
||||||
self.model = getattr(view, 'model', None) or self.model
|
self.model = getattr(view, 'model', None) or self.model
|
||||||
|
|
||||||
def retrieve(self, request, *args, **kwargs):
|
def retrieve(self, *args, **kwargs):
|
||||||
"""
|
"""
|
||||||
Return a model instance or None.
|
Return a model instance or None.
|
||||||
"""
|
"""
|
||||||
|
@ -339,7 +330,7 @@ class ModelResource(FormResource):
|
||||||
self.instance = instance
|
self.instance = instance
|
||||||
return self.instance
|
return self.instance
|
||||||
|
|
||||||
def create(self, request, *args, **kwargs):
|
def create(self, *args, **kwargs):
|
||||||
model = self.get_model()
|
model = self.get_model()
|
||||||
kwargs = self._clean_url_kwargs(kwargs)
|
kwargs = self._clean_url_kwargs(kwargs)
|
||||||
|
|
||||||
|
@ -347,8 +338,12 @@ class ModelResource(FormResource):
|
||||||
self.instance.save()
|
self.instance.save()
|
||||||
return self.instance
|
return self.instance
|
||||||
|
|
||||||
@bound_resource_required
|
def update(self, data, *args, **kwargs):
|
||||||
def update(self, data, request, *args, **kwargs):
|
# The resource needs to be bound to an
|
||||||
|
# instance, or updating is not possible
|
||||||
|
if not self.is_bound():
|
||||||
|
raise Exception("resource needs to be bound") #TODO: what exception?
|
||||||
|
|
||||||
model = self.get_model()
|
model = self.get_model()
|
||||||
kwargs = self._clean_url_kwargs(kwargs)
|
kwargs = self._clean_url_kwargs(kwargs)
|
||||||
data = dict(data, **kwargs)
|
data = dict(data, **kwargs)
|
||||||
|
@ -382,12 +377,16 @@ class ModelResource(FormResource):
|
||||||
self.instance.save()
|
self.instance.save()
|
||||||
return self.instance
|
return self.instance
|
||||||
|
|
||||||
@bound_resource_required
|
def delete(self, *args, **kwargs):
|
||||||
def delete(self, request, *args, **kwargs):
|
# The resource needs to be bound to an
|
||||||
|
# instance, or deleting is not possible
|
||||||
|
if not self.is_bound():
|
||||||
|
raise Exception("resource needs to be bound") #TODO: what exception?
|
||||||
|
|
||||||
self.instance.delete()
|
self.instance.delete()
|
||||||
return self.instance
|
return self.instance
|
||||||
|
|
||||||
def list(self, request, *args, **kwargs):
|
def list(self, *args, **kwargs):
|
||||||
# TODO: QuerysetResource instead !?
|
# TODO: QuerysetResource instead !?
|
||||||
kwargs = self._clean_url_kwargs(kwargs)
|
kwargs = self._clean_url_kwargs(kwargs)
|
||||||
queryset = self.get_queryset()
|
queryset = self.get_queryset()
|
||||||
|
@ -397,7 +396,6 @@ class ModelResource(FormResource):
|
||||||
queryset = queryset.order_by(ordering)
|
queryset = queryset.order_by(ordering)
|
||||||
return queryset.filter(**kwargs)
|
return queryset.filter(**kwargs)
|
||||||
|
|
||||||
@bound_resource_required
|
|
||||||
def get_url(self):
|
def get_url(self):
|
||||||
"""
|
"""
|
||||||
Attempts to reverse resolve the url of the given model *instance* for
|
Attempts to reverse resolve the url of the given model *instance* for
|
||||||
|
@ -409,6 +407,10 @@ class ModelResource(FormResource):
|
||||||
This method can be overridden if you need to set the resource url
|
This method can be overridden if you need to set the resource url
|
||||||
reversing explicitly.
|
reversing explicitly.
|
||||||
"""
|
"""
|
||||||
|
# The resource needs to be bound to an
|
||||||
|
# instance, or getting url is not possible
|
||||||
|
if not self.is_bound():
|
||||||
|
raise Exception("resource needs to be bound") #TODO: what exception?
|
||||||
|
|
||||||
if not hasattr(self, 'view_callable'):
|
if not hasattr(self, 'view_callable'):
|
||||||
raise _SkipField
|
raise _SkipField
|
||||||
|
|
Loading…
Reference in New Issue
Block a user