__all__ = ["wsgiwrap", "restart", "cookie", "formdata"]
-from .util import wsgiwrap, persession
+from .util import wsgiwrap, persession, sessiondata
from .dispatch import restart
from . import cookie
from .form import formdata
return result
return [str(result)]
+class iterproxy(object):
+ # Makes sure iter(real).next() is called immediately, in order to
+ # let generator code run.
+ def __init__(self, real):
+ self.bk = real
+ self.bki = iter(real)
+ self._next = [None]
+ self.next()
+
+ def __iter__(self):
+ return self
+
+ def next(self):
+ if self._next is None:
+ raise StopIteration()
+ ret = self._next[0]
+ try:
+ self._next[:] = [self.bki.next()]
+ except StopIteration:
+ self._next = None
+ return ret
+
+ def close(self):
+ if hasattr(self.bk, "close"):
+ self.bk.close()
+
def handle(req, startreq, handler):
try:
resp = [""]
while True:
try:
- resp = handler(req)
+ resp = iterproxy(handler(req))
break
except restart as i:
handler = i.handle
sess.atime = now
except KeyError:
sess = session()
- self.live[sess.id] = sess
new = True
def ckfreeze(req):
if sess.dirty():
+ if new:
+ cookie.add(req, self.cookiename, sess.id, self.path)
+ with self.lock:
+ self.live[sess.id] = sess
try:
- if new:
- cookie.add(req, self.cookiename, sess.id, self.path)
self.freeze(sess)
except:
pass
return sess[callable].handle(req)
return wrapper
return dec
+
+class sessiondata(object):
+ @classmethod
+ def get(cls, req):
+ sess = cls.sessdb().get(req)
+ with sess.lock:
+ try:
+ return sess[cls]
+ except KeyError:
+ ret = cls(req)
+ ret._is_dirty = [False]
+ sess[cls] = ret
+ return ret
+
+ @classmethod
+ def sessdb(cls):
+ return session.default
+
+ def sessfrozen(self):
+ self._is_dirty[:] = [False]
+
+ def sessdirty(self):
+ return self._is_dirty[0]
+
+ def __setattr__(self, name, value):
+ if hasattr(self, "_is_dirty"):
+ self._is_dirty[:] = [True]
+ super().__setattr__(name, value)
+
+ def __delattr__(self, name):
+ super().__delattr__(name)
+ self._is_dirty[:] = [True]