Minor fix in sessions backend (remove pk from Session.expire_date)

Author: r4fek

django_cassandra_engine/sessions/backends/db.py

@@ -1,6 +1,11 @@
+from datetime import datetime
+import logging
+
 from django.contrib.sessions.backends.db import (
     SessionStore as DjangoSessionStore
 )
+from django.core.exceptions import SuspiciousOperation
+from django.utils.encoding import force_text
 from django.utils.functional import cached_property
 from django.contrib.sessions.backends import db
 
@@ -36,6 +41,21 @@ def create_model_instance(self, data):
             expire_date=self.get_expiry_date(),
         )
 
+    def load(self):
+        try:
+            s = self.model.objects.get(session_key=self.session_key)
+            if s.expire_date <= datetime.now():
+                s.delete()
+                raise SuspiciousOperation('old session detected')
+            return self.decode(s.session_data)
+        except (self.model.DoesNotExist, SuspiciousOperation) as e:
+            if isinstance(e, SuspiciousOperation):
+                logger = logging.getLogger('django.security.%s' %
+                                           e.__class__.__name__)
+                logger.warning(force_text(e))
+            self.create()
+            return {}
+
     def exists(self, session_key):
         try:
             self.model.objects.get(session_key=session_key)

django_cassandra_engine/sessions/models.py

@@ -6,7 +6,7 @@ class AbstractBaseSession(Model):
     __abstract__ = True
 
     session_key = columns.Text(primary_key=True, max_length=40)
-    expire_date = columns.DateTime(primary_key=True, clustering_order='DESC')
+    expire_date = columns.DateTime()
 
     session_data = columns.Text()