djresttoolkit 0.3.0__py3-none-any.whl → 0.4.0__py3-none-any.whl

{djresttoolkit-0.3.0.dist-info → djresttoolkit-0.4.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: djresttoolkit
-Version: 0.3.0
+Version: 0.4.0
 Summary: A collection of Django and DRF utilities to simplify API development.
 Project-URL: Homepage, https://github.com/shaileshpandit141/djresttoolkit
 Project-URL: Documentation, https://shaileshpandit141.github.io/djresttoolkit

{djresttoolkit-0.3.0.dist-info → djresttoolkit-0.4.0.dist-info}/RECORD

@@ -8,13 +8,15 @@ src/djresttoolkit/mail/__init__.py,sha256=tB9SdMlhfWQ640q4aobZ0H1c7fTWalpDL2I-on
 src/djresttoolkit/mail/_email_sender.py,sha256=vvTPZzSAfX2FXHv1IY0ST8dEsq8M4wAxkihm0JbRh1Y,3136
 src/djresttoolkit/mail/_models.py,sha256=_41pH3xC0jP8SHSty2FkxvRh2_ddKk-4peT11OHcBBE,1462
 src/djresttoolkit/mail/_types.py,sha256=est1mrN80vB_4-j-2yuAr_l_7rNzO4AJlqpq74zO5ow,682
+src/djresttoolkit/middlewares/__init__.py,sha256=GZHU3Yy4xXoEi62tHn0UJNxN6XgGM2_HES8Bt5AS5Lk,100
+src/djresttoolkit/middlewares/_response_time_middleware.py,sha256=1wCwdkW5Ng6HJo8zx0F7ylms84OGP-1K0kbyG6Vacuk,908
 src/djresttoolkit/migrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/djresttoolkit/views/__init__.py,sha256=XrxBrs6sH4HmUzp41omcmy_y94pSaXAVn01ttQ022-4,76
 src/djresttoolkit/views/_exceptions/__init__.py,sha256=DrCUxuPNyBR4WhzNutn5HDxLa--q51ykIxSG7_bFsOI,83
 src/djresttoolkit/views/_exceptions/_exception_handler.py,sha256=lg6kfAABex-UbAsF78uX-M-ZnJ3u1vK1eIXytb-4KSw,2375
-djresttoolkit-0.3.0.dist-info/METADATA,sha256=LKqn4ZSUVD-yo03t6klk-QHW1q_VwxpCJk6DfTSIDMo,4374
-djresttoolkit-0.3.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-djresttoolkit-0.3.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
-djresttoolkit-0.3.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
-djresttoolkit-0.3.0.dist-info/RECORD,,
+djresttoolkit-0.4.0.dist-info/METADATA,sha256=FGDP5PR0AZoaPvEyrG95DVC516vGOWlXk2ZgyLfm8kE,4374
+djresttoolkit-0.4.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+djresttoolkit-0.4.0.dist-info/entry_points.txt,sha256=YMhfTF-7mYppO8QqqWnvR_hyMWvoYxD6XI94_ViFu3k,60
+djresttoolkit-0.4.0.dist-info/licenses/LICENSE,sha256=8-oZM3yuuTRjySMbVKX9YXYA7Y4M_KhQNBYXPFjeWUo,1074
+djresttoolkit-0.4.0.dist-info/RECORD,,

src/djresttoolkit/middlewares/__init__.py

@@ -0,0 +1,3 @@
+from ._response_time_middleware import ResponseTimeMiddleware
+
+__all__ = ["ResponseTimeMiddleware"]

src/djresttoolkit/middlewares/_response_time_middleware.py

@@ -0,0 +1,32 @@
+import logging
+import time
+from collections.abc import Callable
+
+from django.http import HttpRequest, HttpResponse
+
+# Get logger from logging.
+logger = logging.getLogger(__name__)
+
+
+class ResponseTimeMiddleware:
+    """Calculte response response time."""
+
+    def __init__(
+        self,
+        get_response: Callable[[HttpRequest], HttpResponse],
+    ) -> None:
+        "Initilize response time middleware."
+        self.get_response = get_response
+
+    def __call__(self, request: HttpRequest) -> HttpResponse:
+        """Handle to response response time calculation."""
+        start_time = time.perf_counter()
+        response = self.get_response(request)
+        end_time = time.perf_counter()
+
+        response_time = f"{round(end_time - start_time, 5)} seconds"
+        response["X-Response-Time"] = response_time
+
+        logger.info(f"Request processed in {response_time}")
+
+        return response