Fixing redissession

magfest · May 2, 2024 · 71c842e · 71c842e
1 parent 92370b3
commit 71c842e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/sideboard/lib/_cp.py b/sideboard/lib/_cp.py
@@ -15,6 +15,7 @@
 except ImportError:
     from cherrypy.lib.sessions import Session
     import redis
+    import pickle
     from redis import Sentinel
 
     class RedisSession(Session):
@@ -62,15 +63,16 @@ def _exists(self):
 
         def _load(self):
             try:
-            return pickle.loads(self.cache.get(self.prefix+self.id))
+                return pickle.loads(self.cache.get(self.prefix+self.id))
             except TypeError:
             # if id not defined pickle can't load None and raise TypeError
-            return None
+                return None
 
         def _save(self, expiration_time):
             pickled_data = pickle.dumps(
                 (self._data, expiration_time),
-                pickle.HIGHEST_PROTOCOL)
+                pickle.HIGHEST_PROTOCOL
+            )
 
             result = self.cache.setex(self.prefix+self.id, self.timeout * 60, pickled_data)