Split to Api Docs & Api Endpoint

Author: Emmanouil Konstantinidis

drfdocs/api_docs.py

@@ -1,22 +1,24 @@
 from django.conf import settings
 from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
+from drfdocs.api_endpoint import ApiEndpoint
 
 
 class ApiDocumentation(object):
     excluded_apps = ["admin", "drfdocs"]
     excluded_endpoints = ["serve"]
-    root_urlconf = __import__(settings.ROOT_URLCONF)
 
     def __init__(self):
-        self.view_names = []
-        self.get_all_view_names(self.root_urlconf.urls.urlpatterns)
+        self.endpoints = []
+        root_urlconf = __import__(settings.ROOT_URLCONF)
+        self.get_all_view_names(root_urlconf.urls.urlpatterns)
 
     def get_all_view_names(self, urlpatterns):
         for pattern in urlpatterns:
             if isinstance(pattern, RegexURLResolver) and (pattern.app_name not in self.excluded_apps):
                 self.get_all_view_names(pattern.url_patterns)
             elif isinstance(pattern, RegexURLPattern) and (pattern.callback.__name__ not in self.excluded_endpoints):
-                self.view_names.append(pattern.callback.__name__)
+                api_endpoint = ApiEndpoint(pattern)
+                self.endpoints.append(api_endpoint)
 
-    def get_views(self):
-        return self.view_names
+    def get_endpoints(self):
+        return self.endpoints

drfdocs/api_endpoint.py

@@ -0,0 +1,18 @@
+from django.contrib.admindocs.views import simplify_regex
+from rest_framework.views import APIView
+
+
+class ApiEndpoint(object):
+
+    def __init__(self, pattern):
+        self.pattern = pattern
+        self.regex = simplify_regex(pattern._regex)
+        self.view_name = pattern.callback.__name__
+        # self._get_api_callback(pattern)
+
+    def _get_api_callback(self, pattern):
+        """
+        Verifies that pattern callback is a subclass of APIView, and returns the class
+        """
+        if (hasattr(pattern.callback, 'cls') and issubclass(pattern.callback.cls, APIView)):
+            return pattern.callback.cls

drfdocs/templates/drfdocs/home.html

@@ -2,8 +2,10 @@
 
 {% block content %}
   <ul>
-  {% for view in views %}
-    <li>{{ view }}</li>
+  {% for endpoint in endpoints %}
+    <li>
+      <pre>{{ endpoint.regex }}</pre>
+      <p>{{ endpoint.view_name }}</p>
   {% endfor %}
   </ul>
 {% endblock %}

drfdocs/views.py

@@ -9,5 +9,5 @@ class DRFDocsView(TemplateView):
     def get_context_data(self, **kwargs):
         context = super(DRFDocsView, self).get_context_data(**kwargs)
         docs = ApiDocumentation()
-        context['views'] = docs.get_views()
+        context['endpoints'] = docs.get_endpoints()
         return context