add middleware for loggin

Author: Sublian

myproject/middleware.py

@@ -0,0 +1,28 @@
+import logging
+import time
+
+logger = logging.getLogger("django.request")
+
+
+class RequestLoggingMiddleware:
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        start_time = time.time()
+
+        response = self.get_response(request)
+
+        duration = round(time.time() - start_time, 3)
+
+        user = request.user if request.user.is_authenticated else "Anonymous"
+        ip = request.META.get("REMOTE_ADDR")
+        path = request.path
+        method = request.method
+        status = response.status_code
+
+        logger.warning(
+            f"{method} {path} | status={status} | user={user} | ip={ip} | {duration}s"
+        )
+
+        return response

myproject/settings.py

@@ -45,6 +45,7 @@
     'django.contrib.messages.middleware.MessageMiddleware',
     'django.middleware.clickjacking.XFrameOptionsMiddleware',
 ]
+MIDDLEWARE.append("myproject.middleware.RequestLoggingMiddleware")
 
 ROOT_URLCONF = 'myproject.urls'
 
@@ -184,4 +185,40 @@
 })
 
 if "pytest" in sys.argv[0]:
-    REST_FRAMEWORK["DEFAULT_THROTTLE_CLASSES"] = []
+    REST_FRAMEWORK["DEFAULT_THROTTLE_CLASSES"] = []
+    
+LOGGING = {
+    "version": 1,
+    "disable_existing_loggers": False,
+
+    "formatters": {
+        "verbose": {
+            "format": "[{levelname}] {asctime} {name} | {message}",
+            "style": "{",
+        },
+        "simple": {
+            "format": "[{levelname}] {message}",
+            "style": "{",
+        },
+    },
+
+    "handlers": {
+        "console": {
+            "class": "logging.StreamHandler",
+            "formatter": "verbose",
+        },
+    },
+
+    "loggers": {
+        "django": {
+            "handlers": ["console"],
+            "level": "INFO",
+        },
+        "django.request": {
+            "handlers": ["console"],
+            "level": "WARNING",
+            "propagate": False,
+        },
+    },
+}
+    

products/views.py

@@ -3,6 +3,9 @@
 from .models import Product
 from .serializers import ProductSerializer
 from users.permissions import IsAdmin, IsStaff
+import logging
+
+logger = logging.getLogger("django.request")
 
 class ProductViewSet(viewsets.ModelViewSet):
     serializer_class = ProductSerializer

users/views.py

@@ -14,12 +14,27 @@
 from rest_framework_simplejwt.views import TokenObtainPairView
 from .throttles import LoginRateThrottle
 import sys
+import logging
+
+logger = logging.getLogger("django.request")
+
 
 class CustomTokenObtainPairView(TokenObtainPairView):
     if "pytest" in sys.argv[0]:
         throttle_classes = []
     else:
         throttle_classes = [LoginRateThrottle]
+        
+    def post(self, request, *args, **kwargs):
+        ip = request.META.get("REMOTE_ADDR")
+        response = super().post(request, *args, **kwargs)
+
+        if response.status_code == 200:
+            logger.warning(f"LOGIN OK | user={request.data.get('username')} | ip={ip}")
+        else:
+            logger.warning(f"LOGIN FAIL | user={request.data.get('username')} | ip={ip}")
+
+        return response
 
 class UserViewSet(viewsets.ModelViewSet):
     """