/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

Update the dirstate tests to lock and unlock properly.
Change the lock.WriteLock functions so that they open in 'rb+' mode instead
of automatically truncating the file.
test_dirstate.py now passes, but we need to figure out or locking mechanism for
WorkingTree4. (does it just lock at the same time as Tree.lock_write()?)

Show diffs side-by-side

added added

removed removed

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