/brz/remove-bazaar

To get this branch, use:
bzr branch http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

« back to all changes in this revision

Viewing changes to breezy/status.py

  • Committer: Jelmer Vernooij
  • Date: 2017-11-11 13:10:32 UTC
  • mto: This revision was merged to the branch mainline in revision 6804.
  • Revision ID: jelmer@jelmer.uk-20171111131032-31lgi8qmvlz8363d
Fix typos.

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
from __future__ import absolute_import
 
18
 
 
19
import sys
 
20
 
 
21
from . import (
 
22
    delta as _mod_delta,
 
23
    hooks as _mod_hooks,
 
24
    log,
 
25
    osutils,
 
26
    tsort,
 
27
    revision as _mod_revision,
 
28
    )
 
29
from . import errors as errors
 
30
from .trace import mutter, warning
 
31
from .workingtree import ShelvingUnsupported
 
32
 
 
33
 
 
34
# TODO: when showing single-line logs, truncate to the width of the terminal
 
35
# if known, but only if really going to the terminal (not into a file)
 
36
 
 
37
 
 
38
def report_changes(to_file, old, new, specific_files,
 
39
                   show_short_reporter, show_long_callback,
 
40
                   short=False, want_unchanged=False,
 
41
                   want_unversioned=False, show_ids=False, classify=True):
 
42
    """Display summary of changes.
 
43
 
 
44
    This compares two trees with regards to a list of files, and delegates
 
45
    the display to underlying elements.
 
46
 
 
47
    For short output, it creates an iterator on all changes, and lets a given
 
48
    reporter display these changes.
 
49
 
 
50
    For stantard output, it creates a delta of the changes, and forwards it
 
51
    to a callback
 
52
 
 
53
    :param to_file: If set, write to this file (default stdout.)
 
54
    :param old: Start tree for the comparison
 
55
    :param end: End tree for the comparison
 
56
    :param specific_files: If set, a list of filenames whose status should be
 
57
        shown.  It is an error to give a filename that is not in the working
 
58
        tree, or in the working inventory or in the basis inventory.
 
59
    :param show_short_reporter: Reporter in charge of display for short output
 
60
    :param show_long_callback: Callback in charge of display for normal output
 
61
    :param short: If True, gives short SVN-style status lines.
 
62
    :param want_unchanged: Deprecated parameter. If set, includes unchanged
 
63
        files.
 
64
    :param show_ids: If set, includes each file's id.
 
65
    :param want_unversioned: If False, only shows versioned files.
 
66
    :param classify: Add special symbols to indicate file kind.
 
67
    """
 
68
 
 
69
    if short:
 
70
        changes = new.iter_changes(old, want_unchanged, specific_files,
 
71
            require_versioned=False, want_unversioned=want_unversioned)
 
72
        _mod_delta.report_changes(changes, show_short_reporter)
 
73
    else:
 
74
        delta = new.changes_from(old, want_unchanged=want_unchanged,
 
75
                              specific_files=specific_files,
 
76
                              want_unversioned=want_unversioned)
 
77
        # filter out unknown files. We may want a tree method for
 
78
        # this
 
79
        delta.unversioned = [unversioned for unversioned in
 
80
            delta.unversioned if not new.is_ignored(unversioned[0])]
 
81
        show_long_callback(to_file, delta,
 
82
                           show_ids=show_ids,
 
83
                           show_unchanged=want_unchanged,
 
84
                           classify=classify)
 
85
 
 
86
 
 
87
def show_tree_status(wt, show_unchanged=None,
 
88
                     specific_files=None,
 
89
                     show_ids=False,
 
90
                     to_file=None,
 
91
                     show_pending=True,
 
92
                     revision=None,
 
93
                     short=False,
 
94
                     verbose=False,
 
95
                     versioned=False,
 
96
                     classify=True,
 
97
                     show_long_callback=_mod_delta.report_delta):
 
98
    """Display summary of changes.
 
99
 
 
100
    By default this compares the working tree to a previous revision.
 
101
    If the revision argument is given, summarizes changes between the
 
102
    working tree and another, or between two revisions.
 
103
 
 
104
    The result is written out as Unicode and to_file should be able
 
105
    to encode that.
 
106
 
 
107
    If showing the status of a working tree, extra information is included
 
108
    about unknown files, conflicts, and pending merges.
 
109
 
 
110
    :param show_unchanged: Deprecated parameter. If set, includes unchanged
 
111
        files.
 
112
    :param specific_files: If set, a list of filenames whose status should be
 
113
        shown.  It is an error to give a filename that is not in the working
 
114
        tree, or in the working inventory or in the basis inventory.
 
115
    :param show_ids: If set, includes each file's id.
 
116
    :param to_file: If set, write to this file (default stdout.)
 
117
    :param show_pending: If set, write pending merges.
 
118
    :param revision: If None, compare latest revision with working tree
 
119
        If not None, it must be a RevisionSpec list.
 
120
        If one revision, compare with working tree.
 
121
        If two revisions, show status between first and second.
 
122
    :param short: If True, gives short SVN-style status lines.
 
123
    :param verbose: If True, show all merged revisions, not just
 
124
        the merge tips
 
125
    :param versioned: If True, only shows versioned files.
 
126
    :param classify: Add special symbols to indicate file kind.
 
127
    :param show_long_callback: A callback: message = show_long_callback(to_file, delta,
 
128
        show_ids, show_unchanged, indent, filter), only used with the long output
 
129
    """
 
130
    if show_unchanged is not None:
 
131
        warn("show_tree_status with show_unchanged has been deprecated "
 
132
             "since breezy 0.9", DeprecationWarning, stacklevel=2)
 
133
 
 
134
    if to_file is None:
 
135
        to_file = sys.stdout
 
136
 
 
137
    wt.lock_read()
 
138
    try:
 
139
        new_is_working_tree = True
 
140
        if revision is None:
 
141
            if wt.last_revision() != wt.branch.last_revision():
 
142
                warning("working tree is out of date, run 'brz update'")
 
143
            new = wt
 
144
            old = new.basis_tree()
 
145
        elif len(revision) > 0:
 
146
            try:
 
147
                old = revision[0].as_tree(wt.branch)
 
148
            except errors.NoSuchRevision as e:
 
149
                raise errors.BzrCommandError(str(e))
 
150
            if (len(revision) > 1) and (revision[1].spec is not None):
 
151
                try:
 
152
                    new = revision[1].as_tree(wt.branch)
 
153
                    new_is_working_tree = False
 
154
                except errors.NoSuchRevision as e:
 
155
                    raise errors.BzrCommandError(str(e))
 
156
            else:
 
157
                new = wt
 
158
        old.lock_read()
 
159
        new.lock_read()
 
160
        try:
 
161
            for hook in hooks['pre_status']:
 
162
                hook(StatusHookParams(old, new, to_file, versioned,
 
163
                    show_ids, short, verbose, specific_files=specific_files))
 
164
 
 
165
            specific_files, nonexistents \
 
166
                = _filter_nonexistent(specific_files, old, new)
 
167
            want_unversioned = not versioned
 
168
 
 
169
            # Reporter used for short outputs
 
170
            reporter = _mod_delta._ChangeReporter(output_file=to_file,
 
171
                unversioned_filter=new.is_ignored, classify=classify)
 
172
            report_changes(to_file, old, new, specific_files,
 
173
                           reporter, show_long_callback,
 
174
                           short=short, want_unchanged=show_unchanged,
 
175
                           want_unversioned=want_unversioned, show_ids=show_ids,
 
176
                           classify=classify)
 
177
 
 
178
            # show the ignored files among specific files (i.e. show the files
 
179
            # identified from input that we choose to ignore).
 
180
            if specific_files is not None:
 
181
                # Ignored files is sorted because specific_files is already sorted
 
182
                ignored_files = [specific for specific in
 
183
                    specific_files if new.is_ignored(specific)]
 
184
                if len(ignored_files) > 0 and not short:
 
185
                    to_file.write("ignored:\n")
 
186
                    prefix = ' '
 
187
                else:
 
188
                    prefix = 'I  '
 
189
                for ignored_file in ignored_files:
 
190
                    to_file.write("%s %s\n" % (prefix, ignored_file))
 
191
 
 
192
            # show the new conflicts only for now. XXX: get them from the
 
193
            # delta.
 
194
            conflicts = new.conflicts()
 
195
            if specific_files is not None:
 
196
                conflicts = conflicts.select_conflicts(new, specific_files,
 
197
                    ignore_misses=True, recurse=True)[1]
 
198
            if len(conflicts) > 0 and not short:
 
199
                to_file.write("conflicts:\n")
 
200
            for conflict in conflicts:
 
201
                if short:
 
202
                    prefix = 'C  '
 
203
                else:
 
204
                    prefix = ' '
 
205
                to_file.write("%s %s\n" % (prefix, unicode(conflict)))
 
206
            # Show files that were requested but don't exist (and are
 
207
            # not versioned).  We don't involve delta in this; these
 
208
            # paths are really the province of just the status
 
209
            # command, since they have more to do with how it was
 
210
            # invoked than with the tree it's operating on.
 
211
            if nonexistents and not short:
 
212
                to_file.write("nonexistent:\n")
 
213
            for nonexistent in nonexistents:
 
214
                # We could calculate prefix outside the loop but, given
 
215
                # how rarely this ought to happen, it's OK and arguably
 
216
                # slightly faster to do it here (ala conflicts above)
 
217
                if short:
 
218
                    prefix = 'X  '
 
219
                else:
 
220
                    prefix = ' '
 
221
                to_file.write("%s %s\n" % (prefix, nonexistent))
 
222
            if (new_is_working_tree and show_pending):
 
223
                show_pending_merges(new, to_file, short, verbose=verbose)
 
224
            if nonexistents:
 
225
                raise errors.PathsDoNotExist(nonexistents)
 
226
            for hook in hooks['post_status']:
 
227
                hook(StatusHookParams(old, new, to_file, versioned,
 
228
                    show_ids, short, verbose, specific_files=specific_files))
 
229
        finally:
 
230
            old.unlock()
 
231
            new.unlock()
 
232
    finally:
 
233
        wt.unlock()
 
234
 
 
235
 
 
236
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
 
237
    """Get an iterator which will return the revisions in merge sorted order.
 
238
 
 
239
    This will build up a list of all nodes, such that only nodes in the list
 
240
    are referenced. It then uses MergeSorter to return them in 'merge-sorted'
 
241
    order.
 
242
 
 
243
    :param revision_ids: A set of revision_ids
 
244
    :param parent_map: The parent information for each node. Revisions which
 
245
        are considered ghosts should not be present in the map.
 
246
    :return: iterator from MergeSorter.iter_topo_order()
 
247
    """
 
248
    # MergeSorter requires that all nodes be present in the graph, so get rid
 
249
    # of any references pointing outside of this graph.
 
250
    parent_graph = {}
 
251
    for revision_id in revision_ids:
 
252
        if revision_id not in parent_map: # ghost
 
253
            parent_graph[revision_id] = []
 
254
        else:
 
255
            # Only include parents which are in this sub-graph
 
256
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
 
257
                                            if p in revision_ids]
 
258
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
 
259
    return sorter.iter_topo_order()
 
260
 
 
261
 
 
262
def show_pending_merges(new, to_file, short=False, verbose=False):
 
263
    """Write out a display of pending merges in a working tree."""
 
264
    parents = new.get_parent_ids()
 
265
    if len(parents) < 2:
 
266
        return
 
267
 
 
268
    term_width = osutils.terminal_width()
 
269
    if term_width is not None:
 
270
        # we need one extra space for terminals that wrap on last char
 
271
        term_width = term_width - 1
 
272
    if short:
 
273
        first_prefix = 'P   '
 
274
        sub_prefix = 'P.   '
 
275
    else:
 
276
        first_prefix = '  '
 
277
        sub_prefix = '    '
 
278
 
 
279
    def show_log_message(rev, prefix):
 
280
        if term_width is None:
 
281
            width = term_width
 
282
        else:
 
283
            width = term_width - len(prefix)
 
284
        log_message = log_formatter.log_string(None, rev, width, prefix=prefix)
 
285
        to_file.write(log_message + '\n')
 
286
 
 
287
    pending = parents[1:]
 
288
    branch = new.branch
 
289
    last_revision = parents[0]
 
290
    if not short:
 
291
        if verbose:
 
292
            to_file.write('pending merges:\n')
 
293
        else:
 
294
            to_file.write('pending merge tips:'
 
295
                          ' (use -v to see all merge revisions)\n')
 
296
    graph = branch.repository.get_graph()
 
297
    other_revisions = [last_revision]
 
298
    log_formatter = log.LineLogFormatter(to_file)
 
299
    for merge in pending:
 
300
        try:
 
301
            rev = branch.repository.get_revision(merge)
 
302
        except errors.NoSuchRevision:
 
303
            # If we are missing a revision, just print out the revision id
 
304
            to_file.write(first_prefix + '(ghost) ' + merge + '\n')
 
305
            other_revisions.append(merge)
 
306
            continue
 
307
 
 
308
        # Log the merge, as it gets a slightly different formatting
 
309
        show_log_message(rev, first_prefix)
 
310
        if not verbose:
 
311
            continue
 
312
 
 
313
        # Find all of the revisions in the merge source, which are not in the
 
314
        # last committed revision.
 
315
        merge_extra = graph.find_unique_ancestors(merge, other_revisions)
 
316
        other_revisions.append(merge)
 
317
        merge_extra.discard(_mod_revision.NULL_REVISION)
 
318
 
 
319
        # Get a handle to all of the revisions we will need
 
320
        revisions = dict(branch.repository.iter_revisions(merge_extra))
 
321
 
 
322
        # Display the revisions brought in by this merge.
 
323
        rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
 
324
                            branch.repository.get_parent_map(merge_extra))
 
325
        # Skip the first node
 
326
        num, first, depth, eom = next(rev_id_iterator)
 
327
        if first != merge:
 
328
            raise AssertionError('Somehow we misunderstood how'
 
329
                ' iter_topo_order works %s != %s' % (first, merge))
 
330
        for num, sub_merge, depth, eom in rev_id_iterator:
 
331
            rev = revisions[sub_merge]
 
332
            if rev is None:
 
333
                to_file.write(sub_prefix + '(ghost) ' + sub_merge + '\n')
 
334
                continue
 
335
            show_log_message(revisions[sub_merge], sub_prefix)
 
336
 
 
337
 
 
338
def _filter_nonexistent(orig_paths, old_tree, new_tree):
 
339
    """Convert orig_paths to two sorted lists and return them.
 
340
 
 
341
    The first is orig_paths paths minus the items in the second list,
 
342
    and the second list is paths that are not in either inventory or
 
343
    tree (they don't qualify if they exist in the tree's inventory, or
 
344
    if they exist in the tree but are not versioned.)
 
345
 
 
346
    If either of the two lists is empty, return it as an empty list.
 
347
 
 
348
    This can be used by operations such as brz status that can accept
 
349
    unknown or ignored files.
 
350
    """
 
351
    mutter("check paths: %r", orig_paths)
 
352
    if not orig_paths:
 
353
        return orig_paths, []
 
354
    s = old_tree.filter_unversioned_files(orig_paths)
 
355
    s = new_tree.filter_unversioned_files(s)
 
356
    nonexistent = [path for path in s if not new_tree.has_filename(path)]
 
357
    remaining   = [path for path in orig_paths if not path in nonexistent]
 
358
    # Sorting the 'remaining' list doesn't have much effect in
 
359
    # practice, since the various status output sections will sort
 
360
    # their groups individually.  But for consistency of this
 
361
    # function's API, it's better to sort both than just 'nonexistent'.
 
362
    return sorted(remaining), sorted(nonexistent)
 
363
 
 
364
 
 
365
class StatusHooks(_mod_hooks.Hooks):
 
366
    """A dictionary mapping hook name to a list of callables for status hooks.
 
367
 
 
368
    e.g. ['post_status'] Is the list of items to be called when the
 
369
    status command has finished printing the status.
 
370
    """
 
371
 
 
372
    def __init__(self):
 
373
        """Create the default hooks.
 
374
 
 
375
        These are all empty initially, because by default nothing should get
 
376
        notified.
 
377
        """
 
378
        _mod_hooks.Hooks.__init__(self, "breezy.status", "hooks")
 
379
        self.add_hook('post_status',
 
380
            "Called with argument StatusHookParams after Bazaar has "
 
381
            "displayed the status. StatusHookParams has the attributes "
 
382
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
 
383
            "verbose). The last four arguments correspond to the command "
 
384
            "line options specified by the user for the status command. "
 
385
            "to_file is the output stream for writing.",
 
386
            (2, 3))
 
387
        self.add_hook('pre_status',
 
388
            "Called with argument StatusHookParams before Bazaar "
 
389
            "displays the status. StatusHookParams has the attributes "
 
390
            "(old_tree, new_tree, to_file, versioned, show_ids, short, "
 
391
            "verbose). The last four arguments correspond to the command "
 
392
            "line options specified by the user for the status command. "
 
393
            "to_file is the output stream for writing.",
 
394
            (2, 3))
 
395
 
 
396
 
 
397
class StatusHookParams(object):
 
398
    """Object holding parameters passed to post_status hooks.
 
399
 
 
400
    :ivar old_tree: Start tree (basis tree) for comparison.
 
401
    :ivar new_tree: Working tree.
 
402
    :ivar to_file: If set, write to this file.
 
403
    :ivar versioned: Show only versioned files.
 
404
    :ivar show_ids: Show internal object ids.
 
405
    :ivar short: Use short status indicators.
 
406
    :ivar verbose: Verbose flag.
 
407
    """
 
408
 
 
409
    def __init__(self, old_tree, new_tree, to_file, versioned, show_ids,
 
410
            short, verbose, specific_files=None):
 
411
        """Create a group of post_status hook parameters.
 
412
 
 
413
        :param old_tree: Start tree (basis tree) for comparison.
 
414
        :param new_tree: Working tree.
 
415
        :param to_file: If set, write to this file.
 
416
        :param versioned: Show only versioned files.
 
417
        :param show_ids: Show internal object ids.
 
418
        :param short: Use short status indicators.
 
419
        :param verbose: Verbose flag.
 
420
        :param specific_files: If set, a list of filenames whose status should be
 
421
            shown.  It is an error to give a filename that is not in the working
 
422
            tree, or in the working inventory or in the basis inventory.
 
423
        """
 
424
        self.old_tree = old_tree
 
425
        self.new_tree = new_tree
 
426
        self.to_file = to_file
 
427
        self.versioned = versioned
 
428
        self.show_ids = show_ids
 
429
        self.short = short
 
430
        self.verbose = verbose
 
431
        self.specific_files = specific_files
 
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, %s)>" % (self.__class__.__name__,
 
438
            self.old_tree, self.new_tree, self.to_file, self.versioned,
 
439
            self.show_ids, self.short, self.verbose, self.specific_files)
 
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
    # Don't show shelves if status of specific files is being shown, only if
 
448
    # no file arguments have been passed
 
449
    if params.specific_files:
 
450
        return
 
451
    get_shelf_manager = getattr(params.new_tree, 'get_shelf_manager', None)
 
452
    if get_shelf_manager is None:
 
453
        return
 
454
    try:
 
455
        manager = get_shelf_manager()
 
456
    except ShelvingUnsupported:
 
457
        mutter('shelving not supported by tree, not displaying shelves.')
 
458
    else:
 
459
        shelves = manager.active_shelves()
 
460
        if shelves:
 
461
            singular = '%d shelf exists. '
 
462
            plural = '%d shelves exist. '
 
463
            if len(shelves) == 1:
 
464
                fmt = singular
 
465
            else:
 
466
                fmt = plural
 
467
            params.to_file.write(fmt % len(shelves))
 
468
            params.to_file.write('See "brz shelve --list" for details.\n')
 
469
 
 
470
 
 
471
hooks = StatusHooks()
 
472
 
 
473
 
 
474
hooks.install_named_hook('post_status', _show_shelve_summary,
 
475
    'brz status')
 
476