Files
@ cf78d302d441
Branch filter:
Location: kallithea/rhodecode/lib/oset.py - annotation
cf78d302d441
1.8 KiB
text/x-python
#47 implemented deleting of empty groups. Fixed problem with full paths on nested groups
37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 37625d304a16 | KEY, PREV, NEXT = range(3)
import collections
class OrderedSet(collections.MutableSet):
def __init__(self, iterable=None):
self.end = end = []
end += [None, end, end] # sentinel node for doubly linked list
self.map = {} # key --> [key, prev, next]
if iterable is not None:
self |= iterable
def __len__(self):
return len(self.map)
def __contains__(self, key):
return key in self.map
def add(self, key):
if key not in self.map:
end = self.end
curr = end[PREV]
curr[NEXT] = end[PREV] = self.map[key] = [key, curr, end]
def discard(self, key):
if key in self.map:
key, prev, next = self.map.pop(key)
prev[NEXT] = next
next[PREV] = prev
def __iter__(self):
end = self.end
curr = end[NEXT]
while curr is not end:
yield curr[KEY]
curr = curr[NEXT]
def __reversed__(self):
end = self.end
curr = end[PREV]
while curr is not end:
yield curr[KEY]
curr = curr[PREV]
def pop(self, last=True):
if not self:
raise KeyError('set is empty')
key = next(reversed(self)) if last else next(iter(self))
self.discard(key)
return key
def __repr__(self):
if not self:
return '%s()' % (self.__class__.__name__,)
return '%s(%r)' % (self.__class__.__name__, list(self))
def __eq__(self, other):
if isinstance(other, OrderedSet):
return len(self) == len(other) and list(self) == list(other)
return set(self) == set(other)
def __del__(self):
self.clear() # remove circular references
|