← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/bundle/commands.py

  • Committer: John Arbash Meinel
  • Date: 2011-05-11 11:35:28 UTC
  • mto: This revision was merged to the branch mainline in revision 5851.
  • Revision ID: john@arbash-meinel.com-20110511113528-qepibuwxicjrbb2h
Break compatibility with python <2.6.

This includes auditing the code for places where we were doing
explicit 'sys.version' checks and removing them as appropriate.

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
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
from cStringIO import StringIO
 
24
 
 
25
from bzrlib.lazy_import import lazy_import
 
26
lazy_import(globals(), """
 
27
from bzrlib import (
 
28
    branch,
 
29
    errors,
 
30
    merge_directive,
 
31
    revision as _mod_revision,
 
32
    urlutils,
 
33
    transport,
 
34
    )
 
35
""")
 
36
 
 
37
from bzrlib.commands import Command
 
38
 
 
39
 
 
40
class cmd_bundle_info(Command):
 
41
    __doc__ = """Output interesting stats about a bundle"""
 
42
 
 
43
    hidden = True
 
44
    takes_args = ['location']
 
45
    takes_options = ['verbose']
 
46
    encoding_type = 'exact'
 
47
 
 
48
    def run(self, location, verbose=False):
 
49
        from bzrlib.bundle.serializer import read_bundle
 
50
        from bzrlib.bundle import read_mergeable_from_url
 
51
        from bzrlib import osutils
 
52
        term_encoding = osutils.get_terminal_encoding()
 
53
        bundle_info = read_mergeable_from_url(location)
 
54
        if isinstance(bundle_info, merge_directive.BaseMergeDirective):
 
55
            bundle_file = StringIO(bundle_info.get_raw_bundle())
 
56
            bundle_info = read_bundle(bundle_file)
 
57
        else:
 
58
            if verbose:
 
59
                raise errors.BzrCommandError('--verbose requires a merge'
 
60
                    ' directive')
 
61
        reader_method = getattr(bundle_info, 'get_bundle_reader', None)
 
62
        if reader_method is None:
 
63
            raise errors.BzrCommandError('Bundle format not supported')
 
64
 
 
65
        by_kind = {}
 
66
        file_ids = set()
 
67
        for bytes, parents, repo_kind, revision_id, file_id\
 
68
            in reader_method().iter_records():
 
69
            by_kind.setdefault(repo_kind, []).append(
 
70
                (bytes, parents, repo_kind, revision_id, file_id))
 
71
            if file_id is not None:
 
72
                file_ids.add(file_id)
 
73
        self.outf.write('Records\n')
 
74
        for kind, records in sorted(by_kind.iteritems()):
 
75
            multiparent = sum(1 for b, m, k, r, f in records if
 
76
                              len(m.get('parents', [])) > 1)
 
77
            self.outf.write('%s: %d (%d multiparent)\n' % \
 
78
                (kind, len(records), multiparent))
 
79
        self.outf.write('unique files: %d\n' % len(file_ids))
 
80
        self.outf.write('\n')
 
81
        nicks = set()
 
82
        committers = set()
 
83
        for revision in bundle_info.real_revisions:
 
84
            if 'branch-nick' in revision.properties:
 
85
                nicks.add(revision.properties['branch-nick'])
 
86
            committers.add(revision.committer)
 
87
 
 
88
        self.outf.write('Revisions\n')
 
89
        self.outf.write(('nicks: %s\n'
 
90
            % ', '.join(sorted(nicks))).encode(term_encoding, 'replace'))
 
91
        self.outf.write(('committers: \n%s\n' %
 
92
        '\n'.join(sorted(committers)).encode(term_encoding, 'replace')))
 
93
        if verbose:
 
94
            self.outf.write('\n')
 
95
            bundle_file.seek(0)
 
96
            line = bundle_file.readline()
 
97
            line = bundle_file.readline()
 
98
            content = bundle_file.read().decode('bz2')
 
99
            self.outf.write("Decoded contents\n")
 
100
            self.outf.write(content)
 
101
            self.outf.write('\n')