base

plane-src/apps/api/plane/db/models/session.py

@@ -0,0 +1,56 @@
+# Copyright (c) 2023-present Plane Software, Inc. and contributors
+# SPDX-License-Identifier: AGPL-3.0-only
+# See the LICENSE file for details.
+
+# Python imports
+import string
+
+# Django imports
+from django.contrib.sessions.backends.db import SessionStore as DBSessionStore
+from django.contrib.sessions.base_session import AbstractBaseSession
+from django.db import models
+from django.utils.crypto import get_random_string
+
+VALID_KEY_CHARS = string.ascii_lowercase + string.digits
+
+
+class Session(AbstractBaseSession):
+    device_info = models.JSONField(null=True, blank=True, default=None)
+    session_key = models.CharField(max_length=128, primary_key=True)
+    user_id = models.CharField(null=True, max_length=50, db_index=True)
+
+    @classmethod
+    def get_session_store_class(cls):
+        return SessionStore
+
+    class Meta(AbstractBaseSession.Meta):
+        db_table = "sessions"
+
+
+class SessionStore(DBSessionStore):
+    @classmethod
+    def get_model_class(cls):
+        return Session
+
+    def _get_new_session_key(self):
+        """
+        Return a new session key that is not present in the current backend.
+        Override this method to use a custom session key generation mechanism.
+        """
+        while True:
+            session_key = get_random_string(128, VALID_KEY_CHARS)
+            if not self.exists(session_key):
+                return session_key
+
+    def create_model_instance(self, data):
+        obj = super().create_model_instance(data)
+        try:
+            user_id = data.get("_auth_user_id")
+        except (ValueError, TypeError):
+            user_id = None
+        obj.user_id = user_id
+
+        # Save the device info
+        device_info = data.get("device_info")
+        obj.device_info = device_info if isinstance(device_info, dict) else None
+        return obj