Mercurial > public > mercurial-scm > hg-stable
diff mercurial/obsutil.py @ 33151:0a370b93cca2
obsutil: move 'allsuccessors' to the new modules
We have a new 'obsutil' module now. We move the high level utility there to bring
'obsolete.py' back to a more reasonable size.
| author | Pierre-Yves David <pierre-yves.david@octobus.net> |
|---|---|
| date | Tue, 27 Jun 2017 01:36:20 +0200 |
| parents | d0e5bf12f314 |
| children | 7017567ebdf2 |
line wrap: on
line diff
--- a/mercurial/obsutil.py Tue Jun 27 01:31:18 2017 +0200 +++ b/mercurial/obsutil.py Tue Jun 27 01:36:20 2017 +0200 @@ -57,6 +57,27 @@ seen.add(suc) remaining.add(suc) +def allsuccessors(obsstore, nodes, ignoreflags=0): + """Yield node for every successor of <nodes>. + + Some successors may be unknown locally. + + This is a linear yield unsuited to detecting split changesets. It includes + initial nodes too.""" + remaining = set(nodes) + seen = set(remaining) + while remaining: + current = remaining.pop() + yield current + for mark in obsstore.successors.get(current, ()): + # ignore marker flagged with specified flag + if mark[2] & ignoreflags: + continue + for suc in mark[1]: + if suc not in seen: + seen.add(suc) + remaining.add(suc) + def _filterprunes(markers): """return a set with no prune markers""" return set(m for m in markers if m[1])