takahe/users/middleware.py

from users.models import Domain


class DomainMiddleware:
    """
    Tries to attach a Domain object to every incoming request, if one matches.
    """

    def __init__(self, get_response):
        self.get_response = get_response

    def __call__(self, request):
        request.domain = None
        if "host" in request.headers:
            request.domain = Domain.get_domain(request.headers["host"])
        response = self.get_response(request)
        return response