Get a list with all the views

Emmanouil Konstantinidis · Emmanouil Konstantinidis · commit ed0f7b8b65c4 · 2015-12-12T00:47:48.000Z
diff --git a/drfdocs/templates/drfdocs/base.html b/drfdocs/templates/drfdocs/base.html
@@ -7,5 +7,7 @@
   <body>
     <h1>Django Rest Frameworks Docs</h1>
 
+    {% block content %}{% endblock %}
+
   </body>
 </html>
diff --git a/drfdocs/templates/drfdocs/home.html b/drfdocs/templates/drfdocs/home.html
@@ -1 +1,9 @@
 {% extends "drfdocs/base.html" %}
+
+{% block content %}
+  <ul>
+  {% for view in views %}
+    <li>{{ view }}</li>
+  {% endfor %}
+  </ul>
+{% endblock %}
diff --git a/drfdocs/views.py b/drfdocs/views.py
@@ -1,11 +1,26 @@
+from django.conf import settings
 from django.views.generic.base import TemplateView
+from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
 
 
 class DRFDocsView(TemplateView):
 
+    root_urlconf = __import__(settings.ROOT_URLCONF)
     template_name = "drfdocs/home.html"
+    VIEW_NAMES = []
+
+    def get_all_view_names(self, urlpatterns):
+        for pattern in urlpatterns:
+            if isinstance(pattern, RegexURLResolver):
+                self.get_all_view_names(pattern.url_patterns)
+            elif isinstance(pattern, RegexURLPattern):
+                view_name = pattern.callback.__name__
+                self.VIEW_NAMES.append(view_name)
+        return self.VIEW_NAMES
 
     def get_context_data(self, **kwargs):
+        self.VIEW_NAMES = []
         context = super(DRFDocsView, self).get_context_data(**kwargs)
-        context['example'] = True
+        all_urlpatterns = self.root_urlconf.urls.urlpatterns
+        context['views'] = self.get_all_view_names(all_urlpatterns)
         return context
