40 lines
1.4 KiB
Python
40 lines
1.4 KiB
Python
|
from functools import wraps
|
||
|
|
|
||
|
|
from django.contrib.auth.views import redirect_to_login
|
||
|
|
from django.http import HttpResponseRedirect
|
||
|
|
|
||
|
|
from users.models import Identity
|
||
|
|
|
||
|
|
|
||
|
|
def identity_required(function):
|
||
|
|
"""
|
||
|
|
Decorator for views that ensures an active identity is selected.
|
||
|
|
"""
|
||
|
|
|
||
|
|
@wraps(function)
|
||
|
|
def inner(request, *args, **kwargs):
|
||
|
|
# They do have to be logged in
|
||
|
|
if not request.user.is_authenticated:
|
||
|
|
return redirect_to_login(next=request.get_full_path())
|
||
|
|
# Try to retrieve their active identity
|
||
|
|
identity_id = request.session.get("identity_id")
|
||
|
|
if not identity_id:
|
||
|
|
identity = None
|
||
|
|
else:
|
||
|
|
try:
|
||
|
|
identity = Identity.objects.get(id=identity_id)
|
||
|
|
except Identity.DoesNotExist:
|
||
|
|
identity = None
|
||
|
|
# If there's no active one, try to auto-select one
|
||
|
|
if identity is None:
|
||
|
|
possible_identities = list(request.user.identities.all())
|
||
|
|
if len(possible_identities) != 1:
|
||
|
|
# OK, send them to the identity selection page to select/create one
|
||
|
|
return HttpResponseRedirect("/identity/select/")
|
||
|
|
identity = possible_identities[0]
|
||
|
|
request.identity = identity
|
||
|
|
request.session["identity_id"] = identity.pk
|
||
|
|
return function(request, *args, **kwargs)
|
||
|
|
|
||
|
|
return inner
|