← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/push.py

https://launchpad.net/bugs/524184
NEWS section template into a separate file

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/push.py
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
"""UI helper for the push command."""
18
18
 
19
 
from bzrlib import builtins, bzrdir, errors, transport
20
 
from bzrlib.trace import note, warning
 
19
from bzrlib import (
 
20
    builtins,
 
21
    branch,
 
22
    bzrdir,
 
23
    errors,
 
24
    revision as _mod_revision,
 
25
    transport,
 
26
    )
 
27
from bzrlib.trace import (
 
28
    note,
 
29
    warning,
 
30
    )
 
31
 
 
32
 
 
33
class PushResult(object):
 
34
    """Result of a push operation.
 
35
 
 
36
    :ivar branch_push_result: Result of a push between branches
 
37
    :ivar target_branch: The target branch
 
38
    :ivar stacked_on: URL of the branch on which the result is stacked
 
39
    :ivar workingtree_updated: Whether or not the target workingtree was updated.
 
40
    """
 
41
 
 
42
    def __init__(self):
 
43
        self.branch_push_result = None
 
44
        self.stacked_on = None
 
45
        self.workingtree_updated = None
 
46
        self.target_branch = None
 
47
 
 
48
    def report(self, to_file):
 
49
        """Write a human-readable description of the result."""
 
50
        if self.branch_push_result is None:
 
51
            if self.stacked_on is not None:
 
52
                note('Created new stacked branch referring to %s.' %
 
53
                    self.stacked_on)
 
54
            else:
 
55
                note('Created new branch.')
 
56
        else:
 
57
            self.branch_push_result.report(to_file)
21
58
 
22
59
 
23
60
def _show_push_branch(br_from, revision_id, location, to_file, verbose=False,
24
 
    overwrite=False, remember=False, reference=None, create_prefix=False,
 
61
    overwrite=False, remember=False, stacked_on=None, create_prefix=False,
25
62
    use_existing_dir=False):
26
63
    """Push a branch to a location.
27
64
 
34
71
        have diverged from the source, otherwise the push fails
35
72
    :param remember: if True, store the location as the push location for
36
73
        the source branch
37
 
    :param reference: the url of the branch, if any, to stack on;
 
74
    :param stacked_on: the url of the branch, if any, to stack on;
38
75
        if set, only the revisions not in that branch are pushed
39
76
    :param create_prefix: if True, create the necessary parent directories
40
77
        at the destination if they don't already exist
42
79
        directory exists without a current .bzr directory in it
43
80
    """
44
81
    to_transport = transport.get_transport(location)
45
 
    br_to = repository_to = dir_to = None
46
82
    try:
47
83
        dir_to = bzrdir.BzrDir.open_from_transport(to_transport)
48
84
    except errors.NotBranchError:
49
 
        pass # Didn't find anything
50
 
    else:
51
 
        # If we can open a branch, use its direct repository, otherwise see
52
 
        # if there is a repository without a branch.
53
 
        try:
54
 
            br_to = dir_to.open_branch()
55
 
        except errors.NotBranchError:
56
 
            # Didn't find a branch, can we find a repository?
57
 
            try:
58
 
                repository_to = dir_to.find_repository()
59
 
            except errors.NoRepositoryPresent:
60
 
                pass
61
 
        else:
62
 
            # Found a branch, so we must have found a repository
63
 
            repository_to = br_to.repository
 
85
        # Didn't find anything
 
86
        dir_to = None
64
87
 
65
 
    push_result = None
66
 
    if verbose:
67
 
        old_rh = []
68
88
    if dir_to is None:
69
 
        # The destination doesn't exist; create it.
70
 
        # XXX: Refactor the create_prefix/no_create_prefix code into a
71
 
        #      common helper function
72
 
 
73
 
        def make_directory(transport):
74
 
            transport.mkdir('.')
75
 
            return transport
76
 
 
77
 
        def redirected(redirected_transport, e, redirection_notice):
78
 
            return transport.get_transport(e.get_target_url())
79
 
 
80
89
        try:
81
 
            to_transport = transport.do_catching_redirections(
82
 
                make_directory, to_transport, redirected)
83
 
        except errors.FileExists:
 
90
            br_to = br_from.create_clone_on_transport(to_transport,
 
91
                revision_id=revision_id, stacked_on=stacked_on,
 
92
                create_prefix=create_prefix, use_existing_dir=use_existing_dir)
 
93
        except errors.FileExists, err:
 
94
            if err.path.endswith('/.bzr'):
 
95
                raise errors.BzrCommandError(
 
96
                    "Target directory %s already contains a .bzr directory, "
 
97
                    "but it is not valid." % (location,))
84
98
            if not use_existing_dir:
85
99
                raise errors.BzrCommandError("Target directory %s"
86
 
                     " already exists, but does not have a valid .bzr"
 
100
                     " already exists, but does not have a .bzr"
87
101
                     " directory. Supply --use-existing-dir to push"
88
102
                     " there anyway." % location)
 
103
            # This shouldn't occur, but if it does the FileExists error will be
 
104
            # more informative than an UnboundLocalError for br_to.
 
105
            raise
89
106
        except errors.NoSuchFile:
90
107
            if not create_prefix:
91
108
                raise errors.BzrCommandError("Parent directory of %s"
93
110
                    "\nYou may supply --create-prefix to create all"
94
111
                    " leading parent directories."
95
112
                    % location)
96
 
            builtins._create_prefix(to_transport)
 
113
            # This shouldn't occur (because create_prefix is true, so
 
114
            # create_clone_on_transport should be catching NoSuchFile and
 
115
            # creating the missing directories) but if it does the original
 
116
            # NoSuchFile error will be more informative than an
 
117
            # UnboundLocalError for br_to.
 
118
            raise
97
119
        except errors.TooManyRedirections:
98
120
            raise errors.BzrCommandError("Too many redirections trying "
99
121
                                         "to make %s." % location)
100
 
 
101
 
        # Now the target directory exists, but doesn't have a .bzr
102
 
        # directory. So we need to create it, along with any work to create
103
 
        # all of the dependent branches, etc.
104
 
        if reference is not None:
105
 
            # This should be buried in the clone method itself. TODO.
106
 
            try:
107
 
                # if the from format is stackable, this will either work or
108
 
                # trigger NotStacked. If it's not, an error will be given to
109
 
                # the user.
110
 
                br_from.get_stacked_on()
111
 
            except errors.NotStacked:
112
 
                pass
113
 
            # now we need to sprout the repository,
114
 
            dir_to = br_from.bzrdir._format.initialize_on_transport(to_transport)
115
 
            br_from.repository._format.initialize(dir_to)
116
 
            br_to = br_from._format.initialize(dir_to)
117
 
            br_to.set_stacked_on(reference)
118
 
            # and copy the data up.
119
 
            br_from.push(br_to)
120
 
        else:
121
 
            dir_to = br_from.bzrdir.clone_on_transport(to_transport,
122
 
                revision_id=revision_id)
123
 
        br_to = dir_to.open_branch()
 
122
        push_result = PushResult()
124
123
        # TODO: Some more useful message about what was copied
125
 
        if reference is not None:
126
 
            note('Created new stacked branch referring to %s.' % reference)
127
 
        else:
128
 
            note('Created new branch.')
129
 
        # We successfully created the target, remember it
 
124
        try:
 
125
            push_result.stacked_on = br_to.get_stacked_on_url()
 
126
        except (errors.UnstackableBranchFormat,
 
127
                errors.UnstackableRepositoryFormat,
 
128
                errors.NotStacked):
 
129
            push_result.stacked_on = None
 
130
        push_result.target_branch = br_to
 
131
        push_result.old_revid = _mod_revision.NULL_REVISION
 
132
        push_result.old_revno = 0
130
133
        if br_from.get_push_location() is None or remember:
131
134
            br_from.set_push_location(br_to.base)
132
 
    elif repository_to is None:
133
 
        # we have a bzrdir but no branch or repository
134
 
        # XXX: Figure out what to do other than complain.
135
 
        raise errors.BzrCommandError("At %s you have a valid .bzr control"
136
 
            " directory, but not a branch or repository. This is an"
137
 
            " unsupported configuration. Please move the target directory"
138
 
            " out of the way and try again."
139
 
            % location)
140
 
    elif br_to is None:
141
 
        # We have a repository but no branch, copy the revisions, and then
142
 
        # create a branch.
143
 
        if reference is not None:
 
135
    else:
 
136
        if stacked_on is not None:
144
137
            warning("Ignoring request for a stacked branch as repository "
145
138
                    "already exists at the destination location.")
146
 
        repository_to.fetch(br_from.repository, revision_id=revision_id)
147
 
        br_to = br_from.clone(dir_to, revision_id=revision_id)
148
 
        note('Created new branch.')
149
 
        if br_from.get_push_location() is None or remember:
150
 
            br_from.set_push_location(br_to.base)
151
 
    else: # We have a valid to branch
152
 
        if reference is not None:
153
 
            warning("Ignoring request for a stacked branch as branch "
154
 
                    "already exists at the destination location.")
155
 
        # We were able to connect to the remote location, so remember it.
156
 
        # (We don't need to successfully push because of possible divergence.)
157
 
        if br_from.get_push_location() is None or remember:
158
 
            br_from.set_push_location(br_to.base)
159
 
        if verbose:
160
 
            old_rh = br_to.revision_history()
161
139
        try:
162
 
            try:
163
 
                tree_to = dir_to.open_workingtree()
164
 
            except errors.NotLocalUrl:
165
 
                warning("This transport does not update the working " 
166
 
                        "tree of: %s. See 'bzr help working-trees' for "
167
 
                        "more information." % br_to.base)
168
 
                push_result = br_from.push(br_to, overwrite,
169
 
                                           stop_revision=revision_id)
170
 
            except errors.NoWorkingTree:
171
 
                push_result = br_from.push(br_to, overwrite,
172
 
                                           stop_revision=revision_id)
173
 
            else:
174
 
                tree_to.lock_write()
175
 
                try:
176
 
                    push_result = br_from.push(tree_to.branch, overwrite,
177
 
                                               stop_revision=revision_id)
178
 
                    tree_to.update()
179
 
                finally:
180
 
                    tree_to.unlock()
 
140
            push_result = dir_to.push_branch(br_from, revision_id, overwrite, 
 
141
                remember, create_prefix)
181
142
        except errors.DivergedBranches:
182
143
            raise errors.BzrCommandError('These branches have diverged.'
183
 
                                    '  Try using "merge" and then "push".')
184
 
    if push_result is not None:
185
 
        push_result.report(to_file)
186
 
    elif verbose:
187
 
        new_rh = br_to.revision_history()
188
 
        if old_rh != new_rh:
189
 
            # Something changed
190
 
            from bzrlib.log import show_changed_revisions
191
 
            show_changed_revisions(br_to, old_rh, new_rh,
192
 
                                   to_file=to_file)
193
 
    else:
194
 
        # we probably did a clone rather than a push, so a message was
195
 
        # emitted above
196
 
        pass
 
144
                                    '  See "bzr help diverged-branches"'
 
145
                                    ' for more information.')
 
146
        except errors.NoRepositoryPresent:
 
147
            # we have a bzrdir but no branch or repository
 
148
            # XXX: Figure out what to do other than complain.
 
149
            raise errors.BzrCommandError("At %s you have a valid .bzr"
 
150
                " control directory, but not a branch or repository. This"
 
151
                " is an unsupported configuration. Please move the target"
 
152
                " directory out of the way and try again." % location)
 
153
        if push_result.workingtree_updated == False:
 
154
            warning("This transport does not update the working " 
 
155
                    "tree of: %s. See 'bzr help working-trees' for "
 
156
                    "more information." % push_result.target_branch.base)
 
157
    push_result.report(to_file)
 
158
    if verbose:
 
159
        br_to = push_result.target_branch
 
160
        br_to.lock_read()
 
161
        try:
 
162
            from bzrlib.log import show_branch_change
 
163
            show_branch_change(br_to, to_file, push_result.old_revno, 
 
164
                               push_result.old_revid)
 
165
        finally:
 
166
            br_to.unlock()
 
167
 
 
168