← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: John Arbash Meinel
  • Date: 2007-07-12 18:10:59 UTC
  • mto: This revision was merged to the branch mainline in revision 2643.
  • Revision ID: john@arbash-meinel.com-20070712181059-xnomv3tzzsb2hpx5
Add NEWS entries for performance improvements.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
1
 
# Copyright (C) 2005-2011 Canonical Ltd
 
1
# Copyright (C) 2006 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
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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
17
16
"""\
18
17
This is an attempt to take the internal delta object, and represent
19
18
it as a single-file text-only changeset.
21
20
and for applying a changeset.
22
21
"""
23
22
 
24
 
from __future__ import absolute_import
25
 
 
26
 
from cStringIO import StringIO
 
23
import sys
27
24
 
28
25
from bzrlib.lazy_import import lazy_import
29
26
lazy_import(globals(), """
30
27
from bzrlib import (
31
28
    branch,
32
29
    errors,
33
 
    merge_directive,
34
30
    revision as _mod_revision,
35
31
    urlutils,
36
 
    transport,
37
32
    )
38
 
from bzrlib.i18n import gettext
39
33
""")
40
34
 
41
35
from bzrlib.commands import Command
42
 
 
43
 
 
44
 
class cmd_bundle_info(Command):
45
 
    __doc__ = """Output interesting stats about a bundle"""
46
 
 
47
 
    hidden = True
48
 
    takes_args = ['location']
49
 
    takes_options = ['verbose']
 
36
from bzrlib.option import Option
 
37
from bzrlib.trace import note
 
38
 
 
39
 
 
40
class cmd_send_changeset(Command):
 
41
    """Send a bundled up changset via mail.
 
42
 
 
43
    If no revision has been specified, the last commited change will
 
44
    be sent.
 
45
 
 
46
    Subject of the mail can be specified by the --message option,
 
47
    otherwise information from the changeset log will be used.
 
48
 
 
49
    A editor will be spawned where the user may enter a description
 
50
    of the changeset.  The description can be read from a file with
 
51
    the --file FILE option.
 
52
    """
 
53
    takes_options = ['revision', 'message', 'file']
 
54
    takes_args = ['to?']
 
55
 
 
56
    def run(self, to=None, message=None, revision=None, file=None):
 
57
        from bzrlib.errors import BzrCommandError
 
58
        from send_changeset import send_changeset
 
59
        
 
60
        if isinstance(revision, (list, tuple)):
 
61
            if len(revision) > 1:
 
62
                raise BzrCommandError('We do not support rollup-changesets yet.')
 
63
            revision = revision[0]
 
64
 
 
65
        b = branch.Branch.open_containing('.')
 
66
 
 
67
        if not to:
 
68
            try:
 
69
                to = b.controlfile('x-send-address', 'rb').read().strip('\n')
 
70
            except:
 
71
                raise BzrCommandError('destination address is not known')
 
72
 
 
73
        if not isinstance(revision, (list, tuple)):
 
74
            revision = [revision]
 
75
 
 
76
        send_changeset(b, revision, to, message, file)
 
77
 
 
78
 
 
79
class cmd_bundle_revisions(Command):
 
80
    """Generate a revision bundle.
 
81
 
 
82
    This bundle contains all of the meta-information of a
 
83
    diff, rather than just containing the patch information.
 
84
 
 
85
    You can apply it to another tree using 'bzr merge'.
 
86
 
 
87
    bzr bundle-revisions
 
88
        - Generate a bundle relative to a remembered location
 
89
 
 
90
    bzr bundle-revisions BASE
 
91
        - Bundle to apply the current tree into BASE
 
92
 
 
93
    bzr bundle-revisions --revision A
 
94
        - Bundle to apply revision A to remembered location 
 
95
 
 
96
    bzr bundle-revisions --revision A..B
 
97
        - Bundle to transform A into B
 
98
    """
 
99
    takes_options = ['revision', 'remember',
 
100
                     Option("output",
 
101
                            help="Write bundle to specified file.",
 
102
                            type=unicode),
 
103
                     ]
 
104
    takes_args = ['base?']
 
105
    aliases = ['bundle']
50
106
    encoding_type = 'exact'
51
107
 
52
 
    def run(self, location, verbose=False):
53
 
        from bzrlib.bundle.serializer import read_bundle
54
 
        from bzrlib.bundle import read_mergeable_from_url
55
 
        from bzrlib import osutils
56
 
        term_encoding = osutils.get_terminal_encoding()
57
 
        bundle_info = read_mergeable_from_url(location)
58
 
        if isinstance(bundle_info, merge_directive.BaseMergeDirective):
59
 
            bundle_file = StringIO(bundle_info.get_raw_bundle())
60
 
            bundle_info = read_bundle(bundle_file)
61
 
        else:
62
 
            if verbose:
63
 
                raise errors.BzrCommandError(gettext(
64
 
                            '--verbose requires a merge directive'))
65
 
        reader_method = getattr(bundle_info, 'get_bundle_reader', None)
66
 
        if reader_method is None:
67
 
            raise errors.BzrCommandError(gettext('Bundle format not supported'))
68
 
 
69
 
        by_kind = {}
70
 
        file_ids = set()
71
 
        for bytes, parents, repo_kind, revision_id, file_id\
72
 
            in reader_method().iter_records():
73
 
            by_kind.setdefault(repo_kind, []).append(
74
 
                (bytes, parents, repo_kind, revision_id, file_id))
75
 
            if file_id is not None:
76
 
                file_ids.add(file_id)
77
 
        self.outf.write(gettext('Records\n'))
78
 
        for kind, records in sorted(by_kind.iteritems()):
79
 
            multiparent = sum(1 for b, m, k, r, f in records if
80
 
                              len(m.get('parents', [])) > 1)
81
 
            self.outf.write(gettext('{0}: {1} ({2} multiparent)\n').format(
82
 
                kind, len(records), multiparent))
83
 
        self.outf.write(gettext('unique files: %d\n') % len(file_ids))
84
 
        self.outf.write('\n')
85
 
        nicks = set()
86
 
        committers = set()
87
 
        for revision in bundle_info.real_revisions:
88
 
            if 'branch-nick' in revision.properties:
89
 
                nicks.add(revision.properties['branch-nick'])
90
 
            committers.add(revision.committer)
91
 
 
92
 
        self.outf.write(gettext('Revisions\n'))
93
 
        self.outf.write((gettext('nicks: %s\n')
94
 
            % ', '.join(sorted(nicks))).encode(term_encoding, 'replace'))
95
 
        self.outf.write((gettext('committers: \n%s\n') %
96
 
        '\n'.join(sorted(committers)).encode(term_encoding, 'replace')))
97
 
        if verbose:
98
 
            self.outf.write('\n')
99
 
            bundle_file.seek(0)
100
 
            line = bundle_file.readline()
101
 
            line = bundle_file.readline()
102
 
            content = bundle_file.read().decode('bz2')
103
 
            self.outf.write(gettext("Decoded contents\n"))
104
 
            self.outf.write(content)
105
 
            self.outf.write('\n')
 
108
    def run(self, base=None, revision=None, output=None, remember=False):
 
109
        from bzrlib import user_encoding
 
110
        from bzrlib.bundle.serializer import write_bundle
 
111
 
 
112
        target_branch = branch.Branch.open_containing(u'.')[0]
 
113
 
 
114
        if base is None:
 
115
            base_specified = False
 
116
        else:
 
117
            base_specified = True
 
118
 
 
119
        if revision is None:
 
120
            target_revision = target_branch.last_revision()
 
121
        elif len(revision) < 3:
 
122
            target_revision = revision[-1].in_history(target_branch).rev_id
 
123
            if len(revision) == 2:
 
124
                if base_specified:
 
125
                    raise errors.BzrCommandError('Cannot specify base as well'
 
126
                                                 ' as two revision arguments.')
 
127
                base_revision = revision[0].in_history(target_branch).rev_id
 
128
        else:
 
129
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
 
130
 
 
131
        if revision is None or len(revision) < 2:
 
132
            submit_branch = target_branch.get_submit_branch()
 
133
            if base is None:
 
134
                base = submit_branch
 
135
            if base is None:
 
136
                base = target_branch.get_parent()
 
137
            if base is None:
 
138
                raise errors.BzrCommandError("No base branch known or"
 
139
                                             " specified.")
 
140
            elif not base_specified:
 
141
                # FIXME:
 
142
                # note() doesn't pay attention to terminal_encoding() so
 
143
                # we must format with 'ascii' to be safe
 
144
                note('Using saved location: %s',
 
145
                     urlutils.unescape_for_display(base, 'ascii'))
 
146
            base_branch = branch.Branch.open(base)
 
147
 
 
148
            # We don't want to lock the same branch across
 
149
            # 2 different branches
 
150
            if target_branch.base == base_branch.base:
 
151
                base_branch = target_branch 
 
152
            if submit_branch is None or remember:
 
153
                if base_specified:
 
154
                    target_branch.set_submit_branch(base_branch.base)
 
155
                elif remember:
 
156
                    raise errors.BzrCommandError('--remember requires a branch'
 
157
                                                 ' to be specified.')
 
158
            target_branch.repository.fetch(base_branch.repository, 
 
159
                                           base_branch.last_revision())
 
160
            graph = target_branch.repository.get_graph()
 
161
            base_revision = graph.find_unique_lca(
 
162
                _mod_revision.ensure_null(base_branch.last_revision()),
 
163
                _mod_revision.ensure_null(target_revision))
 
164
 
 
165
        if output is not None:
 
166
            fileobj = file(output, 'wb')
 
167
        else:
 
168
            fileobj = sys.stdout
 
169
        target_branch.repository.lock_read()
 
170
        try:
 
171
            write_bundle(target_branch.repository, target_revision,
 
172
                         base_revision, fileobj)
 
173
        finally:
 
174
            target_branch.repository.unlock()
 
175
 
 
176
 
 
177
class cmd_verify_changeset(Command):
 
178
    """Read a written changeset, and make sure it is valid.
 
179
 
 
180
    """
 
181
    takes_args = ['filename?']
 
182
 
 
183
    def run(self, filename=None):
 
184
        from read_changeset import read_changeset
 
185
        #from bzrlib.xml import serializer_v4
 
186
 
 
187
        b, relpath = branch.Branch.open_containing('.')
 
188
 
 
189
        if filename is None or filename == '-':
 
190
            f = sys.stdin
 
191
        else:
 
192
            f = open(filename, 'U')
 
193
 
 
194
        cset_info, cset_tree = read_changeset(f, b)
 
195
        # print cset_info
 
196
        # print cset_tree
 
197
        #serializer_v4.write(cset_tree.inventory, sys.stdout)