~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/status.py

  • Committer: Martin Pool
  • Date: 2005-06-30 07:58:00 UTC
  • mto: This revision was merged to the branch mainline in revision 852.
  • Revision ID: mbp@sourcefrog.net-20050630075800-9af4341e177e121a
Remove VerInfo class; just store sets directly in the list of  
versions.

Add tests for serialize/deserialize.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2010 Canonical Ltd
2
 
#
3
 
# This program is free software; you can redistribute it and/or modify
4
 
# it under the terms of the GNU General Public License as published by
5
 
# the Free Software Foundation; either version 2 of the License, or
6
 
# (at your option) any later version.
7
 
#
8
 
# This program is distributed in the hope that it will be useful,
9
 
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
 
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
 
# GNU General Public License for more details.
12
 
#
13
 
# You should have received a copy of the GNU General Public License
14
 
# along with this program; if not, write to the Free Software
15
 
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
16
 
 
17
 
import sys
18
 
 
19
 
from bzrlib import (
20
 
    delta as _mod_delta,
21
 
    hooks as _mod_hooks,
22
 
    log,
23
 
    osutils,
24
 
    tsort,
25
 
    revision as _mod_revision,
26
 
    )
27
 
import bzrlib.errors as errors
28
 
from bzrlib.trace import mutter, warning
29
 
 
30
 
# TODO: when showing single-line logs, truncate to the width of the terminal
31
 
# if known, but only if really going to the terminal (not into a file)
32
 
 
33
 
 
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):
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
 
    """
63
 
 
64
 
    if short:
65
 
        changes = new.iter_changes(old, want_unchanged, specific_files,
66
 
            require_versioned=False, want_unversioned=want_unversioned)
67
 
        _mod_delta.report_changes(changes, show_short_reporter)
68
 
        
69
 
    else:
70
 
        delta = new.changes_from(old, want_unchanged=want_unchanged,
71
 
                              specific_files=specific_files,
72
 
                              want_unversioned=want_unversioned)
73
 
        # filter out unknown files. We may want a tree method for
74
 
        # this
75
 
        delta.unversioned = [unversioned for unversioned in
76
 
            delta.unversioned if not new.is_ignored(unversioned[0])]
77
 
        show_long_callback(to_file, delta, 
78
 
                           show_ids=show_ids,
79
 
                           show_unchanged=want_unchanged)
80
 
 
81
 
 
82
 
def show_tree_status(wt, show_unchanged=None,
83
 
                     specific_files=None,
84
 
                     show_ids=False,
85
 
                     to_file=None,
86
 
                     show_pending=True,
87
 
                     revision=None,
88
 
                     short=False,
89
 
                     verbose=False,
90
 
                     versioned=False,
91
 
                     show_long_callback=_mod_delta.report_delta):
92
 
    """Display summary of changes.
93
 
 
94
 
    By default this compares the working tree to a previous revision.
95
 
    If the revision argument is given, summarizes changes between the
96
 
    working tree and another, or between two revisions.
97
 
 
98
 
    The result is written out as Unicode and to_file should be able
99
 
    to encode that.
100
 
 
101
 
    If showing the status of a working tree, extra information is included
102
 
    about unknown files, conflicts, and pending merges.
103
 
 
104
 
    :param show_unchanged: Deprecated parameter. If set, includes unchanged
105
 
        files.
106
 
    :param specific_files: If set, a list of filenames whose status should be
107
 
        shown.  It is an error to give a filename that is not in the working
108
 
        tree, or in the working inventory or in the basis inventory.
109
 
    :param show_ids: If set, includes each file's id.
110
 
    :param to_file: If set, write to this file (default stdout.)
111
 
    :param show_pending: If set, write pending merges.
112
 
    :param revision: If None, compare latest revision with working tree
113
 
        If not None, it must be a RevisionSpec list.
114
 
        If one revision, compare with working tree.
115
 
        If two revisions, show status between first and second.
116
 
    :param short: If True, gives short SVN-style status lines.
117
 
    :param verbose: If True, show all merged revisions, not just
118
 
        the merge tips
119
 
    :param versioned: If True, only shows versioned files.
120
 
    :param show_long_callback: A callback: message = show_long_callback(to_file, delta, 
121
 
        show_ids, show_unchanged, indent, filter), only used with the long output
122
 
    """
123
 
    if show_unchanged is not None:
124
 
        warn("show_tree_status with show_unchanged has been deprecated "
125
 
             "since bzrlib 0.9", DeprecationWarning, stacklevel=2)
126
 
 
127
 
    if to_file is None:
128
 
        to_file = sys.stdout
129
 
 
130
 
    wt.lock_read()
131
 
    try:
132
 
        new_is_working_tree = True
133
 
        if revision is None:
134
 
            if wt.last_revision() != wt.branch.last_revision():
135
 
                warning("working tree is out of date, run 'bzr update'")
136
 
            new = wt
137
 
            old = new.basis_tree()
138
 
        elif len(revision) > 0:
139
 
            try:
140
 
                old = revision[0].as_tree(wt.branch)
141
 
            except errors.NoSuchRevision, e:
142
 
                raise errors.BzrCommandError(str(e))
143
 
            if (len(revision) > 1) and (revision[1].spec is not None):
144
 
                try:
145
 
                    new = revision[1].as_tree(wt.branch)
146
 
                    new_is_working_tree = False
147
 
                except errors.NoSuchRevision, e:
148
 
                    raise errors.BzrCommandError(str(e))
149
 
            else:
150
 
                new = wt
151
 
        old.lock_read()
152
 
        new.lock_read()
153
 
        try:
154
 
            for hook in hooks['pre_status']:
155
 
                hook(StatusHookParams(old, new, to_file, versioned,
156
 
                    show_ids, short, verbose))
157
 
 
158
 
            specific_files, nonexistents \
159
 
                = _filter_nonexistent(specific_files, old, new)
160
 
            want_unversioned = not versioned
161
 
 
162
 
            # Reporter used for short outputs
163
 
            reporter = _mod_delta._ChangeReporter(output_file=to_file,
164
 
                unversioned_filter=new.is_ignored)
165
 
            report_changes(to_file, old, new, specific_files, 
166
 
                           reporter, show_long_callback, 
167
 
                           short=short, want_unchanged=show_unchanged, 
168
 
                           want_unversioned=want_unversioned, show_ids=show_ids)
169
 
 
170
 
            # show the ignored files among specific files (i.e. show the files
171
 
            # identified from input that we choose to ignore). 
172
 
            if specific_files is not None:
173
 
                # Ignored files is sorted because specific_files is already sorted
174
 
                ignored_files = [specific for specific in
175
 
                    specific_files if new.is_ignored(specific)]
176
 
                if len(ignored_files) > 0 and not short:
177
 
                    to_file.write("ignored:\n")
178
 
                    prefix = ' '
179
 
                else:
180
 
                    prefix = 'I  '
181
 
                for ignored_file in ignored_files:
182
 
                    to_file.write("%s %s\n" % (prefix, ignored_file))
183
 
 
184
 
            # show the new conflicts only for now. XXX: get them from the
185
 
            # delta.
186
 
            conflicts = new.conflicts()
187
 
            if specific_files is not None:
188
 
                conflicts = conflicts.select_conflicts(new, specific_files,
189
 
                    ignore_misses=True, recurse=True)[1]
190
 
            if len(conflicts) > 0 and not short:
191
 
                to_file.write("conflicts:\n")
192
 
            for conflict in conflicts:
193
 
                if short:
194
 
                    prefix = 'C  '
195
 
                else:
196
 
                    prefix = ' '
197
 
                to_file.write("%s %s\n" % (prefix, conflict))
198
 
            # Show files that were requested but don't exist (and are
199
 
            # not versioned).  We don't involve delta in this; these
200
 
            # paths are really the province of just the status
201
 
            # command, since they have more to do with how it was
202
 
            # invoked than with the tree it's operating on.
203
 
            if nonexistents and not short:
204
 
                to_file.write("nonexistent:\n")
205
 
            for nonexistent in nonexistents:
206
 
                # We could calculate prefix outside the loop but, given
207
 
                # how rarely this ought to happen, it's OK and arguably
208
 
                # slightly faster to do it here (ala conflicts above)
209
 
                if short:
210
 
                    prefix = 'X  '
211
 
                else:
212
 
                    prefix = ' '
213
 
                to_file.write("%s %s\n" % (prefix, nonexistent))
214
 
            if (new_is_working_tree and show_pending):
215
 
                show_pending_merges(new, to_file, short, verbose=verbose)
216
 
            if nonexistents:
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))
221
 
        finally:
222
 
            old.unlock()
223
 
            new.unlock()
224
 
    finally:
225
 
        wt.unlock()
226
 
 
227
 
 
228
 
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
229
 
    """Get an iterator which will return the revisions in merge sorted order.
230
 
 
231
 
    This will build up a list of all nodes, such that only nodes in the list
232
 
    are referenced. It then uses MergeSorter to return them in 'merge-sorted'
233
 
    order.
234
 
 
235
 
    :param revision_ids: A set of revision_ids
236
 
    :param parent_map: The parent information for each node. Revisions which
237
 
        are considered ghosts should not be present in the map.
238
 
    :return: iterator from MergeSorter.iter_topo_order()
239
 
    """
240
 
    # MergeSorter requires that all nodes be present in the graph, so get rid
241
 
    # of any references pointing outside of this graph.
242
 
    parent_graph = {}
243
 
    for revision_id in revision_ids:
244
 
        if revision_id not in parent_map: # ghost
245
 
            parent_graph[revision_id] = []
246
 
        else:
247
 
            # Only include parents which are in this sub-graph
248
 
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
249
 
                                            if p in revision_ids]
250
 
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
251
 
    return sorter.iter_topo_order()
252
 
 
253
 
 
254
 
def show_pending_merges(new, to_file, short=False, verbose=False):
255
 
    """Write out a display of pending merges in a working tree."""
256
 
    parents = new.get_parent_ids()
257
 
    if len(parents) < 2:
258
 
        return
259
 
 
260
 
    term_width = osutils.terminal_width()
261
 
    if term_width is not None:
262
 
        # we need one extra space for terminals that wrap on last char
263
 
        term_width = term_width - 1
264
 
    if short:
265
 
        first_prefix = 'P   '
266
 
        sub_prefix = 'P.   '
267
 
    else:
268
 
        first_prefix = '  '
269
 
        sub_prefix = '    '
270
 
 
271
 
    def show_log_message(rev, prefix):
272
 
        if term_width is None:
273
 
            width = term_width
274
 
        else:
275
 
            width = term_width - len(prefix)
276
 
        log_message = log_formatter.log_string(None, rev, width, prefix=prefix)
277
 
        to_file.write(log_message + '\n')
278
 
 
279
 
    pending = parents[1:]
280
 
    branch = new.branch
281
 
    last_revision = parents[0]
282
 
    if not short:
283
 
        if verbose:
284
 
            to_file.write('pending merges:\n')
285
 
        else:
286
 
            to_file.write('pending merge tips:'
287
 
                          ' (use -v to see all merge revisions)\n')
288
 
    graph = branch.repository.get_graph()
289
 
    other_revisions = [last_revision]
290
 
    log_formatter = log.LineLogFormatter(to_file)
291
 
    for merge in pending:
292
 
        try:
293
 
            rev = branch.repository.get_revisions([merge])[0]
294
 
        except errors.NoSuchRevision:
295
 
            # If we are missing a revision, just print out the revision id
296
 
            to_file.write(first_prefix + '(ghost) ' + merge + '\n')
297
 
            other_revisions.append(merge)
298
 
            continue
299
 
 
300
 
        # Log the merge, as it gets a slightly different formatting
301
 
        show_log_message(rev, first_prefix)
302
 
        if not verbose:
303
 
            continue
304
 
 
305
 
        # Find all of the revisions in the merge source, which are not in the
306
 
        # last committed revision.
307
 
        merge_extra = graph.find_unique_ancestors(merge, other_revisions)
308
 
        other_revisions.append(merge)
309
 
        merge_extra.discard(_mod_revision.NULL_REVISION)
310
 
 
311
 
        # Get a handle to all of the revisions we will need
312
 
        try:
313
 
            revisions = dict((rev.revision_id, rev) for rev in
314
 
                             branch.repository.get_revisions(merge_extra))
315
 
        except errors.NoSuchRevision:
316
 
            # One of the sub nodes is a ghost, check each one
317
 
            revisions = {}
318
 
            for revision_id in merge_extra:
319
 
                try:
320
 
                    rev = branch.repository.get_revisions([revision_id])[0]
321
 
                except errors.NoSuchRevision:
322
 
                    revisions[revision_id] = None
323
 
                else:
324
 
                    revisions[revision_id] = rev
325
 
 
326
 
        # Display the revisions brought in by this merge.
327
 
        rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
328
 
                            branch.repository.get_parent_map(merge_extra))
329
 
        # Skip the first node
330
 
        num, first, depth, eom = rev_id_iterator.next()
331
 
        if first != merge:
332
 
            raise AssertionError('Somehow we misunderstood how'
333
 
                ' iter_topo_order works %s != %s' % (first, merge))
334
 
        for num, sub_merge, depth, eom in rev_id_iterator:
335
 
            rev = revisions[sub_merge]
336
 
            if rev is None:
337
 
                to_file.write(sub_prefix + '(ghost) ' + sub_merge + '\n')
338
 
                continue
339
 
            show_log_message(revisions[sub_merge], sub_prefix)
340
 
 
341
 
 
342
 
def _filter_nonexistent(orig_paths, old_tree, new_tree):
343
 
    """Convert orig_paths to two sorted lists and return them.
344
 
 
345
 
    The first is orig_paths paths minus the items in the second list,
346
 
    and the second list is paths that are not in either inventory or
347
 
    tree (they don't qualify if they exist in the tree's inventory, or
348
 
    if they exist in the tree but are not versioned.)
349
 
 
350
 
    If either of the two lists is empty, return it as an empty list.
351
 
 
352
 
    This can be used by operations such as bzr status that can accept
353
 
    unknown or ignored files.
354
 
    """
355
 
    mutter("check paths: %r", orig_paths)
356
 
    if not orig_paths:
357
 
        return orig_paths, []
358
 
    s = old_tree.filter_unversioned_files(orig_paths)
359
 
    s = new_tree.filter_unversioned_files(s)
360
 
    nonexistent = [path for path in s if not new_tree.has_filename(path)]
361
 
    remaining   = [path for path in orig_paths if not path in nonexistent]
362
 
    # Sorting the 'remaining' list doesn't have much effect in
363
 
    # practice, since the various status output sections will sort
364
 
    # their groups individually.  But for consistency of this
365
 
    # function's API, it's better to sort both than just 'nonexistent'.
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