From: Fredrik Tolf Date: Tue, 26 Jun 2012 18:00:45 +0000 (+0200) Subject: Tried to create a more advanced sessiondata class for more precise control. X-Git-Url: http://git.dolda2000.com/gitweb/?a=commitdiff_plain;h=df5f786840523bc4faf8e0f2168df0d3b2638473;p=wrw.git Tried to create a more advanced sessiondata class for more precise control. --- diff --git a/wrw/util.py b/wrw/util.py index c2af5aa..d7b52cb 100644 --- a/wrw/util.py +++ b/wrw/util.py @@ -89,3 +89,93 @@ class manudirty(object): def dirty(self): self.__dirty = True + +class specslot(object): + __slots__ = ["nm", "idx", "dirty"] + unbound = object() + + def __init__(self, nm, idx, dirty): + self.nm = nm + self.idx = idx + self.dirty = dirty + + @staticmethod + def slist(ins): + # Avoid calling __getattribute__ + return specdirty.__sslots__.__get__(ins, type(ins)) + + def __get__(self, ins, cls): + val = self.slist(ins)[self.idx] + if val is specslot.unbound: + raise AttributeError("specslot %r is unbound" % self.nm) + return val + + def __set__(self, ins, val): + self.slist(ins)[self.idx] = val + if self.dirty: + ins.dirty() + + def __delete__(self, ins): + self.slist(ins)[self.idx] = specslot.unbound + ins.dirty() + +class specclass(type): + def __init__(self, name, bases, tdict): + super(specclass, self).__init__(name, bases, tdict) + sslots = set() + dslots = set() + for cls in self.__mro__: + css = cls.__dict__.get("__saveslots__", ()) + sslots.update(css) + dslots.update(cls.__dict__.get("__dirtyslots__", css)) + self.__sslots_l__ = list(sslots) + self.__sslots_a__ = list(sslots | dslots) + for i, slot in enumerate(self.__sslots_a__): + setattr(self, slot, specslot(slot, i, slot in dslots)) + +class specdirty(sessiondata): + __metaclass__ = specclass + __slots__ = ["session", "__sslots__", "_is_dirty"] + + def __specinit__(self): + pass + + @staticmethod + def __new__(cls, req, sess): + self = super(specdirty, cls).__new__(cls) + self.session = sess + self.__sslots__ = [specslot.unbound] * len(cls.__sslots_a__) + self.__specinit__() + self._is_dirty = False + return self + + def __getnewargs__(self): + return (None, self.session) + + def dirty(self): + self._is_dirty = True + + def sessfrozen(self): + self._is_dirty = False + + def sessdirty(self): + return self._is_dirty + + def __getstate__(self): + ret = {} + for nm, val in zip(type(self).__sslots_a__, specslot.slist(self)): + if val is specslot.unbound: + ret[nm] = False, None + else: + ret[nm] = True, val + return ret + + def __setstate__(self, st): + ss = specslot.slist(self) + for i, nm in enumerate(type(self).__sslots_a__): + bound, val = st.pop(nm, (False, None)) + print i, nm, bound, val + if not bound: + ss[i] = specslot.unbound + else: + ss[i] = val