← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/send.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-01 08:02:42 UTC
  • mfrom: (5390.3.3 faster-revert-593560)
  • Revision ID: pqm@pqm.ubuntu.com-20100901080242-esg62ody4frwmy66
(spiv) Avoid repeatedly calling self.target.all_file_ids() in
 InterTree.iter_changes. (Andrew Bennetts)

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
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
 
 
18
import os
18
19
import time
19
20
 
20
21
from bzrlib import (
21
22
    bzrdir,
22
23
    errors,
23
 
    merge_directive,
24
24
    osutils,
25
25
    registry,
26
26
    trace,
77
77
                       submit_branch)
78
78
 
79
79
        if mail_to is None or format is None:
 
80
            # TODO: jam 20090716 we open the submit_branch here, but we *don't*
 
81
            #       pass it down into the format creation, so it will have to
 
82
            #       open it again
80
83
            submit_br = Branch.open(submit_branch)
81
84
            submit_config = submit_br.get_config()
82
85
            if mail_to is None:
107
110
            if len(revision) == 2:
108
111
                base_revision_id = revision[0].as_revision_id(branch)
109
112
        if revision_id is None:
110
 
            if strict is None:
111
 
                strict = branch.get_config(
112
 
                    ).get_user_option_as_bool('send_strict')
113
 
            if strict is None: strict = True # default value
114
 
            if strict and tree is not None:
115
 
                if (tree.has_changes(tree.basis_tree())
116
 
                    or len(tree.get_parent_ids()) > 1):
117
 
                    raise errors.UncommittedChanges(
118
 
                        tree, more='Use --no-strict to force the send.')
119
 
                if tree.last_revision() != tree.branch.last_revision():
120
 
                    # The tree has lost sync with its branch, there is little
121
 
                    # chance that the user is aware of it but he can still force
122
 
                    # the push with --no-strict
123
 
                    raise errors.OutOfDateTree(
124
 
                        tree, more='Use --no-strict to force the send.')
 
113
            if tree is not None:
 
114
                tree.check_changed_or_out_of_date(
 
115
                    strict, 'send_strict',
 
116
                    more_error='Use --no-strict to force the send.',
 
117
                    more_warning='Uncommitted changes will not be sent.')
125
118
            revision_id = branch.last_revision()
126
119
        if revision_id == NULL_REVISION:
127
120
            raise errors.BzrCommandError('No revisions to submit.')
128
121
        if format is None:
129
 
            # TODO: Query submit branch for its preferred format
130
122
            format = format_registry.get()
131
123
        directive = format(branch, revision_id, submit_branch,
132
124
            public_branch, no_patch, no_bundle, message, base_revision_id)
134
126
            directive.compose_merge_request(mail_client, mail_to, body,
135
127
                                            branch, tree)
136
128
        else:
137
 
            if output == '-':
138
 
                outfile = to_file
 
129
            if directive.multiple_output_files:
 
130
                if output == '-':
 
131
                    raise errors.BzrCommandError('- not supported for '
 
132
                        'merge directives that use more than one output file.')
 
133
                if not os.path.exists(output):
 
134
                    os.mkdir(output, 0755)
 
135
                for (filename, lines) in directive.to_files():
 
136
                    path = os.path.join(output, filename)
 
137
                    outfile = open(path, 'wb')
 
138
                    try:
 
139
                        outfile.writelines(lines)
 
140
                    finally:
 
141
                        outfile.close()
139
142
            else:
140
 
                outfile = open(output, 'wb')
141
 
            try:
142
 
                outfile.writelines(directive.to_lines())
143
 
            finally:
144
 
                if outfile is not to_file:
145
 
                    outfile.close()
 
143
                if output == '-':
 
144
                    outfile = to_file
 
145
                else:
 
146
                    outfile = open(output, 'wb')
 
147
                try:
 
148
                    outfile.writelines(directive.to_lines())
 
149
                finally:
 
150
                    if outfile is not to_file:
 
151
                        outfile.close()
146
152
    finally:
147
153
        branch.unlock()
148
154
 
149
155
 
150
156
def _send_4(branch, revision_id, submit_branch, public_branch,
151
157
            no_patch, no_bundle, message, base_revision_id):
 
158
    from bzrlib import merge_directive
152
159
    return merge_directive.MergeDirective2.from_objects(
153
160
        branch.repository, revision_id, time.time(),
154
161
        osutils.local_time_offset(), submit_branch,
170
177
            patch_type = 'diff'
171
178
        else:
172
179
            patch_type = None
 
180
    from bzrlib import merge_directive
173
181
    return merge_directive.MergeDirective.from_objects(
174
182
        branch.repository, revision_id, time.time(),
175
183
        osutils.local_time_offset(), submit_branch,