From dbe57075d386d7474bafc208b654507d9a2d769e Mon Sep 17 00:00:00 2001
From: Andrew Godwin
Date: Sun, 6 Nov 2022 13:48:04 -0700
Subject: Rework to a domains model for better vhosting

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

(limited to 'users/middleware.py')

diff --git a/users/middleware.py b/users/middleware.py
new file mode 100644
index 0000000..aa22178
--- /dev/null
+++ b/users/middleware.py
@@ -0,0 +1,24 @@
+from users.models import Identity
+
+
+class IdentityMiddleware:
+    """
+    Adds a request.identity object which is either the current session's
+    identity, or None if they have not picked one yet/it's invalid.
+    """
+
+    def __init__(self, get_response):
+        self.get_response = get_response
+
+    def __call__(self, request):
+        identity_id = request.session.get("identity_id")
+        if not identity_id:
+            request.identity = None
+        else:
+            try:
+                request.identity = Identity.objects.get(id=identity_id)
+            except Identity.DoesNotExist:
+                request.identity = None
+
+        response = self.get_response(request)
+        return response
-- 
cgit v1.2.3