← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: Martin Pool
  • Date: 2009-07-10 07:14:02 UTC
  • mto: (4525.1.1 integration)
  • mto: This revision was merged to the branch mainline in revision 4526.
  • Revision ID: mbp@sourcefrog.net-20090710071402-dga5lhvdup5jeere
Rename remaining *_implementations tests

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/bundle/commands.py
 
1
# Copyright (C) 2006 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
This is an attempt to take the internal delta object, and represent
 
18
it as a single-file text-only changeset.
 
19
This should have commands for both generating a changeset,
 
20
and for applying a changeset.
 
21
"""
 
22
 
 
23
import sys
 
24
from cStringIO import StringIO
 
25
 
 
26
from bzrlib.lazy_import import lazy_import
 
27
lazy_import(globals(), """
 
28
from bzrlib import (
 
29
    branch,
 
30
    errors,
 
31
    merge_directive,
 
32
    revision as _mod_revision,
 
33
    urlutils,
 
34
    transport,
 
35
    )
 
36
""")
 
37
 
 
38
from bzrlib.commands import Command
 
39
from bzrlib.option import Option
 
40
from bzrlib.trace import note
 
41
 
 
42
 
 
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']
 
49
    encoding_type = 'exact'
 
50
 
 
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')