comparison hgext3rd/topic/stack.py @ 2916:17749d9d3968

stack: move data computation on the object

We gather multiple data about the stack on the object. Let us move this on the
object and keep it cached there.

2915:b3abdb3d819e

    def index(self, item):
        return self.revs.index(item)

    @util.propertycache
    def revs(self):
        revs = _orderrevs(self._repo, self._revs)
        if revs:
            pt1 = self._repo[revs[0]].p1()
            if pt1.obsolete():
                pt1 = self._repo[_singlesuccessor(self._repo, pt1)]
            revs.insert(0, pt1.rev())
        return revs

def labelsgen(prefix, labelssuffix):
    """ Takes a label prefix and a list of suffixes. Returns a string of the prefix
    formatted with each suffix separated with a space.
    """

    :troubledcount: number on troubled changesets
    :headcount: number of heads on the topic
    :behindcount: number of changeset on rebase destination
    """
    data = {}
    revs = stack(repo, branch, topic)[1:]
    data['changesetcount'] = len(revs)
    data['troubledcount'] = len([r for r in revs if repo[r].isunstable()])
    deps, rdeps = builddependencies(repo, revs)
    data['headcount'] = len([r for r in revs if not rdeps[r]])
    data['behindcount'] = 0
    if revs:
        minroot = [min(r for r in revs if not deps[r])]
        try:
            dest = destutil.destmerge(repo, action='rebase',
                                      sourceset=minroot,
                                      onheadcheck=False)
            data['behindcount'] = len(repo.revs("only(%d, %ld)", dest,
                                                minroot))
        except error.NoMergeDestAbort:
            data['behindcount'] = 0
        except error.ManyMergeDestAbort:
            data['behindcount'] = -1
    data['branches'] = sorted(set(repo[r].branch() for r in revs))

    return data

2916:17749d9d3968

    def index(self, item):
        return self.revs.index(item)

    @util.propertycache
    def _dependencies(self):
        return builddependencies(self._repo, self.revs[1:])

    @util.propertycache
    def revs(self):
        revs = _orderrevs(self._repo, self._revs)
        if revs:
            pt1 = self._repo[revs[0]].p1()
            if pt1.obsolete():
                pt1 = self._repo[_singlesuccessor(self._repo, pt1)]
            revs.insert(0, pt1.rev())
        return revs

    @util.propertycache
    def changesetcount(self):
        return len(self._revs)

    @util.propertycache
    def troubledcount(self):
        return len([r for r in self._revs if self._repo[r].isunstable()])

    @util.propertycache
    def heads(self):
        revs = self.revs[1:]
        deps, rdeps = self._dependencies
        return [r for r in revs if not rdeps[r]]

    @util.propertycache
    def behindcount(self):
        revs = self.revs[1:]
        deps, rdeps = self._dependencies
        if revs:
            minroot = [min(r for r in revs if not deps[r])]
            try:
                dest = destutil.destmerge(self._repo, action='rebase',
                                          sourceset=minroot,
                                          onheadcheck=False)
                return len(self._repo.revs("only(%d, %ld)", dest, minroot))
            except error.NoMergeDestAbort:
                return 0
            except error.ManyMergeDestAbort:
                return -1
        return 0

    @util.propertycache
    def branches(self):
        return sorted(set(self._repo[r].branch() for r in self._revs))

def labelsgen(prefix, labelssuffix):
    """ Takes a label prefix and a list of suffixes. Returns a string of the prefix
    formatted with each suffix separated with a space.
    """

    :troubledcount: number on troubled changesets
    :headcount: number of heads on the topic
    :behindcount: number of changeset on rebase destination
    """
    data = {}
    current = stack(repo, branch, topic)
    data['changesetcount'] = current.changesetcount
    data['troubledcount'] = current.troubledcount
    data['headcount'] = len(current.heads)
    data['behindcount'] = current.behindcount
    data['branches'] = current.branches
    return data