~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Aaron Bentley
  • Date: 2007-12-30 20:09:59 UTC
  • mto: This revision was merged to the branch mainline in revision 3164.
  • Revision ID: aaron.bentley@utoronto.ca-20071230200959-21ms4szwf591q3cv
Add NEWS entry

Show diffs side-by-side

added added

removed removed

Lines of Context:
21
21
"""
22
22
 
23
23
import sys
 
24
from cStringIO import StringIO
24
25
 
25
26
from bzrlib.lazy_import import lazy_import
26
27
lazy_import(globals(), """
27
28
from bzrlib import (
28
29
    branch,
29
30
    errors,
 
31
    merge_directive,
 
32
    revision as _mod_revision,
30
33
    urlutils,
 
34
    transport,
31
35
    )
32
 
from bzrlib.revision import common_ancestor
33
36
""")
34
37
 
35
38
from bzrlib.commands import Command
37
40
from bzrlib.trace import note
38
41
 
39
42
 
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", help="write bundle to specified file",
101
 
                            type=unicode)]
102
 
    takes_args = ['base?']
103
 
    aliases = ['bundle']
 
43
class cmd_bundle_info(Command):
 
44
    """Output interesting stats about a bundle"""
 
45
 
 
46
    hidden = True
 
47
    takes_args = ['location']
 
48
    takes_options = ['verbose']
104
49
    encoding_type = 'exact'
105
50
 
106
 
    def run(self, base=None, revision=None, output=None, remember=False):
107
 
        from bzrlib import user_encoding
108
 
        from bzrlib.bundle.serializer import write_bundle
109
 
 
110
 
        target_branch = branch.Branch.open_containing(u'.')[0]
111
 
 
112
 
        if base is None:
113
 
            base_specified = False
114
 
        else:
115
 
            base_specified = True
116
 
 
117
 
        if revision is None:
118
 
            target_revision = target_branch.last_revision()
119
 
        elif len(revision) < 3:
120
 
            target_revision = revision[-1].in_history(target_branch).rev_id
121
 
            if len(revision) == 2:
122
 
                if base_specified:
123
 
                    raise errors.BzrCommandError('Cannot specify base as well'
124
 
                                                 ' as two revision arguments.')
125
 
                base_revision = revision[0].in_history(target_branch).rev_id
126
 
        else:
127
 
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
128
 
 
129
 
        if revision is None or len(revision) < 2:
130
 
            submit_branch = target_branch.get_submit_branch()
131
 
            if base is None:
132
 
                base = submit_branch
133
 
            if base is None:
134
 
                base = target_branch.get_parent()
135
 
            if base is None:
136
 
                raise errors.BzrCommandError("No base branch known or"
137
 
                                             " specified.")
138
 
            elif not base_specified:
139
 
                # FIXME:
140
 
                # note() doesn't pay attention to terminal_encoding() so
141
 
                # we must format with 'ascii' to be safe
142
 
                note('Using saved location: %s',
143
 
                     urlutils.unescape_for_display(base, 'ascii'))
144
 
            base_branch = branch.Branch.open(base)
145
 
 
146
 
            # We don't want to lock the same branch across
147
 
            # 2 different branches
148
 
            if target_branch.base == base_branch.base:
149
 
                base_branch = target_branch 
150
 
            if submit_branch is None or remember:
151
 
                if base_specified:
152
 
                    target_branch.set_submit_branch(base_branch.base)
153
 
                elif remember:
154
 
                    raise errors.BzrCommandError('--remember requires a branch'
155
 
                                                 ' to be specified.')
156
 
            target_branch.repository.fetch(base_branch.repository, 
157
 
                                           base_branch.last_revision())
158
 
            base_revision = common_ancestor(base_branch.last_revision(),
159
 
                                            target_revision,
160
 
                                            target_branch.repository)
161
 
 
162
 
 
163
 
        if output is not None:
164
 
            fileobj = file(output, 'wb')
165
 
        else:
166
 
            fileobj = sys.stdout
167
 
        target_branch.repository.lock_read()
168
 
        try:
169
 
            write_bundle(target_branch.repository, target_revision,
170
 
                         base_revision, fileobj)
171
 
        finally:
172
 
            target_branch.repository.unlock()
173
 
 
174
 
 
175
 
class cmd_verify_changeset(Command):
176
 
    """Read a written changeset, and make sure it is valid.
177
 
 
178
 
    """
179
 
    takes_args = ['filename?']
180
 
 
181
 
    def run(self, filename=None):
182
 
        from read_changeset import read_changeset
183
 
        #from bzrlib.xml import serializer_v4
184
 
 
185
 
        b, relpath = branch.Branch.open_containing('.')
186
 
 
187
 
        if filename is None or filename == '-':
188
 
            f = sys.stdin
189
 
        else:
190
 
            f = open(filename, 'U')
191
 
 
192
 
        cset_info, cset_tree = read_changeset(f, b)
193
 
        # print cset_info
194
 
        # print cset_tree
195
 
        #serializer_v4.write(cset_tree.inventory, sys.stdout)
 
51
    def run(self, location, verbose=False):
 
52
        from bzrlib.bundle.serializer import read_bundle
 
53
        from bzrlib.bundle import read_mergeable_from_url
 
54
        from bzrlib import osutils
 
55
        term_encoding = osutils.get_terminal_encoding()
 
56
        bundle_info = read_mergeable_from_url(location)
 
57
        if isinstance(bundle_info, merge_directive._BaseMergeDirective):
 
58
            bundle_file = StringIO(bundle_info.get_raw_bundle())
 
59
            bundle_info = read_bundle(bundle_file)
 
60
        else:
 
61
            if verbose:
 
62
                raise errors.BzrCommandError('Verbose requires a merge'
 
63
                                             ' directive')
 
64
        reader_method = getattr(bundle_info, 'get_bundle_reader', None)
 
65
        if reader_method is None:
 
66
            raise errors.BzrCommandError('Bundle format not supported')
 
67
 
 
68
        by_kind = {}
 
69
        file_ids = set()
 
70
        for bytes, parents, repo_kind, revision_id, file_id\
 
71
            in reader_method().iter_records():
 
72
            by_kind.setdefault(repo_kind, []).append(
 
73
                (bytes, parents, repo_kind, revision_id, file_id))
 
74
            if file_id is not None:
 
75
                file_ids.add(file_id)
 
76
        self.outf.write('Records\n')
 
77
        for kind, records in sorted(by_kind.iteritems()):
 
78
            multiparent = sum(1 for b, m, k, r, f in records if
 
79
                              len(m.get('parents', [])) > 1)
 
80
            self.outf.write('%s: %d (%d multiparent)\n' % \
 
81
                (kind, len(records), multiparent))
 
82
        self.outf.write('unique files: %d\n' % len(file_ids))
 
83
        self.outf.write('\n')
 
84
        nicks = set()
 
85
        committers = set()
 
86
        for revision in bundle_info.real_revisions:
 
87
            if 'branch-nick' in revision.properties:
 
88
                nicks.add(revision.properties['branch-nick'])
 
89
            committers.add(revision.committer)
 
90
 
 
91
        self.outf.write('Revisions\n')
 
92
        self.outf.write(('nicks: %s\n'
 
93
            % ', '.join(sorted(nicks))).encode(term_encoding, 'replace'))
 
94
        self.outf.write(('committers: \n%s\n' %
 
95
        '\n'.join(sorted(committers)).encode(term_encoding, 'replace')))
 
96
        if verbose:
 
97
            self.outf.write('\n')
 
98
            bundle_file.seek(0)
 
99
            line = bundle_file.readline()
 
100
            line = bundle_file.readline()
 
101
            content = bundle_file.read().decode('bz2')
 
102
            self.outf.write("Decoded contents\n")
 
103
            self.outf.write(content)
 
104
            self.outf.write('\n')