mirror of
https://github.com/jwinterm/modoboa.git
synced 2026-10-28 06:28:43 +01:00
40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
"""Custom middlewares."""
|
|
|
|
from django.http import JsonResponse
|
|
from django.utils.deprecation import MiddlewareMixin
|
|
from django.utils.encoding import smart_str
|
|
|
|
from modoboa.lib import signals as lib_signals
|
|
from modoboa.lib.exceptions import ModoboaException
|
|
from modoboa.lib.web_utils import _render_error
|
|
|
|
|
|
def is_ajax(request):
|
|
return request.META.get("HTTP_X_REQUESTED_WITH") == "XMLHttpRequest"
|
|
|
|
|
|
class CommonExceptionCatcher(MiddlewareMixin):
|
|
"""Modoboa exceptions catcher."""
|
|
|
|
def process_exception(self, request, exception):
|
|
if not isinstance(exception, ModoboaException):
|
|
return None
|
|
|
|
if "/api/" in request.path:
|
|
http_code = exception.http_code if exception.http_code else 500
|
|
return JsonResponse({"error": str(exception)}, status=http_code)
|
|
|
|
return _render_error(request, user_context={"error": smart_str(exception)})
|
|
|
|
|
|
class RequestCatcherMiddleware(MiddlewareMixin):
|
|
"""Simple middleware to store the current request."""
|
|
|
|
def process_request(self, request):
|
|
lib_signals.set_current_request(request)
|
|
|
|
def process_response(self, request, response):
|
|
"""Empty store."""
|
|
lib_signals.set_current_request(None)
|
|
return response
|