/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 bzrlib/status.py

  • Committer: Jelmer Vernooij
  • Date: 2010-04-30 11:03:59 UTC
  • mto: This revision was merged to the branch mainline in revision 5197.
  • Revision ID: jelmer@samba.org-20100430110359-ow3e3grh7sxy93pa
Remove more unused imports.

Show diffs side-by-side

added added

removed removed

Lines of Context:
bzrlib/status.py
 
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
    log,
 
22
    osutils,
 
23
    tsort,
 
24
    revision as _mod_revision,
 
25
    )
 
26
import bzrlib.errors as errors
 
27
from bzrlib.trace import mutter, warning
 
28
 
 
29
# TODO: when showing single-line logs, truncate to the width of the terminal
 
30
# if known, but only if really going to the terminal (not into a file)
 
31
 
 
32
 
 
33
def show_tree_status(wt, show_unchanged=None,
 
34
                     specific_files=None,
 
35
                     show_ids=False,
 
36
                     to_file=None,
 
37
                     show_pending=True,
 
38
                     revision=None,
 
39
                     short=False,
 
40
                     verbose=False,
 
41
                     versioned=False):
 
42
    """Display summary of changes.
 
43
 
 
44
    By default this compares the working tree to a previous revision.
 
45
    If the revision argument is given, summarizes changes between the
 
46
    working tree and another, or between two revisions.
 
47
 
 
48
    The result is written out as Unicode and to_file should be able
 
49
    to encode that.
 
50
 
 
51
    If showing the status of a working tree, extra information is included
 
52
    about unknown files, conflicts, and pending merges.
 
53
 
 
54
    :param show_unchanged: Deprecated parameter. If set, includes unchanged
 
55
        files.
 
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_ids: If set, includes each file's id.
 
60
    :param to_file: If set, write to this file (default stdout.)
 
61
    :param show_pending: If set, write pending merges.
 
62
    :param revision: If None, compare latest revision with working tree
 
63
        If not None, it must be a RevisionSpec list.
 
64
        If one revision, compare with working tree.
 
65
        If two revisions, show status between first and second.
 
66
    :param short: If True, gives short SVN-style status lines.
 
67
    :param verbose: If True, show all merged revisions, not just
 
68
        the merge tips
 
69
    :param versioned: If True, only shows versioned files.
 
70
    """
 
71
    if show_unchanged is not None:
 
72
        warn("show_tree_status with show_unchanged has been deprecated "
 
73
             "since bzrlib 0.9", DeprecationWarning, stacklevel=2)
 
74
 
 
75
    if to_file is None:
 
76
        to_file = sys.stdout
 
77
 
 
78
    wt.lock_read()
 
79
    try:
 
80
        new_is_working_tree = True
 
81
        if revision is None:
 
82
            if wt.last_revision() != wt.branch.last_revision():
 
83
                warning("working tree is out of date, run 'bzr update'")
 
84
            new = wt
 
85
            old = new.basis_tree()
 
86
        elif len(revision) > 0:
 
87
            try:
 
88
                old = revision[0].as_tree(wt.branch)
 
89
            except errors.NoSuchRevision, e:
 
90
                raise errors.BzrCommandError(str(e))
 
91
            if (len(revision) > 1) and (revision[1].spec is not None):
 
92
                try:
 
93
                    new = revision[1].as_tree(wt.branch)
 
94
                    new_is_working_tree = False
 
95
                except errors.NoSuchRevision, e:
 
96
                    raise errors.BzrCommandError(str(e))
 
97
            else:
 
98
                new = wt
 
99
        old.lock_read()
 
100
        new.lock_read()
 
101
        try:
 
102
            specific_files, nonexistents \
 
103
                = _filter_nonexistent(specific_files, old, new)
 
104
            want_unversioned = not versioned
 
105
            if short:
 
106
                changes = new.iter_changes(old, show_unchanged, specific_files,
 
107
                    require_versioned=False, want_unversioned=want_unversioned)
 
108
                reporter = _mod_delta._ChangeReporter(output_file=to_file,
 
109
                    unversioned_filter=new.is_ignored)
 
110
                _mod_delta.report_changes(changes, reporter)
 
111
            else:
 
112
                delta = new.changes_from(old, want_unchanged=show_unchanged,
 
113
                                      specific_files=specific_files,
 
114
                                      want_unversioned=want_unversioned)
 
115
                # filter out unknown files. We may want a tree method for
 
116
                # this
 
117
                delta.unversioned = [unversioned for unversioned in
 
118
                    delta.unversioned if not new.is_ignored(unversioned[0])]
 
119
                delta.show(to_file,
 
120
                           show_ids=show_ids,
 
121
                           show_unchanged=show_unchanged,
 
122
                           short_status=False)
 
123
            # show the new conflicts only for now. XXX: get them from the
 
124
            # delta.
 
125
            conflicts = new.conflicts()
 
126
            if specific_files is not None:
 
127
                conflicts = conflicts.select_conflicts(new, specific_files,
 
128
                    ignore_misses=True, recurse=True)[1]
 
129
            if len(conflicts) > 0 and not short:
 
130
                to_file.write("conflicts:\n")
 
131
            for conflict in conflicts:
 
132
                if short:
 
133
                    prefix = 'C  '
 
134
                else:
 
135
                    prefix = ' '
 
136
                to_file.write("%s %s\n" % (prefix, conflict))
 
137
            # Show files that were requested but don't exist (and are
 
138
            # not versioned).  We don't involve delta in this; these
 
139
            # paths are really the province of just the status
 
140
            # command, since they have more to do with how it was
 
141
            # invoked than with the tree it's operating on.
 
142
            if nonexistents and not short:
 
143
                to_file.write("nonexistent:\n")
 
144
            for nonexistent in nonexistents:
 
145
                # We could calculate prefix outside the loop but, given
 
146
                # how rarely this ought to happen, it's OK and arguably
 
147
                # slightly faster to do it here (ala conflicts above)
 
148
                if short:
 
149
                    prefix = 'X  '
 
150
                else:
 
151
                    prefix = ' '
 
152
                to_file.write("%s %s\n" % (prefix, nonexistent))
 
153
            if (new_is_working_tree and show_pending):
 
154
                show_pending_merges(new, to_file, short, verbose=verbose)
 
155
            if nonexistents:
 
156
                raise errors.PathsDoNotExist(nonexistents)
 
157
        finally:
 
158
            old.unlock()
 
159
            new.unlock()
 
160
    finally:
 
161
        wt.unlock()
 
162
 
 
163
 
 
164
def _get_sorted_revisions(tip_revision, revision_ids, parent_map):
 
165
    """Get an iterator which will return the revisions in merge sorted order.
 
166
 
 
167
    This will build up a list of all nodes, such that only nodes in the list
 
168
    are referenced. It then uses MergeSorter to return them in 'merge-sorted'
 
169
    order.
 
170
 
 
171
    :param revision_ids: A set of revision_ids
 
172
    :param parent_map: The parent information for each node. Revisions which
 
173
        are considered ghosts should not be present in the map.
 
174
    :return: iterator from MergeSorter.iter_topo_order()
 
175
    """
 
176
    # MergeSorter requires that all nodes be present in the graph, so get rid
 
177
    # of any references pointing outside of this graph.
 
178
    parent_graph = {}
 
179
    for revision_id in revision_ids:
 
180
        if revision_id not in parent_map: # ghost
 
181
            parent_graph[revision_id] = []
 
182
        else:
 
183
            # Only include parents which are in this sub-graph
 
184
            parent_graph[revision_id] = [p for p in parent_map[revision_id]
 
185
                                            if p in revision_ids]
 
186
    sorter = tsort.MergeSorter(parent_graph, tip_revision)
 
187
    return sorter.iter_topo_order()
 
188
 
 
189
 
 
190
def show_pending_merges(new, to_file, short=False, verbose=False):
 
191
    """Write out a display of pending merges in a working tree."""
 
192
    parents = new.get_parent_ids()
 
193
    if len(parents) < 2:
 
194
        return
 
195
 
 
196
    term_width = osutils.terminal_width()
 
197
    if term_width is not None:
 
198
        # we need one extra space for terminals that wrap on last char
 
199
        term_width = term_width - 1
 
200
    if short:
 
201
        first_prefix = 'P   '
 
202
        sub_prefix = 'P.   '
 
203
    else:
 
204
        first_prefix = '  '
 
205
        sub_prefix = '    '
 
206
 
 
207
    def show_log_message(rev, prefix):
 
208
        if term_width is None:
 
209
            width = term_width
 
210
        else:
 
211
            width = term_width - len(prefix)
 
212
        log_message = log_formatter.log_string(None, rev, width, prefix=prefix)
 
213
        to_file.write(log_message + '\n')
 
214
 
 
215
    pending = parents[1:]
 
216
    branch = new.branch
 
217
    last_revision = parents[0]
 
218
    if not short:
 
219
        if verbose:
 
220
            to_file.write('pending merges:\n')
 
221
        else:
 
222
            to_file.write('pending merge tips:'
 
223
                          ' (use -v to see all merge revisions)\n')
 
224
    graph = branch.repository.get_graph()
 
225
    other_revisions = [last_revision]
 
226
    log_formatter = log.LineLogFormatter(to_file)
 
227
    for merge in pending:
 
228
        try:
 
229
            rev = branch.repository.get_revisions([merge])[0]
 
230
        except errors.NoSuchRevision:
 
231
            # If we are missing a revision, just print out the revision id
 
232
            to_file.write(first_prefix + '(ghost) ' + merge + '\n')
 
233
            other_revisions.append(merge)
 
234
            continue
 
235
 
 
236
        # Log the merge, as it gets a slightly different formatting
 
237
        show_log_message(rev, first_prefix)
 
238
        if not verbose:
 
239
            continue
 
240
 
 
241
        # Find all of the revisions in the merge source, which are not in the
 
242
        # last committed revision.
 
243
        merge_extra = graph.find_unique_ancestors(merge, other_revisions)
 
244
        other_revisions.append(merge)
 
245
        merge_extra.discard(_mod_revision.NULL_REVISION)
 
246
 
 
247
        # Get a handle to all of the revisions we will need
 
248
        try:
 
249
            revisions = dict((rev.revision_id, rev) for rev in
 
250
                             branch.repository.get_revisions(merge_extra))
 
251
        except errors.NoSuchRevision:
 
252
            # One of the sub nodes is a ghost, check each one
 
253
            revisions = {}
 
254
            for revision_id in merge_extra:
 
255
                try:
 
256
                    rev = branch.repository.get_revisions([revision_id])[0]
 
257
                except errors.NoSuchRevision:
 
258
                    revisions[revision_id] = None
 
259
                else:
 
260
                    revisions[revision_id] = rev
 
261
 
 
262
        # Display the revisions brought in by this merge.
 
263
        rev_id_iterator = _get_sorted_revisions(merge, merge_extra,
 
264
                            branch.repository.get_parent_map(merge_extra))
 
265
        # Skip the first node
 
266
        num, first, depth, eom = rev_id_iterator.next()
 
267
        if first != merge:
 
268
            raise AssertionError('Somehow we misunderstood how'
 
269
                ' iter_topo_order works %s != %s' % (first, merge))
 
270
        for num, sub_merge, depth, eom in rev_id_iterator:
 
271
            rev = revisions[sub_merge]
 
272
            if rev is None:
 
273
                to_file.write(sub_prefix + '(ghost) ' + sub_merge + '\n')
 
274
                continue
 
275
            show_log_message(revisions[sub_merge], sub_prefix)
 
276
 
 
277
 
 
278
def _filter_nonexistent(orig_paths, old_tree, new_tree):
 
279
    """Convert orig_paths to two sorted lists and return them.
 
280
 
 
281
    The first is orig_paths paths minus the items in the second list,
 
282
    and the second list is paths that are not in either inventory or
 
283
    tree (they don't qualify if they exist in the tree's inventory, or
 
284
    if they exist in the tree but are not versioned.)
 
285
 
 
286
    If either of the two lists is empty, return it as an empty list.
 
287
 
 
288
    This can be used by operations such as bzr status that can accept
 
289
    unknown or ignored files.
 
290
    """
 
291
    mutter("check paths: %r", orig_paths)
 
292
    if not orig_paths:
 
293
        return orig_paths, []
 
294
    s = old_tree.filter_unversioned_files(orig_paths)
 
295
    s = new_tree.filter_unversioned_files(s)
 
296
    nonexistent = [path for path in s if not new_tree.has_filename(path)]
 
297
    remaining   = [path for path in orig_paths if not path in nonexistent]
 
298
    # Sorting the 'remaining' list doesn't have much effect in
 
299
    # practice, since the various status output sections will sort
 
300
    # their groups individually.  But for consistency of this
 
301
    # function's API, it's better to sort both than just 'nonexistent'.
 
302
    return sorted(remaining), sorted(nonexistent)