← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to cbranch.py

  • Committer: Robert Collins
  • Date: 2005-09-28 05:43:19 UTC
  • mto: (147.2.6) (364.1.3 bzrtools)
  • mto: This revision was merged to the branch mainline in revision 324.
  • Revision ID: robertc@robertcollins.net-20050928054319-2c2e9e3048bbc215
find_branch -> open_containing change

Show diffs side-by-side

added added

removed removed

Lines of Context:
cbranch.py
1
 
# Copyright (C) 2006, 2008 Aaron Bentley
2
 
# <aaron@aaronbentley.com>
3
 
#
4
 
#    This program is free software; you can redistribute it and/or modify
5
 
#    it under the terms of the GNU General Public License as published by
6
 
#    the Free Software Foundation; either version 2 of the License, or
7
 
#    (at your option) any later version.
8
 
#
9
 
#    This program is distributed in the hope that it will be useful,
10
 
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
11
 
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12
 
#    GNU General Public License for more details.
13
 
#
14
 
#    You should have received a copy of the GNU General Public License
15
 
#    along with this program; if not, write to the Free Software
16
 
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
 
 
18
 
from bzrlib import ui
19
 
from bzrlib.bzrdir import BzrDir
20
 
from bzrlib.branch import Branch
21
 
from bzrlib.config import LocationConfig
22
 
from bzrlib.errors import BzrCommandError, NoSuchFile
23
 
from bzrlib.osutils import pathjoin, basename, abspath
24
 
from bzrlib.transport import get_transport
25
 
from bzrlib.workingtree import WorkingTree
26
 
from bzrlib.urlutils import derive_to_location
27
 
 
28
 
def cbranch(from_location, to_location=None, revision=None,
29
 
            lightweight=False, files_from=None, hardlink=False):
30
 
    if to_location is None:
31
 
        to_location = derive_to_location(from_location)
32
 
    config = LocationConfig(abspath(to_location))
33
 
    b_loc = config.get_user_option("cbranch_target")
34
 
    if b_loc is None:
35
 
        b_root = config.get_user_option("cbranch_root")
36
 
        if b_root is None:
37
 
            raise BzrCommandError("Can't find cbranch_target in"
38
 
                                  " locations.conf")
39
 
        b_loc = pathjoin(b_root, basename(to_location))
40
 
    accelerator_tree, old_branch = BzrDir.open_tree_or_branch(from_location)
41
 
    if files_from is not None:
42
 
        accelerator_tree = WorkingTree.open(files_from)
43
 
    if revision is None or len(revision) == 0:
44
 
        revision_id = old_branch.last_revision()
45
 
    elif len(revision) == 1:
46
 
        revision_id = revision[0].in_history(old_branch)[1]
47
 
    else:
48
 
        raise BzrCommandError('At most one revision may be supplied.')
49
 
    b_transport = get_transport(b_loc)
50
 
    ensure_base_recursive(b_transport.clone('..'))
51
 
    pb = ui.ui_factory.nested_progress_bar()
52
 
    try:
53
 
        pb.update('Creating branch', 0, 2)
54
 
        new_branch = old_branch.bzrdir.sprout(b_loc, revision_id,
55
 
            accelerator_tree=accelerator_tree).open_branch()
56
 
        pb.update('Creating checkout', 1, 2)
57
 
        new_branch.create_checkout(to_location, lightweight=lightweight,
58
 
                                   accelerator_tree=accelerator_tree,
59
 
                                   hardlink=hardlink)
60
 
    finally:
61
 
        pb.finished()
62
 
 
63
 
 
64
 
def ensure_base_recursive(transport):
65
 
    """Ensure that the transport base and any its parents exist"""
66
 
    pending_transports = [transport]
67
 
    while len(pending_transports) > 0:
68
 
        transport = pending_transports.pop()
69
 
        try:
70
 
            transport.ensure_base()
71
 
        except NoSuchFile, e:
72
 
            pending_transports.append(transport)
73
 
            parent = transport.clone('..')
74
 
            if parent.base == transport.base:
75
 
                raise e
76
 
            pending_transports.append(parent)