mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-09-16 09:12:29 +03:00
Moved ModelMixin's model instance retreival from .get() .put() .delete() to a .model_instance lazy property
This commit is contained in:
parent
da8187d2c2
commit
52ac21ca3e
|
@ -527,6 +527,28 @@ class ModelMixin(object):
|
||||||
"""
|
"""
|
||||||
return self.get_queryset().get(**kwargs)
|
return self.get_queryset().get(**kwargs)
|
||||||
|
|
||||||
|
@property
|
||||||
|
def model_instance(self):
|
||||||
|
"""
|
||||||
|
Returns the model instance for read/update/delete, or None if does not exit.
|
||||||
|
"""
|
||||||
|
|
||||||
|
if hasattr(self, '_model_instance'): return self._model_instance
|
||||||
|
|
||||||
|
model = self.resource.model
|
||||||
|
request = self.request
|
||||||
|
args = self.args
|
||||||
|
kwargs = self.kwargs
|
||||||
|
|
||||||
|
query_kwargs = self.get_query_kwargs(request, *args, **kwargs)
|
||||||
|
|
||||||
|
try:
|
||||||
|
self._model_instance = self.get_instance(**kwargs)
|
||||||
|
except model.DoesNotExist:
|
||||||
|
pass
|
||||||
|
|
||||||
|
return self._model_instance
|
||||||
|
|
||||||
def get_queryset(self):
|
def get_queryset(self):
|
||||||
"""
|
"""
|
||||||
Return the queryset for this view.
|
Return the queryset for this view.
|
||||||
|
@ -546,12 +568,8 @@ class ReadModelMixin(ModelMixin):
|
||||||
Behavior to read a `model` instance on GET requests
|
Behavior to read a `model` instance on GET requests
|
||||||
"""
|
"""
|
||||||
def get(self, request, *args, **kwargs):
|
def get(self, request, *args, **kwargs):
|
||||||
model = self.resource.model
|
|
||||||
query_kwargs = self.get_query_kwargs(request, *args, **kwargs)
|
|
||||||
|
|
||||||
try:
|
if self.model_instance == None:
|
||||||
self.model_instance = self.get_instance(**query_kwargs)
|
|
||||||
except model.DoesNotExist:
|
|
||||||
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
raise ErrorResponse(status.HTTP_404_NOT_FOUND)
|
||||||
|
|
||||||
return self.model_instance
|
return self.model_instance
|
||||||
|
@ -602,18 +620,17 @@ class UpdateModelMixin(ModelMixin):
|
||||||
Behavior to update a `model` instance on PUT requests
|
Behavior to update a `model` instance on PUT requests
|
||||||
"""
|
"""
|
||||||
def put(self, request, *args, **kwargs):
|
def put(self, request, *args, **kwargs):
|
||||||
|
|
||||||
model = self.resource.model
|
model = self.resource.model
|
||||||
query_kwargs = self.get_query_kwargs(request, *args, **kwargs)
|
|
||||||
|
|
||||||
# TODO: update on the url of a non-existing resource url doesn't work
|
# TODO: update on the url of a non-existing resource url doesn't work
|
||||||
# correctly at the moment - will end up with a new url
|
# correctly at the moment - will end up with a new url
|
||||||
try:
|
if self.model_instance == None:
|
||||||
self.model_instance = self.get_instance(**query_kwargs)
|
self.model_instance = model(**self.get_instance_data(model, self.CONTENT, *args, **kwargs))
|
||||||
|
else:
|
||||||
for (key, val) in self.CONTENT.items():
|
for (key, val) in self.CONTENT.items():
|
||||||
setattr(self.model_instance, key, val)
|
setattr(self.model_instance, key, val)
|
||||||
except model.DoesNotExist:
|
|
||||||
self.model_instance = model(**self.get_instance_data(model, self.CONTENT, *args, **kwargs))
|
|
||||||
self.model_instance.save()
|
self.model_instance.save()
|
||||||
return self.model_instance
|
return self.model_instance
|
||||||
|
|
||||||
|
@ -623,15 +640,9 @@ class DeleteModelMixin(ModelMixin):
|
||||||
Behavior to delete a `model` instance on DELETE requests
|
Behavior to delete a `model` instance on DELETE requests
|
||||||
"""
|
"""
|
||||||
def delete(self, request, *args, **kwargs):
|
def delete(self, request, *args, **kwargs):
|
||||||
model = self.resource.model
|
|
||||||
query_kwargs = self.get_query_kwargs(request, *args, **kwargs)
|
|
||||||
|
|
||||||
try:
|
if self.model_instance == None: raise ErrorResponse(status.HTTP_404_NOT_FOUND, None, {})
|
||||||
instance = self.get_instance(**query_kwargs)
|
self.model_instance.delete()
|
||||||
except model.DoesNotExist:
|
|
||||||
raise ErrorResponse(status.HTTP_404_NOT_FOUND, None, {})
|
|
||||||
|
|
||||||
instance.delete()
|
|
||||||
return
|
return
|
||||||
|
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user