2 from . import store, lib
3 from .store import storedesc
5 class cursor(lib.closable):
6 def __init__(self, bk, st):
18 return k, self.st.get(id)
23 class base(storedesc):
24 def __init__(self, store, indextype, name, datatype, default):
26 self.indextype = indextype
29 self.default = default
31 self.lk = threading.Lock()
32 self.mattr = "__idx_%s_new" % name
33 self.iattr = "__idx_%s_cur" % name
38 self.idx = self.indextype(self.store.db(), self.name, self.typ)
41 def __get__(self, obj, cls):
42 if obj is None: return self
43 return getattr(obj, self.mattr, self.default)
45 def __set__(self, obj, val):
46 setattr(obj, self.mattr, val)
48 def __delete__(self, obj):
49 delattr(obj, self.mattr)
51 def get(self, **kwargs):
52 return cursor(self.index().get(**kwargs), self.store)
55 def __init__(self, store, indextype, name, datatype, default=None):
56 super().__init__(store, indextype, name, datatype, default)
58 def register(self, id, obj, tx):
59 val = self.__get__(obj, None)
60 self.index().put(val, id, tx=tx)
61 tx.postcommit(lambda: setattr(obj, self.iattr, val))
63 def unregister(self, id, obj, tx):
64 self.index().remove(getattr(obj, self.iattr), id, tx=tx)
65 tx.postcommit(lambda: delattr(obj, self.iattr))
67 def update(self, id, obj, tx):
68 val = self.__get__(obj, None)
69 ival = getattr(obj, self.iattr)
72 idx.remove(ival, id, tx=tx)
73 idx.put(val, id, tx=tx)
74 tx.postcommit(lambda: setattr(obj, self.iattr, val))
77 def __init__(self, store, indextype, name, datatype):
78 super().__init__(store, indextype, name, datatype, ())
80 def register(self, id, obj, tx):
81 vals = frozenset(self.__get__(obj, None))
84 idx.put(val, id, tx=tx)
85 tx.postcommit(lambda: setattr(obj, self.iattr, vals))
87 def unregister(self, id, obj, tx):
89 for val in getattr(obj, self.iattr):
90 idx.remove(val, id, tx=tx)
91 tx.postcommit(lambda: delattr(obj, self.iattr))
93 def update(self, id, obj, tx):
94 vals = frozenset(self.__get__(obj, None))
95 ivals = getattr(obj, self.iattr)
98 for val in ivals - vals:
99 idx.remove(val, id, tx=tx)
100 for val in vals - ivals:
101 idx.put(val, id, tx=tx)
102 tx.postcommit(lambda: setattr(obj, self.iattr, vals))