From cab556ee3fde2d0260ba887023346601781aa862 Mon Sep 17 00:00:00 2001 From: Tom Christie Date: Tue, 18 Sep 2012 19:03:59 +0100 Subject: [PATCH] Initial resources code --- djangorestframework/resources.py | 87 ++++++++++++++++++++++++++++++++ 1 file changed, 87 insertions(+) create mode 100644 djangorestframework/resources.py diff --git a/djangorestframework/resources.py b/djangorestframework/resources.py new file mode 100644 index 000000000..41b7c1c6a --- /dev/null +++ b/djangorestframework/resources.py @@ -0,0 +1,87 @@ +from functools import update_wrapper +import inspect +from django.utils.decorators import classonlymethod +from djanorestframework import views, generics + + +def wrapped(source, dest): + """ + Copy public, non-method attributes from source to dest, and return dest. + """ + for attr in [attr for attr in dir(source) + if not attr.startswith('_') and not inspect.ismethod(attr)]: + setattr(dest, attr, getattr(source, attr)) + return dest + + +class ResourceMixin(object): + """ + Clone Django's `View.as_view()` behaviour *except* using REST framework's + 'method -> action' binding for resources. + """ + + @classonlymethod + def as_view(cls, actions, **initkwargs): + """ + Main entry point for a request-response process. + """ + # sanitize keyword arguments + for key in initkwargs: + if key in cls.http_method_names: + raise TypeError("You tried to pass in the %s method name as a " + "keyword argument to %s(). Don't do that." + % (key, cls.__name__)) + if not hasattr(cls, key): + raise TypeError("%s() received an invalid keyword %r" % ( + cls.__name__, key)) + + def view(request, *args, **kwargs): + self = cls(**initkwargs) + + # Bind methods to actions + for method, action in actions.items(): + handler = getattr(self, action) + setattr(self, method, handler) + + # As you were, solider. + if hasattr(self, 'get') and not hasattr(self, 'head'): + self.head = self.get + return self.dispatch(request, *args, **kwargs) + + # take name and docstring from class + update_wrapper(view, cls, updated=()) + + # and possible attributes set by decorators + # like csrf_exempt from dispatch + update_wrapper(view, cls.dispatch, assigned=()) + return view + + +class Resource(ResourceMixin, views.APIView): + pass + + +class ModelResource(ResourceMixin, views.APIView): + root_class = generics.RootAPIView + detail_class = generics.InstanceAPIView + + def root_view(self): + return wrapped(self, self.root_class()) + + def detail_view(self): + return wrapped(self, self.detail_class()) + + def list(self, request, *args, **kwargs): + return self.root_view().list(request, args, kwargs) + + def create(self, request, *args, **kwargs): + return self.root_view().create(request, args, kwargs) + + def retrieve(self, request, *args, **kwargs): + return self.detail_view().retrieve(request, args, kwargs) + + def update(self, request, *args, **kwargs): + return self.detail_view().update(request, args, kwargs) + + def destroy(self, request, *args, **kwargs): + return self.detail_view().destroy(request, args, kwargs)