← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/send.py

  • Committer: John Arbash Meinel
  • Date: 2010-02-10 17:52:08 UTC
  • mfrom: (5021 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5023.
  • Revision ID: john@arbash-meinel.com-20100210175208-bubuwav4uqigu291
Merge bzr.dev 5021 to resolve NEWS

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/send.py
1
 
# Copyright (C) 2009, 2010 Canonical Ltd
 
1
# Copyright (C) 2009 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
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
18
 
import os
19
18
import time
20
19
 
21
20
from bzrlib import (
22
 
    controldir,
 
21
    bzrdir,
23
22
    errors,
 
23
    merge_directive,
24
24
    osutils,
25
25
    registry,
26
26
    trace,
27
27
    )
28
 
from bzrlib.i18n import gettext
29
28
from bzrlib.branch import (
30
29
    Branch,
31
30
    )
40
39
def send(submit_branch, revision, public_branch, remember, format,
41
40
         no_bundle, no_patch, output, from_, mail_to, message, body,
42
41
         to_file, strict=None):
43
 
    tree, branch = controldir.ControlDir.open_containing_tree_or_branch(
44
 
        from_)[:2]
 
42
    tree, branch = bzrdir.BzrDir.open_containing_tree_or_branch(from_)[:2]
45
43
    # we may need to write data into branch's repository to calculate
46
44
    # the data to send.
47
45
    branch.lock_write()
53
51
            mail_client = config.get_mail_client()
54
52
            if (not getattr(mail_client, 'supports_body', False)
55
53
                and body is not None):
56
 
                raise errors.BzrCommandError(gettext(
57
 
                    'Mail client "%s" does not support specifying body') %
 
54
                raise errors.BzrCommandError(
 
55
                    'Mail client "%s" does not support specifying body' %
58
56
                    mail_client.__class__.__name__)
59
57
        if remember and submit_branch is None:
60
 
            raise errors.BzrCommandError(gettext(
61
 
                '--remember requires a branch to be specified.'))
 
58
            raise errors.BzrCommandError(
 
59
                '--remember requires a branch to be specified.')
62
60
        stored_submit_branch = branch.get_submit_branch()
63
61
        remembered_submit_branch = None
64
62
        if submit_branch is None:
65
63
            submit_branch = stored_submit_branch
66
64
            remembered_submit_branch = "submit"
67
65
        else:
68
 
            # Remembers if asked explicitly or no previous location is set
69
 
            if remember or (remember is None and stored_submit_branch is None):
 
66
            if stored_submit_branch is None or remember:
70
67
                branch.set_submit_branch(submit_branch)
71
68
        if submit_branch is None:
72
69
            submit_branch = branch.get_parent()
73
70
            remembered_submit_branch = "parent"
74
71
        if submit_branch is None:
75
 
            raise errors.BzrCommandError(gettext('No submit branch known or'
76
 
                                         ' specified'))
 
72
            raise errors.BzrCommandError('No submit branch known or'
 
73
                                         ' specified')
77
74
        if remembered_submit_branch is not None:
78
 
            trace.note(gettext('Using saved {0} location "{1}" to determine '
79
 
                       'what changes to submit.').format(
80
 
                                    remembered_submit_branch, submit_branch))
 
75
            trace.note('Using saved %s location "%s" to determine what '
 
76
                       'changes to submit.', remembered_submit_branch,
 
77
                       submit_branch)
81
78
 
82
79
        if mail_to is None or format is None:
83
80
            # TODO: jam 20090716 we open the submit_branch here, but we *don't*
92
89
                try:
93
90
                    format = format_registry.get(formatname)
94
91
                except KeyError:
95
 
                    raise errors.BzrCommandError(gettext("No such send format '%s'.") % 
 
92
                    raise errors.BzrCommandError("No such send format '%s'." % 
96
93
                                                 formatname)
97
94
 
98
95
        stored_public_branch = branch.get_public_branch()
99
96
        if public_branch is None:
100
97
            public_branch = stored_public_branch
101
 
        # Remembers if asked explicitly or no previous location is set
102
 
        elif (remember
103
 
              or (remember is None and stored_public_branch is None)):
 
98
        elif stored_public_branch is None or remember:
104
99
            branch.set_public_branch(public_branch)
105
100
        if no_bundle and public_branch is None:
106
 
            raise errors.BzrCommandError(gettext('No public branch specified or'
107
 
                                         ' known'))
 
101
            raise errors.BzrCommandError('No public branch specified or'
 
102
                                         ' known')
108
103
        base_revision_id = None
109
104
        revision_id = None
110
105
        if revision is not None:
111
106
            if len(revision) > 2:
112
 
                raise errors.BzrCommandError(gettext('bzr send takes '
113
 
                    'at most two one revision identifiers'))
 
107
                raise errors.BzrCommandError('bzr send takes '
 
108
                    'at most two one revision identifiers')
114
109
            revision_id = revision[-1].as_revision_id(branch)
115
110
            if len(revision) == 2:
116
111
                base_revision_id = revision[0].as_revision_id(branch)
117
112
        if revision_id is None:
118
 
            if tree is not None:
119
 
                tree.check_changed_or_out_of_date(
120
 
                    strict, 'send_strict',
121
 
                    more_error='Use --no-strict to force the send.',
122
 
                    more_warning='Uncommitted changes will not be sent.')
 
113
            if strict is None:
 
114
                strict = branch.get_config(
 
115
                    ).get_user_option_as_bool('send_strict')
 
116
            if strict is None: strict = True # default value
 
117
            if strict and tree is not None:
 
118
                if (tree.has_changes()):
 
119
                    raise errors.UncommittedChanges(
 
120
                        tree, more='Use --no-strict to force the send.')
 
121
                if tree.last_revision() != tree.branch.last_revision():
 
122
                    # The tree has lost sync with its branch, there is little
 
123
                    # chance that the user is aware of it but he can still force
 
124
                    # the send with --no-strict
 
125
                    raise errors.OutOfDateTree(
 
126
                        tree, more='Use --no-strict to force the send.')
123
127
            revision_id = branch.last_revision()
124
128
        if revision_id == NULL_REVISION:
125
 
            raise errors.BzrCommandError(gettext('No revisions to submit.'))
 
129
            raise errors.BzrCommandError('No revisions to submit.')
126
130
        if format is None:
127
131
            format = format_registry.get()
128
132
        directive = format(branch, revision_id, submit_branch,
131
135
            directive.compose_merge_request(mail_client, mail_to, body,
132
136
                                            branch, tree)
133
137
        else:
134
 
            if directive.multiple_output_files:
135
 
                if output == '-':
136
 
                    raise errors.BzrCommandError(gettext('- not supported for '
137
 
                        'merge directives that use more than one output file.'))
138
 
                if not os.path.exists(output):
139
 
                    os.mkdir(output, 0755)
140
 
                for (filename, lines) in directive.to_files():
141
 
                    path = os.path.join(output, filename)
142
 
                    outfile = open(path, 'wb')
143
 
                    try:
144
 
                        outfile.writelines(lines)
145
 
                    finally:
146
 
                        outfile.close()
 
138
            if output == '-':
 
139
                outfile = to_file
147
140
            else:
148
 
                if output == '-':
149
 
                    outfile = to_file
150
 
                else:
151
 
                    outfile = open(output, 'wb')
152
 
                try:
153
 
                    outfile.writelines(directive.to_lines())
154
 
                finally:
155
 
                    if outfile is not to_file:
156
 
                        outfile.close()
 
141
                outfile = open(output, 'wb')
 
142
            try:
 
143
                outfile.writelines(directive.to_lines())
 
144
            finally:
 
145
                if outfile is not to_file:
 
146
                    outfile.close()
157
147
    finally:
158
148
        branch.unlock()
159
149
 
160
150
 
161
151
def _send_4(branch, revision_id, submit_branch, public_branch,
162
152
            no_patch, no_bundle, message, base_revision_id):
163
 
    from bzrlib import merge_directive
164
153
    return merge_directive.MergeDirective2.from_objects(
165
154
        branch.repository, revision_id, time.time(),
166
155
        osutils.local_time_offset(), submit_branch,
175
164
        if not no_patch:
176
165
            patch_type = 'bundle'
177
166
        else:
178
 
            raise errors.BzrCommandError(gettext('Format 0.9 does not'
179
 
                ' permit bundle with no patch'))
 
167
            raise errors.BzrCommandError('Format 0.9 does not'
 
168
                ' permit bundle with no patch')
180
169
    else:
181
170
        if not no_patch:
182
171
            patch_type = 'diff'
183
172
        else:
184
173
            patch_type = None
185
 
    from bzrlib import merge_directive
186
174
    return merge_directive.MergeDirective.from_objects(
187
175
        branch.repository, revision_id, time.time(),
188
176
        osutils.local_time_offset(), submit_branch,