Skip to content

Small improvements #47

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 6 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 12 additions & 3 deletions demo/project/organisations/urls.py
Original file line number Diff line number Diff line change
@@ -1,11 +1,20 @@
import django
from django.conf.urls import url
from project.organisations import views


urlpatterns = [

organisations_urlpatterns = [
url(r'^create/$', view=views.CreateOrganisationView.as_view(), name="create"),
url(r'^(?P<slug>[\w-]+)/members/$', view=views.OrganisationMembersView.as_view(), name="members"),
url(r'^(?P<slug>[\w-]+)/leave/$', view=views.LeaveOrganisationView.as_view(), name="leave")
]

members_urlpatterns = [
url(r'^(?P<slug>[\w-]+)/members/$', view=views.OrganisationMembersView.as_view(), name="members"),
]

# Django 1.9 Support for the app_name argument is deprecated
# https://docs.djangoproject.com/en/1.9/ref/urls/#include
django_version = django.VERSION
if django.VERSION[:2] >= (1, 9, ):
organisations_urlpatterns = (organisations_urlpatterns, 'organisations_app', )
members_urlpatterns = (members_urlpatterns, 'organisations_app', )
1 change: 0 additions & 1 deletion demo/project/settings.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@

'project.accounts',
'project.organisations',

)

MIDDLEWARE_CLASSES = (
Expand Down
17 changes: 16 additions & 1 deletion demo/project/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,14 +13,29 @@
1. Add an import: from blog import urls as blog_urls
2. Add a URL to urlpatterns: url(r'^blog/', include(blog_urls))
"""
import django
from django.conf.urls import include, url
from django.contrib import admin
from .organisations.urls import organisations_urlpatterns, members_urlpatterns

urlpatterns = [
url(r'^admin/', include(admin.site.urls)),
url(r'^docs/', include('rest_framework_docs.urls')),

# API
url(r'^accounts/', view=include('project.accounts.urls', namespace='accounts')),
url(r'^organisations/', view=include('project.organisations.urls', namespace='organisations')),
]

# Django 1.9 Support for the app_name argument is deprecated
# https://docs.djangoproject.com/en/1.9/ref/urls/#include
django_version = django.VERSION
if django.VERSION[:2] >= (1, 9, ):
urlpatterns.extend([
url(r'^organisations/', view=include(organisations_urlpatterns, namespace='organisations')),
url(r'^members/', view=include(members_urlpatterns, namespace='members')),
])
else:
urlpatterns.extend([
url(r'^organisations/', view=include(organisations_urlpatterns, namespace='organisations', app_name='organisations_app')),
url(r'^members/', view=include(members_urlpatterns, namespace='members', app_name='organisations_app')),
])
23 changes: 14 additions & 9 deletions rest_framework_docs/api_docs.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
from operator import attrgetter
from django.conf import settings
from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
from rest_framework.views import APIView
Expand All @@ -6,25 +7,29 @@

class ApiDocumentation(object):

def __init__(self):
def __init__(self, filter_param=None):
"""
:param filter_param: namespace or app_name
"""
self.endpoints = []
root_urlconf = __import__(settings.ROOT_URLCONF)
if hasattr(root_urlconf, 'urls'):
self.get_all_view_names(root_urlconf.urls.urlpatterns)
self.get_all_view_names(root_urlconf.urls.urlpatterns, filter_param=filter_param)
else:
self.get_all_view_names(root_urlconf.urlpatterns)
self.get_all_view_names(root_urlconf.urlpatterns, filter_param=filter_param)

def get_all_view_names(self, urlpatterns, parent_pattern=None):
def get_all_view_names(self, urlpatterns, parent_pattern=None, filter_param=None):
for pattern in urlpatterns:
if isinstance(pattern, RegexURLResolver):
self.get_all_view_names(urlpatterns=pattern.url_patterns, parent_pattern=pattern)
if isinstance(pattern, RegexURLResolver) and (not filter_param or filter_param in [pattern.app_name, pattern.namespace]):
self.get_all_view_names(urlpatterns=pattern.url_patterns, parent_pattern=pattern, filter_param=filter_param)
elif isinstance(pattern, RegexURLPattern) and self._is_drf_view(pattern):
api_endpoint = ApiEndpoint(pattern, parent_pattern)
self.endpoints.append(api_endpoint)
if not filter_param or (parent_pattern and filter_param in [parent_pattern.app_name, parent_pattern.namespace]):
api_endpoint = ApiEndpoint(pattern, parent_pattern)
self.endpoints.append(api_endpoint)

def _is_drf_view(self, pattern):
# Should check whether a pattern inherits from DRF's APIView
return hasattr(pattern.callback, 'cls') and issubclass(pattern.callback.cls, APIView)

def get_endpoints(self):
return self.endpoints
return sorted(self.endpoints, key=attrgetter('name'))
17 changes: 13 additions & 4 deletions rest_framework_docs/api_endpoint.py
Original file line number Diff line number Diff line change
@@ -1,27 +1,36 @@
import json
import inspect
from django.contrib.admindocs.views import simplify_regex
from rest_framework.viewsets import ModelViewSet


class ApiEndpoint(object):

def __init__(self, pattern, parent_pattern=None):
self.pattern = pattern
self.callback = pattern.callback
# self.name = pattern.name
self.docstring = self.__get_docstring__()
self.name_parent = simplify_regex(parent_pattern.regex.pattern).replace('/', '') if parent_pattern else None
if parent_pattern:
self.name_parent = parent_pattern.namespace or parent_pattern.app_name or \
simplify_regex(parent_pattern.regex.pattern).replace('/', '-')
self.name = self.name_parent
if hasattr(pattern.callback, 'cls') and issubclass(pattern.callback.cls, ModelViewSet):
self.name = '%s (REST)' % self.name_parent
else:
self.name_parent = ''
self.name = ''
# self.labels = (self.name_parent, self.name, slugify(self.name))
self.labels = dict(parent=self.name_parent, name=self.name)
self.path = self.__get_path__(parent_pattern)
self.allowed_methods = self.__get_allowed_methods__()
# self.view_name = pattern.callback.__name__
self.errors = None
self.fields = self.__get_serializer_fields__()
self.fields_json = self.__get_serializer_fields_json__()
self.permissions = self.__get_permissions_class__()

def __get_path__(self, parent_pattern):
if parent_pattern:
return "/{0}{1}".format(self.name_parent, simplify_regex(self.pattern.regex.pattern))
return simplify_regex(parent_pattern.regex.pattern + self.pattern.regex.pattern)
return simplify_regex(self.pattern.regex.pattern)

def __get_allowed_methods__(self):
Expand Down
12 changes: 6 additions & 6 deletions rest_framework_docs/templates/rest_framework_docs/base.html
Original file line number Diff line number Diff line change
Expand Up @@ -54,12 +54,12 @@
</div><!-- /.container-fluid -->
</nav>

{% block jumbotron %}
<div class="jumbotron">
<h1>DRF Docs</h1>
<h3>Document Web APIs made with <a href="http://www.django-rest-framework.org/" target="_blank">Django REST Framework</a>.</h3>
</div>
{% endblock %}
<!--{% block jumbotron %}-->
<!--<div class="jumbotron">-->
<!--<h1>DRF Docs</h1>-->
<!--<h3>Document Web APIs made with <a href="http://www.django-rest-framework.org/" target="_blank">Django REST Framework</a>.</h3>-->
<!--</div>-->
<!--{% endblock %}-->

{% block content %}{% endblock %}

Expand Down
31 changes: 17 additions & 14 deletions rest_framework_docs/templates/rest_framework_docs/home.html
Original file line number Diff line number Diff line change
@@ -1,26 +1,29 @@
{% extends "rest_framework_docs/docs.html" %}

{% block apps_menu %}
{% regroup endpoints by name_parent as endpoints_grouped %}
<li class="dropdown">
<a href="#" class="dropdown-toggle" data-toggle="dropdown" role="button" aria-haspopup="true" aria-expanded="false">Jump To <span class="caret"></span></a>
<ul class="dropdown-menu">
{% for group in endpoints_grouped %}
<li><a href="#{{ group.grouper|lower }}-group">{{ group.grouper }}</a></li>
{% endfor %}
</ul>
</li>
{% regroup endpoints by labels as endpoints_grouped %}
{% if endpoints_grouped|length > 1 %}
<li class="dropdown">
<a href="#" class="dropdown-toggle" data-toggle="dropdown" role="button" aria-haspopup="true" aria-expanded="false">Jump To <span class="caret"></span></a>
<ul class="dropdown-menu">
{% for group in endpoints_grouped %}
<li><a href="#{{ group.grouper.name|slugify }}-group">{{ group.grouper.name }}</a></li>
{% endfor %}
</ul>
</li>
{% endif %}
{% endblock %}


{% block content %}

{% regroup endpoints by name_parent as endpoints_grouped %}

{% regroup endpoints by labels as endpoints_grouped %}
{% if endpoints_grouped %}
{% for group in endpoints_grouped %}

<h1 id="{{ group.grouper|lower }}-group">{{group.grouper}}</h1>
<h1 id="{{ group.grouper.name|slugify }}-group">
{% if group.grouper.parent %}
<a href="{% url 'drfdocs-filter' group.grouper.parent %}">{{ group.grouper.name }}</a>
{% endif %}
</h1>

<div class="panel-group" role="tablist">

Expand Down
2 changes: 2 additions & 0 deletions rest_framework_docs/urls.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,6 @@
urlpatterns = [
# Url to view the API Docs
url(r'^$', DRFDocsView.as_view(), name='drfdocs'),
# Url to view the API Docs with a specific namespace or app_name
url(r'^(?P<filter_param>[\w-]+)/$', DRFDocsView.as_view(), name='drfdocs-filter'),
]
4 changes: 2 additions & 2 deletions rest_framework_docs/views.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,13 +8,13 @@ class DRFDocsView(TemplateView):

template_name = "rest_framework_docs/home.html"

def get_context_data(self, **kwargs):
def get_context_data(self, filter_param=None, **kwargs):
settings = DRFSettings().settings
if settings["HIDDEN"]:
raise Http404("Django Rest Framework Docs are hidden. Check your settings.")

context = super(DRFDocsView, self).get_context_data(**kwargs)
docs = ApiDocumentation()
docs = ApiDocumentation(filter_param=filter_param)
endpoints = docs.get_endpoints()

query = self.request.GET.get("search", "")
Expand Down
1 change: 1 addition & 0 deletions runtests.py
Original file line number Diff line number Diff line change
Expand Up @@ -53,6 +53,7 @@ def run_tests_coverage():
cov.report()
cov.html_report(directory='covhtml')


exit_on_failure(flake8_main(FLAKE8_ARGS))
exit_on_failure(run_tests_eslint())
exit_on_failure(run_tests_coverage())
81 changes: 73 additions & 8 deletions tests/tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -30,16 +30,16 @@ def test_index_view_with_endpoints(self):
self.assertEqual(len(response.context["endpoints"]), 10)

# Test the login view
self.assertEqual(response.context["endpoints"][0].name_parent, "accounts")
self.assertEqual(response.context["endpoints"][0].allowed_methods, ['POST', 'OPTIONS'])
self.assertEqual(response.context["endpoints"][0].path, "/accounts/login/")
self.assertEqual(response.context["endpoints"][0].docstring, "A view that allows users to login providing their username and password.")
self.assertEqual(len(response.context["endpoints"][0].fields), 2)
self.assertEqual(response.context["endpoints"][0].fields[0]["type"], "CharField")
self.assertTrue(response.context["endpoints"][0].fields[0]["required"])
self.assertEqual(response.context["endpoints"][1].name_parent, "accounts")
self.assertEqual(response.context["endpoints"][1].allowed_methods, ['POST', 'OPTIONS'])
self.assertEqual(response.context["endpoints"][1].path, "/accounts/login/")
self.assertEqual(response.context["endpoints"][1].docstring, "A view that allows users to login providing their username and password.")
self.assertEqual(len(response.context["endpoints"][1].fields), 2)
self.assertEqual(response.context["endpoints"][1].fields[0]["type"], "CharField")
self.assertTrue(response.context["endpoints"][1].fields[0]["required"])

# The view "OrganisationErroredView" (organisations/(?P<slug>[\w-]+)/errored/) should contain an error.
self.assertEqual(str(response.context["endpoints"][8].errors), "'test_value'")
self.assertEqual(str(response.context["endpoints"][9].errors), "'test_value'")

def test_index_search_with_endpoints(self):
response = self.client.get("%s?search=reset-password" % reverse("drfdocs"))
Expand All @@ -59,3 +59,68 @@ def test_index_view_docs_hidden(self):

self.assertEqual(response.status_code, 404)
self.assertEqual(response.reason_phrase.upper(), "NOT FOUND")

def test_index_view_with_existent_namespace(self):
"""
Should load the drf docs view with all the endpoints contained in the specified namespace.
NOTE: Views that do **not** inherit from DRF's "APIView" are not included.
"""
# Test 'accounts' namespace
response = self.client.get(reverse('drfdocs-filter', args=['accounts']))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 5)

# Test the login view
self.assertEqual(response.context["endpoints"][0].name_parent, "accounts")
self.assertEqual(response.context["endpoints"][0].allowed_methods, ['POST', 'OPTIONS'])
self.assertEqual(response.context["endpoints"][0].path, "/accounts/login/")

# Test 'organisations' namespace
response = self.client.get(reverse('drfdocs-filter', args=['organisations']))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 3)

# The view "OrganisationErroredView" (organisations/(?P<slug>[\w-]+)/errored/) should contain an error.
self.assertEqual(str(response.context["endpoints"][2].errors), "'test_value'")

# Test 'members' namespace
response = self.client.get(reverse('drfdocs-filter', args=['members']))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 1)

def test_index_search_with_existent_namespace(self):
response = self.client.get("%s?search=reset-password" % reverse('drfdocs-filter', args=['accounts']))

self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 2)
self.assertEqual(response.context["endpoints"][1].path, "/accounts/reset-password/confirm/")
self.assertEqual(len(response.context["endpoints"][1].fields), 3)

def test_index_view_with_existent_app_name(self):
"""
Should load the drf docs view with all the endpoints contained in the specified app_name.
NOTE: Views that do **not** inherit from DRF's "APIView" are not included.
"""
# Test 'organisations_app' app_name
response = self.client.get(reverse('drfdocs-filter', args=['organisations_app']))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 4)
parents_name = [e.name_parent for e in response.context["endpoints"]]
self.assertEquals(parents_name.count('organisations'), 3)
self.assertEquals(parents_name.count('members'), 1)

def test_index_search_with_existent_app_name(self):
response = self.client.get("%s?search=create" % reverse('drfdocs-filter', args=['organisations_app']))

self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 1)
self.assertEqual(response.context["endpoints"][0].path, "/organisations/create/")
self.assertEqual(len(response.context["endpoints"][0].fields), 2)

def test_index_view_with_non_existent_namespace_or_app_name(self):
"""
Should load the drf docs view with no endpoint.
"""
response = self.client.get(reverse('drfdocs-filter', args=['non-existent-ns-or-app-name']))
self.assertEqual(response.status_code, 200)
self.assertEqual(len(response.context["endpoints"]), 0)
24 changes: 21 additions & 3 deletions tests/urls.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
from __future__ import absolute_import, division, print_function

import django
from django.conf.urls import include, url
from django.contrib import admin
from tests import views
Expand All @@ -17,19 +18,36 @@

organisations_urls = [
url(r'^create/$', view=views.CreateOrganisationView.as_view(), name="create"),
url(r'^(?P<slug>[\w-]+)/members/$', view=views.OrganisationMembersView.as_view(), name="members"),
url(r'^(?P<slug>[\w-]+)/leave/$', view=views.LeaveOrganisationView.as_view(), name="leave"),
url(r'^(?P<slug>[\w-]+)/errored/$', view=views.OrganisationErroredView.as_view(), name="errored")
]

members_urls = [
url(r'^(?P<slug>[\w-]+)/members/$', view=views.OrganisationMembersView.as_view(), name="members"),
]

urlpatterns = [
url(r'^admin/', include(admin.site.urls)),
url(r'^docs/', include('rest_framework_docs.urls')),

# API
url(r'^accounts/', view=include(accounts_urls, namespace='accounts')),
url(r'^organisations/', view=include(organisations_urls, namespace='organisations')),

# Endpoints without parents/namespaces
url(r'^another-login/$', views.LoginView.as_view(), name="login"),
]

# Django 1.9 Support for the app_name argument is deprecated
# https://docs.djangoproject.com/en/1.9/ref/urls/#include
django_version = django.VERSION
if django.VERSION[:2] >= (1, 9, ):
organisations_urls = (organisations_urls, 'organisations_app', )
members_urls = (members_urls, 'organisations_app', )
urlpatterns.extend([
url(r'^organisations/', view=include(organisations_urls, namespace='organisations')),
url(r'^members/', view=include(members_urls, namespace='members')),
])
else:
urlpatterns.extend([
url(r'^organisations/', view=include(organisations_urls, namespace='organisations', app_name='organisations_app')),
url(r'^members/', view=include(members_urls, namespace='members', app_name='organisations_app')),
])