comparison mercurial/localrepo.py @ 10772:1e819576e926 stable

push: refactor and optimize prepush

10771:01f097c4ae66

        inc = self.findincoming(remote, common, remote_heads, force=force)

        update, updated_heads = self.findoutgoing(remote, common, remote_heads)
        msng_cl, bases, heads = self.changelog.nodesbetween(update, revs)

        def checkbranch(lheads, rheads, updatelb, branchname=None):
            '''
            check whether there are more local heads than remote heads on
            a specific branch.

            lheads: local branch heads
            rheads: remote branch heads
            updatelb: outgoing local branch bases
            '''

            warn = 0

            if not revs and len(lheads) > len(rheads):
                warn = 1
            else:
                # add local heads involved in the push
                updatelheads = [self.changelog.heads(x, lheads)
                                for x in updatelb]
                newheads = set(sum(updatelheads, [])) & set(lheads)

                if not newheads:
                    return True

                # add heads we don't have or that are not involved in the push
                for r in rheads:
                    if r in self.changelog.nodemap:
                        desc = self.changelog.heads(r, heads)
                        l = [h for h in heads if h in desc]
                        if not l:

                        self.ui.status(_("(use 'hg push -f' to force)\n"))
                        return None, 0
                    for branch, lheads in localbrheads.iteritems():
                        if branch in remotebrheads:
                            rheads = remotebrheads[branch]
                            if not checkbranch(lheads, rheads, update, branch):
                                return None, 0
                else:
                    if not checkbranch(heads, remote_heads, update):
                        return None, 0

            if inc:
                self.ui.warn(_("note: unsynced remote changes!\n"))

10772:1e819576e926

        inc = self.findincoming(remote, common, remote_heads, force=force)

        update, updated_heads = self.findoutgoing(remote, common, remote_heads)
        msng_cl, bases, heads = self.changelog.nodesbetween(update, revs)

        def checkbranch(lheads, rheads, branchname=None):
            '''
            check whether there are more local heads than remote heads on
            a specific branch.

            lheads: local branch heads
            rheads: remote branch heads
            '''

            warn = 0

            if len(lheads) > len(rheads):
                warn = 1
            else:
                # add heads we don't have or that are not involved in the push
                newheads = set(lheads)
                for r in rheads:
                    if r in self.changelog.nodemap:
                        desc = self.changelog.heads(r, heads)
                        l = [h for h in heads if h in desc]
                        if not l:

                        self.ui.status(_("(use 'hg push -f' to force)\n"))
                        return None, 0
                    for branch, lheads in localbrheads.iteritems():
                        if branch in remotebrheads:
                            rheads = remotebrheads[branch]
                            if not checkbranch(lheads, rheads, branch):
                                return None, 0
                else:
                    if not checkbranch(heads, remote_heads):
                        return None, 0

            if inc:
                self.ui.warn(_("note: unsynced remote changes!\n"))