← Back to branch summary

~abentley/bzrtools/bzrtools.dev

« back to all changes in this revision

Viewing changes to cbranch.py

  • Committer: Aaron Bentley
  • Date: 2007-01-08 17:27:48 UTC
  • Revision ID: abentley@panoramicfeedback.com-20070108172748-1b22qtszaadoby89
Improve bzr import docs

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>
 
1
# Copyright (C) 2006 Aaron Bentley
 
2
# <aaron.bentley@utoronto.ca>
3
3
#
4
4
#    This program is free software; you can redistribute it and/or modify
5
5
#    it under the terms of the GNU General Public License as published by
15
15
#    along with this program; if not, write to the Free Software
16
16
#    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
17
 
18
 
from bzrlib import ui
19
 
from bzrlib.bzrdir import BzrDir
20
18
from bzrlib.branch import Branch
 
19
from bzrlib.builtins import cmd_branch, cmd_checkout
21
20
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
 
21
from bzrlib.errors import BzrCommandError
 
22
from bzrlib.osutils import pathjoin, basename, abspath, getcwd
 
23
from bzrlib.urlutils import local_path_to_url
27
24
 
28
25
def cbranch(from_location, to_location=None, revision=None,
29
 
            lightweight=False, files_from=None, hardlink=False):
 
26
            lightweight=False):
 
27
    from_location = from_location.rstrip('/')
30
28
    if to_location is None:
31
 
        to_location = derive_to_location(from_location)
 
29
        to_location = pathjoin(getcwd(), basename(from_location))
32
30
    config = LocationConfig(abspath(to_location))
33
31
    b_loc = config.get_user_option("cbranch_target")
34
32
    if b_loc is None:
37
35
            raise BzrCommandError("Can't find cbranch_target in"
38
36
                                  " locations.conf")
39
37
        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)
 
38
    cmd_branch().run(from_location, b_loc, revision=revision)
 
39
    cmd_checkout().run(b_loc, to_location, lightweight=lightweight)
 
40
    b = Branch.open(to_location)