~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/info.py

  • Committer: Martin von Gagern
  • Date: 2010-04-20 08:47:38 UTC
  • mfrom: (5167 +trunk)
  • mto: This revision was merged to the branch mainline in revision 5195.
  • Revision ID: martin.vgagern@gmx.net-20100420084738-ygymnqmdllzrhpfn
merge trunk

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
 
 
1
# Copyright (C) 2005-2010 Canonical Ltd
 
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
5
5
# the Free Software Foundation; either version 2 of the License, or
12
12
#
13
13
# You should have received a copy of the GNU General Public License
14
14
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
17
__all__ = ['show_bzrdir_info']
18
18
 
 
19
from cStringIO import StringIO
19
20
import os
20
21
import time
21
22
import sys
24
25
    bzrdir,
25
26
    diff,
26
27
    errors,
 
28
    hooks as _mod_hooks,
27
29
    osutils,
28
30
    urlutils,
29
31
    )
261
263
 
262
264
    dir_cnt = 0
263
265
    for file_id in work_inv:
264
 
        if (work_inv.get_file_kind(file_id) == 'directory' and 
 
266
        if (work_inv.get_file_kind(file_id) == 'directory' and
265
267
            not work_inv.is_root(file_id)):
266
268
            dir_cnt += 1
267
269
    outfile.write('  %8d versioned %s\n' % (dir_cnt,
299
301
            'the repository.\n')
300
302
 
301
303
 
302
 
def _show_repository_stats(stats, outfile):
 
304
def _show_repository_stats(repository, stats, outfile):
303
305
    """Show statistics about a repository."""
304
 
    if 'revisions' in stats or 'size' in stats:
305
 
        outfile.write('\n')
306
 
        outfile.write('Repository:\n')
 
306
    f = StringIO()
307
307
    if 'revisions' in stats:
308
308
        revisions = stats['revisions']
309
 
        outfile.write('  %8d revision%s\n' % (revisions, plural(revisions)))
 
309
        f.write('  %8d revision%s\n' % (revisions, plural(revisions)))
310
310
    if 'size' in stats:
311
 
        outfile.write('  %8d KiB\n' % (stats['size']/1024))
 
311
        f.write('  %8d KiB\n' % (stats['size']/1024))
 
312
    for hook in hooks['repository']:
 
313
        hook(repository, stats, f)
 
314
    if f.getvalue() != "":
 
315
        outfile.write('\n')
 
316
        outfile.write('Repository:\n')
 
317
        outfile.write(f.getvalue())
312
318
 
313
319
 
314
320
def show_bzrdir_info(a_bzrdir, verbose=False, outfile=None):
376
382
    elif branch is not None:
377
383
        _show_missing_revisions_branch(branch, outfile)
378
384
    if branch is not None:
379
 
        stats = _show_branch_stats(branch, verbose==2, outfile)
 
385
        show_committers = verbose >= 2
 
386
        stats = _show_branch_stats(branch, show_committers, outfile)
380
387
    else:
381
388
        stats = repository.gather_stats()
382
389
    if branch is None and working is None:
383
390
        _show_repository_info(repository, outfile)
384
 
    _show_repository_stats(stats, outfile)
 
391
    _show_repository_stats(repository, stats, outfile)
385
392
 
386
393
 
387
394
def describe_layout(repository=None, branch=None, tree=None):
471
478
        # do.
472
479
        candidates = new_candidates
473
480
    return ' or '.join(candidates)
 
481
 
 
482
 
 
483
class InfoHooks(_mod_hooks.Hooks):
 
484
    """Hooks for the info command."""
 
485
 
 
486
    def __init__(self):
 
487
        super(InfoHooks, self).__init__()
 
488
        self.create_hook(_mod_hooks.HookPoint('repository',
 
489
            "Invoked when displaying the statistics for a repository. "
 
490
            "repository is called with a statistics dictionary as returned "
 
491
            "by the repository and a file-like object to write to.", (1, 15), 
 
492
            None))
 
493
 
 
494
 
 
495
hooks = InfoHooks()