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

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

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

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

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "utg-base"
-version = "1.5.6"
+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__)

{utg_base-1.5.6 → utg_base-1.6.0}/src/utg_base/references_api/utils.py

@@ -102,7 +102,7 @@ def create_view_set(model: Model):
 
 
 def create_serializer_for_create(model: Model):
-    @extend_schema_serializer(component_name='Admin' + get_serializer_name(model))
+    @extend_schema_serializer(component_name='Admin1' + get_serializer_name(model))
     class Serializer(serializers.ModelSerializer):
         class Meta:
             fields = '__all__'
@@ -114,7 +114,7 @@ def create_serializer_for_create(model: Model):
 def create_serializer_for_partial_update(model: Model):
     fields_list = get_model_fields_list(model)
 
-    @extend_schema_serializer(component_name='Admin' + get_serializer_name(model))
+    @extend_schema_serializer(component_name='Admin2' + get_serializer_name(model))
     class Serializer(serializers.ModelSerializer):
         class Meta:
             fields = '__all__'
@@ -147,7 +147,7 @@ def create_serializer(model: Model, depth=0):
             parsed_url = urlparse(value.url)
             return f'{parsed_url.path}?{parsed_url.query}'
 
-    @extend_schema_serializer(component_name='Admin' + get_serializer_name(model))
+    @extend_schema_serializer(component_name='Admin3' + get_serializer_name(model))
     class Serializer(serializers.ModelSerializer):
         def __init__(self, *args, **kwargs):
             super().__init__(*args, **kwargs)