~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2007-07-19 16:09:34 UTC
  • mfrom: (2520.4.135 bzr.mpbundle)
  • Revision ID: pqm@pqm.ubuntu.com-20070719160934-d51fyijw69oto88p
Add new bundle and merge-directive formats

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,
30
32
    revision as _mod_revision,
31
33
    urlutils,
 
34
    transport,
32
35
    )
33
36
""")
34
37
 
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",
101
 
                            help="Write bundle to specified file.",
102
 
                            type=unicode),
103
 
                     ]
104
 
    takes_args = ['base?']
105
 
    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 = []
106
49
    encoding_type = 'exact'
107
50
 
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 = _mod_revision.ensure_null(
121
 
                target_branch.last_revision())
122
 
        elif len(revision) < 3:
123
 
            target_revision = revision[-1].in_history(target_branch).rev_id
124
 
            if len(revision) == 2:
125
 
                if base_specified:
126
 
                    raise errors.BzrCommandError('Cannot specify base as well'
127
 
                                                 ' as two revision arguments.')
128
 
                base_revision = revision[0].in_history(target_branch).rev_id
129
 
        else:
130
 
            raise errors.BzrCommandError('--revision takes 1 or 2 parameters')
131
 
 
132
 
        if revision is None or len(revision) < 2:
133
 
            submit_branch = target_branch.get_submit_branch()
134
 
            if base is None:
135
 
                base = submit_branch
136
 
            if base is None:
137
 
                base = target_branch.get_parent()
138
 
            if base is None:
139
 
                raise errors.BzrCommandError("No base branch known or"
140
 
                                             " specified.")
141
 
            elif not base_specified:
142
 
                # FIXME:
143
 
                # note() doesn't pay attention to terminal_encoding() so
144
 
                # we must format with 'ascii' to be safe
145
 
                note('Using saved location: %s',
146
 
                     urlutils.unescape_for_display(base, 'ascii'))
147
 
            base_branch = branch.Branch.open(base)
148
 
 
149
 
            # We don't want to lock the same branch across
150
 
            # 2 different branches
151
 
            if target_branch.base == base_branch.base:
152
 
                base_branch = target_branch 
153
 
            if submit_branch is None or remember:
154
 
                if base_specified:
155
 
                    target_branch.set_submit_branch(base_branch.base)
156
 
                elif remember:
157
 
                    raise errors.BzrCommandError('--remember requires a branch'
158
 
                                                 ' to be specified.')
159
 
            base_last_revision = _mod_revision.ensure_null(
160
 
                base_branch.last_revision())
161
 
            target_branch.repository.fetch(base_branch.repository, 
162
 
                base_last_revision)
163
 
            graph = target_branch.repository.get_graph()
164
 
            base_revision = graph.find_unique_lca(base_last_revision,
165
 
                                                  target_revision)
166
 
 
167
 
        if output is not None:
168
 
            fileobj = file(output, 'wb')
169
 
        else:
170
 
            fileobj = sys.stdout
171
 
        target_branch.repository.lock_read()
172
 
        try:
173
 
            write_bundle(target_branch.repository, target_revision,
174
 
                         base_revision, fileobj)
175
 
        finally:
176
 
            target_branch.repository.unlock()
177
 
 
178
 
 
179
 
class cmd_verify_changeset(Command):
180
 
    """Read a written changeset, and make sure it is valid.
181
 
 
182
 
    """
183
 
    takes_args = ['filename?']
184
 
 
185
 
    def run(self, filename=None):
186
 
        from read_changeset import read_changeset
187
 
        #from bzrlib.xml import serializer_v4
188
 
 
189
 
        b, relpath = branch.Branch.open_containing('.')
190
 
 
191
 
        if filename is None or filename == '-':
192
 
            f = sys.stdin
193
 
        else:
194
 
            f = open(filename, 'U')
195
 
 
196
 
        cset_info, cset_tree = read_changeset(f, b)
197
 
        # print cset_info
198
 
        # print cset_tree
199
 
        #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_info = read_bundle(StringIO(bundle_info.get_raw_bundle()))
 
59
        reader_method = getattr(bundle_info, 'get_bundle_reader', None)
 
60
        if reader_method is None:
 
61
            raise errors.BzrCommandError('Bundle format not supported')
 
62
 
 
63
        by_kind = {}
 
64
        file_ids = set()
 
65
        for bytes, parents, repo_kind, revision_id, file_id\
 
66
            in reader_method().iter_records():
 
67
            by_kind.setdefault(repo_kind, []).append(
 
68
                (bytes, parents, repo_kind, revision_id, file_id))
 
69
            if file_id is not None:
 
70
                file_ids.add(file_id)
 
71
        print >> self.outf, 'Records'
 
72
        for kind, records in sorted(by_kind.iteritems()):
 
73
            multiparent = sum(1 for b, p, k, r, f in records if len(p) > 1)
 
74
            print >> self.outf, '%s: %d (%d multiparent)' % \
 
75
                (kind, len(records), multiparent)
 
76
        print >> self.outf, 'unique files: %d' % len(file_ids)
 
77
        print >> self.outf
 
78
        nicks = set()
 
79
        committers = set()
 
80
        for revision in bundle_info.real_revisions:
 
81
            if 'branch-nick' in revision.properties:
 
82
                nicks.add(revision.properties['branch-nick'])
 
83
            committers.add(revision.committer)
 
84
 
 
85
        print >> self.outf, 'Revisions'
 
86
        print >> self.outf, ('nicks: %s'
 
87
            % ', '.join(sorted(nicks))).encode(term_encoding, 'replace')
 
88
        print >> self.outf, ('committers: \n%s' %
 
89
        '\n'.join(sorted(committers)).encode(term_encoding, 'replace'))
 
90
        if verbose:
 
91
            print >> self.outf
 
92
            bundle_file.seek(0)
 
93
            line = bundle_file.readline()
 
94
            line = bundle_file.readline()
 
95
            content = bundle_file.read().decode('bz2')
 
96
            print >> self.outf, "Decoded contents"
 
97
            self.outf.write(content)
 
98
            print >> self.outf