X-Git-Url: http://dolda2000.com/gitweb/?a=blobdiff_plain;f=wrw%2Futil.py;h=db74e623606d9d7f04504f68a0ce05a02ac420b7;hb=d13a1a574515e4c4a40ed0090c8c684cc9e523f2;hp=b757e14a724c8f9aec56d568a4bea96b099c90fc;hpb=fed807e31c96fef8d941b34a25a1f5044b2d4e8d;p=wrw.git diff --git a/wrw/util.py b/wrw/util.py index b757e14..db74e62 100644 --- a/wrw/util.py +++ b/wrw/util.py @@ -19,3 +19,56 @@ def persession(data = None): 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