encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2024-11-28 20:44:03 +03:00
Code Issues Packages Projects Releases Wiki Activity

refactor the way URLs works to be more like the admin site, where there is a subobject for each resource.

Browse Source
This commit is contained in:
Craig Blaszczyk 2011-08-01 12:49:54 +01:00
parent c18d8f02d0
commit 98ec89a575
1 changed files with 46 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

74
djangorestframework/builtins.py
View File

@ -1,7 +1,43 @@
from djangorestframework.mixins import ListModelMixin, InstanceMixin from djangorestframework.mixins import ListModelMixin, InstanceMixin
from django.conf.urls.defaults import patterns, url from django.conf.urls.defaults import patterns, url, include
from django.views.decorators.csrf import csrf_exempt from django.views.decorators.csrf import csrf_exempt
class ApiEntry(object):
def __init__(self, resource, view, prefix, resource_name):
self.resource, self.view = resource, view
self.prefix, self.resource_name = prefix, resource_name
if self.prefix is None:
self.prefix = ''
def get_urls(self):
from django.conf.urls.defaults import patterns, url
if self.prefix == '':
url_prefix = ''
else:
url_prefix = self.prefix + '/'
if issubclass(self.view, ListModelMixin):
urlpatterns = patterns('',
url(r'^%s%s/$' % (url_prefix, self.resource_name),
self.view.as_view(resource=self.resource),
name=self.resource_name,
)
)
elif issubclass(self.view, InstanceMixin):
urlpatterns = patterns('',
url(r'^%s%s/(?P<pk>[0-9]+)/$' % (url_prefix, self.resource_name),
self.view.as_view(resource=self.resource),
name=self.resource_name + '_change',
)
)
return urlpatterns
def urls(self):
return self.get_urls(), 'api', self.prefix
urls = property(urls)
class DjangoRestFrameworkSite(object): class DjangoRestFrameworkSite(object):
app_name = 'api' app_name = 'api'
name = 'api' name = 'api'
@ -17,13 +53,16 @@ class DjangoRestFrameworkSite(object):
else: else:
resource_name = resource.__name__.lower() resource_name = resource.__name__.lower()
resource.resource_name = resource_name
if prefix not in self._registry: if prefix not in self._registry:
self._registry[prefix] = {} self._registry[prefix] = {}
if resource_name not in self._registry[prefix]: if resource_name not in self._registry[prefix]:
self._registry[prefix][resource_name] = [] self._registry[prefix][resource_name] = []
self._registry[prefix][resource_name].append((resource, view)) api_entry = ApiEntry(resource, view, prefix, resource_name)
self._registry[prefix][resource_name].append(api_entry)
# def unregister(self, prefix=None, resource_name=None, resource=None): # def unregister(self, prefix=None, resource_name=None, resource=None):
@ -57,34 +96,11 @@ class DjangoRestFrameworkSite(object):
# Add in each resource's views. # Add in each resource's views.
for prefix in self._registry.keys(): for prefix in self._registry.keys():
for resource_name in self._registry[prefix].keys(): for resource_name in self._registry[prefix].keys():
for resource, view in self._registry[prefix][resource_name]: for api_entry in self._registry[prefix][resource_name]:
urlpatterns += self.__get_urlpatterns( urlpatterns += patterns('',
prefix, resource_name, resource, view url(r'^', include(api_entry.urls))
) )
return urlpatterns return urlpatterns
def __get_urlpatterns(self, prefix, resource_name, resource, view):
"""
Calculates the URL pattern for a given resource and view
"""
if prefix is None:
prefix = ''
else:
prefix += '/'
if issubclass(view, ListModelMixin):
urlpatterns = patterns('',
url(r'^%s%s/$' % (prefix,resource_name),
view.as_view(resource=resource),
name=resource_name
)
)
elif issubclass(view, InstanceMixin):
urlpatterns = patterns('',
url(r'^%s%s/(?P<pk>[0-9]+)/$' % (prefix,resource_name),
view.as_view(resource=resource),
name=resource_name + '_change'
)
)
return urlpatterns