← Back to branch summary

~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/status.py

  • Committer: Martin
  • Date: 2010-05-03 20:57:39 UTC
  • mto: This revision was merged to the branch mainline in revision 5204.
  • Revision ID: gzlist@googlemail.com-20100503205739-n326zdvevv0rmruh
Retain original stack and error message when translating to ValueError in bencode

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/status.py
14
14
# along with this program; if not, write to the Free Software
15
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
16
 
17
 
from __future__ import absolute_import
18
 
 
19
17
import sys
20
18
 
21
19
from bzrlib import (
22
20
    delta as _mod_delta,
23
 
    hooks as _mod_hooks,
24
21
    log,
25
22
    osutils,
26
23
    tsort,
36
33
def report_changes(to_file, old, new, specific_files, 
37
34
                   show_short_reporter, show_long_callback, 
38
35
                   short=False, want_unchanged=False, 
39
 
                   want_unversioned=False, show_ids=False, classify=True):
 
36
                   want_unversioned=False, show_ids=False):
40
37
    """Display summary of changes.
41
38
 
42
39
    This compares two trees with regards to a list of files, and delegates 
61
58
        files.
62
59
    :param show_ids: If set, includes each file's id.
63
60
    :param want_unversioned: If False, only shows versioned files.
64
 
    :param classify: Add special symbols to indicate file kind.
65
61
    """
66
62
 
67
63
    if short:
79
75
            delta.unversioned if not new.is_ignored(unversioned[0])]
80
76
        show_long_callback(to_file, delta, 
81
77
                           show_ids=show_ids,
82
 
                           show_unchanged=want_unchanged,
83
 
                           classify=classify)
 
78
                           show_unchanged=want_unchanged)
84
79
 
85
80
 
86
81
def show_tree_status(wt, show_unchanged=None,
92
87
                     short=False,
93
88
                     verbose=False,
94
89
                     versioned=False,
95
 
                     classify=True,
96
90
                     show_long_callback=_mod_delta.report_delta):
97
91
    """Display summary of changes.
98
92
 
122
116
    :param verbose: If True, show all merged revisions, not just
123
117
        the merge tips
124
118
    :param versioned: If True, only shows versioned files.
125
 
    :param classify: Add special symbols to indicate file kind.
126
119
    :param show_long_callback: A callback: message = show_long_callback(to_file, delta, 
127
120
        show_ids, show_unchanged, indent, filter), only used with the long output
128
121
    """
157
150
        old.lock_read()
158
151
        new.lock_read()
159
152
        try:
160
 
            for hook in hooks['pre_status']:
161
 
                hook(StatusHookParams(old, new, to_file, versioned,
162
 
                    show_ids, short, verbose, specific_files=specific_files))
163
 
 
164
153
            specific_files, nonexistents \
165
154
                = _filter_nonexistent(specific_files, old, new)
166
155
            want_unversioned = not versioned
167
156
 
168
157
            # Reporter used for short outputs
169
158
            reporter = _mod_delta._ChangeReporter(output_file=to_file,
170
 
                unversioned_filter=new.is_ignored, classify=classify)
 
159
                unversioned_filter=new.is_ignored)
171
160
            report_changes(to_file, old, new, specific_files, 
172
161
                           reporter, show_long_callback, 
173
162
                           short=short, want_unchanged=show_unchanged, 
174
 
                           want_unversioned=want_unversioned, show_ids=show_ids,
175
 
                           classify=classify)
 
163
                           want_unversioned=want_unversioned, show_ids=show_ids)
176
164
 
177
165
            # show the ignored files among specific files (i.e. show the files
178
166
            # identified from input that we choose to ignore). 
201
189
                    prefix = 'C  '
202
190
                else:
203
191
                    prefix = ' '
204
 
                to_file.write("%s %s\n" % (prefix, unicode(conflict)))
 
192
                to_file.write("%s %s\n" % (prefix, conflict))
205
193
            # Show files that were requested but don't exist (and are
206
194
            # not versioned).  We don't involve delta in this; these
207
195
            # paths are really the province of just the status
222
210
                show_pending_merges(new, to_file, short, verbose=verbose)
223
211
            if nonexistents:
224
212
                raise errors.PathsDoNotExist(nonexistents)
225
 
            for hook in hooks['post_status']:
226
 
                hook(StatusHookParams(old, new, to_file, versioned,
227
 
                    show_ids, short, verbose, specific_files=specific_files))
228
213
        finally:
229
214
            old.unlock()
230
215
            new.unlock()
371
356
    # their groups individually.  But for consistency of this
372
357
    # function's API, it's better to sort both than just 'nonexistent'.
373
358
    return sorted(remaining), sorted(nonexistent)
374
 
 
375
 
 
376
 
class StatusHooks(_mod_hooks.Hooks):
377
 
    """A dictionary mapping hook name to a list of callables for status hooks.
378
 
 
379
 
    e.g. ['post_status'] Is the list of items to be called when the
380
 
    status command has finished printing the status.
381
 
    """
382
 
 
383
 
    def __init__(self):
384
 
        """Create the default hooks.
385
 
 
386
 
        These are all empty initially, because by default nothing should get
387
 
        notified.
388
 
        """
389
 
        _mod_hooks.Hooks.__init__(self, "bzrlib.status", "hooks")
390
 
        self.add_hook('post_status',
391
 
            "Called with argument StatusHookParams after Bazaar has "
392
 
            "displayed the status. StatusHookParams has the attributes "
393
 
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
394
 
            "verbose). The last four arguments correspond to the command "
395
 
            "line options specified by the user for the status command. "
396
 
            "to_file is the output stream for writing.",
397
 
            (2, 3))
398
 
        self.add_hook('pre_status',
399
 
            "Called with argument StatusHookParams before Bazaar "
400
 
            "displays the status. StatusHookParams has the attributes "
401
 
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
402
 
            "verbose). The last four arguments correspond to the command "
403
 
            "line options specified by the user for the status command. "
404
 
            "to_file is the output stream for writing.",
405
 
            (2, 3))
406
 
 
407
 
 
408
 
class StatusHookParams(object):
409
 
    """Object holding parameters passed to post_status hooks.
410
 
 
411
 
    :ivar old_tree: Start tree (basis tree) for comparison.
412
 
    :ivar new_tree: Working tree.
413
 
    :ivar to_file: If set, write to this file.
414
 
    :ivar versioned: Show only versioned files.
415
 
    :ivar show_ids: Show internal object ids.
416
 
    :ivar short: Use short status indicators.
417
 
    :ivar verbose: Verbose flag.
418
 
    """
419
 
 
420
 
    def __init__(self, old_tree, new_tree, to_file, versioned, show_ids,
421
 
            short, verbose, specific_files=None):
422
 
        """Create a group of post_status hook parameters.
423
 
 
424
 
        :param old_tree: Start tree (basis tree) for comparison.
425
 
        :param new_tree: Working tree.
426
 
        :param to_file: If set, write to this file.
427
 
        :param versioned: Show only versioned files.
428
 
        :param show_ids: Show internal object ids.
429
 
        :param short: Use short status indicators.
430
 
        :param verbose: Verbose flag.
431
 
        :param specific_files: If set, a list of filenames whose status should be
432
 
            shown.  It is an error to give a filename that is not in the working
433
 
            tree, or in the working inventory or in the basis inventory.
434
 
        """
435
 
        self.old_tree = old_tree
436
 
        self.new_tree = new_tree
437
 
        self.to_file = to_file
438
 
        self.versioned = versioned
439
 
        self.show_ids = show_ids
440
 
        self.short = short
441
 
        self.verbose = verbose
442
 
        self.specific_files = specific_files
443
 
 
444
 
    def __eq__(self, other):
445
 
        return self.__dict__ == other.__dict__
446
 
 
447
 
    def __repr__(self):
448
 
        return "<%s(%s, %s, %s, %s, %s, %s, %s, %s)>" % (self.__class__.__name__,
449
 
            self.old_tree, self.new_tree, self.to_file, self.versioned,
450
 
            self.show_ids, self.short, self.verbose, self.specific_files)
451
 
 
452
 
 
453
 
def _show_shelve_summary(params):
454
 
    """post_status hook to display a summary of shelves.
455
 
 
456
 
    :param params: StatusHookParams.
457
 
    """
458
 
    # Don't show shelves if status of specific files is being shown, only if
459
 
    # no file arguments have been passed
460
 
    if params.specific_files:
461
 
        return
462
 
    get_shelf_manager = getattr(params.new_tree, 'get_shelf_manager', None)
463
 
    if get_shelf_manager is None:
464
 
        return
465
 
    manager = get_shelf_manager()
466
 
    shelves = manager.active_shelves()
467
 
    if shelves:
468
 
        singular = '%d shelf exists. '
469
 
        plural = '%d shelves exist. '
470
 
        if len(shelves) == 1:
471
 
            fmt = singular
472
 
        else:
473
 
            fmt = plural
474
 
        params.to_file.write(fmt % len(shelves))
475
 
        params.to_file.write('See "bzr shelve --list" for details.\n')
476
 
 
477
 
 
478
 
hooks = StatusHooks()
479
 
 
480
 
 
481
 
hooks.install_named_hook('post_status', _show_shelve_summary,
482
 
    'bzr status')
483