Source code for django_tomselect.middleware

"""Middleware for django_tomselect to manage request objects."""

__all__ = [
    "TomSelectMiddleware",
    "get_current_request",
]

try:
    from asgiref.local import Local as local  # noqa: N813
except ImportError:
    from threading import local

from django.http import HttpRequest

from django_tomselect.logging import get_logger

logger = get_logger(__name__)

# Create a single local instance for storing request
_request_local = local()




def get_current_request() -> HttpRequest | None:
    """Get the current request from thread/async-local storage."""
    return getattr(_request_local, "request", None)





class TomSelectMiddleware:
    """Stores the request object in thread/async-local storage.

    Compatible with both WSGI and ASGI deployments.
    """



    def __init__(self, get_response):
        """Initialize the middleware with the get_response callable."""
        self.get_response = get_response




    def __call__(self, request):
        """Handle sync requests in WSGI deployments."""
        # Store request in local storage
        _request_local.request = request

        try:
            response = self.get_response(request)
            logger.debug("Request object stored in local storage.")
            return response
        finally:
            # Always clean up the local storage
            if hasattr(_request_local, "request"):
                del _request_local.request




    async def __acall__(self, request):
        """Handle async requests in ASGI deployments."""
        # Store request in local storage
        _request_local.request = request

        try:
            response = await self.get_response(request)
            logger.debug("Request object stored in local storage.")
            return response
        finally:
            # Always clean up the local storage
            if hasattr(_request_local, "request"):
                del _request_local.request