← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/status.py

  • Committer: Canonical.com Patch Queue Manager
  • Date: 2010-09-29 22:03:03 UTC
  • mfrom: (5416.2.6 jam-integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100929220303-cr95h8iwtggco721
(mbp) Add 'break-lock --force'

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/status.py
18
18
 
19
19
from bzrlib import (
20
20
    delta as _mod_delta,
 
21
    hooks as _mod_hooks,
21
22
    log,
22
23
    osutils,
23
24
    tsort,
150
151
        old.lock_read()
151
152
        new.lock_read()
152
153
        try:
 
154
            for hook in hooks['pre_status']:
 
155
                hook(StatusHookParams(old, new, to_file, versioned,
 
156
                    show_ids, short, verbose))
 
157
 
153
158
            specific_files, nonexistents \
154
159
                = _filter_nonexistent(specific_files, old, new)
155
160
            want_unversioned = not versioned
210
215
                show_pending_merges(new, to_file, short, verbose=verbose)
211
216
            if nonexistents:
212
217
                raise errors.PathsDoNotExist(nonexistents)
 
218
            for hook in hooks['post_status']:
 
219
                hook(StatusHookParams(old, new, to_file, versioned,
 
220
                    show_ids, short, verbose))
213
221
        finally:
214
222
            old.unlock()
215
223
            new.unlock()
356
364
    # their groups individually.  But for consistency of this
357
365
    # function's API, it's better to sort both than just 'nonexistent'.
358
366
    return sorted(remaining), sorted(nonexistent)
 
367
 
 
368
 
 
369
class StatusHooks(_mod_hooks.Hooks):
 
370
    """A dictionary mapping hook name to a list of callables for status hooks.
 
371
 
 
372
    e.g. ['post_status'] Is the list of items to be called when the
 
373
    status command has finished printing the status.
 
374
    """
 
375
 
 
376
    def __init__(self):
 
377
        """Create the default hooks.
 
378
 
 
379
        These are all empty initially, because by default nothing should get
 
380
        notified.
 
381
        """
 
382
        _mod_hooks.Hooks.__init__(self)
 
383
        self.create_hook(_mod_hooks.HookPoint('post_status',
 
384
            "Called with argument StatusHookParams after Bazaar has "
 
385
            "displayed the status. StatusHookParams has the attributes "
 
386
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
 
387
            "verbose). The last four arguments correspond to the command "
 
388
            "line options specified by the user for the status command. "
 
389
            "to_file is the output stream for writing.",
 
390
            (2, 3), None))
 
391
        self.create_hook(_mod_hooks.HookPoint('pre_status',
 
392
            "Called with argument StatusHookParams before Bazaar "
 
393
            "displays the status. StatusHookParams has the attributes "
 
394
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
 
395
            "verbose). The last four arguments correspond to the command "
 
396
            "line options specified by the user for the status command. "
 
397
            "to_file is the output stream for writing.",
 
398
            (2, 3), None))
 
399
 
 
400
 
 
401
class StatusHookParams(object):
 
402
    """Object holding parameters passed to post_status hooks.
 
403
 
 
404
    :ivar old_tree: Start tree (basis tree) for comparison.
 
405
    :ivar new_tree: Working tree.
 
406
    :ivar to_file: If set, write to this file.
 
407
    :ivar versioned: Show only versioned files.
 
408
    :ivar show_ids: Show internal object ids.
 
409
    :ivar short: Use short status indicators.
 
410
    :ivar verbose: Verbose flag.
 
411
    """
 
412
 
 
413
    def __init__(self, old_tree, new_tree, to_file, versioned, show_ids,
 
414
            short, verbose):
 
415
        """Create a group of post_status hook parameters.
 
416
 
 
417
        :param old_tree: Start tree (basis tree) for comparison.
 
418
        :param new_tree: Working tree.
 
419
        :param to_file: If set, write to this file.
 
420
        :param versioned: Show only versioned files.
 
421
        :param show_ids: Show internal object ids.
 
422
        :param short: Use short status indicators.
 
423
        :param verbose: Verbose flag.
 
424
        """
 
425
        self.old_tree = old_tree
 
426
        self.new_tree = new_tree
 
427
        self.to_file = to_file
 
428
        self.versioned = versioned
 
429
        self.show_ids = show_ids
 
430
        self.short = short
 
431
        self.verbose = verbose
 
432
 
 
433
    def __eq__(self, other):
 
434
        return self.__dict__ == other.__dict__
 
435
 
 
436
    def __repr__(self):
 
437
        return "<%s(%s, %s, %s, %s, %s, %s, %s)>" % (self.__class__.__name__,
 
438
            self.old_tree, self.new_tree, self.to_file, self.versioned,
 
439
            self.show_ids, self.short, self.verbose)
 
440
 
 
441
 
 
442
def _show_shelve_summary(params):
 
443
    """post_status hook to display a summary of shelves.
 
444
 
 
445
    :param params: StatusHookParams.
 
446
    """
 
447
    manager = params.new_tree.get_shelf_manager()
 
448
    shelves = manager.active_shelves()
 
449
    if shelves:
 
450
        params.to_file.write('%d shelves exist. '
 
451
            'See "bzr shelve --list" for details.\n' % len(shelves))
 
452
 
 
453
 
 
454
hooks = StatusHooks()
 
455
 
 
456
 
 
457
hooks.install_named_hook('post_status', _show_shelve_summary,
 
458
    'bzr status')
 
459