~abentley/bzrtools/bzrtools.dev

# Copyright (C) 2005 by Aaron Bentley

# This program is free software; you can redistribute it and/or modify

# it under the terms of the GNU General Public License as published by

# the Free Software Foundation; either version 2 of the License, or

# (at your option) any later version.

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU General Public License for more details.

# You should have received a copy of the GNU General Public License

# along with this program; if not, write to the Free Software

# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

from bzrlib.branch import find_branch

from bzrlib.fetch import greedy_fetch

from bzrlib.errors import NoSuchRevision, InstallFailed

def fetch_missing(branch):

    """Install the revisions of missing ancestors from another branch."""

    this_branch = find_branch('.')

    if branch is None:

        try:

            branch = this_branch.controlfile("x-pull", "rb").read().rstrip('\n')

            print "Using last location: %s" % branch

        except IOError, e:

            if e.errno != errno.ENOENT:

                raise

            else:

                raise BzrCommandError("No pull location known or specified.")

    other_branch = find_branch(branch)

    installed = []

    failed = []

    # Because iter_missing_ancestors tests for existence after our last fetch

    # is complete, it won't falsely report an ancestor as missing.

    # Yay iterators!

    missing = iter_missing_ancestors(this_branch)

    for revision in missing:

        try:

            greedy_fetch(this_branch, other_branch, revision)

            installed.append(revision)

        except InstallFailed:

            failed.append(revision)

    if len(installed) > 0:

        print "Installed:"

    for rev in installed:

        print rev

    if len(failed) > 0:

        print "Still missing:"

    for rev in failed:

        print rev

def iter_missing_ancestors(branch):

    """Find all ancestors that aren't stored in this branch."""

    seen = set()

    lines = [branch.last_patch()]

    while len(lines) > 0:

        new_lines = []

        for line in lines:

            if line in seen:

                continue

            seen.add(line)

            try:

                revision = branch.get_revision(line)

                new_lines.extend([r.revision_id for r in revision.parents])

            except NoSuchRevision:

                yield line

        lines = new_lines