Pass the current session to sessiondata constructor.

[wrw.git] / wrw / util.py

diff --git a/wrw/util.py b/wrw/util.py
index 7a29ea8..c2af5aa 100644 (file)
--- a/wrw/util.py
+++ b/wrw/util.py
@@ -36,19 +36,21 @@ def persession(data = None):
 
 class sessiondata(object):
     @classmethod
-    def get(cls, req):
+    def get(cls, req, create = True):
         sess = cls.sessdb().get(req)
         with sess.lock:
             try:
                 return sess[cls]
             except KeyError:
-                ret = cls(req)
+                if not create:
+                    return None
+                ret = cls(req, sess)
                 sess[cls] = ret
                 return ret
 
     @classmethod
     def sessdb(cls):
-        return session.default
+        return session.default.val
 
 class autodirty(sessiondata):
     @classmethod