djresttoolkit 0.3.0__tar.gz → 0.4.0__tar.gz

{djresttoolkit-0.3.0 → djresttoolkit-0.4.0}/PKG-INFO

@@ -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 → djresttoolkit-0.4.0}/pyproject.toml

@@ -4,7 +4,7 @@
 
 [project]
 name = "djresttoolkit"
-version = "0.3.0"
+version = "0.4.0"
 description = "A collection of Django and DRF utilities to simplify API development."
 readme = { file = "README.md", content-type = "text/markdown" }
 license = { file = "LICENSE" }

djresttoolkit-0.4.0/src/djresttoolkit/middlewares/__init__.py

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

djresttoolkit-0.4.0/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