← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/send.py

  • Committer: Jelmer Vernooij
  • Date: 2012-02-20 14:15:25 UTC
  • mto: (6471.1.4 iter-child-entries)
  • mto: This revision was merged to the branch mainline in revision 6472.
  • Revision ID: jelmer@samba.org-20120220141525-9azkfei62st8yc7w
Use inventories directly in fewer places.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/send.py
1
 
# Copyright (C) 2009 Canonical Ltd
 
1
# Copyright (C) 2009, 2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
 
17
from __future__ import absolute_import
17
18
 
 
19
import os
18
20
import time
19
21
 
20
22
from bzrlib import (
21
 
    bzrdir,
 
23
    controldir,
22
24
    errors,
23
 
    merge_directive,
24
25
    osutils,
25
26
    registry,
26
27
    trace,
27
28
    )
 
29
from bzrlib.i18n import gettext
28
30
from bzrlib.branch import (
29
31
    Branch,
30
32
    )
36
38
format_registry = registry.Registry()
37
39
 
38
40
 
39
 
def send(submit_branch, revision, public_branch, remember, format,
40
 
         no_bundle, no_patch, output, from_, mail_to, message, body, 
41
 
         to_file):
42
 
    tree, branch = bzrdir.BzrDir.open_containing_tree_or_branch(from_)[:2]
 
41
def send(target_branch, revision, public_branch, remember,
 
42
         format, no_bundle, no_patch, output, from_, mail_to, message, body,
 
43
         to_file, strict=None):
 
44
    possible_transports = []
 
45
    tree, branch = controldir.ControlDir.open_containing_tree_or_branch(
 
46
        from_, possible_transports=possible_transports)[:2]
43
47
    # we may need to write data into branch's repository to calculate
44
48
    # the data to send.
45
49
    branch.lock_write()
46
50
    try:
47
51
        if output is None:
48
 
            config = branch.get_config()
 
52
            config_stack = branch.get_config_stack()
49
53
            if mail_to is None:
50
 
                mail_to = config.get_user_option('submit_to')
51
 
            mail_client = config.get_mail_client()
 
54
                mail_to = config_stack.get('submit_to')
 
55
            mail_client = config_stack.get('mail_client')(config_stack)
52
56
            if (not getattr(mail_client, 'supports_body', False)
53
57
                and body is not None):
54
 
                raise errors.BzrCommandError(
55
 
                    'Mail client "%s" does not support specifying body' %
 
58
                raise errors.BzrCommandError(gettext(
 
59
                    'Mail client "%s" does not support specifying body') %
56
60
                    mail_client.__class__.__name__)
57
 
        if remember and submit_branch is None:
58
 
            raise errors.BzrCommandError(
59
 
                '--remember requires a branch to be specified.')
60
 
        stored_submit_branch = branch.get_submit_branch()
61
 
        remembered_submit_branch = None
62
 
        if submit_branch is None:
63
 
            submit_branch = stored_submit_branch
64
 
            remembered_submit_branch = "submit"
 
61
        if remember and target_branch is None:
 
62
            raise errors.BzrCommandError(gettext(
 
63
                '--remember requires a branch to be specified.'))
 
64
        stored_target_branch = branch.get_submit_branch()
 
65
        remembered_target_branch = None
 
66
        if target_branch is None:
 
67
            target_branch = stored_target_branch
 
68
            remembered_target_branch = "submit"
65
69
        else:
66
 
            if stored_submit_branch is None or remember:
67
 
                branch.set_submit_branch(submit_branch)
68
 
        if submit_branch is None:
69
 
            submit_branch = branch.get_parent()
70
 
            remembered_submit_branch = "parent"
71
 
        if submit_branch is None:
72
 
            raise errors.BzrCommandError('No submit branch known or'
73
 
                                         ' specified')
74
 
        if remembered_submit_branch is not None:
75
 
            trace.note('Using saved %s location "%s" to determine what '
76
 
                       'changes to submit.', remembered_submit_branch,
77
 
                       submit_branch)
 
70
            # Remembers if asked explicitly or no previous location is set
 
71
            if remember or (
 
72
                    remember is None and stored_target_branch is None):
 
73
                branch.set_submit_branch(target_branch)
 
74
        if target_branch is None:
 
75
            target_branch = branch.get_parent()
 
76
            remembered_target_branch = "parent"
 
77
        if target_branch is None:
 
78
            raise errors.BzrCommandError(gettext('No submit branch known or'
 
79
                                         ' specified'))
 
80
        if remembered_target_branch is not None:
 
81
            trace.note(gettext('Using saved {0} location "{1}" to determine '
 
82
                       'what changes to submit.').format(
 
83
                                    remembered_target_branch,
 
84
                                    target_branch))
78
85
 
 
86
        submit_branch = Branch.open(target_branch,
 
87
            possible_transports=possible_transports)
 
88
        possible_transports.append(submit_branch.bzrdir.root_transport)
79
89
        if mail_to is None or format is None:
80
 
            submit_br = Branch.open(submit_branch)
81
 
            submit_config = submit_br.get_config()
82
90
            if mail_to is None:
83
 
                mail_to = submit_config.get_user_option("child_submit_to")
 
91
                mail_to = submit_branch.get_config_stack().get(
 
92
                    'child_submit_to')
84
93
            if format is None:
85
 
                formatname = submit_br.get_child_submit_format()
 
94
                formatname = submit_branch.get_child_submit_format()
86
95
                try:
87
96
                    format = format_registry.get(formatname)
88
97
                except KeyError:
89
 
                    raise errors.BzrCommandError("No such send format '%s'." % 
90
 
                                                 formatname)
 
98
                    raise errors.BzrCommandError(
 
99
                        gettext("No such send format '%s'.") % formatname)
91
100
 
92
101
        stored_public_branch = branch.get_public_branch()
93
102
        if public_branch is None:
94
103
            public_branch = stored_public_branch
95
 
        elif stored_public_branch is None or remember:
 
104
        # Remembers if asked explicitly or no previous location is set
 
105
        elif (remember
 
106
              or (remember is None and stored_public_branch is None)):
96
107
            branch.set_public_branch(public_branch)
97
108
        if no_bundle and public_branch is None:
98
 
            raise errors.BzrCommandError('No public branch specified or'
99
 
                                         ' known')
 
109
            raise errors.BzrCommandError(gettext('No public branch specified or'
 
110
                                         ' known'))
100
111
        base_revision_id = None
101
112
        revision_id = None
102
113
        if revision is not None:
103
114
            if len(revision) > 2:
104
 
                raise errors.BzrCommandError('bzr send takes '
105
 
                    'at most two one revision identifiers')
 
115
                raise errors.BzrCommandError(gettext('bzr send takes '
 
116
                    'at most two one revision identifiers'))
106
117
            revision_id = revision[-1].as_revision_id(branch)
107
118
            if len(revision) == 2:
108
119
                base_revision_id = revision[0].as_revision_id(branch)
109
120
        if revision_id is None:
 
121
            if tree is not None:
 
122
                tree.check_changed_or_out_of_date(
 
123
                    strict, 'send_strict',
 
124
                    more_error='Use --no-strict to force the send.',
 
125
                    more_warning='Uncommitted changes will not be sent.')
110
126
            revision_id = branch.last_revision()
111
127
        if revision_id == NULL_REVISION:
112
 
            raise errors.BzrCommandError('No revisions to submit.')
 
128
            raise errors.BzrCommandError(gettext('No revisions to submit.'))
113
129
        if format is None:
114
 
            # TODO: Query submit branch for its preferred format
115
130
            format = format_registry.get()
116
 
        directive = format(branch, revision_id, submit_branch, 
117
 
            public_branch, no_patch, no_bundle, message, base_revision_id)
 
131
        directive = format(branch, revision_id, target_branch,
 
132
            public_branch, no_patch, no_bundle, message, base_revision_id,
 
133
            submit_branch)
118
134
        if output is None:
119
135
            directive.compose_merge_request(mail_client, mail_to, body,
120
136
                                            branch, tree)
121
137
        else:
122
 
            if output == '-':
123
 
                outfile = to_file
 
138
            if directive.multiple_output_files:
 
139
                if output == '-':
 
140
                    raise errors.BzrCommandError(gettext('- not supported for '
 
141
                        'merge directives that use more than one output file.'))
 
142
                if not os.path.exists(output):
 
143
                    os.mkdir(output, 0755)
 
144
                for (filename, lines) in directive.to_files():
 
145
                    path = os.path.join(output, filename)
 
146
                    outfile = open(path, 'wb')
 
147
                    try:
 
148
                        outfile.writelines(lines)
 
149
                    finally:
 
150
                        outfile.close()
124
151
            else:
125
 
                outfile = open(output, 'wb')
126
 
            try:
127
 
                outfile.writelines(directive.to_lines())
128
 
            finally:
129
 
                if outfile is not to_file:
130
 
                    outfile.close()
 
152
                if output == '-':
 
153
                    outfile = to_file
 
154
                else:
 
155
                    outfile = open(output, 'wb')
 
156
                try:
 
157
                    outfile.writelines(directive.to_lines())
 
158
                finally:
 
159
                    if outfile is not to_file:
 
160
                        outfile.close()
131
161
    finally:
132
162
        branch.unlock()
133
163
 
134
164
 
135
 
def _send_4(branch, revision_id, submit_branch, public_branch,
136
 
            no_patch, no_bundle, message, base_revision_id):
 
165
def _send_4(branch, revision_id, target_branch, public_branch,
 
166
            no_patch, no_bundle, message,
 
167
            base_revision_id, local_target_branch=None):
 
168
    from bzrlib import merge_directive
137
169
    return merge_directive.MergeDirective2.from_objects(
138
170
        branch.repository, revision_id, time.time(),
139
 
        osutils.local_time_offset(), submit_branch,
140
 
        public_branch=public_branch, include_patch=not no_patch,
 
171
        osutils.local_time_offset(), target_branch,
 
172
        public_branch=public_branch,
 
173
        include_patch=not no_patch,
141
174
        include_bundle=not no_bundle, message=message,
142
 
        base_revision_id=base_revision_id)
 
175
        base_revision_id=base_revision_id,
 
176
        local_target_branch=local_target_branch)
143
177
 
144
178
 
145
179
def _send_0_9(branch, revision_id, submit_branch, public_branch,
146
 
              no_patch, no_bundle, message, base_revision_id):
 
180
              no_patch, no_bundle, message,
 
181
              base_revision_id, local_target_branch=None):
147
182
    if not no_bundle:
148
183
        if not no_patch:
149
184
            patch_type = 'bundle'
150
185
        else:
151
 
            raise errors.BzrCommandError('Format 0.9 does not'
152
 
                ' permit bundle with no patch')
 
186
            raise errors.BzrCommandError(gettext('Format 0.9 does not'
 
187
                ' permit bundle with no patch'))
153
188
    else:
154
189
        if not no_patch:
155
190
            patch_type = 'diff'
156
191
        else:
157
192
            patch_type = None
 
193
    from bzrlib import merge_directive
158
194
    return merge_directive.MergeDirective.from_objects(
159
195
        branch.repository, revision_id, time.time(),
160
196
        osutils.local_time_offset(), submit_branch,
161
197
        public_branch=public_branch, patch_type=patch_type,
162
 
        message=message)
163
 
 
164
 
 
165
 
format_registry.register('4', 
 
198
        message=message, local_target_branch=local_target_branch)
 
199
 
 
200
 
 
201
format_registry.register('4',
166
202
    _send_4, 'Bundle format 4, Merge Directive 2 (default)')
167
203
format_registry.register('0.9',
168
204
    _send_0_9, 'Bundle format 0.9, Merge Directive 1')