← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

merge merge tweaks from aaron, which includes latest .dev

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
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
 
 
17
 
"""\
18
 
This is an attempt to take the internal delta object, and represent
19
 
it as a single-file text-only changeset.
20
 
This should have commands for both generating a changeset,
21
 
and for applying a changeset.
22
 
"""
23
 
 
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')