Settings - Hidden

Author: Emmanouil Konstantinidis

README.md

@@ -38,6 +38,12 @@ Finally include the `rest_framework_docs` urls in your `urls.py`:
         url(r'^docs/', include('rest_framework_docs.urls')),
     ]
 
+### Settings
+
+    REST_FRAMEWORK_DOCS = {
+        'HIDDEN': True  # Default: False
+    }
+
 ### Roadmap
 
   - [ ] Support Python 2 & Python 3

rest_framework_docs/settings.py

@@ -0,0 +1,19 @@
+from django.conf import settings
+
+
+class DRFSettings(object):
+
+    def __init__(self):
+        self.drf_settings = {
+            "HIDDEN": self.get_setting("HIDDEN") or False
+        }
+
+    def get_setting(self, name):
+        try:
+            return settings.REST_FRAMEWORK_DOCS[name]
+        except:
+            return None
+
+    @property
+    def settings(self):
+        return self.drf_settings

rest_framework_docs/views.py

@@ -1,12 +1,18 @@
+from django.http import Http404
 from django.views.generic.base import TemplateView
 from rest_framework_docs.api_docs import ApiDocumentation
+from rest_framework_docs.settings import DRFSettings
 
 
 class DRFDocsView(TemplateView):
 
     template_name = "rest_framework_docs/home.html"
 
     def get_context_data(self, **kwargs):
+        settings = DRFSettings().settings
+        if settings["HIDDEN"]:
+            raise Http404("Django Rest Framework Docs are hidden. Check you settings.")
+
         context = super(DRFDocsView, self).get_context_data(**kwargs)
         docs = ApiDocumentation()
         context['endpoints'] = docs.get_endpoints()