~bzr-pqm/bzr/bzr.dev

4797.19.2 by John Arbash Meinel
bring in the latest 2.1 changes
1
# Copyright (C) 2008, 2009, 2010 Canonical Ltd
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
2
#
3
# This program is free software; you can redistribute it and/or modify
4
# it under the terms of the GNU General Public License as published by
5
# the Free Software Foundation; either version 2 of the License, or
6
# (at your option) any later version.
7
#
8
# This program is distributed in the hope that it will be useful,
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
# GNU General Public License for more details.
12
#
13
# You should have received a copy of the GNU General Public License
14
# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob
update FSF mailing address
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
16
17
"""UI helper for the push command."""
18
3978.3.13 by Jelmer Vernooij
Store the pushed branches' URL rather than the URL of the branch itself.
19
from bzrlib import (
20
    bzrdir,
21
    errors,
22
    revision as _mod_revision,
23
    transport,
24
    )
25
from bzrlib.trace import (
26
    note,
27
    warning,
28
    )
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
29
30
4053.3.2 by Jelmer Vernooij
Use PushResult in push.py.
31
class PushResult(object):
32
    """Result of a push operation.
33
34
    :ivar branch_push_result: Result of a push between branches
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
35
    :ivar target_branch: The target branch
4053.3.2 by Jelmer Vernooij
Use PushResult in push.py.
36
    :ivar stacked_on: URL of the branch on which the result is stacked
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
37
    :ivar workingtree_updated: Whether or not the target workingtree was updated.
4053.3.2 by Jelmer Vernooij
Use PushResult in push.py.
38
    """
39
40
    def __init__(self):
41
        self.branch_push_result = None
42
        self.stacked_on = None
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
43
        self.workingtree_updated = None
44
        self.target_branch = None
4053.3.2 by Jelmer Vernooij
Use PushResult in push.py.
45
46
    def report(self, to_file):
47
        """Write a human-readable description of the result."""
48
        if self.branch_push_result is None:
49
            if self.stacked_on is not None:
50
                note('Created new stacked branch referring to %s.' %
51
                    self.stacked_on)
52
            else:
53
                note('Created new branch.')
54
        else:
55
            self.branch_push_result.report(to_file)
56
57
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
58
def _show_push_branch(br_from, revision_id, location, to_file, verbose=False,
3549.1.1 by Martin Pool
rename push --reference to --stacked-on
59
    overwrite=False, remember=False, stacked_on=None, create_prefix=False,
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
60
    use_existing_dir=False):
61
    """Push a branch to a location.
62
63
    :param br_from: the source branch
64
    :param revision_id: the revision-id to push up to
65
    :param location: the url of the destination
66
    :param to_file: the output stream
67
    :param verbose: if True, display more output than normal
68
    :param overwrite: if False, a current branch at the destination may not
69
        have diverged from the source, otherwise the push fails
70
    :param remember: if True, store the location as the push location for
71
        the source branch
3549.1.1 by Martin Pool
rename push --reference to --stacked-on
72
    :param stacked_on: the url of the branch, if any, to stack on;
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
73
        if set, only the revisions not in that branch are pushed
74
    :param create_prefix: if True, create the necessary parent directories
75
        at the destination if they don't already exist
76
    :param use_existing_dir: if True, proceed even if the destination
77
        directory exists without a current .bzr directory in it
78
    """
79
    to_transport = transport.get_transport(location)
80
    try:
81
        dir_to = bzrdir.BzrDir.open_from_transport(to_transport)
82
    except errors.NotBranchError:
4294.2.1 by Robert Collins
Move directory checking for bzr push options into Branch.create_clone_on_transport.
83
        # Didn't find anything
84
        dir_to = None
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
85
86
    if dir_to is None:
4294.2.6 by Robert Collins
Report errors back in the UI layer for push, to use the url the user gave us.
87
        try:
88
            br_to = br_from.create_clone_on_transport(to_transport,
89
                revision_id=revision_id, stacked_on=stacked_on,
90
                create_prefix=create_prefix, use_existing_dir=use_existing_dir)
4634.105.2 by Andrew Bennetts
Clearer code and error message.
91
        except errors.FileExists, err:
92
            if err.path.endswith('/.bzr'):
93
                raise errors.BzrCommandError(
94
                    "Target directory %s already contains a .bzr directory, "
95
                    "but it is not valid." % (location,))
4294.2.6 by Robert Collins
Report errors back in the UI layer for push, to use the url the user gave us.
96
            if not use_existing_dir:
97
                raise errors.BzrCommandError("Target directory %s"
4634.105.1 by Andrew Bennetts
Fix traceback when doing 'bzr push --use-existing-dir' into a dir with an invalid .bzr directory.
98
                     " already exists, but does not have a .bzr"
4294.2.6 by Robert Collins
Report errors back in the UI layer for push, to use the url the user gave us.
99
                     " directory. Supply --use-existing-dir to push"
100
                     " there anyway." % location)
4634.105.2 by Andrew Bennetts
Clearer code and error message.
101
            # This shouldn't occur, but if it does the FileExists error will be
102
            # more informative than an UnboundLocalError for br_to.
103
            raise
4294.2.6 by Robert Collins
Report errors back in the UI layer for push, to use the url the user gave us.
104
        except errors.NoSuchFile:
105
            if not create_prefix:
106
                raise errors.BzrCommandError("Parent directory of %s"
107
                    " does not exist."
108
                    "\nYou may supply --create-prefix to create all"
109
                    " leading parent directories."
110
                    % location)
4634.129.2 by Andrew Bennetts
Add NEWS entry and elaborate comment, as suggested in Martin's review.
111
            # This shouldn't occur (because create_prefix is true, so
112
            # create_clone_on_transport should be catching NoSuchFile and
113
            # creating the missing directories) but if it does the original
114
            # NoSuchFile error will be more informative than an
115
            # UnboundLocalError for br_to.
4634.129.1 by Andrew Bennetts
Propagate the NoSuchFile error in show_push_branch when create_prefix is True, because a user reports that it can happen somehow.
116
            raise
4294.2.6 by Robert Collins
Report errors back in the UI layer for push, to use the url the user gave us.
117
        except errors.TooManyRedirections:
118
            raise errors.BzrCommandError("Too many redirections trying "
119
                                         "to make %s." % location)
4294.2.1 by Robert Collins
Move directory checking for bzr push options into Branch.create_clone_on_transport.
120
        push_result = PushResult()
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
121
        # TODO: Some more useful message about what was copied
3650.5.2 by Aaron Bentley
Always show stacked message if stacking done.
122
        try:
3978.3.3 by Jelmer Vernooij
Use PushResult everywhere.
123
            push_result.stacked_on = br_to.get_stacked_on_url()
3650.5.2 by Aaron Bentley
Always show stacked message if stacking done.
124
        except (errors.UnstackableBranchFormat,
125
                errors.UnstackableRepositoryFormat,
126
                errors.NotStacked):
3978.3.3 by Jelmer Vernooij
Use PushResult everywhere.
127
            push_result.stacked_on = None
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
128
        push_result.target_branch = br_to
3978.3.13 by Jelmer Vernooij
Store the pushed branches' URL rather than the URL of the branch itself.
129
        push_result.old_revid = _mod_revision.NULL_REVISION
130
        push_result.old_revno = 0
3978.3.4 by Jelmer Vernooij
Fix remaining tests; set push location of source branch, not target branch.
131
        if br_from.get_push_location() is None or remember:
3978.3.13 by Jelmer Vernooij
Store the pushed branches' URL rather than the URL of the branch itself.
132
            br_from.set_push_location(br_to.base)
3978.3.2 by Jelmer Vernooij
Move most of push to IterGenericBranchBzrDir.
133
    else:
3549.1.1 by Martin Pool
rename push --reference to --stacked-on
134
        if stacked_on is not None:
3221.19.4 by Ian Clatworthy
shallow -> stacked
135
            warning("Ignoring request for a stacked branch as repository "
3221.19.2 by Ian Clatworthy
tweaks to ui during review by igc
136
                    "already exists at the destination location.")
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
137
        try:
3978.3.15 by Jelmer Vernooij
Rename BzrDir.push() to BzrDir.push_branch().
138
            push_result = dir_to.push_branch(br_from, revision_id, overwrite, 
4416.8.1 by Jelmer Vernooij
Pass create_prefix paremeter to BzrDir.push_branch.
139
                remember, create_prefix)
3221.19.1 by Ian Clatworthy
refactor cmd_push to use a helper function
140
        except errors.DivergedBranches:
141
            raise errors.BzrCommandError('These branches have diverged.'
4416.4.1 by Neil Martinsen-Burrell
Fix 269477 Add documentation about diverged branches and refer to it when pushing fails
142
                                    '  See "bzr help diverged-branches"'
143
                                    ' for more information.')
4925.1.1 by Jelmer Vernooij
Print a proper error when attempting to push to a foreign VCS for which
144
        except errors.NoRoundtrippingSupport, e:
145
            raise errors.BzrCommandError("It is not possible to losslessly "
146
                "push to %s. You may want to use dpush instead." % 
147
                    e.target_branch.mapping.vcs.abbreviation)
3978.3.13 by Jelmer Vernooij
Store the pushed branches' URL rather than the URL of the branch itself.
148
        except errors.NoRepositoryPresent:
149
            # we have a bzrdir but no branch or repository
150
            # XXX: Figure out what to do other than complain.
151
            raise errors.BzrCommandError("At %s you have a valid .bzr"
152
                " control directory, but not a branch or repository. This"
153
                " is an unsupported configuration. Please move the target"
154
                " directory out of the way and try again." % location)
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
155
        if push_result.workingtree_updated == False:
3978.3.2 by Jelmer Vernooij
Move most of push to IterGenericBranchBzrDir.
156
            warning("This transport does not update the working " 
157
                    "tree of: %s. See 'bzr help working-trees' for "
158
                    "more information." % push_result.target_branch.base)
3978.3.3 by Jelmer Vernooij
Use PushResult everywhere.
159
    push_result.report(to_file)
3848.1.19 by Aaron Bentley
Show log for non-initial push -v
160
    if verbose:
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
161
        br_to = push_result.target_branch
3848.1.18 by Aaron Bentley
Update push to use show_branch_change
162
        br_to.lock_read()
163
        try:
164
            from bzrlib.log import show_branch_change
3978.3.13 by Jelmer Vernooij
Store the pushed branches' URL rather than the URL of the branch itself.
165
            show_branch_change(br_to, to_file, push_result.old_revno, 
166
                               push_result.old_revid)
3848.1.18 by Aaron Bentley
Update push to use show_branch_change
167
        finally:
168
            br_to.unlock()
3978.3.11 by Jelmer Vernooij
Move InterBranchBzrDir to bzrlib.push.
169
170