X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Futil.py;h=db74e623606d9d7f04504f68a0ce05a02ac420b7;hb=d13a1a574515e4c4a40ed0090c8c684cc9e523f2;hp=22131eae2b60208cefe2e591897a5a87931d0ceb;hpb=b409a33843abb3221edd27016558c39cf33a6510;p=wrw.git diff --git a/wrw/util.py b/wrw/util.py index 22131ea..db74e62 100644 --- a/wrw/util.py +++ b/wrw/util.py @@ -1,6 +1,74 @@ -import req, dispatch +from . import req, dispatch, session def wsgiwrap(callable): def wrapper(env, startreq): - return dispatch.handle(req.request(env), startreq, callable) + return dispatch.handle(req.origrequest(env), startreq, callable) return wrapper + +def persession(data = None): + def dec(callable): + def wrapper(req): + sess = session.get(req) + if callable not in sess: + if data is None: + sess[callable] = callable() + else: + if data not in sess: + sess[data] = data() + sess[callable] = callable(data) + 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) + sess[cls] = ret + return ret + + @classmethod + def sessdb(cls): + return session.default + +class autodirty(sessiondata): + @classmethod + def get(cls, req): + ret = super().get(req) + if "_is_dirty" not in ret.__dict__: + ret.__dict__["_is_dirty"] = False + + def sessfrozen(self): + self.__dict__["_is_dirty"] = False + + def sessdirty(self): + return self._is_dirty + + def __setattr__(self, name, value): + super().__setattr__(name, value) + if "_is_dirty" in self.__dict__: + self.__dict__["_is_dirty"] = True + + def __delattr__(self, name): + super().__delattr__(name, value) + if "_is_dirty" in self.__dict__: + self.__dict__["_is_dirty"] = True + +class manudirty(object): + def __init__(self, *args, **kwargs): + super().__init__(*args, **kwargs) + self.__dirty = False + + def sessfrozen(self): + self.__dirty = False + + def sessdirty(self): + return self.__dirty + + def dirty(self): + self.__dirty = True