45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
from django.urls import reverse
|
|
from django_scopes import scope, scopes_disabled
|
|
from rest_framework.authtoken.models import Token
|
|
|
|
from cookbook.views import views
|
|
|
|
|
|
class ScopeMiddleware:
|
|
def __init__(self, get_response):
|
|
self.get_response = get_response
|
|
|
|
def __call__(self, request):
|
|
if request.user.is_authenticated:
|
|
|
|
if request.path.startswith('/admin/'):
|
|
with scopes_disabled():
|
|
return self.get_response(request)
|
|
|
|
if request.path.startswith('/signup/') or request.path.startswith('/invite/'):
|
|
return self.get_response(request)
|
|
|
|
if request.path.startswith('/accounts/'):
|
|
return self.get_response(request)
|
|
|
|
with scopes_disabled():
|
|
if request.user.userpreference.space is None and not reverse('account_logout') in request.path:
|
|
return views.no_space(request)
|
|
|
|
if request.user.groups.count() == 0 and not reverse('account_logout') in request.path:
|
|
return views.no_groups(request)
|
|
|
|
request.space = request.user.userpreference.space
|
|
# with scopes_disabled():
|
|
with scope(space=request.space):
|
|
return self.get_response(request)
|
|
else:
|
|
if request.path.startswith('/api/'):
|
|
if token := Token.objects.filter(key=request.headers['Authorization'].replace('Token ', '')).first():
|
|
request.space = token.user.userpreference.space
|
|
with scope(space=request.space):
|
|
return self.get_response(request)
|
|
with scopes_disabled():
|
|
request.space = None
|
|
return self.get_response(request)
|