← 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-01 06:45:57 UTC
  • mfrom: (5247.2.41 more-ignored-exceptions)
  • Revision ID: pqm@pqm.ubuntu.com-20100901064557-qsxmjmp195ozbluf
(vila) Catch EPIPE when shutting down test servers. (Vincent Ladeuil)

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,
22
21
    log,
23
22
    osutils,
24
23
    tsort,
151
150
        old.lock_read()
152
151
        new.lock_read()
153
152
        try:
154
 
            for hook in hooks['pre_status']:
155
 
                hook(StatusHookParams(old, new, to_file, versioned,
156
 
                    show_ids, short, verbose))
157
 
 
158
153
            specific_files, nonexistents \
159
154
                = _filter_nonexistent(specific_files, old, new)
160
155
            want_unversioned = not versioned
215
210
                show_pending_merges(new, to_file, short, verbose=verbose)
216
211
            if nonexistents:
217
212
                raise errors.PathsDoNotExist(nonexistents)
218
 
            for hook in hooks['post_status']:
219
 
                hook(StatusHookParams(old, new, to_file, versioned,
220
 
                    show_ids, short, verbose))
221
213
        finally:
222
214
            old.unlock()
223
215
            new.unlock()
364
356
    # their groups individually.  But for consistency of this
365
357
    # function's API, it's better to sort both than just 'nonexistent'.
366
358
    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