-cdef struct entry:
- assoc_base _b
- PyObject *v
-
-cdef entry *_find(assoc_table *t, object key, unsigned *f) except ?NULL:
- cdef void *p
- cdef int n
- cdef entry *e
- cdef atom *a
- a = ATOM_A(atom_pyintern(key))
- if f:
- f[0] = 0
- e = <entry *>assoc_find(t, a, PSIZEOF(e), f)
- if not f[0]:
- e.v = NULL
- else:
- e = <entry *>assoc_find(t, a, 0, NULL)
- return e
-
-cdef _key(entry *e):
- return atom_pywrap(ASSOC_ATOM(<void *>e))
-
-cdef _eget(entry *e):
- Py_INCREF(e.v)
- return <object>e.v
-
-cdef void _eset(entry *e, v):
- if e.v:
- Py_DECREF(e.v)
- e.v = <PyObject *>v
- Py_INCREF(e.v)
-
-cdef void _edel(assoc_table *t, entry *e):
- if e.v:
- Py_DECREF(e.v)
- assoc_remove(t, <void *>e)
-
-cdef class Table:
- cdef assoc_table _t
- def __new__(me, *hunoz, **hukairz):
- assoc_create(&me._t)
- def __init__(me, stuff = None, **kw):
- me.update(stuff, **kw)
- def __getitem__(me, key):
- cdef entry *e
- e = _find(&me._t, key, NULL)
- if not e:
- raise KeyError, key
- return _eget(e)
- def __setitem__(me, key, value):
- cdef unsigned f
- _eset(_find(&me._t, key, &f), value)
- def __delitem__(me, key):
- cdef entry *e
- cdef unsigned f
- e = _find(&me._t, key, &f)
- if not e:
- raise KeyError, key
- _edel(&me._t, e)
- def get(me, key, default = None):
- cdef entry *e
- e = _find(&me._t, key, NULL)
- if not e:
- return default
- return _eget(e)
- def setdefault(me, key, default = None):
- cdef entry *e
- cdef unsigned f
- e = _find(&me._t, key, &f)
- if f:
- return _eget(e)
- else:
- _eset(e, default)
- return default
- def pop(me, key, default = None):
- cdef entry *e
- e = _find(&me._t, key, NULL)
- if not e:
- return default
- rc = _eget(e)
- _edel(&me._t, e)
- return rc
- def popitem(me):
- cdef entry *e
- cdef assoc_iter i
- assoc_mkiter(&i, &me._t)
- e = <entry *>assoc_next(&i)
- if not e:
- raise ValueError, 'popitem(): table is empty'
- return _key(e), _eget(e)
- def keys(me):
- cdef assoc_iter i
- cdef entry *e
- l = []
- assoc_mkiter(&i, &me._t)
- while 1:
- e = <entry *>assoc_next(&i)
- if not e:
- break
- l.append(_key(e))
- return l
- def values(me):
- cdef assoc_iter i
- cdef entry *e
- l = []
- assoc_mkiter(&i, &me._t)
- while 1:
- e = <entry *>assoc_next(&i)
- if not e:
- break
- l.append(_eget(e))
- return l
- def items(me):
- cdef assoc_iter i
- cdef entry *e
- l = []
- assoc_mkiter(&i, &me._t)
- while 1:
- e = <entry *>assoc_next(&i)
- if not e:
- break
- l.append((_key(e), _eget(e)))
- return l
- def clear(me):
- cdef assoc_iter i
- cdef entry *e
- assoc_mkiter(&i, &me._t)
- while 1:
- e = <entry *>assoc_next(&i)
- if not e:
- break
- _edel(&me._t, e)
- return me
- def __dealloc__(me):
- cdef assoc_iter i
- cdef entry *e
- assoc_mkiter(&i, &me._t)
- while 1:
- e = <entry *>assoc_next(&i)
- if not e:
- break
- _edel(&me._t, e)
- assoc_destroy(&me._t)
- def iterkeys(me):
- return KeyIter(me)
- def __iter__(me):
- return KeyIter(me)
- def itervalues(me):
- return ValueIter(me)
- def iteritems(me):
- return ItemIter(me)
- def update(me, stuff = None, **kw):
- if stuff is None:
- pass
- elif hasattr(stuff, 'keys'):
- for k in stuff:
- me[k] = stuff[k]
- else:
- for k, v in stuff:
- me[k] = v
- for k, v in kw.iteritems():
- me[k] = kw[k]
- return me
-
-cdef class KeyIter:
- cdef Table _t
- cdef assoc_iter _i
- def __new__(me, Table t):
- me._t = t
- assoc_mkiter(&me._i, &t._t)
- def __iter__(me):
- return me
- def __next__(me):
- cdef entry *e
- e = <entry *>assoc_next(&me._i)
- if not e:
- raise StopIteration
- return _key(e)
-
-cdef class ValueIter:
- cdef Table _t
- cdef assoc_iter _i
- def __new__(me, Table t):
- me._t = t
- assoc_mkiter(&me._i, &t._t)
- def __iter__(me):
- return me
- def __next__(me):
- cdef entry *e
- e = <entry *>assoc_next(&me._i)
- if not e:
- raise StopIteration
- return _eget(e)
-
-cdef class ItemIter:
- cdef Table _t
- cdef assoc_iter _i
- def __new__(me, Table t):
- me._t = t
- assoc_mkiter(&me._i, &t._t)
- def __iter__(me):
- return me
- def __next__(me):
- cdef entry *e
- e = <entry *>assoc_next(&me._i)
- if not e:
- raise StopIteration
- return _key(e), _eget(e)
-
-Atom = atom_pystartup()
-
-#----- That's all, folks ----------------------------------------------------