takahe/users/middleware.py

18 wiersze
477 B
Python

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