From d77dcf62b4005a0f36ef2fa7ba6d3651d2ef38d7 Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Sat, 5 Nov 2022 14:17:27 -0600
Subject: Initial commit (users and statuses)

---
 users/decorators.py | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)
 create mode 100644 users/decorators.py

(limited to 'users/decorators.py')

diff --git a/users/decorators.py b/users/decorators.py
new file mode 100644
index 0000000..77d633a
--- /dev/null
+++ b/users/decorators.py
@@ -0,0 +1,39 @@
+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
-- 
cgit v1.2.3