2 from . import store, lib
3 from .store import storedesc
5 __all__ = ["simple", "multi"]
7 class cursor(lib.closable):
8 def __init__(self, bk, st):
20 return k, self.st.get(id)
25 class base(storedesc):
26 def __init__(self, store, indextype, name, datatype, default):
28 self.indextype = indextype
31 self.default = default
33 self.lk = threading.Lock()
34 self.mattr = "__idx_%s_new" % name
35 self.iattr = "__idx_%s_cur" % name
40 self.idx = self.indextype(self.store.db(), self.name, self.typ)
43 def __get__(self, obj, cls):
44 if obj is None: return self
45 return getattr(obj, self.mattr, self.default)
47 def __set__(self, obj, val):
48 setattr(obj, self.mattr, val)
50 def __delete__(self, obj):
51 delattr(obj, self.mattr)
53 def get(self, **kwargs):
54 return cursor(self.index().get(**kwargs), self.store)
57 def __init__(self, store, indextype, name, datatype, default=None):
58 super().__init__(store, indextype, name, datatype, default)
60 def register(self, id, obj, tx):
61 val = self.__get__(obj, None)
62 self.index().put(val, id, tx=tx)
63 tx.postcommit(lambda: setattr(obj, self.iattr, val))
65 def unregister(self, id, obj, tx):
66 self.index().remove(getattr(obj, self.iattr), id, tx=tx)
67 tx.postcommit(lambda: delattr(obj, self.iattr))
69 def update(self, id, obj, tx):
70 val = self.__get__(obj, None)
71 ival = getattr(obj, self.iattr)
74 idx.remove(ival, id, tx=tx)
75 idx.put(val, id, tx=tx)
76 tx.postcommit(lambda: setattr(obj, self.iattr, val))
79 def __init__(self, store, indextype, name, datatype):
80 super().__init__(store, indextype, name, datatype, ())
82 def register(self, id, obj, tx):
83 vals = frozenset(self.__get__(obj, None))
86 idx.put(val, id, tx=tx)
87 tx.postcommit(lambda: setattr(obj, self.iattr, vals))
89 def unregister(self, id, obj, tx):
91 for val in getattr(obj, self.iattr):
92 idx.remove(val, id, tx=tx)
93 tx.postcommit(lambda: delattr(obj, self.iattr))
95 def update(self, id, obj, tx):
96 vals = frozenset(self.__get__(obj, None))
97 ivals = getattr(obj, self.iattr)
100 for val in ivals - vals:
101 idx.remove(val, id, tx=tx)
102 for val in vals - ivals:
103 idx.put(val, id, tx=tx)
104 tx.postcommit(lambda: setattr(obj, self.iattr, vals))