✨ Log only selected response status codes.

Author: Hassan-Elseoudy

drf_api_logger/middleware/api_logger_middleware.py

@@ -51,6 +51,12 @@ def __init__(self, get_response):
                     settings.DRF_API_LOGGER_METHODS) is list:
                 self.DRF_API_LOGGER_METHODS = settings.DRF_API_LOGGER_METHODS
 
+        self.DRF_API_LOGGER_STATUS_CODES = []
+        if hasattr(settings, 'DRF_API_LOGGER_STATUS_CODES'):
+            if type(settings.DRF_API_LOGGER_STATUS_CODES) is tuple or type(
+                    settings.DRF_API_LOGGER_STATUS_CODES) is list:
+                self.DRF_API_LOGGER_STATUS_CODES = settings.DRF_API_LOGGER_STATUS_CODES
+
     def __call__(self, request):
 
         # Run only if logger is enabled.
@@ -82,6 +88,10 @@ def __call__(self, request):
             # the view (and later middleware) are called.
             response = self.get_response(request)
 
+            # Only log required status codes if matching
+            if self.DRF_API_LOGGER_STATUS_CODES and response.status_code not in self.DRF_API_LOGGER_STATUS_CODES:
+                return response
+
             # Code to be executed for each request/response after
             # the view is called.