utg-base 1.5.7__tar.gz → 1.6.0__tar.gz

{utg_base-1.5.7 → utg_base-1.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: utg-base
-Version: 1.5.7
+Version: 1.6.0
 Summary: UTG Base Package
 Author: Rovshen
 Author-email: rovshenashirov1619@gmail.com

{utg_base-1.5.7 → utg_base-1.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utg-base"
-version = "1.5.7"
+version = "1.6.0"
 description = "UTG Base Package"
 authors = ["Rovshen <rovshenashirov1619@gmail.com>"]
 readme = "README.md"

utg_base-1.6.0/src/utg_base/api/spectacular.py

@@ -0,0 +1,33 @@
+import os
+
+from django.conf import settings
+from django.http import HttpResponseForbidden
+from django.urls import path
+from drf_spectacular.views import SpectacularAPIView, SpectacularSwaggerView
+from rest_framework.views import APIView
+
+
+class DebugTokenRequiredMixin(APIView):
+    """
+    If Request header X-Debug-Token not found then 403
+    """
+
+    def dispatch(self, request, *args, **kwargs):
+        token = request.headers.get("X-Debug-Token")
+        if not settings.DEBUG and token != os.environ.get("DJANGO_SECRET_KEY"):
+            return HttpResponseForbidden("Forbidden")
+        return super().dispatch(request, *args, **kwargs)
+
+
+class ProtectedSpectacularAPIView(DebugTokenRequiredMixin, SpectacularAPIView):
+    pass
+
+
+class ProtectedSpectacularSwaggerView(DebugTokenRequiredMixin, SpectacularSwaggerView):
+    pass
+
+
+urlpatterns = [
+    path('api/swagger/schema/', ProtectedSpectacularAPIView.as_view(), name='schema'),
+    path('api/docs/', ProtectedSpectacularSwaggerView.as_view(url_name='schema'), name='docs'),
+]

utg_base-1.6.0/src/utg_base/middleware/debug.py

@@ -0,0 +1,10 @@
+import os
+
+from django.utils.deprecation import MiddlewareMixin
+
+
+class DebugOverrideMiddleware(MiddlewareMixin):
+    def process_exception(self, request, exception):
+        if request.headers.get("X-Debug-Token") == os.environ.get("DJANGO_SECRET_KEY"):
+            from django.views import debug
+            return debug.technical_500_response(request, type(exception), exception, exception.__traceback__)