← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/info.py

(jameinel) Allow 'bzr serve' to interpret SIGHUP as a graceful shutdown.
 (bug #795025) (John A Meinel)

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/info.py
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
 
import os
 
19
from cStringIO import StringIO
20
20
import time
21
21
import sys
22
22
 
23
23
from bzrlib import (
24
24
    bzrdir,
25
 
    diff,
26
25
    errors,
 
26
    hooks as _mod_hooks,
27
27
    osutils,
28
28
    urlutils,
29
29
    )
78
78
 
79
79
def gather_location_info(repository, branch=None, working=None):
80
80
    locs = {}
81
 
    repository_path = repository.bzrdir.root_transport.base
 
81
    repository_path = repository.user_url
82
82
    if branch is not None:
83
 
        branch_path = branch.bzrdir.root_transport.base
 
83
        branch_path = branch.user_url
84
84
        master_path = branch.get_bound_location()
85
85
        if master_path is None:
86
86
            master_path = branch_path
88
88
        branch_path = None
89
89
        master_path = None
90
90
    if working:
91
 
        working_path = working.bzrdir.root_transport.base
 
91
        working_path = working.user_url
92
92
        if working_path != branch_path:
93
93
            locs['light checkout root'] = working_path
94
94
        if master_path != branch_path:
221
221
    """Show missing revisions in working tree."""
222
222
    branch = working.branch
223
223
    basis = working.basis_tree()
224
 
    work_inv = working.inventory
225
224
    branch_revno, branch_last_revision = branch.last_revision_info()
226
225
    try:
227
226
        tree_last_id = working.get_parent_ids()[0]
239
238
def _show_working_stats(working, outfile):
240
239
    """Show statistics about a working tree."""
241
240
    basis = working.basis_tree()
242
 
    work_inv = working.inventory
243
241
    delta = working.changes_from(basis, want_unchanged=True)
244
242
 
245
243
    outfile.write('\n')
260
258
    outfile.write('  %8d ignored\n' % ignore_cnt)
261
259
 
262
260
    dir_cnt = 0
263
 
    for file_id in work_inv:
264
 
        if (work_inv.get_file_kind(file_id) == 'directory' and 
265
 
            not work_inv.is_root(file_id)):
 
261
    root_id = working.get_root_id()
 
262
    for path, entry in working.iter_entries_by_dir():
 
263
        if entry.kind == 'directory' and entry.file_id != root_id:
266
264
            dir_cnt += 1
267
265
    outfile.write('  %8d versioned %s\n' % (dir_cnt,
268
266
        plural(dir_cnt, 'subdirectory', 'subdirectories')))
299
297
            'the repository.\n')
300
298
 
301
299
 
302
 
def _show_repository_stats(stats, outfile):
 
300
def _show_repository_stats(repository, stats, outfile):
303
301
    """Show statistics about a repository."""
304
 
    if 'revisions' in stats or 'size' in stats:
305
 
        outfile.write('\n')
306
 
        outfile.write('Repository:\n')
 
302
    f = StringIO()
307
303
    if 'revisions' in stats:
308
304
        revisions = stats['revisions']
309
 
        outfile.write('  %8d revision%s\n' % (revisions, plural(revisions)))
 
305
        f.write('  %8d revision%s\n' % (revisions, plural(revisions)))
310
306
    if 'size' in stats:
311
 
        outfile.write('  %8d KiB\n' % (stats['size']/1024))
 
307
        f.write('  %8d KiB\n' % (stats['size']/1024))
 
308
    for hook in hooks['repository']:
 
309
        hook(repository, stats, f)
 
310
    if f.getvalue() != "":
 
311
        outfile.write('\n')
 
312
        outfile.write('Repository:\n')
 
313
        outfile.write(f.getvalue())
312
314
 
313
315
 
314
316
def show_bzrdir_info(a_bzrdir, verbose=False, outfile=None):
376
378
    elif branch is not None:
377
379
        _show_missing_revisions_branch(branch, outfile)
378
380
    if branch is not None:
379
 
        stats = _show_branch_stats(branch, verbose==2, outfile)
 
381
        show_committers = verbose >= 2
 
382
        stats = _show_branch_stats(branch, show_committers, outfile)
380
383
    else:
381
384
        stats = repository.gather_stats()
382
385
    if branch is None and working is None:
383
386
        _show_repository_info(repository, outfile)
384
 
    _show_repository_stats(stats, outfile)
 
387
    _show_repository_stats(repository, stats, outfile)
385
388
 
386
389
 
387
390
def describe_layout(repository=None, branch=None, tree=None):
413
416
        if branch is None and tree is not None:
414
417
            phrase = "branchless tree"
415
418
        else:
416
 
            if (tree is not None and tree.bzrdir.root_transport.base !=
417
 
                branch.bzrdir.root_transport.base):
 
419
            if (tree is not None and tree.user_url !=
 
420
                branch.user_url):
418
421
                independence = ''
419
422
                phrase = "Lightweight checkout"
420
423
            elif branch.get_bound_location() is not None:
439
442
    """
440
443
    candidates  = []
441
444
    if (branch is not None and tree is not None and
442
 
        branch.bzrdir.root_transport.base !=
443
 
        tree.bzrdir.root_transport.base):
 
445
        branch.user_url != tree.user_url):
444
446
        branch = None
445
447
        repository = None
446
448
    non_aliases = set(bzrdir.format_registry.keys())
471
473
        # do.
472
474
        candidates = new_candidates
473
475
    return ' or '.join(candidates)
 
476
 
 
477
 
 
478
class InfoHooks(_mod_hooks.Hooks):
 
479
    """Hooks for the info command."""
 
480
 
 
481
    def __init__(self):
 
482
        super(InfoHooks, self).__init__("bzrlib.info", "hooks")
 
483
        self.add_hook('repository',
 
484
            "Invoked when displaying the statistics for a repository. "
 
485
            "repository is called with a statistics dictionary as returned "
 
486
            "by the repository and a file-like object to write to.", (1, 15))
 
487
 
 
488
 
 
489
hooks = InfoHooks()