~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-01-20 17:19:44 UTC
  • mfrom: (4973.1.1 integration)
  • Revision ID: pqm@pqm.ubuntu.com-20100120171944-8gkub20gotjcye67
(spiv) Implement a per-file merge hook

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
 
1
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
2
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
18
18
 
19
19
from bzrlib import (
20
20
    delta as _mod_delta,
21
 
    hooks as _mod_hooks,
22
21
    log,
23
22
    osutils,
 
23
    tree,
24
24
    tsort,
25
25
    revision as _mod_revision,
26
26
    )
27
27
import bzrlib.errors as errors
 
28
from bzrlib.osutils import is_inside_any
 
29
from bzrlib.symbol_versioning import (deprecated_function,
 
30
        )
28
31
from bzrlib.trace import mutter, warning
29
32
 
30
33
# TODO: when showing single-line logs, truncate to the width of the terminal
31
34
# if known, but only if really going to the terminal (not into a file)
32
35
 
33
36
 
34
 
def report_changes(to_file, old, new, specific_files, 
35
 
                   show_short_reporter, show_long_callback, 
36
 
                   short=False, want_unchanged=False, 
37
 
                   want_unversioned=False, show_ids=False, classify=True):
38
 
    """Display summary of changes.
39
 
 
40
 
    This compares two trees with regards to a list of files, and delegates 
41
 
    the display to underlying elements.
42
 
 
43
 
    For short output, it creates an iterator on all changes, and lets a given
44
 
    reporter display these changes.
45
 
 
46
 
    For stantard output, it creates a delta of the changes, and forwards it
47
 
    to a callback
48
 
 
49
 
    :param to_file: If set, write to this file (default stdout.)
50
 
    :param old: Start tree for the comparison
51
 
    :param end: End tree for the comparison
52
 
    :param specific_files: If set, a list of filenames whose status should be
53
 
        shown.  It is an error to give a filename that is not in the working
54
 
        tree, or in the working inventory or in the basis inventory.
55
 
    :param show_short_reporter: Reporter in charge of display for short output
56
 
    :param show_long_callback: Callback in charge of display for normal output
57
 
    :param short: If True, gives short SVN-style status lines.
58
 
    :param want_unchanged: Deprecated parameter. If set, includes unchanged
59
 
        files.
60
 
    :param show_ids: If set, includes each file's id.
61
 
    :param want_unversioned: If False, only shows versioned files.
62
 
    :param classify: Add special symbols to indicate file kind.
63
 
    """
64
 
 
65
 
    if short:
66
 
        changes = new.iter_changes(old, want_unchanged, specific_files,
67
 
            require_versioned=False, want_unversioned=want_unversioned)
68
 
        _mod_delta.report_changes(changes, show_short_reporter)
69
 
        
70
 
    else:
71
 
        delta = new.changes_from(old, want_unchanged=want_unchanged,
72
 
                              specific_files=specific_files,
73
 
                              want_unversioned=want_unversioned)
74
 
        # filter out unknown files. We may want a tree method for
75
 
        # this
76
 
        delta.unversioned = [unversioned for unversioned in
77
 
            delta.unversioned if not new.is_ignored(unversioned[0])]
78
 
        show_long_callback(to_file, delta, 
79
 
                           show_ids=show_ids,
80
 
                           show_unchanged=want_unchanged,
81
 
                           classify=classify)
82
 
 
83
 
 
84
37
def show_tree_status(wt, show_unchanged=None,
85
38
                     specific_files=None,
86
39
                     show_ids=False,
89
42
                     revision=None,
90
43
                     short=False,
91
44
                     verbose=False,
92
 
                     versioned=False,
93
 
                     classify=True,
94
 
                     show_long_callback=_mod_delta.report_delta):
 
45
                     versioned=False):
95
46
    """Display summary of changes.
96
47
 
97
48
    By default this compares the working tree to a previous revision.
120
71
    :param verbose: If True, show all merged revisions, not just
121
72
        the merge tips
122
73
    :param versioned: If True, only shows versioned files.
123
 
    :param classify: Add special symbols to indicate file kind.
124
 
    :param show_long_callback: A callback: message = show_long_callback(to_file, delta, 
125
 
        show_ids, show_unchanged, indent, filter), only used with the long output
126
74
    """
127
75
    if show_unchanged is not None:
128
76
        warn("show_tree_status with show_unchanged has been deprecated "
155
103
        old.lock_read()
156
104
        new.lock_read()
157
105
        try:
158
 
            for hook in hooks['pre_status']:
159
 
                hook(StatusHookParams(old, new, to_file, versioned,
160
 
                    show_ids, short, verbose))
161
 
 
162
106
            specific_files, nonexistents \
163
107
                = _filter_nonexistent(specific_files, old, new)
164
108
            want_unversioned = not versioned
165
 
 
166
 
            # Reporter used for short outputs
167
 
            reporter = _mod_delta._ChangeReporter(output_file=to_file,
168
 
                unversioned_filter=new.is_ignored, classify=classify)
169
 
            report_changes(to_file, old, new, specific_files, 
170
 
                           reporter, show_long_callback, 
171
 
                           short=short, want_unchanged=show_unchanged, 
172
 
                           want_unversioned=want_unversioned, show_ids=show_ids,
173
 
                           classify=classify)
174
 
 
175
 
            # show the ignored files among specific files (i.e. show the files
176
 
            # identified from input that we choose to ignore). 
177
 
            if specific_files is not None:
178
 
                # Ignored files is sorted because specific_files is already sorted
179
 
                ignored_files = [specific for specific in
180
 
                    specific_files if new.is_ignored(specific)]
181
 
                if len(ignored_files) > 0 and not short:
182
 
                    to_file.write("ignored:\n")
183
 
                    prefix = ' '
184
 
                else:
185
 
                    prefix = 'I  '
186
 
                for ignored_file in ignored_files:
187
 
                    to_file.write("%s %s\n" % (prefix, ignored_file))
188
 
 
 
109
            if short:
 
110
                changes = new.iter_changes(old, show_unchanged, specific_files,
 
111
                    require_versioned=False, want_unversioned=want_unversioned)
 
112
                reporter = _mod_delta._ChangeReporter(output_file=to_file,
 
113
                    unversioned_filter=new.is_ignored)
 
114
                _mod_delta.report_changes(changes, reporter)
 
115
            else:
 
116
                delta = new.changes_from(old, want_unchanged=show_unchanged,
 
117
                                      specific_files=specific_files,
 
118
                                      want_unversioned=want_unversioned)
 
119
                # filter out unknown files. We may want a tree method for
 
120
                # this
 
121
                delta.unversioned = [unversioned for unversioned in
 
122
                    delta.unversioned if not new.is_ignored(unversioned[0])]
 
123
                delta.show(to_file,
 
124
                           show_ids=show_ids,
 
125
                           show_unchanged=show_unchanged,
 
126
                           short_status=False)
189
127
            # show the new conflicts only for now. XXX: get them from the
190
128
            # delta.
191
129
            conflicts = new.conflicts()
199
137
                    prefix = 'C  '
200
138
                else:
201
139
                    prefix = ' '
202
 
                to_file.write("%s %s\n" % (prefix, unicode(conflict)))
 
140
                to_file.write("%s %s\n" % (prefix, conflict))
203
141
            # Show files that were requested but don't exist (and are
204
142
            # not versioned).  We don't involve delta in this; these
205
143
            # paths are really the province of just the status
220
158
                show_pending_merges(new, to_file, short, verbose=verbose)
221
159
            if nonexistents:
222
160
                raise errors.PathsDoNotExist(nonexistents)
223
 
            for hook in hooks['post_status']:
224
 
                hook(StatusHookParams(old, new, to_file, versioned,
225
 
                    show_ids, short, verbose))
226
161
        finally:
227
162
            old.unlock()
228
163
            new.unlock()
369
304
    # their groups individually.  But for consistency of this
370
305
    # function's API, it's better to sort both than just 'nonexistent'.
371
306
    return sorted(remaining), sorted(nonexistent)
372
 
 
373
 
 
374
 
class StatusHooks(_mod_hooks.Hooks):
375
 
    """A dictionary mapping hook name to a list of callables for status hooks.
376
 
 
377
 
    e.g. ['post_status'] Is the list of items to be called when the
378
 
    status command has finished printing the status.
379
 
    """
380
 
 
381
 
    def __init__(self):
382
 
        """Create the default hooks.
383
 
 
384
 
        These are all empty initially, because by default nothing should get
385
 
        notified.
386
 
        """
387
 
        _mod_hooks.Hooks.__init__(self, "bzrlib.status", "hooks")
388
 
        self.add_hook('post_status',
389
 
            "Called with argument StatusHookParams after Bazaar has "
390
 
            "displayed the status. StatusHookParams has the attributes "
391
 
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
392
 
            "verbose). The last four arguments correspond to the command "
393
 
            "line options specified by the user for the status command. "
394
 
            "to_file is the output stream for writing.",
395
 
            (2, 3))
396
 
        self.add_hook('pre_status',
397
 
            "Called with argument StatusHookParams before Bazaar "
398
 
            "displays the status. StatusHookParams has the attributes "
399
 
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
400
 
            "verbose). The last four arguments correspond to the command "
401
 
            "line options specified by the user for the status command. "
402
 
            "to_file is the output stream for writing.",
403
 
            (2, 3))
404
 
 
405
 
 
406
 
class StatusHookParams(object):
407
 
    """Object holding parameters passed to post_status hooks.
408
 
 
409
 
    :ivar old_tree: Start tree (basis tree) for comparison.
410
 
    :ivar new_tree: Working tree.
411
 
    :ivar to_file: If set, write to this file.
412
 
    :ivar versioned: Show only versioned files.
413
 
    :ivar show_ids: Show internal object ids.
414
 
    :ivar short: Use short status indicators.
415
 
    :ivar verbose: Verbose flag.
416
 
    """
417
 
 
418
 
    def __init__(self, old_tree, new_tree, to_file, versioned, show_ids,
419
 
            short, verbose):
420
 
        """Create a group of post_status hook parameters.
421
 
 
422
 
        :param old_tree: Start tree (basis tree) for comparison.
423
 
        :param new_tree: Working tree.
424
 
        :param to_file: If set, write to this file.
425
 
        :param versioned: Show only versioned files.
426
 
        :param show_ids: Show internal object ids.
427
 
        :param short: Use short status indicators.
428
 
        :param verbose: Verbose flag.
429
 
        """
430
 
        self.old_tree = old_tree
431
 
        self.new_tree = new_tree
432
 
        self.to_file = to_file
433
 
        self.versioned = versioned
434
 
        self.show_ids = show_ids
435
 
        self.short = short
436
 
        self.verbose = verbose
437
 
 
438
 
    def __eq__(self, other):
439
 
        return self.__dict__ == other.__dict__
440
 
 
441
 
    def __repr__(self):
442
 
        return "<%s(%s, %s, %s, %s, %s, %s, %s)>" % (self.__class__.__name__,
443
 
            self.old_tree, self.new_tree, self.to_file, self.versioned,
444
 
            self.show_ids, self.short, self.verbose)
445
 
 
446
 
 
447
 
def _show_shelve_summary(params):
448
 
    """post_status hook to display a summary of shelves.
449
 
 
450
 
    :param params: StatusHookParams.
451
 
    """
452
 
    get_shelf_manager = getattr(params.new_tree, 'get_shelf_manager', None)
453
 
    if get_shelf_manager is None:
454
 
        return
455
 
    manager = get_shelf_manager()
456
 
    shelves = manager.active_shelves()
457
 
    if shelves:
458
 
        singular = '%d shelf exists. '
459
 
        plural = '%d shelves exist. '
460
 
        if len(shelves) == 1:
461
 
            fmt = singular
462
 
        else:
463
 
            fmt = plural
464
 
        params.to_file.write(fmt % len(shelves))
465
 
        params.to_file.write('See "bzr shelve --list" for details.\n')
466
 
 
467
 
 
468
 
hooks = StatusHooks()
469
 
 
470
 
 
471
 
hooks.install_named_hook('post_status', _show_shelve_summary,
472
 
    'bzr status')
473