Clean up

Author: Emmanouil Konstantinidis

drfdocs/api_docs.py

@@ -0,0 +1,22 @@
+from django.conf import settings
+from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
+
+
+class ApiDocumentation(object):
+    excluded_apps = ["admin", "drfdocs"]
+    root_urlconf = __import__(settings.ROOT_URLCONF)
+
+    def __init__(self):
+        self.view_names = []
+        self.get_all_view_names(self.root_urlconf.urls.urlpatterns)
+
+    def get_all_view_names(self, urlpatterns):
+        for pattern in urlpatterns:
+            print(pattern)
+            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):
+                self.view_names.append(pattern.callback.__name__)
+
+    def get_views(self):
+        return self.view_names

drfdocs/api_endpoint.py

@@ -1,28 +1,13 @@
-from django.conf import settings
 from django.views.generic.base import TemplateView
-from django.core.urlresolvers import RegexURLResolver, RegexURLPattern
+from drfdocs.api_docs import ApiDocumentation
 
 
 class DRFDocsView(TemplateView):
 
-    root_urlconf = __import__(settings.ROOT_URLCONF)
     template_name = "drfdocs/home.html"
-    EXCLUDED_APPS = ["admin", "drfdocs"]
-    VIEW_NAMES = []
-
-    def get_all_view_names(self, urlpatterns):
-        for pattern in urlpatterns:
-            if isinstance(pattern, RegexURLResolver):
-                if pattern.app_name not in self.EXCLUDED_APPS:
-                    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)
-        all_urlpatterns = self.root_urlconf.urls.urlpatterns
-        context['views'] = self.get_all_view_names(all_urlpatterns)
+        docs = ApiDocumentation()
+        context['views'] = docs.get_views()
         return context