~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/status.py

Don't encode unicode messages to UTF-8 in mutter() (the stream writer does it).

Use a codec wrapped log file in tests to match the real environment.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# (C) 2005 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
15
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
import sys
18
 
 
19
 
from bzrlib import (
20
 
    delta as _mod_delta,
21
 
    tree,
22
 
    )
23
 
from bzrlib.diff import _raise_if_nonexistent
24
 
import bzrlib.errors as errors
 
18
from bzrlib.osutils import is_inside_any
 
19
from bzrlib.delta import compare_trees
25
20
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
31
 
 
32
 
# TODO: when showing single-line logs, truncate to the width of the terminal
33
 
# if known, but only if really going to the terminal (not into a file)
34
 
 
35
 
 
36
 
@deprecated_function(zero_eight)
37
 
def show_status(branch, show_unchanged=None,
 
21
from bzrlib.errors import NoSuchRevision
 
22
 
 
23
 
 
24
def show_status(branch, show_unchanged=False,
38
25
                specific_files=None,
39
26
                show_ids=False,
40
27
                to_file=None,
41
28
                show_pending=True,
42
29
                revision=None):
43
 
    """Display summary of changes.
44
 
 
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.
 
30
    """Display status for non-ignored working files.
53
31
 
54
32
    show_unchanged
55
33
        If set, includes unchanged files.
71
49
        If one revision show compared it with working tree.
72
50
        If two revisions show status between first and second.
73
51
    """
74
 
    show_tree_status(branch.bzrdir.open_workingtree(), show_unchanged, 
75
 
                     specific_files, show_ids, to_file, show_pending, revision)
76
 
 
77
 
 
78
 
def show_tree_status(wt, show_unchanged=None,
79
 
                     specific_files=None,
80
 
                     show_ids=False,
81
 
                     to_file=None,
82
 
                     show_pending=True,
83
 
                     revision=None,
84
 
                     short=False,
85
 
                     versioned=False):
86
 
    """Display summary of changes.
87
 
 
88
 
    By default this compares the working tree to a previous revision. 
89
 
    If the revision argument is given, summarizes changes between the 
90
 
    working tree and another, or between two revisions.
91
 
 
92
 
    The result is written out as Unicode and to_file should be able 
93
 
    to encode that.
94
 
 
95
 
    If showing the status of a working tree, extra information is included
96
 
    about unknown files, conflicts, and pending merges.
97
 
 
98
 
    :param show_unchanged: Deprecated parameter. If set, includes unchanged 
99
 
        files.
100
 
    :param specific_files: If set, a list of filenames whose status should be
101
 
        shown.  It is an error to give a filename that is not in the working 
102
 
        tree, or in the working inventory or in the basis inventory.
103
 
    :param show_ids: If set, includes each file's id.
104
 
    :param to_file: If set, write to this file (default stdout.)
105
 
    :param show_pending: If set, write pending merges.
106
 
    :param revision: If None the compare latest revision with working tree
107
 
        If not None it must be a RevisionSpec list.
108
 
        If one revision show compared it with working tree.
109
 
        If two revisions show status between first and second.
110
 
    :param short: If True, gives short SVN-style status lines.
111
 
    :param versioned: If True, only shows versioned files.
112
 
    """
113
 
    if show_unchanged is not None:
114
 
        warn("show_status_trees with show_unchanged has been deprecated "
115
 
             "since bzrlib 0.9", DeprecationWarning, stacklevel=2)
116
 
 
117
 
    if to_file is None:
 
52
    if to_file == None:
118
53
        to_file = sys.stdout
119
54
    
120
 
    wt.lock_read()
 
55
    branch.lock_read()
121
56
    try:
122
57
        new_is_working_tree = True
123
58
        if revision is None:
124
 
            if wt.last_revision() != wt.branch.last_revision():
125
 
                warning("working tree is out of date, run 'bzr update'")
126
 
            new = wt
127
 
            old = new.basis_tree()
 
59
            old = branch.basis_tree()
 
60
            new = branch.working_tree()
128
61
        elif len(revision) > 0:
129
62
            try:
130
 
                rev_id = revision[0].in_history(wt.branch).rev_id
131
 
                old = wt.branch.repository.revision_tree(rev_id)
132
 
            except errors.NoSuchRevision, e:
133
 
                raise errors.BzrCommandError(str(e))
134
 
            if (len(revision) > 1) and (revision[1].spec is not None):
 
63
                rev_id = revision[0].in_history(branch).rev_id
 
64
                old = branch.revision_tree(rev_id)
 
65
            except NoSuchRevision, e:
 
66
                raise BzrCommandError(str(e))
 
67
            if len(revision) > 1:
135
68
                try:
136
 
                    rev_id = revision[1].in_history(wt.branch).rev_id
137
 
                    new = wt.branch.repository.revision_tree(rev_id)
 
69
                    rev_id = revision[1].in_history(branch).rev_id
 
70
                    new = branch.revision_tree(rev_id)
138
71
                    new_is_working_tree = False
139
 
                except errors.NoSuchRevision, e:
140
 
                    raise errors.BzrCommandError(str(e))
141
 
            else:
142
 
                new = wt
143
 
        old.lock_read()
144
 
        new.lock_read()
145
 
        try:
146
 
            _raise_if_nonexistent(specific_files, old, new)
147
 
            want_unversioned = not versioned
148
 
            if short:
149
 
                changes = new._iter_changes(old, show_unchanged, specific_files,
150
 
                    require_versioned=False, want_unversioned=want_unversioned)
151
 
                reporter = _mod_delta._ChangeReporter(output_file=to_file,
152
 
                    unversioned_filter=new.is_ignored)
153
 
                _mod_delta.report_changes(changes, reporter)
154
 
            else:
155
 
                delta = new.changes_from(old, want_unchanged=show_unchanged,
156
 
                                      specific_files=specific_files,
157
 
                                      want_unversioned=want_unversioned)
158
 
                # filter out unknown files. We may want a tree method for
159
 
                # this
160
 
                delta.unversioned = [unversioned for unversioned in
161
 
                    delta.unversioned if not new.is_ignored(unversioned[0])]
162
 
                delta.show(to_file,
163
 
                           show_ids=show_ids,
164
 
                           show_unchanged=show_unchanged,
165
 
                           short_status=False)
166
 
            conflict_title = False
167
 
            # show the new conflicts only for now. XXX: get them from the delta.
168
 
            for conflict in new.conflicts():
169
 
                if not short and conflict_title is False:
170
 
                    print >> to_file, "conflicts:"
171
 
                    conflict_title = True
172
 
                if short:
173
 
                    prefix = 'C  '
 
72
                except NoSuchRevision, e:
 
73
                    raise BzrCommandError(str(e))
 
74
            else:
 
75
                new = branch.working_tree()
 
76
                
 
77
 
 
78
        delta = compare_trees(old, new, want_unchanged=show_unchanged,
 
79
                              specific_files=specific_files)
 
80
 
 
81
        delta.show(to_file,
 
82
                   show_ids=show_ids,
 
83
                   show_unchanged=show_unchanged)
 
84
 
 
85
        if new_is_working_tree:
 
86
            conflicts = new.iter_conflicts()
 
87
            unknowns = new.unknowns()
 
88
            list_paths('unknown', unknowns, specific_files, to_file)
 
89
            list_paths('conflicts', conflicts, specific_files, to_file)
 
90
            if show_pending and len(new.pending_merges()) > 0:
 
91
                print >>to_file, 'pending merges:'
 
92
                last_revision = branch.last_revision()
 
93
                if last_revision is not None:
 
94
                    ignore = set(branch.get_ancestry(last_revision))
174
95
                else:
175
 
                    prefix = ' '
176
 
                print >> to_file, "%s %s" % (prefix, conflict)
177
 
            if new_is_working_tree and show_pending:
178
 
                show_pending_merges(new, to_file, short)
179
 
        finally:
180
 
            old.unlock()
181
 
            new.unlock()
 
96
                    ignore = set()
 
97
                for merge in new.pending_merges():
 
98
                    ignore.add(merge)
 
99
                    try:
 
100
                        m_revision = branch.get_revision(merge)
 
101
                        print >> to_file, ' ', line_log(m_revision, 77)
 
102
                        inner_merges = branch.get_ancestry(merge)
 
103
                        inner_merges.reverse()
 
104
                        for mmerge in inner_merges:
 
105
                            if mmerge in ignore:
 
106
                                continue
 
107
                            mm_revision = branch.get_revision(mmerge)
 
108
                            print >> to_file, '   ', line_log(mm_revision, 75)
 
109
                            ignore.add(mmerge)
 
110
                            
 
111
                    except NoSuchRevision:
 
112
                        print >> to_file, ' ', merge 
 
113
                        
182
114
    finally:
183
 
        wt.unlock()
184
 
 
185
 
def show_pending_merges(new, to_file, short=False):
186
 
    """Write out a display of pending merges in a working tree."""
187
 
    parents = new.get_parent_ids()
188
 
    if len(parents) < 2:
189
 
        return
190
 
    pending = parents[1:]
191
 
    branch = new.branch
192
 
    last_revision = parents[0]
193
 
    if not short:
194
 
        print >>to_file, 'pending merges:'
195
 
    if last_revision is not None:
196
 
        try:
197
 
            ignore = set(branch.repository.get_ancestry(last_revision))
198
 
        except errors.NoSuchRevision:
199
 
            # the last revision is a ghost : assume everything is new 
200
 
            # except for it
201
 
            ignore = set([None, last_revision])
202
 
    else:
203
 
        ignore = set([None])
204
 
    # TODO: this could be improved using merge_sorted - we'd get the same 
205
 
    # output rather than one level of indent.
206
 
    for merge in pending:
207
 
        ignore.add(merge)
208
 
        try:
209
 
            from bzrlib.osutils import terminal_width
210
 
            width = terminal_width()
211
 
            m_revision = branch.repository.get_revision(merge)
212
 
            if short:
213
 
                prefix = 'P  '
214
 
            else:
215
 
                prefix = ' '
216
 
            print >> to_file, prefix, line_log(m_revision, width - 4)
217
 
            inner_merges = branch.repository.get_ancestry(merge)
218
 
            assert inner_merges[0] is None
219
 
            inner_merges.pop(0)
220
 
            inner_merges.reverse()
221
 
            for mmerge in inner_merges:
222
 
                if mmerge in ignore:
223
 
                    continue
224
 
                mm_revision = branch.repository.get_revision(mmerge)
225
 
                if short:
226
 
                    prefix = 'P.  '
227
 
                else:
228
 
                    prefix = '   '
229
 
                print >> to_file, prefix, line_log(mm_revision, width - 5)
230
 
                ignore.add(mmerge)
231
 
        except errors.NoSuchRevision:
232
 
            if short:
233
 
                prefix = 'P  '
234
 
            else:
235
 
                prefix = ' '
236
 
            print >> to_file, prefix, merge
 
115
        branch.unlock()
 
116
        
 
117
def list_paths(header, paths, specific_files, to_file):
 
118
    done_header = False
 
119
    for path in paths:
 
120
        if specific_files and not is_inside_any(specific_files, path):
 
121
            continue
 
122
        if not done_header:
 
123
            print >>to_file, '%s:' % header
 
124
            done_header = True
 
125
        print >>to_file, ' ', path