← 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: 2009-06-22 17:11:20 UTC
  • mfrom: (4398.8.10 1.16-commit-fulltext)
  • Revision ID: pqm@pqm.ubuntu.com-20090622171120-fuxez9ylfqpxynqn
(jam) Add VF._add_text and reduce memory overhead during commit (see
        bug #109114)

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
21
    bzrdir,
23
22
    errors,
 
23
    merge_directive,
24
24
    osutils,
25
25
    registry,
26
26
    trace,
37
37
 
38
38
 
39
39
def send(submit_branch, revision, public_branch, remember, format,
40
 
         no_bundle, no_patch, output, from_, mail_to, message, body,
41
 
         to_file, strict=None):
 
40
         no_bundle, no_patch, output, from_, mail_to, message, body, 
 
41
         to_file):
42
42
    tree, branch = bzrdir.BzrDir.open_containing_tree_or_branch(from_)[:2]
43
43
    # we may need to write data into branch's repository to calculate
44
44
    # the data to send.
63
63
            submit_branch = stored_submit_branch
64
64
            remembered_submit_branch = "submit"
65
65
        else:
66
 
            # Remembers if asked explicitly or no previous location is set
67
 
            if remember or (remember is None and stored_submit_branch is None):
 
66
            if stored_submit_branch is None or remember:
68
67
                branch.set_submit_branch(submit_branch)
69
68
        if submit_branch is None:
70
69
            submit_branch = branch.get_parent()
78
77
                       submit_branch)
79
78
 
80
79
        if mail_to is None or format is None:
81
 
            # TODO: jam 20090716 we open the submit_branch here, but we *don't*
82
 
            #       pass it down into the format creation, so it will have to
83
 
            #       open it again
84
80
            submit_br = Branch.open(submit_branch)
85
81
            submit_config = submit_br.get_config()
86
82
            if mail_to is None:
96
92
        stored_public_branch = branch.get_public_branch()
97
93
        if public_branch is None:
98
94
            public_branch = stored_public_branch
99
 
        # Remembers if asked explicitly or no previous location is set
100
 
        elif (remember
101
 
              or (remember is None and stored_public_branch is None)):
 
95
        elif stored_public_branch is None or remember:
102
96
            branch.set_public_branch(public_branch)
103
97
        if no_bundle and public_branch is None:
104
98
            raise errors.BzrCommandError('No public branch specified or'
113
107
            if len(revision) == 2:
114
108
                base_revision_id = revision[0].as_revision_id(branch)
115
109
        if revision_id is None:
116
 
            if tree is not None:
117
 
                tree.check_changed_or_out_of_date(
118
 
                    strict, 'send_strict',
119
 
                    more_error='Use --no-strict to force the send.',
120
 
                    more_warning='Uncommitted changes will not be sent.')
121
110
            revision_id = branch.last_revision()
122
111
        if revision_id == NULL_REVISION:
123
112
            raise errors.BzrCommandError('No revisions to submit.')
124
113
        if format is None:
 
114
            # TODO: Query submit branch for its preferred format
125
115
            format = format_registry.get()
126
 
        directive = format(branch, revision_id, submit_branch,
 
116
        directive = format(branch, revision_id, submit_branch, 
127
117
            public_branch, no_patch, no_bundle, message, base_revision_id)
128
118
        if output is None:
129
119
            directive.compose_merge_request(mail_client, mail_to, body,
130
120
                                            branch, tree)
131
121
        else:
132
 
            if directive.multiple_output_files:
133
 
                if output == '-':
134
 
                    raise errors.BzrCommandError('- not supported for '
135
 
                        'merge directives that use more than one output file.')
136
 
                if not os.path.exists(output):
137
 
                    os.mkdir(output, 0755)
138
 
                for (filename, lines) in directive.to_files():
139
 
                    path = os.path.join(output, filename)
140
 
                    outfile = open(path, 'wb')
141
 
                    try:
142
 
                        outfile.writelines(lines)
143
 
                    finally:
144
 
                        outfile.close()
 
122
            if output == '-':
 
123
                outfile = to_file
145
124
            else:
146
 
                if output == '-':
147
 
                    outfile = to_file
148
 
                else:
149
 
                    outfile = open(output, 'wb')
150
 
                try:
151
 
                    outfile.writelines(directive.to_lines())
152
 
                finally:
153
 
                    if outfile is not to_file:
154
 
                        outfile.close()
 
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()
155
131
    finally:
156
132
        branch.unlock()
157
133
 
158
134
 
159
135
def _send_4(branch, revision_id, submit_branch, public_branch,
160
136
            no_patch, no_bundle, message, base_revision_id):
161
 
    from bzrlib import merge_directive
162
137
    return merge_directive.MergeDirective2.from_objects(
163
138
        branch.repository, revision_id, time.time(),
164
139
        osutils.local_time_offset(), submit_branch,
180
155
            patch_type = 'diff'
181
156
        else:
182
157
            patch_type = None
183
 
    from bzrlib import merge_directive
184
158
    return merge_directive.MergeDirective.from_objects(
185
159
        branch.repository, revision_id, time.time(),
186
160
        osutils.local_time_offset(), submit_branch,