~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/status.py

  • Committer: Jelmer Vernooij
  • Date: 2011-08-19 22:34:02 UTC
  • mto: This revision was merged to the branch mainline in revision 6089.
  • Revision ID: jelmer@samba.org-20110819223402-wjywqb0fa1xxx522
Use get_transport_from_{url,path} in more places.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006 Canonical Ltd
 
1
# Copyright (C) 2005-2010 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
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
import sys
18
18
 
19
19
from bzrlib import (
20
20
    delta as _mod_delta,
21
 
    tree,
 
21
    hooks as _mod_hooks,
 
22
    log,
 
23
    osutils,
 
24
    tsort,
 
25
    revision as _mod_revision,
22
26
    )
23
 
from bzrlib.diff import _raise_if_nonexistent
24
27
import bzrlib.errors as errors
25
 
from bzrlib.log import line_log
26
 
from bzrlib.osutils import is_inside_any
27
 
from bzrlib.symbol_versioning import (deprecated_function,
28
 
        zero_eight,
29
 
        )
30
 
from bzrlib.trace import warning
 
28
from bzrlib.trace import mutter, warning
31
29
 
32
30
# TODO: when showing single-line logs, truncate to the width of the terminal
33
31
# if known, but only if really going to the terminal (not into a file)
34
32
 
35
33
 
36
 
@deprecated_function(zero_eight)
37
 
def show_status(branch, show_unchanged=None,
38
 
                specific_files=None,
39
 
                show_ids=False,
40
 
                to_file=None,
41
 
                show_pending=True,
42
 
                revision=None):
 
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):
43
38
    """Display summary of changes.
44
39
 
45
 
    Please use show_tree_status instead.
46
 
 
47
 
    By default this compares the working tree to a previous revision. 
48
 
    If the revision argument is given, summarizes changes between the 
49
 
    working tree and another, or between two revisions.
50
 
 
51
 
    The result is written out as Unicode and to_file should be able 
52
 
    to encode that.
53
 
 
54
 
    show_unchanged
55
 
        If set, includes unchanged files.
56
 
 
57
 
    specific_files
58
 
        If set, only show the status of files in this list.
59
 
 
60
 
    show_ids
61
 
        If set, includes each file's id.
62
 
 
63
 
    to_file
64
 
        If set, write to this file (default stdout.)
65
 
 
66
 
    show_pending
67
 
        If set, write pending merges.
68
 
 
69
 
    revision
70
 
        If None the compare latest revision with working tree
71
 
        If one revision show compared it with working tree.
72
 
        If two revisions show status between first and second.
 
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.
73
63
    """
74
 
    show_tree_status(branch.bzrdir.open_workingtree(), show_unchanged, 
75
 
                     specific_files, show_ids, to_file, show_pending, revision)
 
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)
76
82
 
77
83
 
78
84
def show_tree_status(wt, show_unchanged=None,
81
87
                     to_file=None,
82
88
                     show_pending=True,
83
89
                     revision=None,
84
 
                     short=False):
 
90
                     short=False,
 
91
                     verbose=False,
 
92
                     versioned=False,
 
93
                     classify=True,
 
94
                     show_long_callback=_mod_delta.report_delta):
85
95
    """Display summary of changes.
86
96
 
87
 
    By default this compares the working tree to a previous revision. 
88
 
    If the revision argument is given, summarizes changes between the 
 
97
    By default this compares the working tree to a previous revision.
 
98
    If the revision argument is given, summarizes changes between the
89
99
    working tree and another, or between two revisions.
90
100
 
91
 
    The result is written out as Unicode and to_file should be able 
 
101
    The result is written out as Unicode and to_file should be able
92
102
    to encode that.
93
103
 
94
104
    If showing the status of a working tree, extra information is included
95
105
    about unknown files, conflicts, and pending merges.
96
106
 
97
 
    :param show_unchanged: Deprecated parameter. If set, includes unchanged 
 
107
    :param show_unchanged: Deprecated parameter. If set, includes unchanged
98
108
        files.
99
109
    :param specific_files: If set, a list of filenames whose status should be
100
 
        shown.  It is an error to give a filename that is not in the working 
 
110
        shown.  It is an error to give a filename that is not in the working
101
111
        tree, or in the working inventory or in the basis inventory.
102
112
    :param show_ids: If set, includes each file's id.
103
113
    :param to_file: If set, write to this file (default stdout.)
104
114
    :param show_pending: If set, write pending merges.
105
 
    :param revision: If None the compare latest revision with working tree
106
 
        If not None it must be a RevisionSpec list.
107
 
        If one revision show compared it with working tree.
108
 
        If two revisions show status between first and second.
109
 
    :param short: If True, gives short SVN-style status lines
 
115
    :param revision: If None, compare latest revision with working tree
 
116
        If not None, it must be a RevisionSpec list.
 
117
        If one revision, compare with working tree.
 
118
        If two revisions, show status between first and second.
 
119
    :param short: If True, gives short SVN-style status lines.
 
120
    :param verbose: If True, show all merged revisions, not just
 
121
        the merge tips
 
122
    :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
110
126
    """
111
127
    if show_unchanged is not None:
112
 
        warn("show_status_trees with show_unchanged has been deprecated "
 
128
        warn("show_tree_status with show_unchanged has been deprecated "
113
129
             "since bzrlib 0.9", DeprecationWarning, stacklevel=2)
114
130
 
115
131
    if to_file is None:
116
132
        to_file = sys.stdout
117
 
    
 
133
 
118
134
    wt.lock_read()
119
135
    try:
120
136
        new_is_working_tree = True
125
141
            old = new.basis_tree()
126
142
        elif len(revision) > 0:
127
143
            try:
128
 
                rev_id = revision[0].in_history(wt.branch).rev_id
129
 
                old = wt.branch.repository.revision_tree(rev_id)
 
144
                old = revision[0].as_tree(wt.branch)
130
145
            except errors.NoSuchRevision, e:
131
146
                raise errors.BzrCommandError(str(e))
132
147
            if (len(revision) > 1) and (revision[1].spec is not None):
133
148
                try:
134
 
                    rev_id = revision[1].in_history(wt.branch).rev_id
135
 
                    new = wt.branch.repository.revision_tree(rev_id)
 
149
                    new = revision[1].as_tree(wt.branch)
136
150
                    new_is_working_tree = False
137
151
                except errors.NoSuchRevision, e:
138
152
                    raise errors.BzrCommandError(str(e))
141
155
        old.lock_read()
142
156
        new.lock_read()
143
157
        try:
144
 
            _raise_if_nonexistent(specific_files, old, new)
145
 
            if short:
146
 
                specific_file_ids = tree.find_ids_across_trees(specific_files,
147
 
                    (old, new), require_versioned=False)
148
 
                changes = new._iter_changes(old, show_unchanged,
149
 
                                            specific_file_ids)
150
 
                reporter = _mod_delta.ChangeReporter(old.inventory,
151
 
                    output_file=to_file)
152
 
                _mod_delta.report_changes(changes, reporter)
153
 
            else:
154
 
                delta = new.changes_from(old, want_unchanged=show_unchanged,
155
 
                                      specific_files=specific_files)
156
 
                delta.show(to_file,
157
 
                           show_ids=show_ids,
158
 
                           show_unchanged=show_unchanged,
159
 
                           short_status=short)
160
 
            short_status_letter = '? '
161
 
            if not short:
162
 
                short_status_letter = ''
163
 
            list_paths('unknown', new.unknowns(), specific_files, to_file,
164
 
                       short_status_letter)
165
 
            conflict_title = False
166
 
            # show the new conflicts only for now. XXX: get them from the delta.
167
 
            for conflict in new.conflicts():
168
 
                if not short and conflict_title is False:
169
 
                    print >> to_file, "conflicts:"
170
 
                    conflict_title = True
 
158
            for hook in hooks['pre_status']:
 
159
                hook(StatusHookParams(old, new, to_file, versioned,
 
160
                    show_ids, short, verbose))
 
161
 
 
162
            specific_files, nonexistents \
 
163
                = _filter_nonexistent(specific_files, old, new)
 
164
            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
 
 
189
            # show the new conflicts only for now. XXX: get them from the
 
190
            # delta.
 
191
            conflicts = new.conflicts()
 
192
            if specific_files is not None:
 
193
                conflicts = conflicts.select_conflicts(new, specific_files,
 
194
                    ignore_misses=True, recurse=True)[1]
 
195
            if len(conflicts) > 0 and not short:
 
196
                to_file.write("conflicts:\n")
 
197
            for conflict in conflicts:
171
198
                if short:
172
199
                    prefix = 'C  '
173
200
                else:
174
201
                    prefix = ' '
175
 
                print >> to_file, "%s %s" % (prefix, conflict)
176
 
            if new_is_working_tree and show_pending:
177
 
                show_pending_merges(new, to_file, short)
 
202
                to_file.write("%s %s\n" % (prefix, unicode(conflict)))
 
203
            # Show files that were requested but don't exist (and are
 
204
            # not versioned).  We don't involve delta in this; these
 
205
            # paths are really the province of just the status
 
206
            # command, since they have more to do with how it was
 
207
            # invoked than with the tree it's operating on.
 
208
            if nonexistents and not short:
 
209
                to_file.write("nonexistent:\n")
 
210
            for nonexistent in nonexistents:
 
211
                # We could calculate prefix outside the loop but, given
 
212
                # how rarely this ought to happen, it's OK and arguably
 
213
                # slightly faster to do it here (ala conflicts above)
 
214
                if short:
 
215
                    prefix = 'X  '
 
216
                else:
 
217
                    prefix = ' '
 
218
                to_file.write("%s %s\n" % (prefix, nonexistent))
 
219
            if (new_is_working_tree and show_pending):
 
220
                show_pending_merges(new, to_file, short, verbose=verbose)
 
221
            if nonexistents:
 
222
                raise errors.PathsDoNotExist(nonexistents)
 
223
            for hook in hooks['post_status']:
 
224
                hook(StatusHookParams(old, new, to_file, versioned,
 
225
                    show_ids, short, verbose))
178
226
        finally:
179
227
            old.unlock()
180
228
            new.unlock()
181
229
    finally:
182
230
        wt.unlock()
183
231
 
184
 
def show_pending_merges(new, to_file, short=False):
 
232
 
 
233
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
 
234
    """Get an iterator which will return the revisions in merge sorted order.
 
235
 
 
236
    This will build up a list of all nodes, such that only nodes in the list
 
237
    are referenced. It then uses MergeSorter to return them in 'merge-sorted'
 
238
    order.
 
239
 
 
240
    :param revision_ids: A set of revision_ids
 
241
    :param parent_map: The parent information for each node. Revisions which
 
242
        are considered ghosts should not be present in the map.
 
243
    :return: iterator from MergeSorter.iter_topo_order()
 
244
    """
 
245
    # MergeSorter requires that all nodes be present in the graph, so get rid
 
246
    # of any references pointing outside of this graph.
 
247
    parent_graph = {}
 
248
    for revision_id in revision_ids:
 
249
        if revision_id not in parent_map: # ghost
 
250
            parent_graph[revision_id] = []
 
251
        else:
 
252
            # Only include parents which are in this sub-graph
 
253
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
 
254
                                            if p in revision_ids]
 
255
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
 
256
    return sorter.iter_topo_order()
 
257
 
 
258
 
 
259
def show_pending_merges(new, to_file, short=False, verbose=False):
185
260
    """Write out a display of pending merges in a working tree."""
186
261
    parents = new.get_parent_ids()
187
262
    if len(parents) < 2:
188
263
        return
 
264
 
 
265
    term_width = osutils.terminal_width()
 
266
    if term_width is not None:
 
267
        # we need one extra space for terminals that wrap on last char
 
268
        term_width = term_width - 1
 
269
    if short:
 
270
        first_prefix = 'P   '
 
271
        sub_prefix = 'P.   '
 
272
    else:
 
273
        first_prefix = '  '
 
274
        sub_prefix = '    '
 
275
 
 
276
    def show_log_message(rev, prefix):
 
277
        if term_width is None:
 
278
            width = term_width
 
279
        else:
 
280
            width = term_width - len(prefix)
 
281
        log_message = log_formatter.log_string(None, rev, width, prefix=prefix)
 
282
        to_file.write(log_message + '\n')
 
283
 
189
284
    pending = parents[1:]
190
285
    branch = new.branch
191
286
    last_revision = parents[0]
192
287
    if not short:
193
 
        print >>to_file, 'pending merges:'
194
 
    if last_revision is not None:
195
 
        try:
196
 
            ignore = set(branch.repository.get_ancestry(last_revision))
197
 
        except errors.NoSuchRevision:
198
 
            # the last revision is a ghost : assume everything is new 
199
 
            # except for it
200
 
            ignore = set([None, last_revision])
201
 
    else:
202
 
        ignore = set([None])
203
 
    # TODO: this could be improved using merge_sorted - we'd get the same 
204
 
    # output rather than one level of indent.
 
288
        if verbose:
 
289
            to_file.write('pending merges:\n')
 
290
        else:
 
291
            to_file.write('pending merge tips:'
 
292
                          ' (use -v to see all merge revisions)\n')
 
293
    graph = branch.repository.get_graph()
 
294
    other_revisions = [last_revision]
 
295
    log_formatter = log.LineLogFormatter(to_file)
205
296
    for merge in pending:
206
 
        ignore.add(merge)
207
 
        try:
208
 
            from bzrlib.osutils import terminal_width
209
 
            width = terminal_width()
210
 
            m_revision = branch.repository.get_revision(merge)
211
 
            if short:
212
 
                prefix = 'P  '
213
 
            else:
214
 
                prefix = ' '
215
 
            print >> to_file, prefix, line_log(m_revision, width - 4)
216
 
            inner_merges = branch.repository.get_ancestry(merge)
217
 
            assert inner_merges[0] is None
218
 
            inner_merges.pop(0)
219
 
            inner_merges.reverse()
220
 
            for mmerge in inner_merges:
221
 
                if mmerge in ignore:
222
 
                    continue
223
 
                mm_revision = branch.repository.get_revision(mmerge)
224
 
                if short:
225
 
                    prefix = 'P.  '
 
297
        try:
 
298
            rev = branch.repository.get_revisions([merge])[0]
 
299
        except errors.NoSuchRevision:
 
300
            # If we are missing a revision, just print out the revision id
 
301
            to_file.write(first_prefix + '(ghost) ' + merge + '\n')
 
302
            other_revisions.append(merge)
 
303
            continue
 
304
 
 
305
        # Log the merge, as it gets a slightly different formatting
 
306
        show_log_message(rev, first_prefix)
 
307
        if not verbose:
 
308
            continue
 
309
 
 
310
        # Find all of the revisions in the merge source, which are not in the
 
311
        # last committed revision.
 
312
        merge_extra = graph.find_unique_ancestors(merge, other_revisions)
 
313
        other_revisions.append(merge)
 
314
        merge_extra.discard(_mod_revision.NULL_REVISION)
 
315
 
 
316
        # Get a handle to all of the revisions we will need
 
317
        try:
 
318
            revisions = dict((rev.revision_id, rev) for rev in
 
319
                             branch.repository.get_revisions(merge_extra))
 
320
        except errors.NoSuchRevision:
 
321
            # One of the sub nodes is a ghost, check each one
 
322
            revisions = {}
 
323
            for revision_id in merge_extra:
 
324
                try:
 
325
                    rev = branch.repository.get_revisions([revision_id])[0]
 
326
                except errors.NoSuchRevision:
 
327
                    revisions[revision_id] = None
226
328
                else:
227
 
                    prefix = '   '
228
 
                print >> to_file, prefix, line_log(mm_revision, width - 5)
229
 
                ignore.add(mmerge)
230
 
        except errors.NoSuchRevision:
231
 
            if short:
232
 
                prefix = 'P  '
233
 
            else:
234
 
                prefix = ' '
235
 
            print >> to_file, prefix, merge
236
 
        
237
 
def list_paths(header, paths, specific_files, to_file, short_status_letter=''):
238
 
    done_header = False
239
 
    for path in paths:
240
 
        if specific_files and not is_inside_any(specific_files, path):
241
 
            continue
242
 
        if not short_status_letter and not done_header:
243
 
            print >>to_file, '%s:' % header
244
 
            done_header = True
245
 
        print >>to_file, '%s  %s' % (short_status_letter, path)
 
329
                    revisions[revision_id] = rev
 
330
 
 
331
        # Display the revisions brought in by this merge.
 
332
        rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
 
333
                            branch.repository.get_parent_map(merge_extra))
 
334
        # Skip the first node
 
335
        num, first, depth, eom = rev_id_iterator.next()
 
336
        if first != merge:
 
337
            raise AssertionError('Somehow we misunderstood how'
 
338
                ' iter_topo_order works %s != %s' % (first, merge))
 
339
        for num, sub_merge, depth, eom in rev_id_iterator:
 
340
            rev = revisions[sub_merge]
 
341
            if rev is None:
 
342
                to_file.write(sub_prefix + '(ghost) ' + sub_merge + '\n')
 
343
                continue
 
344
            show_log_message(revisions[sub_merge], sub_prefix)
 
345
 
 
346
 
 
347
def _filter_nonexistent(orig_paths, old_tree, new_tree):
 
348
    """Convert orig_paths to two sorted lists and return them.
 
349
 
 
350
    The first is orig_paths paths minus the items in the second list,
 
351
    and the second list is paths that are not in either inventory or
 
352
    tree (they don't qualify if they exist in the tree's inventory, or
 
353
    if they exist in the tree but are not versioned.)
 
354
 
 
355
    If either of the two lists is empty, return it as an empty list.
 
356
 
 
357
    This can be used by operations such as bzr status that can accept
 
358
    unknown or ignored files.
 
359
    """
 
360
    mutter("check paths: %r", orig_paths)
 
361
    if not orig_paths:
 
362
        return orig_paths, []
 
363
    s = old_tree.filter_unversioned_files(orig_paths)
 
364
    s = new_tree.filter_unversioned_files(s)
 
365
    nonexistent = [path for path in s if not new_tree.has_filename(path)]
 
366
    remaining   = [path for path in orig_paths if not path in nonexistent]
 
367
    # Sorting the 'remaining' list doesn't have much effect in
 
368
    # practice, since the various status output sections will sort
 
369
    # their groups individually.  But for consistency of this
 
370
    # function's API, it's better to sort both than just 'nonexistent'.
 
371
    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