weakreflist.pyx

cdef inline object _get_object(object x):
    x = PyWeakref_GetObject(x)
    Py_XINCREF(x)
    return x

cdef inline object _get_ref(object x, object self):
    return PyWeakref_NewRef(x, self._remove)


cdef class WeakList(list):

    def __init__(self, items=None):
        items = items or []
        super(WeakList, self).__init__((_get_ref(x, self) for x in items))

    def __contains__(self, item):
        return super(WeakList, self).__contains__(_get_ref(item, self))

    def __getitem__(self, item):
        return _get_object(super(WeakList, self).__getitem__(item))

    def __getslice__(self, i, j):
        return [_get_object(x) for x in super(WeakList, self).__getslice__(i, j)] #slow?

    def __iter__(self):
        for x in super(WeakList, self).__iter__():
            yield _get_object(x)

    def __repr__(self):
        return "WeakList({!r})".format(list(self))

    def __reversed__(self, *args, **kwargs):
        for x in super(WeakList, self).__reversed__(*args, **kwargs):
            yield _get_object(x)

    def __setitem__(self, i, item):
        super(WeakList, self).__setitem__(i, _get_ref(item, self))

    def __setslice__(self, i, j, items):
        super(WeakList, self).__setslice__(i, j, (_get_ref(x, self) for x in items))

    cpdef _remove(self, object item):
        while super(WeakList, self).__contains__(item):
            super(WeakList, self).remove(item)

    def append(self, item):
        super(WeakList, self).append(_get_ref(item, self))

    def count(self, item):
        return super(WeakList, self).count(_get_ref(item, self))

    def extend(self, items):
        super(WeakList, self).extend((_get_ref(x, self) for x in items))

    def index(self, item):
        return super(WeakList, self).index(_get_ref(item, self))

    def insert(self, i, item):
        super(WeakList, self).insert(i, _get_ref(item, self))
        
    def pop(self, i=-1):
        return _get_object(super(WeakList, self).pop(i))

    def remove(self, item):
        super(WeakList, self).remove(_get_ref(item, self))