/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/workingtree.py

[merge] robertc's integration, updated tests to check for retcode=3

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005-2011 Canonical Ltd
2
 
#
 
1
# Copyright (C) 2005 Canonical Ltd
 
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
5
5
# the Free Software Foundation; either version 2 of the License, or
6
6
# (at your option) any later version.
7
 
#
 
7
 
8
8
# This program is distributed in the hope that it will be useful,
9
9
# but WITHOUT ANY WARRANTY; without even the implied warranty of
10
10
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
11
# GNU General Public License for more details.
12
 
#
 
12
 
13
13
# You should have received a copy of the GNU General Public License
14
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
 
15
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
16
16
 
17
17
"""WorkingTree object and friends.
18
18
 
19
19
A WorkingTree represents the editable working copy of a branch.
20
 
Operations which represent the WorkingTree are also done here,
21
 
such as renaming or adding files.
 
20
Operations which represent the WorkingTree are also done here, 
 
21
such as renaming or adding files.  The WorkingTree has an inventory 
 
22
which is updated by these operations.  A commit produces a 
 
23
new revision based on the workingtree and its inventory.
22
24
 
23
25
At the moment every WorkingTree has its own branch.  Remote
24
26
WorkingTrees aren't supported.
25
27
 
26
 
To get a WorkingTree, call controldir.open_workingtree() or
27
 
WorkingTree.open(dir).
 
28
To get a WorkingTree, call Branch.working_tree():
28
29
"""
29
30
 
30
 
from __future__ import absolute_import
31
 
 
32
 
import errno
 
31
 
 
32
# TODO: Don't allow WorkingTrees to be constructed for remote branches if 
 
33
# they don't work.
 
34
 
 
35
# FIXME: I don't know if writing out the cache from the destructor is really a
 
36
# good idea, because destructors are considered poor taste in Python, and it's
 
37
# not predictable when it will be written out.
 
38
 
 
39
# TODO: Give the workingtree sole responsibility for the working inventory;
 
40
# remove the variable and references to it from the branch.  This may require
 
41
# updating the commit code so as to update the inventory within the working
 
42
# copy, and making sure there's only one WorkingTree for any directory on disk.
 
43
# At the momenthey may alias the inventory and have old copies of it in memory.
 
44
 
33
45
import os
34
 
import sys
35
 
 
36
 
import breezy
37
 
 
38
 
from .lazy_import import lazy_import
39
 
lazy_import(globals(), """
40
 
import shutil
41
46
import stat
42
 
 
43
 
from breezy import (
44
 
    cleanup,
45
 
    conflicts as _mod_conflicts,
46
 
    filters as _mod_filters,
47
 
    merge,
48
 
    revision as _mod_revision,
49
 
    transform,
50
 
    transport,
51
 
    views,
52
 
    )
53
 
from breezy.bzr import (
54
 
    generate_ids,
55
 
    )
56
 
""")
57
 
 
58
 
from . import (
59
 
    errors,
60
 
    )
61
 
from .controldir import (
62
 
    ControlComponent,
63
 
    ControlComponentFormatRegistry,
64
 
    ControlComponentFormat,
65
 
    ControlDir,
66
 
    ControlDirFormat,
67
 
    )
68
 
from . import (
69
 
    osutils,
70
 
    )
71
 
from .i18n import gettext
72
 
from . import mutabletree
73
 
from .symbol_versioning import deprecated_method, deprecated_in
74
 
from .trace import mutter, note
75
 
 
76
 
 
77
 
class SettingFileIdUnsupported(errors.BzrError):
78
 
 
79
 
    _fmt = "This format does not support setting file ids."
80
 
 
81
 
 
82
 
class ShelvingUnsupported(errors.BzrError):
83
 
 
84
 
    _fmt = "This format does not support shelving changes."
85
 
 
86
 
 
87
 
class WorkingTree(mutabletree.MutableTree, ControlComponent):
 
47
import fnmatch
 
48
 
 
49
from bzrlib.branch import Branch, needs_read_lock, needs_write_lock, quotefn
 
50
import bzrlib.tree
 
51
from bzrlib.osutils import (appendpath,
 
52
                            file_kind,
 
53
                            isdir,
 
54
                            pumpfile,
 
55
                            splitpath,
 
56
                            relpath)
 
57
from bzrlib.errors import BzrCheckError, DivergedBranches, NotVersionedError
 
58
from bzrlib.trace import mutter
 
59
import bzrlib.xml5
 
60
 
 
61
 
 
62
class TreeEntry(object):
 
63
    """An entry that implements the minium interface used by commands.
 
64
 
 
65
    This needs further inspection, it may be better to have 
 
66
    InventoryEntries without ids - though that seems wrong. For now,
 
67
    this is a parallel hierarchy to InventoryEntry, and needs to become
 
68
    one of several things: decorates to that hierarchy, children of, or
 
69
    parents of it.
 
70
    Another note is that these objects are currently only used when there is
 
71
    no InventoryEntry available - i.e. for unversioned objects.
 
72
    Perhaps they should be UnversionedEntry et al. ? - RBC 20051003
 
73
    """
 
74
 
 
75
    def __eq__(self, other):
 
76
        # yes, this us ugly, TODO: best practice __eq__ style.
 
77
        return (isinstance(other, TreeEntry)
 
78
                and other.__class__ == self.__class__)
 
79
 
 
80
    def kind_character(self):
 
81
        return "???"
 
82
 
 
83
 
 
84
class TreeDirectory(TreeEntry):
 
85
    """See TreeEntry. This is a directory in a working tree."""
 
86
 
 
87
    def __eq__(self, other):
 
88
        return (isinstance(other, TreeDirectory)
 
89
                and other.__class__ == self.__class__)
 
90
 
 
91
    def kind_character(self):
 
92
        return "/"
 
93
 
 
94
 
 
95
class TreeFile(TreeEntry):
 
96
    """See TreeEntry. This is a regular file in a working tree."""
 
97
 
 
98
    def __eq__(self, other):
 
99
        return (isinstance(other, TreeFile)
 
100
                and other.__class__ == self.__class__)
 
101
 
 
102
    def kind_character(self):
 
103
        return ''
 
104
 
 
105
 
 
106
class TreeLink(TreeEntry):
 
107
    """See TreeEntry. This is a symlink in a working tree."""
 
108
 
 
109
    def __eq__(self, other):
 
110
        return (isinstance(other, TreeLink)
 
111
                and other.__class__ == self.__class__)
 
112
 
 
113
    def kind_character(self):
 
114
        return ''
 
115
 
 
116
 
 
117
class WorkingTree(bzrlib.tree.Tree):
88
118
    """Working copy tree.
89
119
 
90
 
    :ivar basedir: The root of the tree on disk. This is a unicode path object
91
 
        (as opposed to a URL).
 
120
    The inventory is held in the `Branch` working-inventory, and the
 
121
    files are in a directory on disk.
 
122
 
 
123
    It is possible for a `WorkingTree` to have a filename which is
 
124
    not listed in the Inventory and vice versa.
92
125
    """
93
126
 
94
 
    # override this to set the strategy for storing views
95
 
    def _make_views(self):
96
 
        return views.DisabledViews(self)
97
 
 
98
 
    def __init__(self, basedir='.',
99
 
                 branch=None,
100
 
                 _internal=False,
101
 
                 _transport=None,
102
 
                 _format=None,
103
 
                 _controldir=None):
104
 
        """Construct a WorkingTree instance. This is not a public API.
105
 
 
106
 
        :param branch: A branch to override probing for the branch.
107
 
        """
108
 
        self._format = _format
109
 
        self.controldir = _controldir
110
 
        if not _internal:
111
 
            raise errors.BzrError("Please use controldir.open_workingtree or "
112
 
                                  "WorkingTree.open() to obtain a WorkingTree.")
113
 
        basedir = osutils.safe_unicode(basedir)
114
 
        mutter("opening working tree %r", basedir)
115
 
        if branch is not None:
116
 
            self._branch = branch
117
 
        else:
118
 
            self._branch = self.controldir.open_branch()
119
 
        self.basedir = osutils.realpath(basedir)
120
 
        self._transport = _transport
121
 
        self._rules_searcher = None
122
 
        self.views = self._make_views()
123
 
 
124
 
    @property
125
 
    def user_transport(self):
126
 
        return self.controldir.user_transport
127
 
 
128
 
    @property
129
 
    def control_transport(self):
130
 
        return self._transport
131
 
 
132
 
    def supports_symlinks(self):
133
 
        return osutils.supports_symlinks(self.basedir)
134
 
 
135
 
    def is_control_filename(self, filename):
136
 
        """True if filename is the name of a control file in this tree.
137
 
 
138
 
        :param filename: A filename within the tree. This is a relative path
139
 
            from the root of this tree.
140
 
 
141
 
        This is true IF and ONLY IF the filename is part of the meta data
142
 
        that bzr controls in this tree. I.E. a random .bzr directory placed
143
 
        on disk will not be a control file for this tree.
144
 
        """
145
 
        return self.controldir.is_control_filename(filename)
146
 
 
147
 
    branch = property(
148
 
        fget=lambda self: self._branch,
149
 
        doc="""The branch this WorkingTree is connected to.
150
 
 
151
 
            This cannot be set - it is reflective of the actual disk structure
152
 
            the working tree has been constructed from.
153
 
            """)
154
 
 
155
 
    def has_versioned_directories(self):
156
 
        """See `Tree.has_versioned_directories`."""
157
 
        return self._format.supports_versioned_directories
158
 
 
159
 
    def supports_merge_modified(self):
160
 
        """Indicate whether this workingtree supports storing merge_modified.
161
 
        """
162
 
        return self._format.supports_merge_modified
163
 
 
164
 
    def _supports_executable(self):
165
 
        return osutils.supports_executable(self.basedir)
166
 
 
167
 
    def break_lock(self):
168
 
        """Break a lock if one is present from another instance.
169
 
 
170
 
        Uses the ui factory to ask for confirmation if the lock may be from
171
 
        an active process.
172
 
 
173
 
        This will probe the repository for its lock as well.
174
 
        """
175
 
        raise NotImplementedError(self.break_lock)
176
 
 
177
 
    def requires_rich_root(self):
178
 
        return self._format.requires_rich_root
179
 
 
180
 
    def supports_tree_reference(self):
181
 
        return False
182
 
 
183
 
    def supports_content_filtering(self):
184
 
        return self._format.supports_content_filtering()
185
 
 
186
 
    def supports_views(self):
187
 
        return self.views.supports_views()
188
 
 
189
 
    def supports_setting_file_ids(self):
190
 
        return self._format.supports_setting_file_ids
191
 
 
192
 
    def get_config_stack(self):
193
 
        """Retrieve the config stack for this tree.
194
 
 
195
 
        :return: A ``breezy.config.Stack``
196
 
        """
197
 
        # For the moment, just provide the branch config stack.
198
 
        return self.branch.get_config_stack()
199
 
 
200
 
    @staticmethod
201
 
    def open(path=None, _unsupported=False):
202
 
        """Open an existing working tree at path.
203
 
 
204
 
        """
205
 
        if path is None:
206
 
            path = osutils.getcwd()
207
 
        control = ControlDir.open(path, _unsupported=_unsupported)
208
 
        return control.open_workingtree(unsupported=_unsupported)
209
 
 
210
 
    @staticmethod
211
 
    def open_containing(path=None):
212
 
        """Open an existing working tree which has its root about path.
213
 
 
214
 
        This probes for a working tree at path and searches upwards from there.
215
 
 
216
 
        Basically we keep looking up until we find the control directory or
217
 
        run into /.  If there isn't one, raises NotBranchError.
218
 
        TODO: give this a new exception.
219
 
        If there is one, it is returned, along with the unused portion of path.
220
 
 
221
 
        :return: The WorkingTree that contains 'path', and the rest of path
222
 
        """
223
 
        if path is None:
224
 
            path = osutils.getcwd()
225
 
        control, relpath = ControlDir.open_containing(path)
226
 
        return control.open_workingtree(), relpath
227
 
 
228
 
    @staticmethod
229
 
    def open_containing_paths(file_list, default_directory=None,
230
 
                              canonicalize=True, apply_view=True):
231
 
        """Open the WorkingTree that contains a set of paths.
232
 
 
233
 
        Fail if the paths given are not all in a single tree.
234
 
 
235
 
        This is used for the many command-line interfaces that take a list of
236
 
        any number of files and that require they all be in the same tree.
237
 
        """
238
 
        if default_directory is None:
239
 
            default_directory = u'.'
240
 
        # recommended replacement for builtins.internal_tree_files
241
 
        if file_list is None or len(file_list) == 0:
242
 
            tree = WorkingTree.open_containing(default_directory)[0]
243
 
            # XXX: doesn't really belong here, and seems to have the strange
244
 
            # side effect of making it return a bunch of files, not the whole
245
 
            # tree -- mbp 20100716
246
 
            if tree.supports_views() and apply_view:
247
 
                view_files = tree.views.lookup_view()
248
 
                if view_files:
249
 
                    file_list = view_files
250
 
                    view_str = views.view_display_str(view_files)
251
 
                    note(gettext("Ignoring files outside view. View is %s") % view_str)
252
 
            return tree, file_list
253
 
        if default_directory == u'.':
254
 
            seed = file_list[0]
255
 
        else:
256
 
            seed = default_directory
257
 
            file_list = [osutils.pathjoin(default_directory, f)
258
 
                         for f in file_list]
259
 
        tree = WorkingTree.open_containing(seed)[0]
260
 
        return tree, tree.safe_relpath_files(file_list, canonicalize,
261
 
                                             apply_view=apply_view)
262
 
 
263
 
    def safe_relpath_files(self, file_list, canonicalize=True, apply_view=True):
264
 
        """Convert file_list into a list of relpaths in tree.
265
 
 
266
 
        :param self: A tree to operate on.
267
 
        :param file_list: A list of user provided paths or None.
268
 
        :param apply_view: if True and a view is set, apply it or check that
269
 
            specified files are within it
270
 
        :return: A list of relative paths.
271
 
        :raises errors.PathNotChild: When a provided path is in a different self
272
 
            than self.
273
 
        """
274
 
        if file_list is None:
275
 
            return None
276
 
        if self.supports_views() and apply_view:
277
 
            view_files = self.views.lookup_view()
278
 
        else:
279
 
            view_files = []
280
 
        new_list = []
281
 
        # self.relpath exists as a "thunk" to osutils, but canonical_relpath
282
 
        # doesn't - fix that up here before we enter the loop.
283
 
        if canonicalize:
284
 
            def fixer(p):
285
 
                return osutils.canonical_relpath(self.basedir, p)
286
 
        else:
287
 
            fixer = self.relpath
288
 
        for filename in file_list:
289
 
            relpath = fixer(osutils.dereference_path(filename))
290
 
            if view_files and not osutils.is_inside_any(view_files, relpath):
291
 
                raise views.FileOutsideView(filename, view_files)
292
 
            new_list.append(relpath)
293
 
        return new_list
294
 
 
295
 
    @staticmethod
296
 
    def open_downlevel(path=None):
297
 
        """Open an unsupported working tree.
298
 
 
299
 
        Only intended for advanced situations like upgrading part of a controldir.
300
 
        """
301
 
        return WorkingTree.open(path, _unsupported=True)
 
127
    def __init__(self, basedir, branch=None):
 
128
        """Construct a WorkingTree for basedir.
 
129
 
 
130
        If the branch is not supplied, it is opened automatically.
 
131
        If the branch is supplied, it must be the branch for this basedir.
 
132
        (branch.base is not cross checked, because for remote branches that
 
133
        would be meaningless).
 
134
        """
 
135
        from bzrlib.hashcache import HashCache
 
136
        from bzrlib.trace import note, mutter
 
137
        assert isinstance(basedir, basestring), \
 
138
            "base directory %r is not a string" % basedir
 
139
        if branch is None:
 
140
            branch = Branch.open(basedir)
 
141
        assert isinstance(branch, Branch), \
 
142
            "branch %r is not a Branch" % branch
 
143
        self.branch = branch
 
144
        self.basedir = basedir
 
145
        self._inventory = self.read_working_inventory()
 
146
        self.path2id = self._inventory.path2id
 
147
 
 
148
        # update the whole cache up front and write to disk if anything changed;
 
149
        # in the future we might want to do this more selectively
 
150
        # two possible ways offer themselves : in self._unlock, write the cache
 
151
        # if needed, or, when the cache sees a change, append it to the hash
 
152
        # cache file, and have the parser take the most recent entry for a
 
153
        # given path only.
 
154
        hc = self._hashcache = HashCache(basedir)
 
155
        hc.read()
 
156
        hc.scan()
 
157
 
 
158
        if hc.needs_write:
 
159
            mutter("write hc")
 
160
            hc.write()
 
161
 
 
162
    def __iter__(self):
 
163
        """Iterate through file_ids for this tree.
 
164
 
 
165
        file_ids are in a WorkingTree if they are in the working inventory
 
166
        and the working file exists.
 
167
        """
 
168
        inv = self._inventory
 
169
        for path, ie in inv.iter_entries():
 
170
            if bzrlib.osutils.lexists(self.abspath(path)):
 
171
                yield ie.file_id
 
172
 
302
173
 
303
174
    def __repr__(self):
304
175
        return "<%s of %s>" % (self.__class__.__name__,
305
176
                               getattr(self, 'basedir', None))
306
177
 
 
178
 
 
179
 
307
180
    def abspath(self, filename):
308
 
        return osutils.pathjoin(self.basedir, filename)
309
 
 
310
 
    def basis_tree(self):
311
 
        """Return RevisionTree for the current last revision.
312
 
 
313
 
        If the left most parent is a ghost then the returned tree will be an
314
 
        empty tree - one obtained by calling
315
 
        repository.revision_tree(NULL_REVISION).
316
 
        """
317
 
        try:
318
 
            revision_id = self.get_parent_ids()[0]
319
 
        except IndexError:
320
 
            # no parents, return an empty revision tree.
321
 
            # in the future this should return the tree for
322
 
            # 'empty:' - the implicit root empty tree.
323
 
            return self.branch.repository.revision_tree(
324
 
                _mod_revision.NULL_REVISION)
325
 
        try:
326
 
            return self.revision_tree(revision_id)
327
 
        except errors.NoSuchRevision:
328
 
            pass
329
 
        # No cached copy available, retrieve from the repository.
330
 
        # FIXME? RBC 20060403 should we cache the tree locally
331
 
        # at this point ?
332
 
        try:
333
 
            return self.branch.repository.revision_tree(revision_id)
334
 
        except (errors.RevisionNotPresent, errors.NoSuchRevision):
335
 
            # the basis tree *may* be a ghost or a low level error may have
336
 
            # occurred. If the revision is present, its a problem, if its not
337
 
            # its a ghost.
338
 
            if self.branch.repository.has_revision(revision_id):
339
 
                raise
340
 
            # the basis tree is a ghost so return an empty tree.
341
 
            return self.branch.repository.revision_tree(
342
 
                _mod_revision.NULL_REVISION)
343
 
 
344
 
    def relpath(self, path):
345
 
        """Return the local path portion from a given path.
346
 
 
347
 
        The path may be absolute or relative. If its a relative path it is
348
 
        interpreted relative to the python current working directory.
349
 
        """
350
 
        return osutils.relpath(self.basedir, path)
 
181
        return os.path.join(self.basedir, filename)
 
182
 
 
183
    def relpath(self, abspath):
 
184
        """Return the local path portion from a given absolute path."""
 
185
        return relpath(self.basedir, abspath)
351
186
 
352
187
    def has_filename(self, filename):
353
 
        return osutils.lexists(self.abspath(filename))
354
 
 
355
 
    def get_file(self, path, filtered=True):
356
 
        return self.get_file_with_stat(path, filtered=filtered)[0]
357
 
 
358
 
    def get_file_with_stat(self, path, filtered=True,
359
 
                           _fstat=osutils.fstat):
360
 
        """See Tree.get_file_with_stat."""
361
 
        abspath = self.abspath(path)
362
 
        try:
363
 
            file_obj = open(abspath, 'rb')
364
 
        except EnvironmentError as e:
365
 
            if e.errno == errno.ENOENT:
366
 
                raise errors.NoSuchFile(path)
367
 
            raise
368
 
        stat_value = _fstat(file_obj.fileno())
369
 
        if filtered and self.supports_content_filtering():
370
 
            filters = self._content_filter_stack(path)
371
 
            if filters:
372
 
                file_obj, size = _mod_filters.filtered_input_file(
373
 
                    file_obj, filters)
374
 
                stat_value = _mod_filters.FilteredStat(
375
 
                    stat_value, st_size=size)
376
 
        return (file_obj, stat_value)
377
 
 
378
 
    def get_file_text(self, path, filtered=True):
379
 
        with self.get_file(path, filtered=filtered) as my_file:
380
 
            return my_file.read()
381
 
 
382
 
    def get_file_lines(self, path, filtered=True):
383
 
        """See Tree.get_file_lines()"""
384
 
        with self.get_file(path, filtered=filtered) as file:
385
 
            return file.readlines()
386
 
 
387
 
    def get_parent_ids(self):
388
 
        """See Tree.get_parent_ids.
389
 
 
390
 
        This implementation reads the pending merges list and last_revision
391
 
        value and uses that to decide what the parents list should be.
392
 
        """
393
 
        last_rev = _mod_revision.ensure_null(self._last_revision())
394
 
        if _mod_revision.NULL_REVISION == last_rev:
395
 
            parents = []
396
 
        else:
397
 
            parents = [last_rev]
398
 
        try:
399
 
            merges_bytes = self._transport.get_bytes('pending-merges')
400
 
        except errors.NoSuchFile:
401
 
            pass
402
 
        else:
403
 
            for l in osutils.split_lines(merges_bytes):
404
 
                revision_id = l.rstrip(b'\n')
405
 
                parents.append(revision_id)
406
 
        return parents
407
 
 
408
 
    def clone(self, to_controldir, revision_id=None):
409
 
        """Duplicate this working tree into to_bzr, including all state.
410
 
 
411
 
        Specifically modified files are kept as modified, but
412
 
        ignored and unknown files are discarded.
413
 
 
414
 
        If you want to make a new line of development, see ControlDir.sprout()
415
 
 
416
 
        revision
417
 
            If not None, the cloned tree will have its last revision set to
418
 
            revision, and difference between the source trees last revision
419
 
            and this one merged in.
420
 
        """
421
 
        with self.lock_read():
422
 
            # assumes the target bzr dir format is compatible.
423
 
            result = to_controldir.create_workingtree()
424
 
            self.copy_content_into(result, revision_id)
425
 
            return result
426
 
 
427
 
    def copy_content_into(self, tree, revision_id=None):
428
 
        """Copy the current content and user files of this tree into tree."""
429
 
        with self.lock_read():
430
 
            tree.set_root_id(self.path2id(''))
431
 
            if revision_id is None:
432
 
                merge.transform_tree(tree, self)
433
 
            else:
434
 
                # TODO now merge from tree.last_revision to revision (to
435
 
                # preserve user local changes)
436
 
                try:
437
 
                    other_tree = self.revision_tree(revision_id)
438
 
                except errors.NoSuchRevision:
439
 
                    other_tree = self.branch.repository.revision_tree(
440
 
                        revision_id)
441
 
 
442
 
                merge.transform_tree(tree, other_tree)
443
 
                if revision_id == _mod_revision.NULL_REVISION:
444
 
                    new_parents = []
445
 
                else:
446
 
                    new_parents = [revision_id]
447
 
                tree.set_parent_ids(new_parents)
448
 
 
449
 
    def get_file_size(self, path):
450
 
        """See Tree.get_file_size"""
451
 
        # XXX: this returns the on-disk size; it should probably return the
452
 
        # canonical size
453
 
        try:
454
 
            return os.path.getsize(self.abspath(path))
455
 
        except OSError as e:
456
 
            if e.errno != errno.ENOENT:
457
 
                raise
458
 
            else:
459
 
                return None
460
 
 
461
 
    def _gather_kinds(self, files, kinds):
462
 
        """See MutableTree._gather_kinds."""
463
 
        with self.lock_tree_write():
464
 
            for pos, f in enumerate(files):
465
 
                if kinds[pos] is None:
466
 
                    fullpath = osutils.normpath(self.abspath(f))
467
 
                    try:
468
 
                        kinds[pos] = osutils.file_kind(fullpath)
469
 
                    except OSError as e:
470
 
                        if e.errno == errno.ENOENT:
471
 
                            raise errors.NoSuchFile(fullpath)
472
 
 
473
 
    def add_parent_tree_id(self, revision_id, allow_leftmost_as_ghost=False):
474
 
        """Add revision_id as a parent.
475
 
 
476
 
        This is equivalent to retrieving the current list of parent ids
477
 
        and setting the list to its value plus revision_id.
478
 
 
479
 
        :param revision_id: The revision id to add to the parent list. It may
480
 
            be a ghost revision as long as its not the first parent to be
481
 
            added, or the allow_leftmost_as_ghost parameter is set True.
482
 
        :param allow_leftmost_as_ghost: Allow the first parent to be a ghost.
483
 
        """
484
 
        with self.lock_write():
485
 
            parents = self.get_parent_ids() + [revision_id]
486
 
            self.set_parent_ids(parents, allow_leftmost_as_ghost=len(parents) > 1
487
 
                                or allow_leftmost_as_ghost)
488
 
 
489
 
    def add_parent_tree(self, parent_tuple, allow_leftmost_as_ghost=False):
490
 
        """Add revision_id, tree tuple as a parent.
491
 
 
492
 
        This is equivalent to retrieving the current list of parent trees
493
 
        and setting the list to its value plus parent_tuple. See also
494
 
        add_parent_tree_id - if you only have a parent id available it will be
495
 
        simpler to use that api. If you have the parent already available, using
496
 
        this api is preferred.
497
 
 
498
 
        :param parent_tuple: The (revision id, tree) to add to the parent list.
499
 
            If the revision_id is a ghost, pass None for the tree.
500
 
        :param allow_leftmost_as_ghost: Allow the first parent to be a ghost.
501
 
        """
502
 
        with self.lock_tree_write():
503
 
            parent_ids = self.get_parent_ids() + [parent_tuple[0]]
504
 
            if len(parent_ids) > 1:
505
 
                # the leftmost may have already been a ghost, preserve that if it
506
 
                # was.
507
 
                allow_leftmost_as_ghost = True
508
 
            self.set_parent_ids(parent_ids,
509
 
                                allow_leftmost_as_ghost=allow_leftmost_as_ghost)
510
 
 
 
188
        return bzrlib.osutils.lexists(self.abspath(filename))
 
189
 
 
190
    def get_file(self, file_id):
 
191
        return self.get_file_byname(self.id2path(file_id))
 
192
 
 
193
    def get_file_byname(self, filename):
 
194
        return file(self.abspath(filename), 'rb')
 
195
 
 
196
    def get_root_id(self):
 
197
        """Return the id of this trees root"""
 
198
        inv = self.read_working_inventory()
 
199
        return inv.root.file_id
 
200
        
 
201
    def _get_store_filename(self, file_id):
 
202
        ## XXX: badly named; this isn't in the store at all
 
203
        return self.abspath(self.id2path(file_id))
 
204
 
 
205
    @needs_write_lock
 
206
    def commit(self, *args, **kw):
 
207
        from bzrlib.commit import Commit
 
208
        Commit().commit(self.branch, *args, **kw)
 
209
        self._inventory = self.read_working_inventory()
 
210
 
 
211
    def id2abspath(self, file_id):
 
212
        return self.abspath(self.id2path(file_id))
 
213
 
 
214
                
 
215
    def has_id(self, file_id):
 
216
        # files that have been deleted are excluded
 
217
        inv = self._inventory
 
218
        if not inv.has_id(file_id):
 
219
            return False
 
220
        path = inv.id2path(file_id)
 
221
        return bzrlib.osutils.lexists(self.abspath(path))
 
222
 
 
223
    def has_or_had_id(self, file_id):
 
224
        if file_id == self.inventory.root.file_id:
 
225
            return True
 
226
        return self.inventory.has_id(file_id)
 
227
 
 
228
    __contains__ = has_id
 
229
    
 
230
 
 
231
    def get_file_size(self, file_id):
 
232
        return os.path.getsize(self.id2abspath(file_id))
 
233
 
 
234
    def get_file_sha1(self, file_id):
 
235
        path = self._inventory.id2path(file_id)
 
236
        return self._hashcache.get_sha1(path)
 
237
 
 
238
 
 
239
    def is_executable(self, file_id):
 
240
        if os.name == "nt":
 
241
            return self._inventory[file_id].executable
 
242
        else:
 
243
            path = self._inventory.id2path(file_id)
 
244
            mode = os.lstat(self.abspath(path)).st_mode
 
245
            return bool(stat.S_ISREG(mode) and stat.S_IEXEC&mode)
 
246
 
 
247
    @needs_write_lock
511
248
    def add_pending_merge(self, *revision_ids):
512
 
        with self.lock_tree_write():
513
 
            # TODO: Perhaps should check at this point that the
514
 
            # history of the revision is actually present?
515
 
            parents = self.get_parent_ids()
516
 
            updated = False
517
 
            for rev_id in revision_ids:
518
 
                if rev_id in parents:
519
 
                    continue
520
 
                parents.append(rev_id)
521
 
                updated = True
522
 
            if updated:
523
 
                self.set_parent_ids(parents, allow_leftmost_as_ghost=True)
524
 
 
525
 
    def path_content_summary(self, path, _lstat=os.lstat,
526
 
                             _mapper=osutils.file_kind_from_stat_mode):
527
 
        """See Tree.path_content_summary."""
528
 
        abspath = self.abspath(path)
529
 
        try:
530
 
            stat_result = _lstat(abspath)
531
 
        except OSError as e:
532
 
            if getattr(e, 'errno', None) == errno.ENOENT:
533
 
                # no file.
534
 
                return ('missing', None, None, None)
535
 
            # propagate other errors
536
 
            raise
537
 
        kind = _mapper(stat_result.st_mode)
538
 
        if kind == 'file':
539
 
            return self._file_content_summary(path, stat_result)
540
 
        elif kind == 'directory':
541
 
            # perhaps it looks like a plain directory, but it's really a
542
 
            # reference.
543
 
            if self._directory_is_tree_reference(path):
544
 
                kind = 'tree-reference'
545
 
            return kind, None, None, None
546
 
        elif kind == 'symlink':
547
 
            target = osutils.readlink(abspath)
548
 
            return ('symlink', None, None, target)
549
 
        else:
550
 
            return (kind, None, None, None)
551
 
 
552
 
    def _file_content_summary(self, path, stat_result):
553
 
        size = stat_result.st_size
554
 
        executable = self._is_executable_from_path_and_stat(path, stat_result)
555
 
        # try for a stat cache lookup
556
 
        return ('file', size, executable, self._sha_from_stat(
557
 
            path, stat_result))
558
 
 
559
 
    def _check_parents_for_ghosts(self, revision_ids, allow_leftmost_as_ghost):
560
 
        """Common ghost checking functionality from set_parent_*.
561
 
 
562
 
        This checks that the left hand-parent exists if there are any
563
 
        revisions present.
564
 
        """
565
 
        if len(revision_ids) > 0:
566
 
            leftmost_id = revision_ids[0]
567
 
            if (not allow_leftmost_as_ghost and not
568
 
                    self.branch.repository.has_revision(leftmost_id)):
569
 
                raise errors.GhostRevisionUnusableHere(leftmost_id)
570
 
 
571
 
    def _set_merges_from_parent_ids(self, parent_ids):
572
 
        merges = parent_ids[1:]
573
 
        self._transport.put_bytes('pending-merges', b'\n'.join(merges),
574
 
                                  mode=self.controldir._get_file_mode())
575
 
 
576
 
    def _filter_parent_ids_by_ancestry(self, revision_ids):
577
 
        """Check that all merged revisions are proper 'heads'.
578
 
 
579
 
        This will always return the first revision_id, and any merged revisions
580
 
        which are
581
 
        """
582
 
        if len(revision_ids) == 0:
583
 
            return revision_ids
584
 
        graph = self.branch.repository.get_graph()
585
 
        heads = graph.heads(revision_ids)
586
 
        new_revision_ids = revision_ids[:1]
587
 
        for revision_id in revision_ids[1:]:
588
 
            if revision_id in heads and revision_id not in new_revision_ids:
589
 
                new_revision_ids.append(revision_id)
590
 
        if new_revision_ids != revision_ids:
591
 
            mutter('requested to set revision_ids = %s,'
592
 
                   ' but filtered to %s', revision_ids, new_revision_ids)
593
 
        return new_revision_ids
594
 
 
595
 
    def set_parent_ids(self, revision_ids, allow_leftmost_as_ghost=False):
596
 
        """Set the parent ids to revision_ids.
597
 
 
598
 
        See also set_parent_trees. This api will try to retrieve the tree data
599
 
        for each element of revision_ids from the trees repository. If you have
600
 
        tree data already available, it is more efficient to use
601
 
        set_parent_trees rather than set_parent_ids. set_parent_ids is however
602
 
        an easier API to use.
603
 
 
604
 
        :param revision_ids: The revision_ids to set as the parent ids of this
605
 
            working tree. Any of these may be ghosts.
606
 
        """
607
 
        with self.lock_tree_write():
608
 
            self._check_parents_for_ghosts(revision_ids,
609
 
                                           allow_leftmost_as_ghost=allow_leftmost_as_ghost)
610
 
            for revision_id in revision_ids:
611
 
                _mod_revision.check_not_reserved_id(revision_id)
612
 
 
613
 
            revision_ids = self._filter_parent_ids_by_ancestry(revision_ids)
614
 
 
615
 
            if len(revision_ids) > 0:
616
 
                self.set_last_revision(revision_ids[0])
617
 
            else:
618
 
                self.set_last_revision(_mod_revision.NULL_REVISION)
619
 
 
620
 
            self._set_merges_from_parent_ids(revision_ids)
621
 
 
 
249
        # TODO: Perhaps should check at this point that the
 
250
        # history of the revision is actually present?
 
251
        p = self.pending_merges()
 
252
        updated = False
 
253
        for rev_id in revision_ids:
 
254
            if rev_id in p:
 
255
                continue
 
256
            p.append(rev_id)
 
257
            updated = True
 
258
        if updated:
 
259
            self.set_pending_merges(p)
 
260
 
 
261
    def pending_merges(self):
 
262
        """Return a list of pending merges.
 
263
 
 
264
        These are revisions that have been merged into the working
 
265
        directory but not yet committed.
 
266
        """
 
267
        cfn = self.branch._rel_controlfilename('pending-merges')
 
268
        if not self.branch._transport.has(cfn):
 
269
            return []
 
270
        p = []
 
271
        for l in self.branch.controlfile('pending-merges', 'r').readlines():
 
272
            p.append(l.rstrip('\n'))
 
273
        return p
 
274
 
 
275
    @needs_write_lock
622
276
    def set_pending_merges(self, rev_list):
623
 
        with self.lock_tree_write():
624
 
            parents = self.get_parent_ids()
625
 
            leftmost = parents[:1]
626
 
            new_parents = leftmost + rev_list
627
 
            self.set_parent_ids(new_parents)
628
 
 
629
 
    def set_merge_modified(self, modified_hashes):
630
 
        """Set the merge modified hashes."""
631
 
        raise NotImplementedError(self.set_merge_modified)
632
 
 
633
 
    def _sha_from_stat(self, path, stat_result):
634
 
        """Get a sha digest from the tree's stat cache.
635
 
 
636
 
        The default implementation assumes no stat cache is present.
637
 
 
638
 
        :param path: The path.
639
 
        :param stat_result: The stat result being looked up.
640
 
        """
641
 
        return None
642
 
 
643
 
    def merge_from_branch(self, branch, to_revision=None, from_revision=None,
644
 
                          merge_type=None, force=False):
645
 
        """Merge from a branch into this working tree.
646
 
 
647
 
        :param branch: The branch to merge from.
648
 
        :param to_revision: If non-None, the merge will merge to to_revision,
649
 
            but not beyond it. to_revision does not need to be in the history
650
 
            of the branch when it is supplied. If None, to_revision defaults to
651
 
            branch.last_revision().
652
 
        """
653
 
        from .merge import Merger, Merge3Merger
654
 
        with self.lock_write():
655
 
            merger = Merger(self.branch, this_tree=self)
656
 
            # check that there are no local alterations
657
 
            if not force and self.has_changes():
658
 
                raise errors.UncommittedChanges(self)
659
 
            if to_revision is None:
660
 
                to_revision = _mod_revision.ensure_null(branch.last_revision())
661
 
            merger.other_rev_id = to_revision
662
 
            if _mod_revision.is_null(merger.other_rev_id):
663
 
                raise errors.NoCommits(branch)
664
 
            self.branch.fetch(branch, stop_revision=merger.other_rev_id)
665
 
            merger.other_basis = merger.other_rev_id
666
 
            merger.other_tree = self.branch.repository.revision_tree(
667
 
                merger.other_rev_id)
668
 
            merger.other_branch = branch
669
 
            if from_revision is None:
670
 
                merger.find_base()
671
 
            else:
672
 
                merger.set_base_revision(from_revision, branch)
673
 
            if merger.base_rev_id == merger.other_rev_id:
674
 
                raise errors.PointlessMerge
675
 
            merger.backup_files = False
676
 
            if merge_type is None:
677
 
                merger.merge_type = Merge3Merger
678
 
            else:
679
 
                merger.merge_type = merge_type
680
 
            merger.set_interesting_files(None)
681
 
            merger.show_base = False
682
 
            merger.reprocess = False
683
 
            conflicts = merger.do_merge()
684
 
            merger.set_pending()
685
 
            return conflicts
686
 
 
687
 
    def merge_modified(self):
688
 
        """Return a dictionary of files modified by a merge.
689
 
 
690
 
        The list is initialized by WorkingTree.set_merge_modified, which is
691
 
        typically called after we make some automatic updates to the tree
692
 
        because of a merge.
693
 
 
694
 
        This returns a map of file_id->sha1, containing only files which are
695
 
        still in the working tree and have that text hash.
696
 
        """
697
 
        raise NotImplementedError(self.merge_modified)
698
 
 
699
 
    def mkdir(self, path, file_id=None):
700
 
        """See MutableTree.mkdir()."""
701
 
        if file_id is None:
702
 
            if self.supports_setting_file_ids():
703
 
                file_id = generate_ids.gen_file_id(os.path.basename(path))
 
277
        self.branch.put_controlfile('pending-merges', '\n'.join(rev_list))
 
278
 
 
279
    def get_symlink_target(self, file_id):
 
280
        return os.readlink(self.id2abspath(file_id))
 
281
 
 
282
    def file_class(self, filename):
 
283
        if self.path2id(filename):
 
284
            return 'V'
 
285
        elif self.is_ignored(filename):
 
286
            return 'I'
704
287
        else:
705
 
            if not self.supports_setting_file_ids():
706
 
                raise SettingFileIdUnsupported()
707
 
        with self.lock_write():
708
 
            os.mkdir(self.abspath(path))
709
 
            self.add(path, file_id, 'directory')
710
 
            return file_id
711
 
 
712
 
    def get_symlink_target(self, path):
713
 
        abspath = self.abspath(path)
714
 
        target = osutils.readlink(abspath)
715
 
        return target
716
 
 
717
 
    def subsume(self, other_tree):
718
 
        raise NotImplementedError(self.subsume)
719
 
 
720
 
    def _directory_is_tree_reference(self, relpath):
721
 
        raise NotImplementedError(self._directory_is_tree_reference)
722
 
 
723
 
    def extract(self, path, format=None):
724
 
        """Extract a subtree from this tree.
725
 
 
726
 
        A new branch will be created, relative to the path for this tree.
727
 
        """
728
 
        raise NotImplementedError(self.extract)
729
 
 
730
 
    def flush(self):
731
 
        """Write the in memory meta data to disk."""
732
 
        raise NotImplementedError(self.flush)
733
 
 
734
 
    def kind(self, relpath):
735
 
        return osutils.file_kind(self.abspath(relpath))
736
 
 
737
 
    def list_files(self, include_root=False, from_dir=None, recursive=True,
738
 
                   recurse_nested=False):
739
 
        """List all files as (path, class, kind, id, entry).
 
288
            return '?'
 
289
 
 
290
 
 
291
    def list_files(self):
 
292
        """Recursively list all files as (path, class, kind, id).
740
293
 
741
294
        Lists, but does not descend into unversioned directories.
 
295
 
742
296
        This does not include files that have been deleted in this
743
 
        tree. Skips the control directory.
744
 
 
745
 
        :param include_root: if True, return an entry for the root
746
 
        :param from_dir: start from this directory or None for the root
747
 
        :param recursive: whether to recurse into subdirectories or not
748
 
        """
749
 
        raise NotImplementedError(self.list_files)
750
 
 
751
 
    def move(self, from_paths, to_dir=None, after=False):
752
 
        """Rename files.
753
 
 
754
 
        to_dir must be known to the working tree.
755
 
 
756
 
        If to_dir exists and is a directory, the files are moved into
757
 
        it, keeping their old names.
758
 
 
759
 
        Note that to_dir is only the last component of the new name;
760
 
        this doesn't change the directory.
761
 
 
762
 
        For each entry in from_paths the move mode will be determined
763
 
        independently.
764
 
 
765
 
        The first mode moves the file in the filesystem and updates the
766
 
        working tree metadata. The second mode only updates the working tree
767
 
        metadata without touching the file on the filesystem.
768
 
 
769
 
        move uses the second mode if 'after == True' and the target is not
770
 
        versioned but present in the working tree.
771
 
 
772
 
        move uses the second mode if 'after == False' and the source is
773
 
        versioned but no longer in the working tree, and the target is not
774
 
        versioned but present in the working tree.
775
 
 
776
 
        move uses the first mode if 'after == False' and the source is
777
 
        versioned and present in the working tree, and the target is not
778
 
        versioned and not present in the working tree.
779
 
 
780
 
        Everything else results in an error.
781
 
 
782
 
        This returns a list of (from_path, to_path) pairs for each
783
 
        entry that is moved.
784
 
        """
785
 
        raise NotImplementedError(self.move)
786
 
 
787
 
    def copy_one(self, from_rel, to_rel):
788
 
        """Copy a file in the tree to a new location.
789
 
 
790
 
        This default implementation just copies the file, then
791
 
        adds the target.
792
 
 
793
 
        :param from_rel: From location (relative to tree root)
794
 
        :param to_rel: Target location (relative to tree root)
795
 
        """
796
 
        shutil.copyfile(self.abspath(from_rel), self.abspath(to_rel))
797
 
        self.add(to_rel)
 
297
        tree.
 
298
 
 
299
        Skips the control directory.
 
300
        """
 
301
        inv = self._inventory
 
302
 
 
303
        def descend(from_dir_relpath, from_dir_id, dp):
 
304
            ls = os.listdir(dp)
 
305
            ls.sort()
 
306
            for f in ls:
 
307
                ## TODO: If we find a subdirectory with its own .bzr
 
308
                ## directory, then that is a separate tree and we
 
309
                ## should exclude it.
 
310
                if bzrlib.BZRDIR == f:
 
311
                    continue
 
312
 
 
313
                # path within tree
 
314
                fp = appendpath(from_dir_relpath, f)
 
315
 
 
316
                # absolute path
 
317
                fap = appendpath(dp, f)
 
318
                
 
319
                f_ie = inv.get_child(from_dir_id, f)
 
320
                if f_ie:
 
321
                    c = 'V'
 
322
                elif self.is_ignored(fp):
 
323
                    c = 'I'
 
324
                else:
 
325
                    c = '?'
 
326
 
 
327
                fk = file_kind(fap)
 
328
 
 
329
                if f_ie:
 
330
                    if f_ie.kind != fk:
 
331
                        raise BzrCheckError("file %r entered as kind %r id %r, "
 
332
                                            "now of kind %r"
 
333
                                            % (fap, f_ie.kind, f_ie.file_id, fk))
 
334
 
 
335
                # make a last minute entry
 
336
                if f_ie:
 
337
                    entry = f_ie
 
338
                else:
 
339
                    if fk == 'directory':
 
340
                        entry = TreeDirectory()
 
341
                    elif fk == 'file':
 
342
                        entry = TreeFile()
 
343
                    elif fk == 'symlink':
 
344
                        entry = TreeLink()
 
345
                    else:
 
346
                        entry = TreeEntry()
 
347
                
 
348
                yield fp, c, fk, (f_ie and f_ie.file_id), entry
 
349
 
 
350
                if fk != 'directory':
 
351
                    continue
 
352
 
 
353
                if c != 'V':
 
354
                    # don't descend unversioned directories
 
355
                    continue
 
356
                
 
357
                for ff in descend(fp, f_ie.file_id, fap):
 
358
                    yield ff
 
359
 
 
360
        for f in descend('', inv.root.file_id, self.basedir):
 
361
            yield f
 
362
            
 
363
 
798
364
 
799
365
    def unknowns(self):
800
 
        """Return all unknown files.
801
 
 
802
 
        These are files in the working directory that are not versioned or
803
 
        control files or ignored.
804
 
        """
805
 
        with self.lock_read():
806
 
            # force the extras method to be fully executed before returning, to
807
 
            # prevent race conditions with the lock
808
 
            return iter(
809
 
                [subp for subp in self.extras() if not self.is_ignored(subp)])
810
 
 
811
 
    def unversion(self, paths):
812
 
        """Remove the path in pahs from the current versioned set.
813
 
 
814
 
        When a path is unversioned, all of its children are automatically
815
 
        unversioned.
816
 
 
817
 
        :param paths: The paths to stop versioning.
818
 
        :raises NoSuchFile: if any path is not currently versioned.
819
 
        """
820
 
        raise NotImplementedError(self.unversion)
821
 
 
822
 
    def pull(self, source, overwrite=False, stop_revision=None,
823
 
             change_reporter=None, possible_transports=None, local=False,
824
 
             show_base=False, tag_selector=None):
825
 
        with self.lock_write(), source.lock_read():
826
 
            old_revision_info = self.branch.last_revision_info()
827
 
            basis_tree = self.basis_tree()
828
 
            count = self.branch.pull(source, overwrite, stop_revision,
829
 
                                     possible_transports=possible_transports,
830
 
                                     local=local, tag_selector=tag_selector)
831
 
            new_revision_info = self.branch.last_revision_info()
832
 
            if new_revision_info != old_revision_info:
833
 
                repository = self.branch.repository
834
 
                if repository._format.fast_deltas:
835
 
                    parent_ids = self.get_parent_ids()
836
 
                    if parent_ids:
837
 
                        basis_id = parent_ids[0]
838
 
                        basis_tree = repository.revision_tree(basis_id)
839
 
                with basis_tree.lock_read():
840
 
                    new_basis_tree = self.branch.basis_tree()
841
 
                    merge.merge_inner(
842
 
                        self.branch,
843
 
                        new_basis_tree,
844
 
                        basis_tree,
845
 
                        this_tree=self,
846
 
                        change_reporter=change_reporter,
847
 
                        show_base=show_base)
848
 
                    basis_root_id = basis_tree.path2id('')
849
 
                    new_root_id = new_basis_tree.path2id('')
850
 
                    if new_root_id is not None and basis_root_id != new_root_id:
851
 
                        self.set_root_id(new_root_id)
852
 
                # TODO - dedup parents list with things merged by pull ?
853
 
                # reuse the revisiontree we merged against to set the new
854
 
                # tree data.
855
 
                parent_trees = []
856
 
                if self.branch.last_revision() != _mod_revision.NULL_REVISION:
857
 
                    parent_trees.append(
858
 
                        (self.branch.last_revision(), new_basis_tree))
859
 
                # we have to pull the merge trees out again, because
860
 
                # merge_inner has set the ids. - this corner is not yet
861
 
                # layered well enough to prevent double handling.
862
 
                # XXX TODO: Fix the double handling: telling the tree about
863
 
                # the already known parent data is wasteful.
864
 
                merges = self.get_parent_ids()[1:]
865
 
                parent_trees.extend([
866
 
                    (parent, repository.revision_tree(parent)) for
867
 
                    parent in merges])
868
 
                self.set_parent_trees(parent_trees)
869
 
            return count
870
 
 
871
 
    def put_file_bytes_non_atomic(self, path, bytes):
872
 
        """See MutableTree.put_file_bytes_non_atomic."""
873
 
        with self.lock_write(), open(self.abspath(path), 'wb') as stream:
874
 
            stream.write(bytes)
 
366
        for subp in self.extras():
 
367
            if not self.is_ignored(subp):
 
368
                yield subp
 
369
 
 
370
    def iter_conflicts(self):
 
371
        conflicted = set()
 
372
        for path in (s[0] for s in self.list_files()):
 
373
            stem = get_conflicted_stem(path)
 
374
            if stem is None:
 
375
                continue
 
376
            if stem not in conflicted:
 
377
                conflicted.add(stem)
 
378
                yield stem
 
379
 
 
380
    @needs_write_lock
 
381
    def pull(self, source, overwrite=False):
 
382
        from bzrlib.merge import merge_inner
 
383
        source.lock_read()
 
384
        try:
 
385
            old_revision_history = self.branch.revision_history()
 
386
            self.branch.pull(source, overwrite)
 
387
            new_revision_history = self.branch.revision_history()
 
388
            if new_revision_history != old_revision_history:
 
389
                if len(old_revision_history):
 
390
                    other_revision = old_revision_history[-1]
 
391
                else:
 
392
                    other_revision = None
 
393
                merge_inner(self.branch,
 
394
                            self.branch.basis_tree(), 
 
395
                            self.branch.revision_tree(other_revision))
 
396
        finally:
 
397
            source.unlock()
875
398
 
876
399
    def extras(self):
877
 
        """Yield all unversioned files in this WorkingTree.
 
400
        """Yield all unknown files in this WorkingTree.
878
401
 
879
 
        If there are any unversioned directories and the file format
880
 
        supports versioning directories, then only the directory is returned,
881
 
        not all its children. But if there are unversioned files under a
882
 
        versioned subdirectory, they are returned.
 
402
        If there are any unknown directories then only the directory is
 
403
        returned, not all its children.  But if there are unknown files
 
404
        under a versioned subdirectory, they are returned.
883
405
 
884
406
        Currently returned depth-first, sorted by name within directories.
885
 
        This is the same order used by 'osutils.walkdirs'.
886
407
        """
887
 
        raise NotImplementedError(self.extras)
 
408
        ## TODO: Work from given directory downwards
 
409
        for path, dir_entry in self.inventory.directories():
 
410
            mutter("search for unknowns in %r", path)
 
411
            dirabs = self.abspath(path)
 
412
            if not isdir(dirabs):
 
413
                # e.g. directory deleted
 
414
                continue
 
415
 
 
416
            fl = []
 
417
            for subf in os.listdir(dirabs):
 
418
                if (subf != '.bzr'
 
419
                    and (subf not in dir_entry.children)):
 
420
                    fl.append(subf)
 
421
            
 
422
            fl.sort()
 
423
            for subf in fl:
 
424
                subp = appendpath(path, subf)
 
425
                yield subp
 
426
 
888
427
 
889
428
    def ignored_files(self):
890
429
        """Yield list of PATH, IGNORE_PATTERN"""
891
430
        for subp in self.extras():
892
431
            pat = self.is_ignored(subp)
893
 
            if pat is not None:
 
432
            if pat != None:
894
433
                yield subp, pat
895
434
 
 
435
 
 
436
    def get_ignore_list(self):
 
437
        """Return list of ignore patterns.
 
438
 
 
439
        Cached in the Tree object after the first call.
 
440
        """
 
441
        if hasattr(self, '_ignorelist'):
 
442
            return self._ignorelist
 
443
 
 
444
        l = bzrlib.DEFAULT_IGNORE[:]
 
445
        if self.has_filename(bzrlib.IGNORE_FILENAME):
 
446
            f = self.get_file_byname(bzrlib.IGNORE_FILENAME)
 
447
            l.extend([line.rstrip("\n\r") for line in f.readlines()])
 
448
        self._ignorelist = l
 
449
        return l
 
450
 
 
451
 
896
452
    def is_ignored(self, filename):
897
453
        r"""Check whether the filename matches an ignore pattern.
898
 
        """
899
 
        raise NotImplementedError(self.is_ignored)
900
 
 
901
 
    def stored_kind(self, path):
902
 
        """See Tree.stored_kind"""
903
 
        raise NotImplementedError(self.stored_kind)
904
 
 
905
 
    def _comparison_data(self, entry, path):
906
 
        abspath = self.abspath(path)
907
 
        try:
908
 
            stat_value = os.lstat(abspath)
909
 
        except OSError as e:
910
 
            if getattr(e, 'errno', None) == errno.ENOENT:
911
 
                stat_value = None
912
 
                kind = None
913
 
                executable = False
 
454
 
 
455
        Patterns containing '/' or '\' need to match the whole path;
 
456
        others match against only the last component.
 
457
 
 
458
        If the file is ignored, returns the pattern which caused it to
 
459
        be ignored, otherwise None.  So this can simply be used as a
 
460
        boolean if desired."""
 
461
 
 
462
        # TODO: Use '**' to match directories, and other extended
 
463
        # globbing stuff from cvs/rsync.
 
464
 
 
465
        # XXX: fnmatch is actually not quite what we want: it's only
 
466
        # approximately the same as real Unix fnmatch, and doesn't
 
467
        # treat dotfiles correctly and allows * to match /.
 
468
        # Eventually it should be replaced with something more
 
469
        # accurate.
 
470
        
 
471
        for pat in self.get_ignore_list():
 
472
            if '/' in pat or '\\' in pat:
 
473
                
 
474
                # as a special case, you can put ./ at the start of a
 
475
                # pattern; this is good to match in the top-level
 
476
                # only;
 
477
                
 
478
                if (pat[:2] == './') or (pat[:2] == '.\\'):
 
479
                    newpat = pat[2:]
 
480
                else:
 
481
                    newpat = pat
 
482
                if fnmatch.fnmatchcase(filename, newpat):
 
483
                    return pat
914
484
            else:
915
 
                raise
 
485
                if fnmatch.fnmatchcase(splitpath(filename)[-1], pat):
 
486
                    return pat
916
487
        else:
917
 
            mode = stat_value.st_mode
918
 
            kind = osutils.file_kind_from_stat_mode(mode)
919
 
            if not self._supports_executable():
920
 
                executable = entry is not None and entry.executable
921
 
            else:
922
 
                executable = bool(stat.S_ISREG(mode) and stat.S_IEXEC & mode)
923
 
        return kind, executable, stat_value
924
 
 
925
 
    def last_revision(self):
926
 
        """Return the last revision of the branch for this tree.
927
 
 
928
 
        This format tree does not support a separate marker for last-revision
929
 
        compared to the branch.
930
 
 
931
 
        See MutableTree.last_revision
932
 
        """
933
 
        return self._last_revision()
934
 
 
935
 
    def _last_revision(self):
936
 
        """helper for get_parent_ids."""
937
 
        with self.lock_read():
938
 
            return _mod_revision.ensure_null(self.branch.last_revision())
939
 
 
940
 
    def is_locked(self):
941
 
        """Check if this tree is locked."""
942
 
        raise NotImplementedError(self.is_locked)
 
488
            return None
 
489
 
 
490
    def kind(self, file_id):
 
491
        return file_kind(self.id2abspath(file_id))
943
492
 
944
493
    def lock_read(self):
945
 
        """Lock the tree for reading.
946
 
 
947
 
        This also locks the branch, and can be unlocked via self.unlock().
948
 
 
949
 
        :return: A breezy.lock.LogicalLockResult.
950
 
        """
951
 
        raise NotImplementedError(self.lock_read)
952
 
 
953
 
    def lock_tree_write(self):
954
 
        """See MutableTree.lock_tree_write, and WorkingTree.unlock.
955
 
 
956
 
        :return: A breezy.lock.LogicalLockResult.
957
 
        """
958
 
        raise NotImplementedError(self.lock_tree_write)
 
494
        """See Branch.lock_read, and WorkingTree.unlock."""
 
495
        return self.branch.lock_read()
959
496
 
960
497
    def lock_write(self):
961
 
        """See MutableTree.lock_write, and WorkingTree.unlock.
962
 
 
963
 
        :return: A breezy.lock.LogicalLockResult.
964
 
        """
965
 
        raise NotImplementedError(self.lock_write)
966
 
 
967
 
    def get_physical_lock_status(self):
968
 
        raise NotImplementedError(self.get_physical_lock_status)
969
 
 
970
 
    def set_last_revision(self, new_revision):
971
 
        """Change the last revision in the working tree."""
972
 
        raise NotImplementedError(self.set_last_revision)
973
 
 
974
 
    def _change_last_revision(self, new_revision):
975
 
        """Template method part of set_last_revision to perform the change.
976
 
 
977
 
        This is used to allow WorkingTree3 instances to not affect branch
978
 
        when their last revision is set.
979
 
        """
980
 
        if _mod_revision.is_null(new_revision):
981
 
            self.branch.set_last_revision_info(0, new_revision)
982
 
            return False
983
 
        _mod_revision.check_not_reserved_id(new_revision)
984
 
        try:
985
 
            self.branch.generate_revision_history(new_revision)
986
 
        except errors.NoSuchRevision:
987
 
            # not present in the repo - dont try to set it deeper than the tip
988
 
            self.branch._set_revision_history([new_revision])
989
 
        return True
990
 
 
991
 
    def remove(self, files, verbose=False, to_file=None, keep_files=True,
992
 
               force=False):
993
 
        """Remove nominated files from the working tree metadata.
994
 
 
995
 
        :files: File paths relative to the basedir.
996
 
        :keep_files: If true, the files will also be kept.
997
 
        :force: Delete files and directories, even if they are changed and
998
 
            even if the directories are not empty.
999
 
        """
1000
 
        raise NotImplementedError(self.remove)
1001
 
 
1002
 
    def revert(self, filenames=None, old_tree=None, backups=True,
1003
 
               pb=None, report_changes=False):
1004
 
        from .conflicts import resolve
1005
 
        with cleanup.ExitStack() as exit_stack:
1006
 
            exit_stack.enter_context(self.lock_tree_write())
1007
 
            if old_tree is None:
1008
 
                basis_tree = self.basis_tree()
1009
 
                exit_stack.enter_context(basis_tree.lock_read())
1010
 
                old_tree = basis_tree
1011
 
            else:
1012
 
                basis_tree = None
1013
 
            conflicts = transform.revert(self, old_tree, filenames, backups, pb,
1014
 
                                         report_changes)
1015
 
            if filenames is None and len(self.get_parent_ids()) > 1:
1016
 
                parent_trees = []
1017
 
                last_revision = self.last_revision()
1018
 
                if last_revision != _mod_revision.NULL_REVISION:
1019
 
                    if basis_tree is None:
1020
 
                        basis_tree = self.basis_tree()
1021
 
                        exit_stack.enter_context(basis_tree.lock_read())
1022
 
                    parent_trees.append((last_revision, basis_tree))
1023
 
                self.set_parent_trees(parent_trees)
1024
 
                resolve(self)
1025
 
            else:
1026
 
                resolve(self, filenames, ignore_misses=True, recursive=True)
1027
 
            return conflicts
1028
 
 
1029
 
    def store_uncommitted(self):
1030
 
        """Store uncommitted changes from the tree in the branch."""
1031
 
        raise NotImplementedError(self.store_uncommitted)
1032
 
 
1033
 
    def restore_uncommitted(self):
1034
 
        """Restore uncommitted changes from the branch into the tree."""
1035
 
        raise NotImplementedError(self.restore_uncommitted)
1036
 
 
1037
 
    def revision_tree(self, revision_id):
1038
 
        """See Tree.revision_tree.
1039
 
 
1040
 
        For trees that can be obtained from the working tree, this
1041
 
        will do so. For other trees, it will fall back to the repository.
1042
 
        """
1043
 
        raise NotImplementedError(self.revision_tree)
1044
 
 
 
498
        """See Branch.lock_write, and WorkingTree.unlock."""
 
499
        return self.branch.lock_write()
 
500
 
 
501
    @needs_read_lock
 
502
    def read_working_inventory(self):
 
503
        """Read the working inventory."""
 
504
        # ElementTree does its own conversion from UTF-8, so open in
 
505
        # binary.
 
506
        f = self.branch.controlfile('inventory', 'rb')
 
507
        return bzrlib.xml5.serializer_v5.read_inventory(f)
 
508
 
 
509
    @needs_write_lock
 
510
    def remove(self, files, verbose=False):
 
511
        """Remove nominated files from the working inventory..
 
512
 
 
513
        This does not remove their text.  This does not run on XXX on what? RBC
 
514
 
 
515
        TODO: Refuse to remove modified files unless --force is given?
 
516
 
 
517
        TODO: Do something useful with directories.
 
518
 
 
519
        TODO: Should this remove the text or not?  Tough call; not
 
520
        removing may be useful and the user can just use use rm, and
 
521
        is the opposite of add.  Removing it is consistent with most
 
522
        other tools.  Maybe an option.
 
523
        """
 
524
        ## TODO: Normalize names
 
525
        ## TODO: Remove nested loops; better scalability
 
526
        if isinstance(files, basestring):
 
527
            files = [files]
 
528
 
 
529
        inv = self.inventory
 
530
 
 
531
        # do this before any modifications
 
532
        for f in files:
 
533
            fid = inv.path2id(f)
 
534
            if not fid:
 
535
                # TODO: Perhaps make this just a warning, and continue?
 
536
                # This tends to happen when 
 
537
                raise NotVersionedError(path=f)
 
538
            mutter("remove inventory entry %s {%s}", quotefn(f), fid)
 
539
            if verbose:
 
540
                # having remove it, it must be either ignored or unknown
 
541
                if self.is_ignored(f):
 
542
                    new_status = 'I'
 
543
                else:
 
544
                    new_status = '?'
 
545
                show_status(new_status, inv[fid].kind, quotefn(f))
 
546
            del inv[fid]
 
547
 
 
548
        self._write_inventory(inv)
 
549
 
 
550
    @needs_write_lock
 
551
    def revert(self, filenames, old_tree=None, backups=True):
 
552
        from bzrlib.merge import merge_inner
 
553
        if old_tree is None:
 
554
            old_tree = self.branch.basis_tree()
 
555
        merge_inner(self.branch, old_tree,
 
556
                    self, ignore_zero=True,
 
557
                    backup_files=backups, 
 
558
                    interesting_files=filenames)
 
559
        if not len(filenames):
 
560
            self.set_pending_merges([])
 
561
 
 
562
    @needs_write_lock
 
563
    def set_inventory(self, new_inventory_list):
 
564
        from bzrlib.inventory import (Inventory,
 
565
                                      InventoryDirectory,
 
566
                                      InventoryEntry,
 
567
                                      InventoryFile,
 
568
                                      InventoryLink)
 
569
        inv = Inventory(self.get_root_id())
 
570
        for path, file_id, parent, kind in new_inventory_list:
 
571
            name = os.path.basename(path)
 
572
            if name == "":
 
573
                continue
 
574
            # fixme, there should be a factory function inv,add_?? 
 
575
            if kind == 'directory':
 
576
                inv.add(InventoryDirectory(file_id, name, parent))
 
577
            elif kind == 'file':
 
578
                inv.add(InventoryFile(file_id, name, parent))
 
579
            elif kind == 'symlink':
 
580
                inv.add(InventoryLink(file_id, name, parent))
 
581
            else:
 
582
                raise BzrError("unknown kind %r" % kind)
 
583
        self._write_inventory(inv)
 
584
 
 
585
    @needs_write_lock
1045
586
    def set_root_id(self, file_id):
1046
587
        """Set the root id for this tree."""
1047
 
        if not self.supports_setting_file_ids():
1048
 
            raise SettingFileIdUnsupported()
1049
 
        with self.lock_tree_write():
1050
 
            # for compatibility
1051
 
            if file_id is None:
1052
 
                raise ValueError(
1053
 
                    'WorkingTree.set_root_id with fileid=None')
1054
 
            file_id = osutils.safe_file_id(file_id)
1055
 
            self._set_root_id(file_id)
1056
 
 
1057
 
    def _set_root_id(self, file_id):
1058
 
        """Set the root id for this tree, in a format specific manner.
1059
 
 
1060
 
        :param file_id: The file id to assign to the root. It must not be
1061
 
            present in the current inventory or an error will occur. It must
1062
 
            not be None, but rather a valid file id.
1063
 
        """
1064
 
        raise NotImplementedError(self._set_root_id)
 
588
        inv = self.read_working_inventory()
 
589
        orig_root_id = inv.root.file_id
 
590
        del inv._byid[inv.root.file_id]
 
591
        inv.root.file_id = file_id
 
592
        inv._byid[inv.root.file_id] = inv.root
 
593
        for fid in inv:
 
594
            entry = inv[fid]
 
595
            if entry.parent_id in (None, orig_root_id):
 
596
                entry.parent_id = inv.root.file_id
 
597
        self._write_inventory(inv)
1065
598
 
1066
599
    def unlock(self):
1067
600
        """See Branch.unlock.
1068
 
 
 
601
        
1069
602
        WorkingTree locking just uses the Branch locking facilities.
1070
603
        This is current because all working trees have an embedded branch
1071
604
        within them. IF in the future, we were to make branch data shareable
1072
 
        between multiple working trees, i.e. via shared storage, then we
 
605
        between multiple working trees, i.e. via shared storage, then we 
1073
606
        would probably want to lock both the local tree, and the branch.
1074
607
        """
1075
 
        raise NotImplementedError(self.unlock)
1076
 
 
1077
 
    _marker = object()
1078
 
 
1079
 
    def update(self, change_reporter=None, possible_transports=None,
1080
 
               revision=None, old_tip=_marker, show_base=False):
1081
 
        """Update a working tree along its branch.
1082
 
 
1083
 
        This will update the branch if its bound too, which means we have
1084
 
        multiple trees involved:
1085
 
 
1086
 
        - The new basis tree of the master.
1087
 
        - The old basis tree of the branch.
1088
 
        - The old basis tree of the working tree.
1089
 
        - The current working tree state.
1090
 
 
1091
 
        Pathologically, all three may be different, and non-ancestors of each
1092
 
        other.  Conceptually we want to:
1093
 
 
1094
 
        - Preserve the wt.basis->wt.state changes
1095
 
        - Transform the wt.basis to the new master basis.
1096
 
        - Apply a merge of the old branch basis to get any 'local' changes from
1097
 
          it into the tree.
1098
 
        - Restore the wt.basis->wt.state changes.
1099
 
 
1100
 
        There isn't a single operation at the moment to do that, so we:
1101
 
 
1102
 
        - Merge current state -> basis tree of the master w.r.t. the old tree
1103
 
          basis.
1104
 
        - Do a 'normal' merge of the old branch basis if it is relevant.
1105
 
 
1106
 
        :param revision: The target revision to update to. Must be in the
1107
 
            revision history.
1108
 
        :param old_tip: If branch.update() has already been run, the value it
1109
 
            returned (old tip of the branch or None). _marker is used
1110
 
            otherwise.
1111
 
        """
1112
 
        if self.branch.get_bound_location() is not None:
1113
 
            self.lock_write()
1114
 
            update_branch = (old_tip is self._marker)
1115
 
        else:
1116
 
            self.lock_tree_write()
1117
 
            update_branch = False
 
608
        return self.branch.unlock()
 
609
 
 
610
    @needs_write_lock
 
611
    def _write_inventory(self, inv):
 
612
        """Write inventory as the current inventory."""
 
613
        from cStringIO import StringIO
 
614
        from bzrlib.atomicfile import AtomicFile
 
615
        sio = StringIO()
 
616
        bzrlib.xml5.serializer_v5.write_inventory(inv, sio)
 
617
        sio.seek(0)
 
618
        f = AtomicFile(self.branch.controlfilename('inventory'))
1118
619
        try:
1119
 
            if update_branch:
1120
 
                old_tip = self.branch.update(possible_transports)
1121
 
            else:
1122
 
                if old_tip is self._marker:
1123
 
                    old_tip = None
1124
 
            return self._update_tree(old_tip, change_reporter, revision, show_base)
 
620
            pumpfile(sio, f)
 
621
            f.commit()
1125
622
        finally:
1126
 
            self.unlock()
1127
 
 
1128
 
    def _update_tree(self, old_tip=None, change_reporter=None, revision=None,
1129
 
                     show_base=False):
1130
 
        """Update a tree to the master branch.
1131
 
 
1132
 
        :param old_tip: if supplied, the previous tip revision the branch,
1133
 
            before it was changed to the master branch's tip.
1134
 
        """
1135
 
        # here if old_tip is not None, it is the old tip of the branch before
1136
 
        # it was updated from the master branch. This should become a pending
1137
 
        # merge in the working tree to preserve the user existing work.  we
1138
 
        # cant set that until we update the working trees last revision to be
1139
 
        # one from the new branch, because it will just get absorbed by the
1140
 
        # parent de-duplication logic.
1141
 
        #
1142
 
        # We MUST save it even if an error occurs, because otherwise the users
1143
 
        # local work is unreferenced and will appear to have been lost.
1144
 
        #
1145
 
        with self.lock_tree_write():
1146
 
            nb_conflicts = 0
1147
 
            try:
1148
 
                last_rev = self.get_parent_ids()[0]
1149
 
            except IndexError:
1150
 
                last_rev = _mod_revision.NULL_REVISION
1151
 
            if revision is None:
1152
 
                revision = self.branch.last_revision()
1153
 
 
1154
 
            old_tip = old_tip or _mod_revision.NULL_REVISION
1155
 
 
1156
 
            if not _mod_revision.is_null(old_tip) and old_tip != last_rev:
1157
 
                # the branch we are bound to was updated
1158
 
                # merge those changes in first
1159
 
                base_tree = self.basis_tree()
1160
 
                other_tree = self.branch.repository.revision_tree(old_tip)
1161
 
                nb_conflicts = merge.merge_inner(self.branch, other_tree,
1162
 
                                                 base_tree, this_tree=self,
1163
 
                                                 change_reporter=change_reporter,
1164
 
                                                 show_base=show_base)
1165
 
                if nb_conflicts:
1166
 
                    self.add_parent_tree((old_tip, other_tree))
1167
 
                    note(gettext('Rerun update after fixing the conflicts.'))
1168
 
                    return nb_conflicts
1169
 
 
1170
 
            if last_rev != _mod_revision.ensure_null(revision):
1171
 
                # the working tree is up to date with the branch
1172
 
                # we can merge the specified revision from master
1173
 
                to_tree = self.branch.repository.revision_tree(revision)
1174
 
                to_root_id = to_tree.path2id('')
1175
 
 
1176
 
                basis = self.basis_tree()
1177
 
                with basis.lock_read():
1178
 
                    if (basis.path2id('') is None or basis.path2id('') != to_root_id):
1179
 
                        self.set_root_id(to_root_id)
1180
 
                        self.flush()
1181
 
 
1182
 
                # determine the branch point
1183
 
                graph = self.branch.repository.get_graph()
1184
 
                base_rev_id = graph.find_unique_lca(self.branch.last_revision(),
1185
 
                                                    last_rev)
1186
 
                base_tree = self.branch.repository.revision_tree(base_rev_id)
1187
 
 
1188
 
                nb_conflicts = merge.merge_inner(self.branch, to_tree, base_tree,
1189
 
                                                 this_tree=self,
1190
 
                                                 change_reporter=change_reporter,
1191
 
                                                 show_base=show_base)
1192
 
                self.set_last_revision(revision)
1193
 
                # TODO - dedup parents list with things merged by pull ?
1194
 
                # reuse the tree we've updated to to set the basis:
1195
 
                parent_trees = [(revision, to_tree)]
1196
 
                merges = self.get_parent_ids()[1:]
1197
 
                # Ideally we ask the tree for the trees here, that way the working
1198
 
                # tree can decide whether to give us the entire tree or give us a
1199
 
                # lazy initialised tree. dirstate for instance will have the trees
1200
 
                # in ram already, whereas a last-revision + basis-inventory tree
1201
 
                # will not, but also does not need them when setting parents.
1202
 
                for parent in merges:
1203
 
                    parent_trees.append(
1204
 
                        (parent, self.branch.repository.revision_tree(parent)))
1205
 
                if not _mod_revision.is_null(old_tip):
1206
 
                    parent_trees.append(
1207
 
                        (old_tip, self.branch.repository.revision_tree(old_tip)))
1208
 
                self.set_parent_trees(parent_trees)
1209
 
                last_rev = parent_trees[0][0]
1210
 
            return nb_conflicts
1211
 
 
1212
 
    def set_conflicts(self, arg):
1213
 
        raise errors.UnsupportedOperation(self.set_conflicts, self)
1214
 
 
1215
 
    def add_conflicts(self, arg):
1216
 
        raise errors.UnsupportedOperation(self.add_conflicts, self)
1217
 
 
1218
 
    def conflicts(self):
1219
 
        raise NotImplementedError(self.conflicts)
1220
 
 
1221
 
    def walkdirs(self, prefix=""):
1222
 
        """Walk the directories of this tree.
1223
 
 
1224
 
        returns a generator which yields items in the form:
1225
 
                ((curren_directory_path, fileid),
1226
 
                 [(file1_path, file1_name, file1_kind, (lstat), file1_id,
1227
 
                   file1_kind), ... ])
1228
 
 
1229
 
        This API returns a generator, which is only valid during the current
1230
 
        tree transaction - within a single lock_read or lock_write duration.
1231
 
 
1232
 
        If the tree is not locked, it may cause an error to be raised,
1233
 
        depending on the tree implementation.
1234
 
        """
1235
 
        raise NotImplementedError(self.walkdirs)
1236
 
 
1237
 
    @deprecated_method(deprecated_in((3, 0, 1)))
1238
 
    def auto_resolve(self):
1239
 
        """Automatically resolve text conflicts according to contents.
1240
 
 
1241
 
        Only text conflicts are auto_resolvable. Files with no conflict markers
1242
 
        are considered 'resolved', because bzr always puts conflict markers
1243
 
        into files that have text conflicts.  The corresponding .THIS .BASE and
1244
 
        .OTHER files are deleted, as per 'resolve'.
1245
 
 
1246
 
        :return: a tuple of ConflictLists: (un_resolved, resolved).
1247
 
        """
1248
 
        with self.lock_tree_write():
1249
 
            un_resolved = _mod_conflicts.ConflictList()
1250
 
            resolved = _mod_conflicts.ConflictList()
1251
 
            for conflict in self.conflicts():
1252
 
                try:
1253
 
                    conflict.action_auto(self)
1254
 
                except NotImplementedError:
1255
 
                    un_resolved.append(conflict)
1256
 
                else:
1257
 
                    conflict.cleanup(self)
1258
 
                    resolved.append(conflict)
1259
 
            self.set_conflicts(un_resolved)
1260
 
            return un_resolved, resolved
1261
 
 
1262
 
    def _validate(self):
1263
 
        """Validate internal structures.
1264
 
 
1265
 
        This is meant mostly for the test suite. To give it a chance to detect
1266
 
        corruption after actions have occurred. The default implementation is a
1267
 
        just a no-op.
1268
 
 
1269
 
        :return: None. An exception should be raised if there is an error.
1270
 
        """
1271
 
        return
1272
 
 
1273
 
    def check_state(self):
1274
 
        """Check that the working state is/isn't valid."""
1275
 
        raise NotImplementedError(self.check_state)
1276
 
 
1277
 
    def reset_state(self, revision_ids=None):
1278
 
        """Reset the state of the working tree.
1279
 
 
1280
 
        This does a hard-reset to a last-known-good state. This is a way to
1281
 
        fix if something got corrupted (like the .bzr/checkout/dirstate file)
1282
 
        """
1283
 
        raise NotImplementedError(self.reset_state)
1284
 
 
1285
 
    def _get_rules_searcher(self, default_searcher):
1286
 
        """See Tree._get_rules_searcher."""
1287
 
        if self._rules_searcher is None:
1288
 
            self._rules_searcher = super(WorkingTree,
1289
 
                                         self)._get_rules_searcher(default_searcher)
1290
 
        return self._rules_searcher
1291
 
 
1292
 
    def get_shelf_manager(self):
1293
 
        """Return the ShelfManager for this WorkingTree."""
1294
 
        raise NotImplementedError(self.get_shelf_manager)
1295
 
 
1296
 
    def get_canonical_paths(self, paths):
1297
 
        """Like get_canonical_path() but works on multiple items.
1298
 
 
1299
 
        :param paths: A sequence of paths relative to the root of the tree.
1300
 
        :return: A list of paths, with each item the corresponding input path
1301
 
            adjusted to account for existing elements that match case
1302
 
            insensitively.
1303
 
        """
1304
 
        with self.lock_read():
1305
 
            for path in paths:
1306
 
                yield path
1307
 
 
1308
 
    def get_canonical_path(self, path):
1309
 
        """Returns the first item in the tree that matches a path.
1310
 
 
1311
 
        This is meant to allow case-insensitive path lookups on e.g.
1312
 
        FAT filesystems.
1313
 
 
1314
 
        If a path matches exactly, it is returned. If no path matches exactly
1315
 
        but more than one path matches according to the underlying file system,
1316
 
        it is implementation defined which is returned.
1317
 
 
1318
 
        If no path matches according to the file system, the input path is
1319
 
        returned, but with as many path entries that do exist changed to their
1320
 
        canonical form.
1321
 
 
1322
 
        If you need to resolve many names from the same tree, you should
1323
 
        use get_canonical_paths() to avoid O(N) behaviour.
1324
 
 
1325
 
        :param path: A paths relative to the root of the tree.
1326
 
        :return: The input path adjusted to account for existing elements
1327
 
        that match case insensitively.
1328
 
        """
1329
 
        with self.lock_read():
1330
 
            return next(self.get_canonical_paths([path]))
1331
 
 
1332
 
    def reference_parent(self, path, branch=None, possible_transports=None):
1333
 
        raise errors.UnsupportedOperation(self.reference_parent, self)
1334
 
 
1335
 
    def get_reference_info(self, path, branch=None):
1336
 
        raise errors.UnsupportedOperation(self.get_reference_info, self)
1337
 
 
1338
 
    def set_reference_info(self, tree_path, branch_location):
1339
 
        raise errors.UnsupportedOperation(self.set_reference_info, self)
1340
 
 
1341
 
 
1342
 
class WorkingTreeFormatRegistry(ControlComponentFormatRegistry):
1343
 
    """Registry for working tree formats."""
1344
 
 
1345
 
    def __init__(self, other_registry=None):
1346
 
        super(WorkingTreeFormatRegistry, self).__init__(other_registry)
1347
 
        self._default_format = None
1348
 
        self._default_format_key = None
1349
 
 
1350
 
    def get_default(self):
1351
 
        """Return the current default format."""
1352
 
        if (self._default_format_key is not None and
1353
 
                self._default_format is None):
1354
 
            self._default_format = self.get(self._default_format_key)
1355
 
        return self._default_format
1356
 
 
1357
 
    def set_default(self, format):
1358
 
        """Set the default format."""
1359
 
        self._default_format = format
1360
 
        self._default_format_key = None
1361
 
 
1362
 
    def set_default_key(self, format_string):
1363
 
        """Set the default format by its format string."""
1364
 
        self._default_format_key = format_string
1365
 
        self._default_format = None
1366
 
 
1367
 
 
1368
 
format_registry = WorkingTreeFormatRegistry()
1369
 
 
1370
 
 
1371
 
class WorkingTreeFormat(ControlComponentFormat):
1372
 
    """An encapsulation of the initialization and open routines for a format.
1373
 
 
1374
 
    Formats provide three things:
1375
 
     * An initialization routine,
1376
 
     * a format string,
1377
 
     * an open routine.
1378
 
 
1379
 
    Formats are placed in an dict by their format string for reference
1380
 
    during workingtree opening. Its not required that these be instances, they
1381
 
    can be classes themselves with class methods - it simply depends on
1382
 
    whether state is needed for a given format or not.
1383
 
 
1384
 
    Once a format is deprecated, just deprecate the initialize and open
1385
 
    methods on the format class. Do not deprecate the object, as the
1386
 
    object will be created every time regardless.
1387
 
    """
1388
 
 
1389
 
    requires_rich_root = False
1390
 
 
1391
 
    upgrade_recommended = False
1392
 
 
1393
 
    requires_normalized_unicode_filenames = False
1394
 
 
1395
 
    case_sensitive_filename = "FoRMaT"
1396
 
 
1397
 
    missing_parent_conflicts = False
1398
 
    """If this format supports missing parent conflicts."""
1399
 
 
1400
 
    supports_versioned_directories = None
1401
 
 
1402
 
    supports_merge_modified = True
1403
 
    """If this format supports storing merge modified hashes."""
1404
 
 
1405
 
    supports_setting_file_ids = True
1406
 
    """If this format allows setting the file id."""
1407
 
 
1408
 
    supports_store_uncommitted = True
1409
 
    """If this format supports shelve-like functionality."""
1410
 
 
1411
 
    supports_leftmost_parent_id_as_ghost = True
1412
 
 
1413
 
    supports_righthand_parent_id_as_ghost = True
1414
 
 
1415
 
    ignore_filename = None
1416
 
    """Name of file with ignore patterns, if any. """
1417
 
 
1418
 
    def initialize(self, controldir, revision_id=None, from_branch=None,
1419
 
                   accelerator_tree=None, hardlink=False):
1420
 
        """Initialize a new working tree in controldir.
1421
 
 
1422
 
        :param controldir: ControlDir to initialize the working tree in.
1423
 
        :param revision_id: allows creating a working tree at a different
1424
 
            revision than the branch is at.
1425
 
        :param from_branch: Branch to checkout
1426
 
        :param accelerator_tree: A tree which can be used for retrieving file
1427
 
            contents more quickly than the revision tree, i.e. a workingtree.
1428
 
            The revision tree will be used for cases where accelerator_tree's
1429
 
            content is different.
1430
 
        :param hardlink: If true, hard-link files from accelerator_tree,
1431
 
            where possible.
1432
 
        """
1433
 
        raise NotImplementedError(self.initialize)
1434
 
 
1435
 
    def __eq__(self, other):
1436
 
        return self.__class__ is other.__class__
1437
 
 
1438
 
    def __ne__(self, other):
1439
 
        return not (self == other)
1440
 
 
1441
 
    def get_format_description(self):
1442
 
        """Return the short description for this format."""
1443
 
        raise NotImplementedError(self.get_format_description)
1444
 
 
1445
 
    def is_supported(self):
1446
 
        """Is this format supported?
1447
 
 
1448
 
        Supported formats can be initialized and opened.
1449
 
        Unsupported formats may not support initialization or committing or
1450
 
        some other features depending on the reason for not being supported.
1451
 
        """
1452
 
        return True
1453
 
 
1454
 
    def supports_content_filtering(self):
1455
 
        """True if this format supports content filtering."""
1456
 
        return False
1457
 
 
1458
 
    def supports_views(self):
1459
 
        """True if this format supports stored views."""
1460
 
        return False
1461
 
 
1462
 
    def get_controldir_for_branch(self):
1463
 
        """Get the control directory format for creating branches.
1464
 
 
1465
 
        This is to support testing of working tree formats that can not exist
1466
 
        in the same control directory as a branch.
1467
 
        """
1468
 
        return self._matchingcontroldir
1469
 
 
1470
 
 
1471
 
format_registry.register_lazy(b"Bazaar Working Tree Format 4 (bzr 0.15)\n",
1472
 
                              "breezy.bzr.workingtree_4", "WorkingTreeFormat4")
1473
 
format_registry.register_lazy(b"Bazaar Working Tree Format 5 (bzr 1.11)\n",
1474
 
                              "breezy.bzr.workingtree_4", "WorkingTreeFormat5")
1475
 
format_registry.register_lazy(b"Bazaar Working Tree Format 6 (bzr 1.14)\n",
1476
 
                              "breezy.bzr.workingtree_4", "WorkingTreeFormat6")
1477
 
format_registry.register_lazy(b"Bazaar-NG Working Tree format 3",
1478
 
                              "breezy.bzr.workingtree_3", "WorkingTreeFormat3")
1479
 
format_registry.set_default_key(b"Bazaar Working Tree Format 6 (bzr 1.14)\n")
 
623
            f.close()
 
624
        mutter('wrote working inventory')
 
625
            
 
626
 
 
627
CONFLICT_SUFFIXES = ('.THIS', '.BASE', '.OTHER')
 
628
def get_conflicted_stem(path):
 
629
    for suffix in CONFLICT_SUFFIXES:
 
630
        if path.endswith(suffix):
 
631
            return path[:-len(suffix)]