django_scopedsites/scopedsites/middleware.py

from django.http import Http404
from django_scopes import scope

from scopedsites.models import Domain


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

    def __call__(self, request):
        d = Domain.get_from_request(request)

        with scope(domain=d):
            response = self.get_response(request)

            return response


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

    def __call__(self, request):
        try:
            Domain.get_from_request(request)
        except Http404:
            Domain.objects.create(fqdn=request.META.get('HTTP_HOST'))