wassname/sloth
1
0
Fork 0
mirror of https://github.com/wassname/sloth.git synced 2026-07-05 11:09:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Support limiting the depth that the model iterator is allowed to descend

Browse Source
This commit is contained in:
Mika Fischer
2011-07-13 17:29:09 +02:00
parent d2e582c383
commit 94a4ddcbb1
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sloth/annotations/model.py
+6 -2
View File
@@ -685,8 +685,9 @@ class AnnotationModel(QAbstractItemModel):
return index.internalPointer()
return self._root
def iterator(self, _class=None, predicate=None, start=None):
def iterator(self, _class=None, predicate=None, start=None, maxlevels=10000):
# Visit all nodes
level = 0
item = start if start is not None else self.root()
while item is not None:
# Return item
@@ -695,13 +696,15 @@ class AnnotationModel(QAbstractItemModel):
yield item
# Get next item
if item.rowCount() > 0:
if item.rowCount() > 0 and level < maxlevels:
level += 1
item = item.childAt(0)
else:
next_sibling = item.getNextSibling()
if next_sibling is not None:
item = next_sibling
else:
level -= 1
ancestor = item.parent()
item = None
while ancestor is not None:
@@ -709,6 +712,7 @@ class AnnotationModel(QAbstractItemModel):
if ancestor_sibling is not None:
item = ancestor_sibling
break
level -= 1
ancestor = ancestor.parent()