← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Patch Queue Manager
  • Date: 2012-03-04 18:21:04 UTC
  • mfrom: (6468.4.3 509275-lp-fixes)
  • Revision ID: pqm@pqm.ubuntu.com-20120304182104-8y1tdavoblrrhflc
(jelmer) Add --link-bug option to 'bzr lp-propose'. (Ross Lagerwall)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
1
 
#!/usr/bin/env python
 
1
# Copyright (C) 2005-2011 Canonical Ltd
 
2
#
 
3
# This program is free software; you can redistribute it and/or modify
 
4
# it under the terms of the GNU General Public License as published by
 
5
# the Free Software Foundation; either version 2 of the License, or
 
6
# (at your option) any later version.
 
7
#
 
8
# This program is distributed in the hope that it will be useful,
 
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
11
# GNU General Public License for more details.
 
12
#
 
13
# You should have received a copy of the GNU General Public License
 
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
 
2
17
"""\
3
18
This is an attempt to take the internal delta object, and represent
4
19
it as a single-file text-only changeset.
6
21
and for applying a changeset.
7
22
"""
8
23
 
9
 
import sys
10
 
 
11
 
from bzrlib.branch import Branch
12
 
from bzrlib.commands import Command, register_command
13
 
import bzrlib.errors as errors
14
 
from bzrlib.option import Option
15
 
from bzrlib.revision import (common_ancestor, MultipleRevisionSources,
16
 
                             NULL_REVISION)
17
 
from bzrlib.revisionspec import RevisionSpec
18
 
from bzrlib.trace import note
19
 
from bzrlib import urlutils
20
 
 
21
 
 
22
 
class cmd_send_changeset(Command):
23
 
    """Send a bundled up changset via mail.
24
 
 
25
 
    If no revision has been specified, the last commited change will
26
 
    be sent.
27
 
 
28
 
    Subject of the mail can be specified by the --message option,
29
 
    otherwise information from the changeset log will be used.
30
 
 
31
 
    A editor will be spawned where the user may enter a description
32
 
    of the changeset.  The description can be read from a file with
33
 
    the --file FILE option.
34
 
    """
35
 
    takes_options = ['revision', 'message', 'file']
36
 
    takes_args = ['to?']
37
 
 
38
 
    def run(self, to=None, message=None, revision=None, file=None):
39
 
        from bzrlib.errors import BzrCommandError
40
 
        from send_changeset import send_changeset
41
 
        
42
 
        if isinstance(revision, (list, tuple)):
43
 
            if len(revision) > 1:
44
 
                raise BzrCommandError('We do not support rollup-changesets yet.')
45
 
            revision = revision[0]
46
 
 
47
 
        b = Branch.open_containing('.')
48
 
 
49
 
        if not to:
50
 
            try:
51
 
                to = b.controlfile('x-send-address', 'rb').read().strip('\n')
52
 
            except:
53
 
                raise BzrCommandError('destination address is not known')
54
 
 
55
 
        if not isinstance(revision, (list, tuple)):
56
 
            revision = [revision]
57
 
 
58
 
        send_changeset(b, revision, to, message, file)
59
 
 
60
 
 
61
 
class cmd_bundle_revisions(Command):
62
 
    """Generate a revision bundle.
63
 
 
64
 
    This bundle contains all of the meta-information of a
65
 
    diff, rather than just containing the patch information.
66
 
 
67
 
    You can apply it to another tree using 'bzr merge'.
68
 
 
69
 
    bzr bundle-revisions
70
 
        - Generate a bundle relative to a remembered location
71
 
    bzr bundle-revisions BASE
72
 
        - Bundle to apply the current tree into BASE
73
 
    bzr bundle-revisions --revision A
74
 
        - Bundle to apply revision A to remembered location 
75
 
    bzr bundle-revisions --revision A..B
76
 
        - Bundle to transform A into B
77
 
    """
78
 
    takes_options = ['verbose', 'revision', 'remember',
79
 
                     Option("output", help="write bundle to specified file",
80
 
                            type=unicode)]
81
 
    takes_args = ['base?']
82
 
    aliases = ['bundle']
83
 
 
84
 
    def run(self, base=None, revision=None, output=None, remember=False):
85
 
        from bzrlib import user_encoding
86
 
        from bzrlib.bundle.serializer import write_bundle
87
 
 
88
 
        target_branch = Branch.open_containing(u'.')[0]
89
 
 
90
 
        if base is None:
91
 
            base_specified = False
92
 
        else:
93
 
            base_specified = True
94
 
 
95
 
        if revision is None:
96
 
            target_revision = target_branch.last_revision()
97
 
        elif len(revision) < 3:
98
 
            target_revision = revision[-1].in_history(target_branch).rev_id
99
 
            if len(revision) == 2:
100
 
                if base_specified:
101
 
                    raise errors.BzrCommandError('Cannot specify base as well'
102
 
                                                 ' as two revision arguments.')
103
 
                base_revision = revision[0].in_history(target_branch).rev_id
104
 
        else:
105
 
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
106
 
 
107
 
        if revision is None or len(revision) < 2:
108
 
            submit_branch = target_branch.get_submit_branch()
109
 
            if base is None:
110
 
                base = submit_branch
111
 
            if base is None:
112
 
                base = target_branch.get_parent()
113
 
            if base is None:
114
 
                raise errors.BzrCommandError("No base branch known or"
115
 
                                             " specified.")
116
 
            elif not base_specified:
117
 
                # FIXME:
118
 
                # note() doesn't pay attention to terminal_encoding() so
119
 
                # we must format with 'ascii' to be safe
120
 
                note('Using saved location: %s',
121
 
                     urlutils.unescape_for_display(base, 'ascii'))
122
 
            base_branch = Branch.open(base)
123
 
 
124
 
            # We don't want to lock the same branch across
125
 
            # 2 different branches
126
 
            if target_branch.base == base_branch.base:
127
 
                base_branch = target_branch 
128
 
            if submit_branch is None or remember:
129
 
                if base_specified:
130
 
                    target_branch.set_submit_branch(base_branch.base)
131
 
                elif remember:
132
 
                    raise errors.BzrCommandError('--remember requires a branch'
133
 
                                                 ' to be specified.')
134
 
            target_branch.repository.fetch(base_branch.repository, 
135
 
                                           base_branch.last_revision())
136
 
            base_revision = common_ancestor(base_branch.last_revision(),
137
 
                                            target_revision,
138
 
                                            target_branch.repository)
139
 
 
140
 
 
141
 
        if output is not None:
142
 
            fileobj = file(output, 'wb')
143
 
        else:
144
 
            fileobj = sys.stdout
145
 
        target_branch.repository.lock_read()
146
 
        try:
147
 
            write_bundle(target_branch.repository, target_revision,
148
 
                         base_revision, fileobj)
149
 
        finally:
150
 
            target_branch.repository.unlock()
151
 
 
152
 
 
153
 
class cmd_verify_changeset(Command):
154
 
    """Read a written changeset, and make sure it is valid.
155
 
 
156
 
    """
157
 
    takes_args = ['filename?']
158
 
 
159
 
    def run(self, filename=None):
160
 
        from read_changeset import read_changeset
161
 
        #from bzrlib.xml import serializer_v4
162
 
 
163
 
        b, relpath = Branch.open_containing('.')
164
 
 
165
 
        if filename is None or filename == '-':
166
 
            f = sys.stdin
167
 
        else:
168
 
            f = open(filename, 'U')
169
 
 
170
 
        cset_info, cset_tree = read_changeset(f, b)
171
 
        # print cset_info
172
 
        # print cset_tree
173
 
        #serializer_v4.write(cset_tree.inventory, sys.stdout)
 
24
from __future__ import absolute_import
 
25
 
 
26
from cStringIO import StringIO
 
27
 
 
28
from bzrlib.lazy_import import lazy_import
 
29
lazy_import(globals(), """
 
30
from bzrlib import (
 
31
    branch,
 
32
    errors,
 
33
    merge_directive,
 
34
    revision as _mod_revision,
 
35
    urlutils,
 
36
    transport,
 
37
    )
 
38
from bzrlib.i18n import gettext
 
39
""")
 
40
 
 
41
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']
 
50
    encoding_type = 'exact'
 
51
 
 
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')