~bzr-pqm/bzr/bzr.dev

« back to all changes in this revision

Viewing changes to bzrlib/workingtree.py

  • Committer: Robert Collins
  • Date: 2010-05-06 23:54:05 UTC
  • mto: This revision was merged to the branch mainline in revision 5223.
  • Revision ID: robertc@robertcollins.net-20100506235405-wii4elupfhzl3jvy
Add __str__ to the new helper classes.

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
 
# Copyright (C) 2005, 2006, 2007 Canonical Ltd
 
1
# Copyright (C) 2005-2010 Canonical Ltd
2
2
#
3
3
# This program is free software; you can redistribute it and/or modify
4
4
# it under the terms of the GNU General Public License as published by
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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
15
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 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.  The WorkingTree has an inventory 
22
 
which is updated by these operations.  A commit produces a 
 
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
23
new revision based on the workingtree and its inventory.
24
24
 
25
25
At the moment every WorkingTree has its own branch.  Remote
29
29
WorkingTree.open(dir).
30
30
"""
31
31
 
32
 
# TODO: Give the workingtree sole responsibility for the working inventory;
33
 
# remove the variable and references to it from the branch.  This may require
34
 
# updating the commit code so as to update the inventory within the working
35
 
# copy, and making sure there's only one WorkingTree for any directory on disk.
36
 
# At the moment they may alias the inventory and have old copies of it in
37
 
# memory.  (Now done? -- mbp 20060309)
38
32
 
39
33
from cStringIO import StringIO
40
34
import os
48
42
import itertools
49
43
import operator
50
44
import stat
51
 
from time import time
52
 
import warnings
53
45
import re
54
46
 
55
47
import bzrlib
57
49
    branch,
58
50
    bzrdir,
59
51
    conflicts as _mod_conflicts,
60
 
    dirstate,
61
52
    errors,
62
53
    generate_ids,
63
54
    globbing,
 
55
    graph as _mod_graph,
64
56
    hashcache,
65
57
    ignores,
 
58
    inventory,
66
59
    merge,
67
 
    osutils,
68
60
    revision as _mod_revision,
69
61
    revisiontree,
70
 
    repository,
71
 
    textui,
72
62
    trace,
73
63
    transform,
74
64
    ui,
75
 
    urlutils,
 
65
    views,
76
66
    xml5,
77
 
    xml6,
78
67
    xml7,
79
68
    )
80
69
import bzrlib.branch
81
70
from bzrlib.transport import get_transport
82
 
import bzrlib.ui
83
 
from bzrlib.workingtree_4 import WorkingTreeFormat4
 
71
from bzrlib.workingtree_4 import (
 
72
    WorkingTreeFormat4,
 
73
    WorkingTreeFormat5,
 
74
    WorkingTreeFormat6,
 
75
    )
84
76
""")
85
77
 
86
78
from bzrlib import symbol_versioning
87
79
from bzrlib.decorators import needs_read_lock, needs_write_lock
88
 
from bzrlib.inventory import InventoryEntry, Inventory, ROOT_ID, TreeReference
89
 
from bzrlib.lockable_files import LockableFiles, TransportLock
 
80
from bzrlib.lockable_files import LockableFiles
90
81
from bzrlib.lockdir import LockDir
91
82
import bzrlib.mutabletree
92
83
from bzrlib.mutabletree import needs_tree_write_lock
 
84
from bzrlib import osutils
93
85
from bzrlib.osutils import (
94
 
    compact_date,
95
86
    file_kind,
96
87
    isdir,
97
88
    normpath,
98
89
    pathjoin,
99
 
    rand_chars,
100
90
    realpath,
101
91
    safe_unicode,
102
92
    splitpath,
103
93
    supports_executable,
104
94
    )
 
95
from bzrlib.filters import filtered_input_file
105
96
from bzrlib.trace import mutter, note
106
97
from bzrlib.transport.local import LocalTransport
107
 
from bzrlib.progress import DummyProgress, ProgressPhase
108
 
from bzrlib.revision import NULL_REVISION, CURRENT_REVISION
 
98
from bzrlib.revision import CURRENT_REVISION
109
99
from bzrlib.rio import RioReader, rio_file, Stanza
110
 
from bzrlib.symbol_versioning import (deprecated_passed,
111
 
        deprecated_method,
112
 
        deprecated_function,
113
 
        DEPRECATED_PARAMETER,
114
 
        zero_eight,
115
 
        zero_eleven,
116
 
        zero_thirteen,
117
 
        )
 
100
from bzrlib.symbol_versioning import (
 
101
    deprecated_passed,
 
102
    DEPRECATED_PARAMETER,
 
103
    )
118
104
 
119
105
 
120
106
MERGE_MODIFIED_HEADER_1 = "BZR merge-modified list format 1"
 
107
# TODO: Modifying the conflict objects or their type is currently nearly
 
108
# impossible as there is no clear relationship between the working tree format
 
109
# and the conflict list file format.
121
110
CONFLICT_HEADER_1 = "BZR conflict list format 1"
122
111
 
123
112
ERROR_PATH_NOT_FOUND = 3    # WindowsError errno code, equivalent to ENOENT
124
113
 
125
114
 
126
 
@deprecated_function(zero_thirteen)
127
 
def gen_file_id(name):
128
 
    """Return new file id for the basename 'name'.
129
 
 
130
 
    Use bzrlib.generate_ids.gen_file_id() instead
131
 
    """
132
 
    return generate_ids.gen_file_id(name)
133
 
 
134
 
 
135
 
@deprecated_function(zero_thirteen)
136
 
def gen_root_id():
137
 
    """Return a new tree-root file id.
138
 
 
139
 
    This has been deprecated in favor of bzrlib.generate_ids.gen_root_id()
140
 
    """
141
 
    return generate_ids.gen_root_id()
142
 
 
143
 
 
144
115
class TreeEntry(object):
145
116
    """An entry that implements the minimum interface used by commands.
146
117
 
147
 
    This needs further inspection, it may be better to have 
 
118
    This needs further inspection, it may be better to have
148
119
    InventoryEntries without ids - though that seems wrong. For now,
149
120
    this is a parallel hierarchy to InventoryEntry, and needs to become
150
121
    one of several things: decorates to that hierarchy, children of, or
153
124
    no InventoryEntry available - i.e. for unversioned objects.
154
125
    Perhaps they should be UnversionedEntry et al. ? - RBC 20051003
155
126
    """
156
 
 
 
127
 
157
128
    def __eq__(self, other):
158
129
        # yes, this us ugly, TODO: best practice __eq__ style.
159
130
        return (isinstance(other, TreeEntry)
160
131
                and other.__class__ == self.__class__)
161
 
 
 
132
 
162
133
    def kind_character(self):
163
134
        return "???"
164
135
 
196
167
        return ''
197
168
 
198
169
 
199
 
class WorkingTree(bzrlib.mutabletree.MutableTree):
 
170
class WorkingTree(bzrlib.mutabletree.MutableTree,
 
171
    bzrdir.ControlComponent):
200
172
    """Working copy tree.
201
173
 
202
174
    The inventory is held in the `Branch` working-inventory, and the
206
178
    not listed in the Inventory and vice versa.
207
179
    """
208
180
 
 
181
    # override this to set the strategy for storing views
 
182
    def _make_views(self):
 
183
        return views.DisabledViews(self)
 
184
 
209
185
    def __init__(self, basedir='.',
210
186
                 branch=DEPRECATED_PARAMETER,
211
187
                 _inventory=None,
222
198
        if not _internal:
223
199
            raise errors.BzrError("Please use bzrdir.open_workingtree or "
224
200
                "WorkingTree.open() to obtain a WorkingTree.")
225
 
        assert isinstance(basedir, basestring), \
226
 
            "base directory %r is not a string" % basedir
227
201
        basedir = safe_unicode(basedir)
228
202
        mutter("opening working tree %r", basedir)
229
203
        if deprecated_passed(branch):
237
211
            self._control_files = self.branch.control_files
238
212
        else:
239
213
            # assume all other formats have their own control files.
240
 
            assert isinstance(_control_files, LockableFiles), \
241
 
                    "_control_files must be a LockableFiles, not %r" \
242
 
                    % _control_files
243
214
            self._control_files = _control_files
 
215
        self._transport = self._control_files._transport
244
216
        # update the whole cache up front and write to disk if anything changed;
245
217
        # in the future we might want to do this more selectively
246
218
        # two possible ways offer themselves : in self._unlock, write the cache
250
222
        wt_trans = self.bzrdir.get_workingtree_transport(None)
251
223
        cache_filename = wt_trans.local_abspath('stat-cache')
252
224
        self._hashcache = hashcache.HashCache(basedir, cache_filename,
253
 
                                              self._control_files._file_mode)
 
225
            self.bzrdir._get_file_mode(),
 
226
            self._content_filter_stack_provider())
254
227
        hc = self._hashcache
255
228
        hc.read()
256
229
        # is this scan needed ? it makes things kinda slow.
270
243
            # the Format factory and creation methods that are
271
244
            # permitted to do this.
272
245
            self._set_inventory(_inventory, dirty=False)
 
246
        self._detect_case_handling()
 
247
        self._rules_searcher = None
 
248
        self.views = self._make_views()
 
249
 
 
250
    @property
 
251
    def user_transport(self):
 
252
        return self.bzrdir.user_transport
 
253
 
 
254
    @property
 
255
    def control_transport(self):
 
256
        return self._transport
 
257
 
 
258
    def _detect_case_handling(self):
 
259
        wt_trans = self.bzrdir.get_workingtree_transport(None)
 
260
        try:
 
261
            wt_trans.stat("FoRMaT")
 
262
        except errors.NoSuchFile:
 
263
            self.case_sensitive = True
 
264
        else:
 
265
            self.case_sensitive = False
 
266
 
 
267
        self._setup_directory_is_tree_reference()
273
268
 
274
269
    branch = property(
275
270
        fget=lambda self: self._branch,
290
285
        self._control_files.break_lock()
291
286
        self.branch.break_lock()
292
287
 
 
288
    def _get_check_refs(self):
 
289
        """Return the references needed to perform a check of this tree.
 
290
        
 
291
        The default implementation returns no refs, and is only suitable for
 
292
        trees that have no local caching and can commit on ghosts at any time.
 
293
 
 
294
        :seealso: bzrlib.check for details about check_refs.
 
295
        """
 
296
        return []
 
297
 
293
298
    def requires_rich_root(self):
294
299
        return self._format.requires_rich_root
295
300
 
296
301
    def supports_tree_reference(self):
297
302
        return False
298
303
 
 
304
    def supports_content_filtering(self):
 
305
        return self._format.supports_content_filtering()
 
306
 
 
307
    def supports_views(self):
 
308
        return self.views.supports_views()
 
309
 
299
310
    def _set_inventory(self, inv, dirty):
300
311
        """Set the internal cached inventory.
301
312
 
306
317
            False then the inventory is the same as that on disk and any
307
318
            serialisation would be unneeded overhead.
308
319
        """
309
 
        assert inv.root is not None
310
320
        self._inventory = inv
311
321
        self._inventory_is_modified = dirty
312
322
 
316
326
 
317
327
        """
318
328
        if path is None:
319
 
            path = os.path.getcwdu()
 
329
            path = osutils.getcwd()
320
330
        control = bzrdir.BzrDir.open(path, _unsupported)
321
331
        return control.open_workingtree(_unsupported)
322
 
        
 
332
 
323
333
    @staticmethod
324
334
    def open_containing(path=None):
325
335
        """Open an existing working tree which has its root about path.
326
 
        
 
336
 
327
337
        This probes for a working tree at path and searches upwards from there.
328
338
 
329
339
        Basically we keep looking up until we find the control directory or
347
357
        """
348
358
        return WorkingTree.open(path, _unsupported=True)
349
359
 
 
360
    @staticmethod
 
361
    def find_trees(location):
 
362
        def list_current(transport):
 
363
            return [d for d in transport.list_dir('') if d != '.bzr']
 
364
        def evaluate(bzrdir):
 
365
            try:
 
366
                tree = bzrdir.open_workingtree()
 
367
            except errors.NoWorkingTree:
 
368
                return True, None
 
369
            else:
 
370
                return True, tree
 
371
        transport = get_transport(location)
 
372
        iterator = bzrdir.BzrDir.find_bzrdirs(transport, evaluate=evaluate,
 
373
                                              list_current=list_current)
 
374
        return [t for t in iterator if t is not None]
 
375
 
350
376
    # should be deprecated - this is slow and in any case treating them as a
351
377
    # container is (we now know) bad style -- mbp 20070302
352
378
    ## @deprecated_method(zero_fifteen)
361
387
            if osutils.lexists(self.abspath(path)):
362
388
                yield ie.file_id
363
389
 
 
390
    def all_file_ids(self):
 
391
        """See Tree.iter_all_file_ids"""
 
392
        return set(self.inventory)
 
393
 
364
394
    def __repr__(self):
365
395
        return "<%s of %s>" % (self.__class__.__name__,
366
396
                               getattr(self, 'basedir', None))
370
400
 
371
401
    def basis_tree(self):
372
402
        """Return RevisionTree for the current last revision.
373
 
        
 
403
 
374
404
        If the left most parent is a ghost then the returned tree will be an
375
 
        empty tree - one obtained by calling repository.revision_tree(None).
 
405
        empty tree - one obtained by calling
 
406
        repository.revision_tree(NULL_REVISION).
376
407
        """
377
408
        try:
378
409
            revision_id = self.get_parent_ids()[0]
380
411
            # no parents, return an empty revision tree.
381
412
            # in the future this should return the tree for
382
413
            # 'empty:' - the implicit root empty tree.
383
 
            return self.branch.repository.revision_tree(None)
 
414
            return self.branch.repository.revision_tree(
 
415
                       _mod_revision.NULL_REVISION)
384
416
        try:
385
417
            return self.revision_tree(revision_id)
386
418
        except errors.NoSuchRevision:
390
422
        # at this point ?
391
423
        try:
392
424
            return self.branch.repository.revision_tree(revision_id)
393
 
        except errors.RevisionNotPresent:
 
425
        except (errors.RevisionNotPresent, errors.NoSuchRevision):
394
426
            # the basis tree *may* be a ghost or a low level error may have
395
 
            # occured. If the revision is present, its a problem, if its not
 
427
            # occurred. If the revision is present, its a problem, if its not
396
428
            # its a ghost.
397
429
            if self.branch.repository.has_revision(revision_id):
398
430
                raise
399
431
            # the basis tree is a ghost so return an empty tree.
400
 
            return self.branch.repository.revision_tree(None)
 
432
            return self.branch.repository.revision_tree(
 
433
                       _mod_revision.NULL_REVISION)
401
434
 
402
435
    def _cleanup(self):
403
436
        self._flush_ignore_list_cache()
404
437
 
405
 
    @staticmethod
406
 
    @deprecated_method(zero_eight)
407
 
    def create(branch, directory):
408
 
        """Create a workingtree for branch at directory.
409
 
 
410
 
        If existing_directory already exists it must have a .bzr directory.
411
 
        If it does not exist, it will be created.
412
 
 
413
 
        This returns a new WorkingTree object for the new checkout.
414
 
 
415
 
        TODO FIXME RBC 20060124 when we have checkout formats in place this
416
 
        should accept an optional revisionid to checkout [and reject this if
417
 
        checking out into the same dir as a pre-checkout-aware branch format.]
418
 
 
419
 
        XXX: When BzrDir is present, these should be created through that 
420
 
        interface instead.
421
 
        """
422
 
        warnings.warn('delete WorkingTree.create', stacklevel=3)
423
 
        transport = get_transport(directory)
424
 
        if branch.bzrdir.root_transport.base == transport.base:
425
 
            # same dir 
426
 
            return branch.bzrdir.create_workingtree()
427
 
        # different directory, 
428
 
        # create a branch reference
429
 
        # and now a working tree.
430
 
        raise NotImplementedError
431
 
 
432
 
    @staticmethod
433
 
    @deprecated_method(zero_eight)
434
 
    def create_standalone(directory):
435
 
        """Create a checkout and a branch and a repo at directory.
436
 
 
437
 
        Directory must exist and be empty.
438
 
 
439
 
        please use BzrDir.create_standalone_workingtree
440
 
        """
441
 
        return bzrdir.BzrDir.create_standalone_workingtree(directory)
442
 
 
443
438
    def relpath(self, path):
444
439
        """Return the local path portion from a given path.
445
 
        
446
 
        The path may be absolute or relative. If its a relative path it is 
 
440
 
 
441
        The path may be absolute or relative. If its a relative path it is
447
442
        interpreted relative to the python current working directory.
448
443
        """
449
444
        return osutils.relpath(self.basedir, path)
451
446
    def has_filename(self, filename):
452
447
        return osutils.lexists(self.abspath(filename))
453
448
 
454
 
    def get_file(self, file_id, path=None):
 
449
    def get_file(self, file_id, path=None, filtered=True):
 
450
        return self.get_file_with_stat(file_id, path, filtered=filtered)[0]
 
451
 
 
452
    def get_file_with_stat(self, file_id, path=None, filtered=True,
 
453
        _fstat=os.fstat):
 
454
        """See Tree.get_file_with_stat."""
455
455
        if path is None:
456
456
            path = self.id2path(file_id)
457
 
        return self.get_file_byname(path)
458
 
 
459
 
    def get_file_text(self, file_id):
460
 
        return self.get_file(file_id).read()
461
 
 
462
 
    def get_file_byname(self, filename):
463
 
        return file(self.abspath(filename), 'rb')
 
457
        file_obj = self.get_file_byname(path, filtered=False)
 
458
        stat_value = _fstat(file_obj.fileno())
 
459
        if filtered and self.supports_content_filtering():
 
460
            filters = self._content_filter_stack(path)
 
461
            file_obj = filtered_input_file(file_obj, filters)
 
462
        return (file_obj, stat_value)
 
463
 
 
464
    def get_file_text(self, file_id, path=None, filtered=True):
 
465
        return self.get_file(file_id, path=path, filtered=filtered).read()
 
466
 
 
467
    def get_file_byname(self, filename, filtered=True):
 
468
        path = self.abspath(filename)
 
469
        f = file(path, 'rb')
 
470
        if filtered and self.supports_content_filtering():
 
471
            filters = self._content_filter_stack(filename)
 
472
            return filtered_input_file(f, filters)
 
473
        else:
 
474
            return f
 
475
 
 
476
    def get_file_lines(self, file_id, path=None, filtered=True):
 
477
        """See Tree.get_file_lines()"""
 
478
        file = self.get_file(file_id, path, filtered=filtered)
 
479
        try:
 
480
            return file.readlines()
 
481
        finally:
 
482
            file.close()
464
483
 
465
484
    @needs_read_lock
466
485
    def annotate_iter(self, file_id, default_revision=CURRENT_REVISION):
473
492
        incorrectly attributed to CURRENT_REVISION (but after committing, the
474
493
        attribution will be correct).
475
494
        """
476
 
        basis = self.basis_tree()
477
 
        basis.lock_read()
478
 
        try:
479
 
            changes = self._iter_changes(basis, True, [self.id2path(file_id)],
480
 
                require_versioned=True).next()
481
 
            changed_content, kind = changes[2], changes[6]
482
 
            if not changed_content:
483
 
                return basis.annotate_iter(file_id)
484
 
            if kind[1] is None:
485
 
                return None
486
 
            import annotate
487
 
            if kind[0] != 'file':
488
 
                old_lines = []
489
 
            else:
490
 
                old_lines = list(basis.annotate_iter(file_id))
491
 
            old = [old_lines]
492
 
            for tree in self.branch.repository.revision_trees(
493
 
                self.get_parent_ids()[1:]):
494
 
                if file_id not in tree:
495
 
                    continue
496
 
                old.append(list(tree.annotate_iter(file_id)))
497
 
            return annotate.reannotate(old, self.get_file(file_id).readlines(),
498
 
                                       default_revision)
499
 
        finally:
500
 
            basis.unlock()
 
495
        maybe_file_parent_keys = []
 
496
        for parent_id in self.get_parent_ids():
 
497
            try:
 
498
                parent_tree = self.revision_tree(parent_id)
 
499
            except errors.NoSuchRevisionInTree:
 
500
                parent_tree = self.branch.repository.revision_tree(parent_id)
 
501
            parent_tree.lock_read()
 
502
            try:
 
503
                if file_id not in parent_tree:
 
504
                    continue
 
505
                ie = parent_tree.inventory[file_id]
 
506
                if ie.kind != 'file':
 
507
                    # Note: this is slightly unnecessary, because symlinks and
 
508
                    # directories have a "text" which is the empty text, and we
 
509
                    # know that won't mess up annotations. But it seems cleaner
 
510
                    continue
 
511
                parent_text_key = (file_id, ie.revision)
 
512
                if parent_text_key not in maybe_file_parent_keys:
 
513
                    maybe_file_parent_keys.append(parent_text_key)
 
514
            finally:
 
515
                parent_tree.unlock()
 
516
        graph = _mod_graph.Graph(self.branch.repository.texts)
 
517
        heads = graph.heads(maybe_file_parent_keys)
 
518
        file_parent_keys = []
 
519
        for key in maybe_file_parent_keys:
 
520
            if key in heads:
 
521
                file_parent_keys.append(key)
 
522
 
 
523
        # Now we have the parents of this content
 
524
        annotator = self.branch.repository.texts.get_annotator()
 
525
        text = self.get_file(file_id).read()
 
526
        this_key =(file_id, default_revision)
 
527
        annotator.add_special_text(this_key, file_parent_keys, text)
 
528
        annotations = [(key[-1], line)
 
529
                       for key, line in annotator.annotate_flat(this_key)]
 
530
        return annotations
501
531
 
502
532
    def _get_ancestors(self, default_revision):
503
533
        ancestors = set([default_revision])
508
538
 
509
539
    def get_parent_ids(self):
510
540
        """See Tree.get_parent_ids.
511
 
        
 
541
 
512
542
        This implementation reads the pending merges list and last_revision
513
543
        value and uses that to decide what the parents list should be.
514
544
        """
518
548
        else:
519
549
            parents = [last_rev]
520
550
        try:
521
 
            merges_file = self._control_files.get('pending-merges')
 
551
            merges_bytes = self._transport.get_bytes('pending-merges')
522
552
        except errors.NoSuchFile:
523
553
            pass
524
554
        else:
525
 
            for l in merges_file.readlines():
 
555
            for l in osutils.split_lines(merges_bytes):
526
556
                revision_id = l.rstrip('\n')
527
557
                parents.append(revision_id)
528
558
        return parents
531
561
    def get_root_id(self):
532
562
        """Return the id of this trees root"""
533
563
        return self._inventory.root.file_id
534
 
        
 
564
 
535
565
    def _get_store_filename(self, file_id):
536
566
        ## XXX: badly named; this is not in the store at all
537
567
        return self.abspath(self.id2path(file_id))
539
569
    @needs_read_lock
540
570
    def clone(self, to_bzrdir, revision_id=None):
541
571
        """Duplicate this working tree into to_bzr, including all state.
542
 
        
 
572
 
543
573
        Specifically modified files are kept as modified, but
544
574
        ignored and unknown files are discarded.
545
575
 
546
576
        If you want to make a new line of development, see bzrdir.sprout()
547
577
 
548
578
        revision
549
 
            If not None, the cloned tree will have its last revision set to 
550
 
            revision, and and difference between the source trees last revision
 
579
            If not None, the cloned tree will have its last revision set to
 
580
            revision, and difference between the source trees last revision
551
581
            and this one merged in.
552
582
        """
553
583
        # assumes the target bzr dir format is compatible.
554
 
        result = self._format.initialize(to_bzrdir)
 
584
        result = to_bzrdir.create_workingtree()
555
585
        self.copy_content_into(result, revision_id)
556
586
        return result
557
587
 
586
616
    __contains__ = has_id
587
617
 
588
618
    def get_file_size(self, file_id):
589
 
        return os.path.getsize(self.id2abspath(file_id))
 
619
        """See Tree.get_file_size"""
 
620
        # XXX: this returns the on-disk size; it should probably return the
 
621
        # canonical size
 
622
        try:
 
623
            return os.path.getsize(self.id2abspath(file_id))
 
624
        except OSError, e:
 
625
            if e.errno != errno.ENOENT:
 
626
                raise
 
627
            else:
 
628
                return None
590
629
 
591
630
    @needs_read_lock
592
631
    def get_file_sha1(self, file_id, path=None, stat_value=None):
601
640
 
602
641
    def _is_executable_from_path_and_stat_from_basis(self, path, stat_result):
603
642
        file_id = self.path2id(path)
 
643
        if file_id is None:
 
644
            # For unversioned files on win32, we just assume they are not
 
645
            # executable
 
646
            return False
604
647
        return self._inventory[file_id].executable
605
648
 
606
649
    def _is_executable_from_path_and_stat_from_stat(self, path, stat_result):
628
671
        """See MutableTree._add."""
629
672
        # TODO: Re-adding a file that is removed in the working copy
630
673
        # should probably put it back with the previous ID.
631
 
        # the read and write working inventory should not occur in this 
 
674
        # the read and write working inventory should not occur in this
632
675
        # function - they should be part of lock_write and unlock.
633
676
        inv = self.inventory
634
677
        for f, file_id, kind in zip(files, ids, kinds):
635
 
            assert kind is not None
636
678
            if file_id is None:
637
679
                inv.add_path(f, kind=kind)
638
680
            else:
703
745
        if updated:
704
746
            self.set_parent_ids(parents, allow_leftmost_as_ghost=True)
705
747
 
706
 
    def path_content_summary(self, path, _lstat=osutils.lstat,
 
748
    def path_content_summary(self, path, _lstat=os.lstat,
707
749
        _mapper=osutils.file_kind_from_stat_mode):
708
750
        """See Tree.path_content_summary."""
709
751
        abspath = self.abspath(path)
717
759
            raise
718
760
        kind = _mapper(stat_result.st_mode)
719
761
        if kind == 'file':
720
 
            size = stat_result.st_size
721
 
            # try for a stat cache lookup
722
 
            executable = self._is_executable_from_path_and_stat(path, stat_result)
723
 
            return (kind, size, executable, self._sha_from_stat(
724
 
                path, stat_result))
 
762
            return self._file_content_summary(path, stat_result)
725
763
        elif kind == 'directory':
726
764
            # perhaps it looks like a plain directory, but it's really a
727
765
            # reference.
729
767
                kind = 'tree-reference'
730
768
            return kind, None, None, None
731
769
        elif kind == 'symlink':
732
 
            return ('symlink', None, None, os.readlink(abspath))
 
770
            target = osutils.readlink(abspath)
 
771
            return ('symlink', None, None, target)
733
772
        else:
734
773
            return (kind, None, None, None)
735
774
 
736
 
    @deprecated_method(zero_eleven)
737
 
    @needs_read_lock
738
 
    def pending_merges(self):
739
 
        """Return a list of pending merges.
740
 
 
741
 
        These are revisions that have been merged into the working
742
 
        directory but not yet committed.
743
 
 
744
 
        As of 0.11 this is deprecated. Please see WorkingTree.get_parent_ids()
745
 
        instead - which is available on all tree objects.
746
 
        """
747
 
        return self.get_parent_ids()[1:]
 
775
    def _file_content_summary(self, path, stat_result):
 
776
        size = stat_result.st_size
 
777
        executable = self._is_executable_from_path_and_stat(path, stat_result)
 
778
        # try for a stat cache lookup
 
779
        return ('file', size, executable, self._sha_from_stat(
 
780
            path, stat_result))
748
781
 
749
782
    def _check_parents_for_ghosts(self, revision_ids, allow_leftmost_as_ghost):
750
783
        """Common ghost checking functionality from set_parent_*.
760
793
 
761
794
    def _set_merges_from_parent_ids(self, parent_ids):
762
795
        merges = parent_ids[1:]
763
 
        self._control_files.put_bytes('pending-merges', '\n'.join(merges))
 
796
        self._transport.put_bytes('pending-merges', '\n'.join(merges),
 
797
            mode=self.bzrdir._get_file_mode())
 
798
 
 
799
    def _filter_parent_ids_by_ancestry(self, revision_ids):
 
800
        """Check that all merged revisions are proper 'heads'.
 
801
 
 
802
        This will always return the first revision_id, and any merged revisions
 
803
        which are
 
804
        """
 
805
        if len(revision_ids) == 0:
 
806
            return revision_ids
 
807
        graph = self.branch.repository.get_graph()
 
808
        heads = graph.heads(revision_ids)
 
809
        new_revision_ids = revision_ids[:1]
 
810
        for revision_id in revision_ids[1:]:
 
811
            if revision_id in heads and revision_id not in new_revision_ids:
 
812
                new_revision_ids.append(revision_id)
 
813
        if new_revision_ids != revision_ids:
 
814
            trace.mutter('requested to set revision_ids = %s,'
 
815
                         ' but filtered to %s', revision_ids, new_revision_ids)
 
816
        return new_revision_ids
764
817
 
765
818
    @needs_tree_write_lock
766
819
    def set_parent_ids(self, revision_ids, allow_leftmost_as_ghost=False):
767
820
        """Set the parent ids to revision_ids.
768
 
        
 
821
 
769
822
        See also set_parent_trees. This api will try to retrieve the tree data
770
823
        for each element of revision_ids from the trees repository. If you have
771
824
        tree data already available, it is more efficient to use
780
833
        for revision_id in revision_ids:
781
834
            _mod_revision.check_not_reserved_id(revision_id)
782
835
 
 
836
        revision_ids = self._filter_parent_ids_by_ancestry(revision_ids)
 
837
 
783
838
        if len(revision_ids) > 0:
784
839
            self.set_last_revision(revision_ids[0])
785
840
        else:
797
852
        self._check_parents_for_ghosts(parent_ids,
798
853
            allow_leftmost_as_ghost=allow_leftmost_as_ghost)
799
854
 
 
855
        parent_ids = self._filter_parent_ids_by_ancestry(parent_ids)
 
856
 
800
857
        if len(parent_ids) == 0:
801
858
            leftmost_parent_id = _mod_revision.NULL_REVISION
802
859
            leftmost_parent_tree = None
842
899
    def _put_rio(self, filename, stanzas, header):
843
900
        self._must_be_locked()
844
901
        my_file = rio_file(stanzas, header)
845
 
        self._control_files.put(filename, my_file)
 
902
        self._transport.put_file(filename, my_file,
 
903
            mode=self.bzrdir._get_file_mode())
846
904
 
847
905
    @needs_write_lock # because merge pulls data into the branch.
848
906
    def merge_from_branch(self, branch, to_revision=None, from_revision=None,
849
 
        merge_type=None):
 
907
                          merge_type=None, force=False):
850
908
        """Merge from a branch into this working tree.
851
909
 
852
910
        :param branch: The branch to merge from.
856
914
            branch.last_revision().
857
915
        """
858
916
        from bzrlib.merge import Merger, Merge3Merger
859
 
        pb = bzrlib.ui.ui_factory.nested_progress_bar()
860
 
        try:
861
 
            merger = Merger(self.branch, this_tree=self, pb=pb)
862
 
            merger.pp = ProgressPhase("Merge phase", 5, pb)
863
 
            merger.pp.next_phase()
864
 
            # check that there are no
865
 
            # local alterations
866
 
            merger.check_basis(check_clean=True, require_commits=False)
867
 
            if to_revision is None:
868
 
                to_revision = _mod_revision.ensure_null(branch.last_revision())
869
 
            merger.other_rev_id = to_revision
870
 
            if _mod_revision.is_null(merger.other_rev_id):
871
 
                raise errors.NoCommits(branch)
872
 
            self.branch.fetch(branch, last_revision=merger.other_rev_id)
873
 
            merger.other_basis = merger.other_rev_id
874
 
            merger.other_tree = self.branch.repository.revision_tree(
875
 
                merger.other_rev_id)
876
 
            merger.other_branch = branch
877
 
            merger.pp.next_phase()
878
 
            if from_revision is None:
879
 
                merger.find_base()
880
 
            else:
881
 
                merger.set_base_revision(from_revision, branch)
882
 
            if merger.base_rev_id == merger.other_rev_id:
883
 
                raise errors.PointlessMerge
884
 
            merger.backup_files = False
885
 
            if merge_type is None:
886
 
                merger.merge_type = Merge3Merger
887
 
            else:
888
 
                merger.merge_type = merge_type
889
 
            merger.set_interesting_files(None)
890
 
            merger.show_base = False
891
 
            merger.reprocess = False
892
 
            conflicts = merger.do_merge()
893
 
            merger.set_pending()
894
 
        finally:
895
 
            pb.finished()
 
917
        merger = Merger(self.branch, this_tree=self)
 
918
        # check that there are no local alterations
 
919
        if not force and self.has_changes():
 
920
            raise errors.UncommittedChanges(self)
 
921
        if to_revision is None:
 
922
            to_revision = _mod_revision.ensure_null(branch.last_revision())
 
923
        merger.other_rev_id = to_revision
 
924
        if _mod_revision.is_null(merger.other_rev_id):
 
925
            raise errors.NoCommits(branch)
 
926
        self.branch.fetch(branch, last_revision=merger.other_rev_id)
 
927
        merger.other_basis = merger.other_rev_id
 
928
        merger.other_tree = self.branch.repository.revision_tree(
 
929
            merger.other_rev_id)
 
930
        merger.other_branch = branch
 
931
        if from_revision is None:
 
932
            merger.find_base()
 
933
        else:
 
934
            merger.set_base_revision(from_revision, branch)
 
935
        if merger.base_rev_id == merger.other_rev_id:
 
936
            raise errors.PointlessMerge
 
937
        merger.backup_files = False
 
938
        if merge_type is None:
 
939
            merger.merge_type = Merge3Merger
 
940
        else:
 
941
            merger.merge_type = merge_type
 
942
        merger.set_interesting_files(None)
 
943
        merger.show_base = False
 
944
        merger.reprocess = False
 
945
        conflicts = merger.do_merge()
 
946
        merger.set_pending()
896
947
        return conflicts
897
948
 
898
949
    @needs_read_lock
899
950
    def merge_modified(self):
900
951
        """Return a dictionary of files modified by a merge.
901
952
 
902
 
        The list is initialized by WorkingTree.set_merge_modified, which is 
 
953
        The list is initialized by WorkingTree.set_merge_modified, which is
903
954
        typically called after we make some automatic updates to the tree
904
955
        because of a merge.
905
956
 
907
958
        still in the working inventory and have that text hash.
908
959
        """
909
960
        try:
910
 
            hashfile = self._control_files.get('merge-hashes')
 
961
            hashfile = self._transport.get('merge-hashes')
911
962
        except errors.NoSuchFile:
912
963
            return {}
913
 
        merge_hashes = {}
914
964
        try:
915
 
            if hashfile.next() != MERGE_MODIFIED_HEADER_1 + '\n':
 
965
            merge_hashes = {}
 
966
            try:
 
967
                if hashfile.next() != MERGE_MODIFIED_HEADER_1 + '\n':
 
968
                    raise errors.MergeModifiedFormatError()
 
969
            except StopIteration:
916
970
                raise errors.MergeModifiedFormatError()
917
 
        except StopIteration:
918
 
            raise errors.MergeModifiedFormatError()
919
 
        for s in RioReader(hashfile):
920
 
            # RioReader reads in Unicode, so convert file_ids back to utf8
921
 
            file_id = osutils.safe_file_id(s.get("file_id"), warn=False)
922
 
            if file_id not in self.inventory:
923
 
                continue
924
 
            text_hash = s.get("hash")
925
 
            if text_hash == self.get_file_sha1(file_id):
926
 
                merge_hashes[file_id] = text_hash
927
 
        return merge_hashes
 
971
            for s in RioReader(hashfile):
 
972
                # RioReader reads in Unicode, so convert file_ids back to utf8
 
973
                file_id = osutils.safe_file_id(s.get("file_id"), warn=False)
 
974
                if file_id not in self.inventory:
 
975
                    continue
 
976
                text_hash = s.get("hash")
 
977
                if text_hash == self.get_file_sha1(file_id):
 
978
                    merge_hashes[file_id] = text_hash
 
979
            return merge_hashes
 
980
        finally:
 
981
            hashfile.close()
928
982
 
929
983
    @needs_write_lock
930
984
    def mkdir(self, path, file_id=None):
936
990
        return file_id
937
991
 
938
992
    def get_symlink_target(self, file_id):
939
 
        return os.readlink(self.id2abspath(file_id))
 
993
        abspath = self.id2abspath(file_id)
 
994
        target = osutils.readlink(abspath)
 
995
        return target
940
996
 
941
997
    @needs_write_lock
942
998
    def subsume(self, other_tree):
980
1036
            other_tree.unlock()
981
1037
        other_tree.bzrdir.retire_bzrdir()
982
1038
 
983
 
    def _directory_is_tree_reference(self, relpath):
984
 
        # as a special case, if a directory contains control files then 
 
1039
    def _setup_directory_is_tree_reference(self):
 
1040
        if self._branch.repository._format.supports_tree_reference:
 
1041
            self._directory_is_tree_reference = \
 
1042
                self._directory_may_be_tree_reference
 
1043
        else:
 
1044
            self._directory_is_tree_reference = \
 
1045
                self._directory_is_never_tree_reference
 
1046
 
 
1047
    def _directory_is_never_tree_reference(self, relpath):
 
1048
        return False
 
1049
 
 
1050
    def _directory_may_be_tree_reference(self, relpath):
 
1051
        # as a special case, if a directory contains control files then
985
1052
        # it's a tree reference, except that the root of the tree is not
986
1053
        return relpath and osutils.isdir(self.abspath(relpath) + u"/.bzr")
987
1054
        # TODO: We could ask all the control formats whether they
998
1065
    @needs_tree_write_lock
999
1066
    def extract(self, file_id, format=None):
1000
1067
        """Extract a subtree from this tree.
1001
 
        
 
1068
 
1002
1069
        A new branch will be created, relative to the path for this tree.
1003
1070
        """
1004
1071
        self.flush()
1009
1076
                transport = transport.clone(name)
1010
1077
                transport.ensure_base()
1011
1078
            return transport
1012
 
            
 
1079
 
1013
1080
        sub_path = self.id2path(file_id)
1014
1081
        branch_transport = mkdirs(sub_path)
1015
1082
        if format is None:
1016
 
            format = bzrdir.format_registry.make_bzrdir('dirstate-with-subtree')
 
1083
            format = self.bzrdir.cloning_metadir()
1017
1084
        branch_transport.ensure_base()
1018
1085
        branch_bzrdir = format.initialize_on_transport(branch_transport)
1019
1086
        try:
1020
1087
            repo = branch_bzrdir.find_repository()
1021
1088
        except errors.NoRepositoryPresent:
1022
1089
            repo = branch_bzrdir.create_repository()
1023
 
            assert repo.supports_rich_root()
1024
 
        else:
1025
 
            if not repo.supports_rich_root():
1026
 
                raise errors.RootNotRich()
 
1090
        if not repo.supports_rich_root():
 
1091
            raise errors.RootNotRich()
1027
1092
        new_branch = branch_bzrdir.create_branch()
1028
1093
        new_branch.pull(self.branch)
1029
1094
        for parent_id in self.get_parent_ids():
1031
1096
        tree_transport = self.bzrdir.root_transport.clone(sub_path)
1032
1097
        if tree_transport.base != branch_transport.base:
1033
1098
            tree_bzrdir = format.initialize_on_transport(tree_transport)
1034
 
            branch.BranchReferenceFormat().initialize(tree_bzrdir, new_branch)
 
1099
            branch.BranchReferenceFormat().initialize(tree_bzrdir,
 
1100
                target_branch=new_branch)
1035
1101
        else:
1036
1102
            tree_bzrdir = branch_bzrdir
1037
 
        wt = tree_bzrdir.create_workingtree(NULL_REVISION)
 
1103
        wt = tree_bzrdir.create_workingtree(_mod_revision.NULL_REVISION)
1038
1104
        wt.set_parent_ids(self.get_parent_ids())
1039
1105
        my_inv = self.inventory
1040
 
        child_inv = Inventory(root_id=None)
 
1106
        child_inv = inventory.Inventory(root_id=None)
1041
1107
        new_root = my_inv[file_id]
1042
1108
        my_inv.remove_recursive_id(file_id)
1043
1109
        new_root.parent_id = None
1061
1127
        sio = StringIO()
1062
1128
        self._serialize(self._inventory, sio)
1063
1129
        sio.seek(0)
1064
 
        self._control_files.put('inventory', sio)
 
1130
        self._transport.put_file('inventory', sio,
 
1131
            mode=self.bzrdir._get_file_mode())
1065
1132
        self._inventory_is_modified = False
1066
1133
 
1067
1134
    def _kind(self, relpath):
1068
1135
        return osutils.file_kind(self.abspath(relpath))
1069
1136
 
1070
 
    def list_files(self, include_root=False):
1071
 
        """Recursively list all files as (path, class, kind, id, entry).
 
1137
    def list_files(self, include_root=False, from_dir=None, recursive=True):
 
1138
        """List all files as (path, class, kind, id, entry).
1072
1139
 
1073
1140
        Lists, but does not descend into unversioned directories.
1074
 
 
1075
1141
        This does not include files that have been deleted in this
1076
 
        tree.
 
1142
        tree. Skips the control directory.
1077
1143
 
1078
 
        Skips the control directory.
 
1144
        :param include_root: if True, return an entry for the root
 
1145
        :param from_dir: start from this directory or None for the root
 
1146
        :param recursive: whether to recurse into subdirectories or not
1079
1147
        """
1080
1148
        # list_files is an iterator, so @needs_read_lock doesn't work properly
1081
1149
        # with it. So callers should be careful to always read_lock the tree.
1083
1151
            raise errors.ObjectNotLocked(self)
1084
1152
 
1085
1153
        inv = self.inventory
1086
 
        if include_root is True:
 
1154
        if from_dir is None and include_root is True:
1087
1155
            yield ('', 'V', 'directory', inv.root.file_id, inv.root)
1088
1156
        # Convert these into local objects to save lookup times
1089
1157
        pathjoin = osutils.pathjoin
1096
1164
        fk_entries = {'directory':TreeDirectory, 'file':TreeFile, 'symlink':TreeLink}
1097
1165
 
1098
1166
        # directory file_id, relative path, absolute path, reverse sorted children
1099
 
        children = os.listdir(self.basedir)
 
1167
        if from_dir is not None:
 
1168
            from_dir_id = inv.path2id(from_dir)
 
1169
            if from_dir_id is None:
 
1170
                # Directory not versioned
 
1171
                return
 
1172
            from_dir_abspath = pathjoin(self.basedir, from_dir)
 
1173
        else:
 
1174
            from_dir_id = inv.root.file_id
 
1175
            from_dir_abspath = self.basedir
 
1176
        children = os.listdir(from_dir_abspath)
1100
1177
        children.sort()
1101
 
        # jam 20060527 The kernel sized tree seems equivalent whether we 
 
1178
        # jam 20060527 The kernel sized tree seems equivalent whether we
1102
1179
        # use a deque and popleft to keep them sorted, or if we use a plain
1103
1180
        # list and just reverse() them.
1104
1181
        children = collections.deque(children)
1105
 
        stack = [(inv.root.file_id, u'', self.basedir, children)]
 
1182
        stack = [(from_dir_id, u'', from_dir_abspath, children)]
1106
1183
        while stack:
1107
1184
            from_dir_id, from_dir_relpath, from_dir_abspath, children = stack[-1]
1108
1185
 
1124
1201
 
1125
1202
                # absolute path
1126
1203
                fap = from_dir_abspath + '/' + f
1127
 
                
 
1204
 
1128
1205
                f_ie = inv.get_child(from_dir_id, f)
1129
1206
                if f_ie:
1130
1207
                    c = 'V'
1162
1239
                    except KeyError:
1163
1240
                        yield fp[1:], c, fk, None, TreeEntry()
1164
1241
                    continue
1165
 
                
 
1242
 
1166
1243
                if fk != 'directory':
1167
1244
                    continue
1168
1245
 
1169
 
                # But do this child first
1170
 
                new_children = os.listdir(fap)
1171
 
                new_children.sort()
1172
 
                new_children = collections.deque(new_children)
1173
 
                stack.append((f_ie.file_id, fp, fap, new_children))
1174
 
                # Break out of inner loop,
1175
 
                # so that we start outer loop with child
1176
 
                break
 
1246
                # But do this child first if recursing down
 
1247
                if recursive:
 
1248
                    new_children = os.listdir(fap)
 
1249
                    new_children.sort()
 
1250
                    new_children = collections.deque(new_children)
 
1251
                    stack.append((f_ie.file_id, fp, fap, new_children))
 
1252
                    # Break out of inner loop,
 
1253
                    # so that we start outer loop with child
 
1254
                    break
1177
1255
            else:
1178
1256
                # if we finished all children, pop it off the stack
1179
1257
                stack.pop()
1185
1263
        to_dir must exist in the inventory.
1186
1264
 
1187
1265
        If to_dir exists and is a directory, the files are moved into
1188
 
        it, keeping their old names.  
 
1266
        it, keeping their old names.
1189
1267
 
1190
1268
        Note that to_dir is only the last component of the new name;
1191
1269
        this doesn't change the directory.
1228
1306
                                       DeprecationWarning)
1229
1307
 
1230
1308
        # check destination directory
1231
 
        assert not isinstance(from_paths, basestring)
 
1309
        if isinstance(from_paths, basestring):
 
1310
            raise ValueError()
1232
1311
        inv = self.inventory
1233
1312
        to_abs = self.abspath(to_dir)
1234
1313
        if not isdir(to_abs):
1318
1397
                only_change_inv = True
1319
1398
            elif self.has_filename(from_rel) and not self.has_filename(to_rel):
1320
1399
                only_change_inv = False
 
1400
            elif (not self.case_sensitive
 
1401
                  and from_rel.lower() == to_rel.lower()
 
1402
                  and self.has_filename(from_rel)):
 
1403
                only_change_inv = False
1321
1404
            else:
1322
1405
                # something is wrong, so lets determine what exactly
1323
1406
                if not self.has_filename(from_rel) and \
1326
1409
                        errors.PathsDoNotExist(paths=(str(from_rel),
1327
1410
                        str(to_rel))))
1328
1411
                else:
1329
 
                    raise errors.RenameFailedFilesExist(from_rel, to_rel,
1330
 
                        extra="(Use --after to update the Bazaar id)")
 
1412
                    raise errors.RenameFailedFilesExist(from_rel, to_rel)
1331
1413
            rename_entry.only_change_inv = only_change_inv
1332
1414
        return rename_entries
1333
1415
 
1353
1435
        inv = self.inventory
1354
1436
        for entry in moved:
1355
1437
            try:
1356
 
                self._move_entry(_RenameEntry(entry.to_rel, entry.from_id,
 
1438
                self._move_entry(WorkingTree._RenameEntry(
 
1439
                    entry.to_rel, entry.from_id,
1357
1440
                    entry.to_tail, entry.to_parent_id, entry.from_rel,
1358
1441
                    entry.from_tail, entry.from_parent_id,
1359
1442
                    entry.only_change_inv))
1410
1493
        from_tail = splitpath(from_rel)[-1]
1411
1494
        from_id = inv.path2id(from_rel)
1412
1495
        if from_id is None:
1413
 
            raise errors.BzrRenameFailedError(from_rel,to_rel,
1414
 
                errors.NotVersionedError(path=str(from_rel)))
1415
 
        from_entry = inv[from_id]
 
1496
            # if file is missing in the inventory maybe it's in the basis_tree
 
1497
            basis_tree = self.branch.basis_tree()
 
1498
            from_id = basis_tree.path2id(from_rel)
 
1499
            if from_id is None:
 
1500
                raise errors.BzrRenameFailedError(from_rel,to_rel,
 
1501
                    errors.NotVersionedError(path=str(from_rel)))
 
1502
            # put entry back in the inventory so we can rename it
 
1503
            from_entry = basis_tree.inventory[from_id].copy()
 
1504
            inv.add(from_entry)
 
1505
        else:
 
1506
            from_entry = inv[from_id]
1416
1507
        from_parent_id = from_entry.parent_id
1417
1508
        to_dir, to_tail = os.path.split(to_rel)
1418
1509
        to_dir_id = inv.path2id(to_dir)
1464
1555
        These are files in the working directory that are not versioned or
1465
1556
        control files or ignored.
1466
1557
        """
1467
 
        # force the extras method to be fully executed before returning, to 
 
1558
        # force the extras method to be fully executed before returning, to
1468
1559
        # prevent race conditions with the lock
1469
1560
        return iter(
1470
1561
            [subp for subp in self.extras() if not self.is_ignored(subp)])
1480
1571
        :raises: NoSuchId if any fileid is not currently versioned.
1481
1572
        """
1482
1573
        for file_id in file_ids:
 
1574
            if file_id not in self._inventory:
 
1575
                raise errors.NoSuchId(self, file_id)
 
1576
        for file_id in file_ids:
1483
1577
            if self._inventory.has_id(file_id):
1484
1578
                self._inventory.remove_recursive_id(file_id)
1485
 
            else:
1486
 
                raise errors.NoSuchId(self, file_id)
1487
1579
        if len(file_ids):
1488
 
            # in the future this should just set a dirty bit to wait for the 
 
1580
            # in the future this should just set a dirty bit to wait for the
1489
1581
            # final unlock. However, until all methods of workingtree start
1490
 
            # with the current in -memory inventory rather than triggering 
 
1582
            # with the current in -memory inventory rather than triggering
1491
1583
            # a read, it is more complex - we need to teach read_inventory
1492
1584
            # to know when to read, and when to not read first... and possibly
1493
1585
            # to save first when the in memory one may be corrupted.
1494
1586
            # so for now, we just only write it if it is indeed dirty.
1495
1587
            # - RBC 20060907
1496
1588
            self._write_inventory(self._inventory)
1497
 
    
1498
 
    @deprecated_method(zero_eight)
1499
 
    def iter_conflicts(self):
1500
 
        """List all files in the tree that have text or content conflicts.
1501
 
        DEPRECATED.  Use conflicts instead."""
1502
 
        return self._iter_conflicts()
1503
1589
 
1504
1590
    def _iter_conflicts(self):
1505
1591
        conflicted = set()
1514
1600
 
1515
1601
    @needs_write_lock
1516
1602
    def pull(self, source, overwrite=False, stop_revision=None,
1517
 
             change_reporter=None, possible_transports=None):
1518
 
        top_pb = bzrlib.ui.ui_factory.nested_progress_bar()
 
1603
             change_reporter=None, possible_transports=None, local=False):
1519
1604
        source.lock_read()
1520
1605
        try:
1521
 
            pp = ProgressPhase("Pull phase", 2, top_pb)
1522
 
            pp.next_phase()
1523
1606
            old_revision_info = self.branch.last_revision_info()
1524
1607
            basis_tree = self.basis_tree()
1525
1608
            count = self.branch.pull(source, overwrite, stop_revision,
1526
 
                                     possible_transports=possible_transports)
 
1609
                                     possible_transports=possible_transports,
 
1610
                                     local=local)
1527
1611
            new_revision_info = self.branch.last_revision_info()
1528
1612
            if new_revision_info != old_revision_info:
1529
 
                pp.next_phase()
1530
1613
                repository = self.branch.repository
1531
 
                pb = bzrlib.ui.ui_factory.nested_progress_bar()
1532
1614
                basis_tree.lock_read()
1533
1615
                try:
1534
1616
                    new_basis_tree = self.branch.basis_tree()
1537
1619
                                new_basis_tree,
1538
1620
                                basis_tree,
1539
1621
                                this_tree=self,
1540
 
                                pb=pb,
 
1622
                                pb=None,
1541
1623
                                change_reporter=change_reporter)
1542
 
                    if (basis_tree.inventory.root is None and
1543
 
                        new_basis_tree.inventory.root is not None):
1544
 
                        self.set_root_id(new_basis_tree.inventory.root.file_id)
 
1624
                    basis_root_id = basis_tree.get_root_id()
 
1625
                    new_root_id = new_basis_tree.get_root_id()
 
1626
                    if basis_root_id != new_root_id:
 
1627
                        self.set_root_id(new_root_id)
1545
1628
                finally:
1546
 
                    pb.finished()
1547
1629
                    basis_tree.unlock()
1548
1630
                # TODO - dedup parents list with things merged by pull ?
1549
1631
                # reuse the revisiontree we merged against to set the new
1550
1632
                # tree data.
1551
1633
                parent_trees = [(self.branch.last_revision(), new_basis_tree)]
1552
 
                # we have to pull the merge trees out again, because 
1553
 
                # merge_inner has set the ids. - this corner is not yet 
 
1634
                # we have to pull the merge trees out again, because
 
1635
                # merge_inner has set the ids. - this corner is not yet
1554
1636
                # layered well enough to prevent double handling.
1555
1637
                # XXX TODO: Fix the double handling: telling the tree about
1556
1638
                # the already known parent data is wasteful.
1562
1644
            return count
1563
1645
        finally:
1564
1646
            source.unlock()
1565
 
            top_pb.finished()
1566
1647
 
1567
1648
    @needs_write_lock
1568
1649
    def put_file_bytes_non_atomic(self, file_id, bytes):
1594
1675
 
1595
1676
            fl = []
1596
1677
            for subf in os.listdir(dirabs):
1597
 
                if subf == '.bzr':
 
1678
                if self.bzrdir.is_control_filename(subf):
1598
1679
                    continue
1599
1680
                if subf not in dir_entry.children:
1600
 
                    subf_norm, can_access = osutils.normalized_filename(subf)
 
1681
                    try:
 
1682
                        (subf_norm,
 
1683
                         can_access) = osutils.normalized_filename(subf)
 
1684
                    except UnicodeDecodeError:
 
1685
                        path_os_enc = path.encode(osutils._fs_enc)
 
1686
                        relpath = path_os_enc + '/' + subf
 
1687
                        raise errors.BadFilenameEncoding(relpath,
 
1688
                                                         osutils._fs_enc)
1601
1689
                    if subf_norm != subf and can_access:
1602
1690
                        if subf_norm not in dir_entry.children:
1603
1691
                            fl.append(subf_norm)
1604
1692
                    else:
1605
1693
                        fl.append(subf)
1606
 
            
 
1694
 
1607
1695
            fl.sort()
1608
1696
            for subf in fl:
1609
1697
                subp = pathjoin(path, subf)
1646
1734
        r"""Check whether the filename matches an ignore pattern.
1647
1735
 
1648
1736
        Patterns containing '/' or '\' need to match the whole path;
1649
 
        others match against only the last component.
 
1737
        others match against only the last component.  Patterns starting
 
1738
        with '!' are ignore exceptions.  Exceptions take precedence
 
1739
        over regular patterns and cause the filename to not be ignored.
1650
1740
 
1651
1741
        If the file is ignored, returns the pattern which caused it to
1652
1742
        be ignored, otherwise None.  So this can simply be used as a
1653
1743
        boolean if desired."""
1654
1744
        if getattr(self, '_ignoreglobster', None) is None:
1655
 
            self._ignoreglobster = globbing.Globster(self.get_ignore_list())
 
1745
            self._ignoreglobster = globbing.ExceptionGlobster(self.get_ignore_list())
1656
1746
        return self._ignoreglobster.match(filename)
1657
1747
 
1658
1748
    def kind(self, file_id):
1659
1749
        return file_kind(self.id2abspath(file_id))
1660
1750
 
 
1751
    def stored_kind(self, file_id):
 
1752
        """See Tree.stored_kind"""
 
1753
        return self.inventory[file_id].kind
 
1754
 
1661
1755
    def _comparison_data(self, entry, path):
1662
1756
        abspath = self.abspath(path)
1663
1757
        try:
1704
1798
            raise errors.ObjectNotLocked(self)
1705
1799
 
1706
1800
    def lock_read(self):
1707
 
        """See Branch.lock_read, and WorkingTree.unlock."""
 
1801
        """Lock the tree for reading.
 
1802
 
 
1803
        This also locks the branch, and can be unlocked via self.unlock().
 
1804
 
 
1805
        :return: An object with an unlock method which will release the lock
 
1806
            obtained.
 
1807
        """
1708
1808
        if not self.is_locked():
1709
1809
            self._reset_data()
1710
1810
        self.branch.lock_read()
1711
1811
        try:
1712
 
            return self._control_files.lock_read()
 
1812
            self._control_files.lock_read()
 
1813
            return self
1713
1814
        except:
1714
1815
            self.branch.unlock()
1715
1816
            raise
1716
1817
 
1717
1818
    def lock_tree_write(self):
1718
 
        """See MutableTree.lock_tree_write, and WorkingTree.unlock."""
 
1819
        """See MutableTree.lock_tree_write, and WorkingTree.unlock.
 
1820
 
 
1821
        :return: An object with an unlock method which will release the lock
 
1822
            obtained.
 
1823
        """
1719
1824
        if not self.is_locked():
1720
1825
            self._reset_data()
1721
1826
        self.branch.lock_read()
1722
1827
        try:
1723
 
            return self._control_files.lock_write()
 
1828
            self._control_files.lock_write()
 
1829
            return self
1724
1830
        except:
1725
1831
            self.branch.unlock()
1726
1832
            raise
1727
1833
 
1728
1834
    def lock_write(self):
1729
 
        """See MutableTree.lock_write, and WorkingTree.unlock."""
 
1835
        """See MutableTree.lock_write, and WorkingTree.unlock.
 
1836
 
 
1837
        :return: An object with an unlock method which will release the lock
 
1838
            obtained.
 
1839
        """
1730
1840
        if not self.is_locked():
1731
1841
            self._reset_data()
1732
1842
        self.branch.lock_write()
1733
1843
        try:
1734
 
            return self._control_files.lock_write()
 
1844
            self._control_files.lock_write()
 
1845
            return self
1735
1846
        except:
1736
1847
            self.branch.unlock()
1737
1848
            raise
1745
1856
    def _reset_data(self):
1746
1857
        """Reset transient data that cannot be revalidated."""
1747
1858
        self._inventory_is_modified = False
1748
 
        result = self._deserialize(self._control_files.get('inventory'))
 
1859
        f = self._transport.get('inventory')
 
1860
        try:
 
1861
            result = self._deserialize(f)
 
1862
        finally:
 
1863
            f.close()
1749
1864
        self._set_inventory(result, dirty=False)
1750
1865
 
1751
1866
    @needs_tree_write_lock
1756
1871
 
1757
1872
    def _change_last_revision(self, new_revision):
1758
1873
        """Template method part of set_last_revision to perform the change.
1759
 
        
 
1874
 
1760
1875
        This is used to allow WorkingTree3 instances to not affect branch
1761
1876
        when their last revision is set.
1762
1877
        """
1772
1887
 
1773
1888
    def _write_basis_inventory(self, xml):
1774
1889
        """Write the basis inventory XML to the basis-inventory file"""
1775
 
        assert isinstance(xml, str), 'serialised xml must be bytestring.'
1776
1890
        path = self._basis_inventory_name()
1777
1891
        sio = StringIO(xml)
1778
 
        self._control_files.put(path, sio)
 
1892
        self._transport.put_file(path, sio,
 
1893
            mode=self.bzrdir._get_file_mode())
1779
1894
 
1780
1895
    def _create_basis_xml_from_inventory(self, revision_id, inventory):
1781
1896
        """Create the text that will be saved in basis-inventory"""
1788
1903
        # as commit already has that ready-to-use [while the format is the
1789
1904
        # same, that is].
1790
1905
        try:
1791
 
            # this double handles the inventory - unpack and repack - 
 
1906
            # this double handles the inventory - unpack and repack -
1792
1907
            # but is easier to understand. We can/should put a conditional
1793
1908
            # in here based on whether the inventory is in the latest format
1794
1909
            # - perhaps we should repack all inventories on a repository
1795
1910
            # upgrade ?
1796
1911
            # the fast path is to copy the raw xml from the repository. If the
1797
 
            # xml contains 'revision_id="', then we assume the right 
 
1912
            # xml contains 'revision_id="', then we assume the right
1798
1913
            # revision_id is set. We must check for this full string, because a
1799
1914
            # root node id can legitimately look like 'revision_id' but cannot
1800
1915
            # contain a '"'.
1801
 
            xml = self.branch.repository.get_inventory_xml(new_revision)
 
1916
            xml = self.branch.repository._get_inventory_xml(new_revision)
1802
1917
            firstline = xml.split('\n', 1)[0]
1803
 
            if (not 'revision_id="' in firstline or 
 
1918
            if (not 'revision_id="' in firstline or
1804
1919
                'format="7"' not in firstline):
1805
 
                inv = self.branch.repository.deserialise_inventory(
1806
 
                    new_revision, xml)
 
1920
                inv = self.branch.repository._serializer.read_inventory_from_string(
 
1921
                    xml, new_revision)
1807
1922
                xml = self._create_basis_xml_from_inventory(new_revision, inv)
1808
1923
            self._write_basis_inventory(xml)
1809
1924
        except (errors.NoSuchRevision, errors.RevisionNotPresent):
1812
1927
    def read_basis_inventory(self):
1813
1928
        """Read the cached basis inventory."""
1814
1929
        path = self._basis_inventory_name()
1815
 
        return self._control_files.get(path).read()
1816
 
        
 
1930
        return self._transport.get_bytes(path)
 
1931
 
1817
1932
    @needs_read_lock
1818
1933
    def read_working_inventory(self):
1819
1934
        """Read the working inventory.
1820
 
        
 
1935
 
1821
1936
        :raises errors.InventoryModified: read_working_inventory will fail
1822
1937
            when the current in memory inventory has been modified.
1823
1938
        """
1824
 
        # conceptually this should be an implementation detail of the tree. 
 
1939
        # conceptually this should be an implementation detail of the tree.
1825
1940
        # XXX: Deprecate this.
1826
1941
        # ElementTree does its own conversion from UTF-8, so open in
1827
1942
        # binary.
1828
1943
        if self._inventory_is_modified:
1829
1944
            raise errors.InventoryModified(self)
1830
 
        result = self._deserialize(self._control_files.get('inventory'))
 
1945
        f = self._transport.get('inventory')
 
1946
        try:
 
1947
            result = self._deserialize(f)
 
1948
        finally:
 
1949
            f.close()
1831
1950
        self._set_inventory(result, dirty=False)
1832
1951
        return result
1833
1952
 
1848
1967
 
1849
1968
        new_files=set()
1850
1969
        unknown_nested_files=set()
 
1970
        if to_file is None:
 
1971
            to_file = sys.stdout
1851
1972
 
1852
1973
        def recurse_directory_to_add_files(directory):
1853
1974
            # Recurse directory and add all files
1854
1975
            # so we can check if they have changed.
1855
1976
            for parent_info, file_infos in\
1856
 
                osutils.walkdirs(self.abspath(directory),
1857
 
                    directory):
1858
 
                for relpath, basename, kind, lstat, abspath in file_infos:
 
1977
                self.walkdirs(directory):
 
1978
                for relpath, basename, kind, lstat, fileid, kind in file_infos:
1859
1979
                    # Is it versioned or ignored?
1860
1980
                    if self.path2id(relpath) or self.is_ignored(relpath):
1861
1981
                        # Add nested content for deletion.
1871
1991
            filename = self.relpath(abspath)
1872
1992
            if len(filename) > 0:
1873
1993
                new_files.add(filename)
1874
 
                if osutils.isdir(abspath):
1875
 
                    recurse_directory_to_add_files(filename)
 
1994
                recurse_directory_to_add_files(filename)
1876
1995
 
1877
1996
        files = list(new_files)
1878
1997
 
1887
2006
            has_changed_files = len(unknown_nested_files) > 0
1888
2007
            if not has_changed_files:
1889
2008
                for (file_id, path, content_change, versioned, parent_id, name,
1890
 
                     kind, executable) in self._iter_changes(self.basis_tree(),
 
2009
                     kind, executable) in self.iter_changes(self.basis_tree(),
1891
2010
                         include_unchanged=True, require_versioned=False,
1892
2011
                         want_unversioned=True, specific_files=files):
1893
 
                    # Check if it's an unknown (but not ignored) OR
1894
 
                    # changed (but not deleted) :
1895
 
                    if not self.is_ignored(path[1]) and (
1896
 
                        versioned == (False, False) or
1897
 
                        content_change and kind[1] != None):
 
2012
                    if versioned == (False, False):
 
2013
                        # The record is unknown ...
 
2014
                        if not self.is_ignored(path[1]):
 
2015
                            # ... but not ignored
 
2016
                            has_changed_files = True
 
2017
                            break
 
2018
                    elif content_change and (kind[1] is not None):
 
2019
                        # Versioned and changed, but not deleted
1898
2020
                        has_changed_files = True
1899
2021
                        break
1900
2022
 
1908
2030
                        tree_delta.unversioned.extend((unknown_file,))
1909
2031
                raise errors.BzrRemoveChangedFilesError(tree_delta)
1910
2032
 
1911
 
        # Build inv_delta and delete files where applicaple,
 
2033
        # Build inv_delta and delete files where applicable,
1912
2034
        # do this before any modifications to inventory.
1913
2035
        for f in files:
1914
2036
            fid = self.path2id(f)
1922
2044
                        new_status = 'I'
1923
2045
                    else:
1924
2046
                        new_status = '?'
1925
 
                    textui.show_status(new_status, self.kind(fid), f,
1926
 
                                       to_file=to_file)
 
2047
                    # XXX: Really should be a more abstract reporter interface
 
2048
                    kind_ch = osutils.kind_marker(self.kind(fid))
 
2049
                    to_file.write(new_status + '       ' + f + kind_ch + '\n')
1927
2050
                # Unversion file
1928
2051
                inv_delta.append((f, None, fid, None))
1929
2052
                message = "removed %s" % (f,)
1952
2075
 
1953
2076
    @needs_tree_write_lock
1954
2077
    def revert(self, filenames=None, old_tree=None, backups=True,
1955
 
               pb=DummyProgress(), report_changes=False):
 
2078
               pb=None, report_changes=False):
1956
2079
        from bzrlib.conflicts import resolve
1957
2080
        if filenames == []:
1958
2081
            filenames = None
1960
2083
                ' as of bzr 0.91.  Please use None (the default) instead.',
1961
2084
                DeprecationWarning, stacklevel=2)
1962
2085
        if old_tree is None:
1963
 
            old_tree = self.basis_tree()
1964
 
        conflicts = transform.revert(self, old_tree, filenames, backups, pb,
1965
 
                                     report_changes)
1966
 
        if filenames is None:
1967
 
            self.set_parent_ids(self.get_parent_ids()[:1])
1968
 
            resolve(self)
 
2086
            basis_tree = self.basis_tree()
 
2087
            basis_tree.lock_read()
 
2088
            old_tree = basis_tree
1969
2089
        else:
1970
 
            resolve(self, filenames, ignore_misses=True)
 
2090
            basis_tree = None
 
2091
        try:
 
2092
            conflicts = transform.revert(self, old_tree, filenames, backups, pb,
 
2093
                                         report_changes)
 
2094
            if filenames is None and len(self.get_parent_ids()) > 1:
 
2095
                parent_trees = []
 
2096
                last_revision = self.last_revision()
 
2097
                if last_revision != _mod_revision.NULL_REVISION:
 
2098
                    if basis_tree is None:
 
2099
                        basis_tree = self.basis_tree()
 
2100
                        basis_tree.lock_read()
 
2101
                    parent_trees.append((last_revision, basis_tree))
 
2102
                self.set_parent_trees(parent_trees)
 
2103
                resolve(self)
 
2104
            else:
 
2105
                resolve(self, filenames, ignore_misses=True, recursive=True)
 
2106
        finally:
 
2107
            if basis_tree is not None:
 
2108
                basis_tree.unlock()
1971
2109
        return conflicts
1972
2110
 
1973
2111
    def revision_tree(self, revision_id):
2000
2138
    def set_inventory(self, new_inventory_list):
2001
2139
        from bzrlib.inventory import (Inventory,
2002
2140
                                      InventoryDirectory,
2003
 
                                      InventoryEntry,
2004
2141
                                      InventoryFile,
2005
2142
                                      InventoryLink)
2006
2143
        inv = Inventory(self.get_root_id())
2008
2145
            name = os.path.basename(path)
2009
2146
            if name == "":
2010
2147
                continue
2011
 
            # fixme, there should be a factory function inv,add_?? 
 
2148
            # fixme, there should be a factory function inv,add_??
2012
2149
            if kind == 'directory':
2013
2150
                inv.add(InventoryDirectory(file_id, name, parent))
2014
2151
            elif kind == 'file':
2022
2159
    @needs_tree_write_lock
2023
2160
    def set_root_id(self, file_id):
2024
2161
        """Set the root id for this tree."""
2025
 
        # for compatability 
 
2162
        # for compatability
2026
2163
        if file_id is None:
2027
 
            symbol_versioning.warn(symbol_versioning.zero_twelve
2028
 
                % 'WorkingTree.set_root_id with fileid=None',
2029
 
                DeprecationWarning,
2030
 
                stacklevel=3)
2031
 
            file_id = ROOT_ID
2032
 
        else:
2033
 
            file_id = osutils.safe_file_id(file_id)
 
2164
            raise ValueError(
 
2165
                'WorkingTree.set_root_id with fileid=None')
 
2166
        file_id = osutils.safe_file_id(file_id)
2034
2167
        self._set_root_id(file_id)
2035
2168
 
2036
2169
    def _set_root_id(self, file_id):
2037
2170
        """Set the root id for this tree, in a format specific manner.
2038
2171
 
2039
 
        :param file_id: The file id to assign to the root. It must not be 
 
2172
        :param file_id: The file id to assign to the root. It must not be
2040
2173
            present in the current inventory or an error will occur. It must
2041
2174
            not be None, but rather a valid file id.
2042
2175
        """
2061
2194
 
2062
2195
    def unlock(self):
2063
2196
        """See Branch.unlock.
2064
 
        
 
2197
 
2065
2198
        WorkingTree locking just uses the Branch locking facilities.
2066
2199
        This is current because all working trees have an embedded branch
2067
2200
        within them. IF in the future, we were to make branch data shareable
2068
 
        between multiple working trees, i.e. via shared storage, then we 
 
2201
        between multiple working trees, i.e. via shared storage, then we
2069
2202
        would probably want to lock both the local tree, and the branch.
2070
2203
        """
2071
2204
        raise NotImplementedError(self.unlock)
2072
2205
 
2073
 
    def update(self, change_reporter=None, possible_transports=None):
 
2206
    _marker = object()
 
2207
 
 
2208
    def update(self, change_reporter=None, possible_transports=None,
 
2209
               revision=None, old_tip=_marker):
2074
2210
        """Update a working tree along its branch.
2075
2211
 
2076
2212
        This will update the branch if its bound too, which means we have
2094
2230
        - Merge current state -> basis tree of the master w.r.t. the old tree
2095
2231
          basis.
2096
2232
        - Do a 'normal' merge of the old branch basis if it is relevant.
 
2233
 
 
2234
        :param revision: The target revision to update to. Must be in the
 
2235
            revision history.
 
2236
        :param old_tip: If branch.update() has already been run, the value it
 
2237
            returned (old tip of the branch or None). _marker is used
 
2238
            otherwise.
2097
2239
        """
2098
 
        if self.branch.get_master_branch(possible_transports) is not None:
 
2240
        if self.branch.get_bound_location() is not None:
2099
2241
            self.lock_write()
2100
 
            update_branch = True
 
2242
            update_branch = (old_tip is self._marker)
2101
2243
        else:
2102
2244
            self.lock_tree_write()
2103
2245
            update_branch = False
2105
2247
            if update_branch:
2106
2248
                old_tip = self.branch.update(possible_transports)
2107
2249
            else:
2108
 
                old_tip = None
2109
 
            return self._update_tree(old_tip, change_reporter)
 
2250
                if old_tip is self._marker:
 
2251
                    old_tip = None
 
2252
            return self._update_tree(old_tip, change_reporter, revision)
2110
2253
        finally:
2111
2254
            self.unlock()
2112
2255
 
2113
2256
    @needs_tree_write_lock
2114
 
    def _update_tree(self, old_tip=None, change_reporter=None):
 
2257
    def _update_tree(self, old_tip=None, change_reporter=None, revision=None):
2115
2258
        """Update a tree to the master branch.
2116
2259
 
2117
2260
        :param old_tip: if supplied, the previous tip revision the branch,
2123
2266
        # cant set that until we update the working trees last revision to be
2124
2267
        # one from the new branch, because it will just get absorbed by the
2125
2268
        # parent de-duplication logic.
2126
 
        # 
 
2269
        #
2127
2270
        # We MUST save it even if an error occurs, because otherwise the users
2128
2271
        # local work is unreferenced and will appear to have been lost.
2129
 
        # 
2130
 
        result = 0
 
2272
        #
 
2273
        nb_conflicts = 0
2131
2274
        try:
2132
2275
            last_rev = self.get_parent_ids()[0]
2133
2276
        except IndexError:
2134
2277
            last_rev = _mod_revision.NULL_REVISION
2135
 
        if last_rev != _mod_revision.ensure_null(self.branch.last_revision()):
2136
 
            # merge tree state up to new branch tip.
 
2278
        if revision is None:
 
2279
            revision = self.branch.last_revision()
 
2280
 
 
2281
        old_tip = old_tip or _mod_revision.NULL_REVISION
 
2282
 
 
2283
        if not _mod_revision.is_null(old_tip) and old_tip != last_rev:
 
2284
            # the branch we are bound to was updated
 
2285
            # merge those changes in first
 
2286
            base_tree  = self.basis_tree()
 
2287
            other_tree = self.branch.repository.revision_tree(old_tip)
 
2288
            nb_conflicts = merge.merge_inner(self.branch, other_tree,
 
2289
                                             base_tree, this_tree=self,
 
2290
                                             change_reporter=change_reporter)
 
2291
            if nb_conflicts:
 
2292
                self.add_parent_tree((old_tip, other_tree))
 
2293
                trace.note('Rerun update after fixing the conflicts.')
 
2294
                return nb_conflicts
 
2295
 
 
2296
        if last_rev != _mod_revision.ensure_null(revision):
 
2297
            # the working tree is up to date with the branch
 
2298
            # we can merge the specified revision from master
 
2299
            to_tree = self.branch.repository.revision_tree(revision)
 
2300
            to_root_id = to_tree.get_root_id()
 
2301
 
2137
2302
            basis = self.basis_tree()
2138
2303
            basis.lock_read()
2139
2304
            try:
2140
 
                to_tree = self.branch.basis_tree()
2141
 
                if basis.inventory.root is None:
2142
 
                    self.set_root_id(to_tree.inventory.root.file_id)
 
2305
                if (basis.inventory.root is None
 
2306
                    or basis.inventory.root.file_id != to_root_id):
 
2307
                    self.set_root_id(to_root_id)
2143
2308
                    self.flush()
2144
 
                result += merge.merge_inner(
2145
 
                                      self.branch,
2146
 
                                      to_tree,
2147
 
                                      basis,
2148
 
                                      this_tree=self,
2149
 
                                      change_reporter=change_reporter)
2150
2309
            finally:
2151
2310
                basis.unlock()
 
2311
 
 
2312
            # determine the branch point
 
2313
            graph = self.branch.repository.get_graph()
 
2314
            base_rev_id = graph.find_unique_lca(self.branch.last_revision(),
 
2315
                                                last_rev)
 
2316
            base_tree = self.branch.repository.revision_tree(base_rev_id)
 
2317
 
 
2318
            nb_conflicts = merge.merge_inner(self.branch, to_tree, base_tree,
 
2319
                                             this_tree=self,
 
2320
                                             change_reporter=change_reporter)
 
2321
            self.set_last_revision(revision)
2152
2322
            # TODO - dedup parents list with things merged by pull ?
2153
2323
            # reuse the tree we've updated to to set the basis:
2154
 
            parent_trees = [(self.branch.last_revision(), to_tree)]
 
2324
            parent_trees = [(revision, to_tree)]
2155
2325
            merges = self.get_parent_ids()[1:]
2156
2326
            # Ideally we ask the tree for the trees here, that way the working
2157
 
            # tree can decide whether to give us teh entire tree or give us a
 
2327
            # tree can decide whether to give us the entire tree or give us a
2158
2328
            # lazy initialised tree. dirstate for instance will have the trees
2159
2329
            # in ram already, whereas a last-revision + basis-inventory tree
2160
2330
            # will not, but also does not need them when setting parents.
2161
2331
            for parent in merges:
2162
2332
                parent_trees.append(
2163
2333
                    (parent, self.branch.repository.revision_tree(parent)))
2164
 
            if (old_tip is not None and not _mod_revision.is_null(old_tip)):
 
2334
            if not _mod_revision.is_null(old_tip):
2165
2335
                parent_trees.append(
2166
2336
                    (old_tip, self.branch.repository.revision_tree(old_tip)))
2167
2337
            self.set_parent_trees(parent_trees)
2168
2338
            last_rev = parent_trees[0][0]
2169
 
        else:
2170
 
            # the working tree had the same last-revision as the master
2171
 
            # branch did. We may still have pivot local work from the local
2172
 
            # branch into old_tip:
2173
 
            if (old_tip is not None and not _mod_revision.is_null(old_tip)):
2174
 
                self.add_parent_tree_id(old_tip)
2175
 
        if (old_tip is not None and not _mod_revision.is_null(old_tip)
2176
 
            and old_tip != last_rev):
2177
 
            # our last revision was not the prior branch last revision
2178
 
            # and we have converted that last revision to a pending merge.
2179
 
            # base is somewhere between the branch tip now
2180
 
            # and the now pending merge
2181
 
 
2182
 
            # Since we just modified the working tree and inventory, flush out
2183
 
            # the current state, before we modify it again.
2184
 
            # TODO: jam 20070214 WorkingTree3 doesn't require this, dirstate
2185
 
            #       requires it only because TreeTransform directly munges the
2186
 
            #       inventory and calls tree._write_inventory(). Ultimately we
2187
 
            #       should be able to remove this extra flush.
2188
 
            self.flush()
2189
 
            graph = self.branch.repository.get_graph()
2190
 
            base_rev_id = graph.find_unique_lca(self.branch.last_revision(),
2191
 
                                                old_tip)
2192
 
            base_tree = self.branch.repository.revision_tree(base_rev_id)
2193
 
            other_tree = self.branch.repository.revision_tree(old_tip)
2194
 
            result += merge.merge_inner(
2195
 
                                  self.branch,
2196
 
                                  other_tree,
2197
 
                                  base_tree,
2198
 
                                  this_tree=self,
2199
 
                                  change_reporter=change_reporter)
2200
 
        return result
 
2339
        return nb_conflicts
2201
2340
 
2202
2341
    def _write_hashcache_if_dirty(self):
2203
2342
        """Write out the hashcache if it is dirty."""
2287
2426
            current_inv = None
2288
2427
            inv_finished = True
2289
2428
        while not inv_finished or not disk_finished:
 
2429
            if current_disk:
 
2430
                ((cur_disk_dir_relpath, cur_disk_dir_path_from_top),
 
2431
                    cur_disk_dir_content) = current_disk
 
2432
            else:
 
2433
                ((cur_disk_dir_relpath, cur_disk_dir_path_from_top),
 
2434
                    cur_disk_dir_content) = ((None, None), None)
2290
2435
            if not disk_finished:
2291
2436
                # strip out .bzr dirs
2292
 
                if current_disk[0][1][top_strip_len:] == '':
2293
 
                    # osutils.walkdirs can be made nicer - 
 
2437
                if (cur_disk_dir_path_from_top[top_strip_len:] == '' and
 
2438
                    len(cur_disk_dir_content) > 0):
 
2439
                    # osutils.walkdirs can be made nicer -
2294
2440
                    # yield the path-from-prefix rather than the pathjoined
2295
2441
                    # value.
2296
 
                    bzrdir_loc = bisect_left(current_disk[1], ('.bzr', '.bzr'))
2297
 
                    if current_disk[1][bzrdir_loc][0] == '.bzr':
 
2442
                    bzrdir_loc = bisect_left(cur_disk_dir_content,
 
2443
                        ('.bzr', '.bzr'))
 
2444
                    if (bzrdir_loc < len(cur_disk_dir_content)
 
2445
                        and self.bzrdir.is_control_filename(
 
2446
                            cur_disk_dir_content[bzrdir_loc][0])):
2298
2447
                        # we dont yield the contents of, or, .bzr itself.
2299
 
                        del current_disk[1][bzrdir_loc]
 
2448
                        del cur_disk_dir_content[bzrdir_loc]
2300
2449
            if inv_finished:
2301
2450
                # everything is unknown
2302
2451
                direction = 1
2304
2453
                # everything is missing
2305
2454
                direction = -1
2306
2455
            else:
2307
 
                direction = cmp(current_inv[0][0], current_disk[0][0])
 
2456
                direction = cmp(current_inv[0][0], cur_disk_dir_relpath)
2308
2457
            if direction > 0:
2309
2458
                # disk is before inventory - unknown
2310
2459
                dirblock = [(relpath, basename, kind, stat, None, None) for
2311
 
                    relpath, basename, kind, stat, top_path in current_disk[1]]
2312
 
                yield (current_disk[0][0], None), dirblock
 
2460
                    relpath, basename, kind, stat, top_path in
 
2461
                    cur_disk_dir_content]
 
2462
                yield (cur_disk_dir_relpath, None), dirblock
2313
2463
                try:
2314
2464
                    current_disk = disk_iterator.next()
2315
2465
                except StopIteration:
2317
2467
            elif direction < 0:
2318
2468
                # inventory is before disk - missing.
2319
2469
                dirblock = [(relpath, basename, 'unknown', None, fileid, kind)
2320
 
                    for relpath, basename, dkind, stat, fileid, kind in 
 
2470
                    for relpath, basename, dkind, stat, fileid, kind in
2321
2471
                    current_inv[1]]
2322
2472
                yield (current_inv[0][0], current_inv[0][1]), dirblock
2323
2473
                try:
2329
2479
                # merge the inventory and disk data together
2330
2480
                dirblock = []
2331
2481
                for relpath, subiterator in itertools.groupby(sorted(
2332
 
                    current_inv[1] + current_disk[1], key=operator.itemgetter(0)), operator.itemgetter(1)):
 
2482
                    current_inv[1] + cur_disk_dir_content,
 
2483
                    key=operator.itemgetter(0)), operator.itemgetter(1)):
2333
2484
                    path_elements = list(subiterator)
2334
2485
                    if len(path_elements) == 2:
2335
2486
                        inv_row, disk_row = path_elements
2388
2539
                relroot = ""
2389
2540
            # FIXME: stash the node in pending
2390
2541
            entry = inv[top_id]
2391
 
            for name, child in entry.sorted_children():
2392
 
                dirblock.append((relroot + name, name, child.kind, None,
2393
 
                    child.file_id, child.kind
2394
 
                    ))
 
2542
            if entry.kind == 'directory':
 
2543
                for name, child in entry.sorted_children():
 
2544
                    dirblock.append((relroot + name, name, child.kind, None,
 
2545
                        child.file_id, child.kind
 
2546
                        ))
2395
2547
            yield (currentdir[0], entry.file_id), dirblock
2396
2548
            # push the user specified dirs from dirblock
2397
2549
            for dir in reversed(dirblock):
2430
2582
        self.set_conflicts(un_resolved)
2431
2583
        return un_resolved, resolved
2432
2584
 
 
2585
    @needs_read_lock
 
2586
    def _check(self, references):
 
2587
        """Check the tree for consistency.
 
2588
 
 
2589
        :param references: A dict with keys matching the items returned by
 
2590
            self._get_check_refs(), and values from looking those keys up in
 
2591
            the repository.
 
2592
        """
 
2593
        tree_basis = self.basis_tree()
 
2594
        tree_basis.lock_read()
 
2595
        try:
 
2596
            repo_basis = references[('trees', self.last_revision())]
 
2597
            if len(list(repo_basis.iter_changes(tree_basis))) > 0:
 
2598
                raise errors.BzrCheckError(
 
2599
                    "Mismatched basis inventory content.")
 
2600
            self._validate()
 
2601
        finally:
 
2602
            tree_basis.unlock()
 
2603
 
2433
2604
    def _validate(self):
2434
2605
        """Validate internal structures.
2435
2606
 
2441
2612
        """
2442
2613
        return
2443
2614
 
 
2615
    def _get_rules_searcher(self, default_searcher):
 
2616
        """See Tree._get_rules_searcher."""
 
2617
        if self._rules_searcher is None:
 
2618
            self._rules_searcher = super(WorkingTree,
 
2619
                self)._get_rules_searcher(default_searcher)
 
2620
        return self._rules_searcher
 
2621
 
 
2622
    def get_shelf_manager(self):
 
2623
        """Return the ShelfManager for this WorkingTree."""
 
2624
        from bzrlib.shelf import ShelfManager
 
2625
        return ShelfManager(self, self._transport)
 
2626
 
2444
2627
 
2445
2628
class WorkingTree2(WorkingTree):
2446
2629
    """This is the Format 2 working tree.
2447
2630
 
2448
 
    This was the first weave based working tree. 
 
2631
    This was the first weave based working tree.
2449
2632
     - uses os locks for locking.
2450
2633
     - uses the branch last-revision.
2451
2634
    """
2461
2644
        if self._inventory is None:
2462
2645
            self.read_working_inventory()
2463
2646
 
 
2647
    def _get_check_refs(self):
 
2648
        """Return the references needed to perform a check of this tree."""
 
2649
        return [('trees', self.last_revision())]
 
2650
 
2464
2651
    def lock_tree_write(self):
2465
2652
        """See WorkingTree.lock_tree_write().
2466
2653
 
2467
2654
        In Format2 WorkingTrees we have a single lock for the branch and tree
2468
2655
        so lock_tree_write() degrades to lock_write().
 
2656
 
 
2657
        :return: An object with an unlock method which will release the lock
 
2658
            obtained.
2469
2659
        """
2470
2660
        self.branch.lock_write()
2471
2661
        try:
2472
 
            return self._control_files.lock_write()
 
2662
            self._control_files.lock_write()
 
2663
            return self
2473
2664
        except:
2474
2665
            self.branch.unlock()
2475
2666
            raise
2484
2675
            if self._inventory_is_modified:
2485
2676
                self.flush()
2486
2677
            self._write_hashcache_if_dirty()
2487
 
                    
 
2678
 
2488
2679
        # reverse order of locking.
2489
2680
        try:
2490
2681
            return self._control_files.unlock()
2506
2697
    def _last_revision(self):
2507
2698
        """See Mutable.last_revision."""
2508
2699
        try:
2509
 
            return self._control_files.get('last-revision').read()
 
2700
            return self._transport.get_bytes('last-revision')
2510
2701
        except errors.NoSuchFile:
2511
2702
            return _mod_revision.NULL_REVISION
2512
2703
 
2513
2704
    def _change_last_revision(self, revision_id):
2514
2705
        """See WorkingTree._change_last_revision."""
2515
 
        if revision_id is None or revision_id == NULL_REVISION:
 
2706
        if revision_id is None or revision_id == _mod_revision.NULL_REVISION:
2516
2707
            try:
2517
 
                self._control_files._transport.delete('last-revision')
 
2708
                self._transport.delete('last-revision')
2518
2709
            except errors.NoSuchFile:
2519
2710
                pass
2520
2711
            return False
2521
2712
        else:
2522
 
            self._control_files.put_bytes('last-revision', revision_id)
 
2713
            self._transport.put_bytes('last-revision', revision_id,
 
2714
                mode=self.bzrdir._get_file_mode())
2523
2715
            return True
2524
2716
 
 
2717
    def _get_check_refs(self):
 
2718
        """Return the references needed to perform a check of this tree."""
 
2719
        return [('trees', self.last_revision())]
 
2720
 
2525
2721
    @needs_tree_write_lock
2526
2722
    def set_conflicts(self, conflicts):
2527
 
        self._put_rio('conflicts', conflicts.to_stanzas(), 
 
2723
        self._put_rio('conflicts', conflicts.to_stanzas(),
2528
2724
                      CONFLICT_HEADER_1)
2529
2725
 
2530
2726
    @needs_tree_write_lock
2537
2733
    @needs_read_lock
2538
2734
    def conflicts(self):
2539
2735
        try:
2540
 
            confile = self._control_files.get('conflicts')
 
2736
            confile = self._transport.get('conflicts')
2541
2737
        except errors.NoSuchFile:
2542
2738
            return _mod_conflicts.ConflictList()
2543
2739
        try:
2544
 
            if confile.next() != CONFLICT_HEADER_1 + '\n':
 
2740
            try:
 
2741
                if confile.next() != CONFLICT_HEADER_1 + '\n':
 
2742
                    raise errors.ConflictFormatError()
 
2743
            except StopIteration:
2545
2744
                raise errors.ConflictFormatError()
2546
 
        except StopIteration:
2547
 
            raise errors.ConflictFormatError()
2548
 
        return _mod_conflicts.ConflictList.from_stanzas(RioReader(confile))
 
2745
            return _mod_conflicts.ConflictList.from_stanzas(RioReader(confile))
 
2746
        finally:
 
2747
            confile.close()
2549
2748
 
2550
2749
    def unlock(self):
2551
2750
        # do non-implementation specific cleanup
2568
2767
            return path[:-len(suffix)]
2569
2768
 
2570
2769
 
2571
 
@deprecated_function(zero_eight)
2572
 
def is_control_file(filename):
2573
 
    """See WorkingTree.is_control_filename(filename)."""
2574
 
    ## FIXME: better check
2575
 
    filename = normpath(filename)
2576
 
    while filename != '':
2577
 
        head, tail = os.path.split(filename)
2578
 
        ## mutter('check %r for control file' % ((head, tail),))
2579
 
        if tail == '.bzr':
2580
 
            return True
2581
 
        if filename == head:
2582
 
            break
2583
 
        filename = head
2584
 
    return False
2585
 
 
2586
 
 
2587
2770
class WorkingTreeFormat(object):
2588
2771
    """An encapsulation of the initialization and open routines for a format.
2589
2772
 
2592
2775
     * a format string,
2593
2776
     * an open routine.
2594
2777
 
2595
 
    Formats are placed in an dict by their format string for reference 
 
2778
    Formats are placed in an dict by their format string for reference
2596
2779
    during workingtree opening. Its not required that these be instances, they
2597
 
    can be classes themselves with class methods - it simply depends on 
 
2780
    can be classes themselves with class methods - it simply depends on
2598
2781
    whether state is needed for a given format or not.
2599
2782
 
2600
2783
    Once a format is deprecated, just deprecate the initialize and open
2601
 
    methods on the format class. Do not deprecate the object, as the 
 
2784
    methods on the format class. Do not deprecate the object, as the
2602
2785
    object will be created every time regardless.
2603
2786
    """
2604
2787
 
2617
2800
        """Return the format for the working tree object in a_bzrdir."""
2618
2801
        try:
2619
2802
            transport = a_bzrdir.get_workingtree_transport(None)
2620
 
            format_string = transport.get("format").read()
 
2803
            format_string = transport.get_bytes("format")
2621
2804
            return klass._formats[format_string]
2622
2805
        except errors.NoSuchFile:
2623
2806
            raise errors.NoWorkingTree(base=transport.base)
2624
2807
        except KeyError:
2625
 
            raise errors.UnknownFormatError(format=format_string)
 
2808
            raise errors.UnknownFormatError(format=format_string,
 
2809
                                            kind="working tree")
2626
2810
 
2627
2811
    def __eq__(self, other):
2628
2812
        return self.__class__ is other.__class__
2647
2831
        """Is this format supported?
2648
2832
 
2649
2833
        Supported formats can be initialized and opened.
2650
 
        Unsupported formats may not support initialization or committing or 
 
2834
        Unsupported formats may not support initialization or committing or
2651
2835
        some other features depending on the reason for not being supported.
2652
2836
        """
2653
2837
        return True
2654
2838
 
 
2839
    def supports_content_filtering(self):
 
2840
        """True if this format supports content filtering."""
 
2841
        return False
 
2842
 
 
2843
    def supports_views(self):
 
2844
        """True if this format supports stored views."""
 
2845
        return False
 
2846
 
2655
2847
    @classmethod
2656
2848
    def register_format(klass, format):
2657
2849
        klass._formats[format.get_format_string()] = format
2662
2854
 
2663
2855
    @classmethod
2664
2856
    def unregister_format(klass, format):
2665
 
        assert klass._formats[format.get_format_string()] is format
2666
2857
        del klass._formats[format.get_format_string()]
2667
2858
 
2668
2859
 
2669
2860
class WorkingTreeFormat2(WorkingTreeFormat):
2670
 
    """The second working tree format. 
 
2861
    """The second working tree format.
2671
2862
 
2672
2863
    This format modified the hash cache from the format 1 hash cache.
2673
2864
    """
2678
2869
        """See WorkingTreeFormat.get_format_description()."""
2679
2870
        return "Working tree format 2"
2680
2871
 
2681
 
    def stub_initialize_remote(self, control_files):
2682
 
        """As a special workaround create critical control files for a remote working tree
2683
 
        
 
2872
    def _stub_initialize_on_transport(self, transport, file_mode):
 
2873
        """Workaround: create control files for a remote working tree.
 
2874
 
2684
2875
        This ensures that it can later be updated and dealt with locally,
2685
 
        since BzrDirFormat6 and BzrDirFormat5 cannot represent dirs with 
 
2876
        since BzrDirFormat6 and BzrDirFormat5 cannot represent dirs with
2686
2877
        no working tree.  (See bug #43064).
2687
2878
        """
2688
2879
        sio = StringIO()
2689
 
        inv = Inventory()
 
2880
        inv = inventory.Inventory()
2690
2881
        xml5.serializer_v5.write_inventory(inv, sio, working=True)
2691
2882
        sio.seek(0)
2692
 
        control_files.put('inventory', sio)
2693
 
 
2694
 
        control_files.put_bytes('pending-merges', '')
2695
 
        
2696
 
 
2697
 
    def initialize(self, a_bzrdir, revision_id=None):
 
2883
        transport.put_file('inventory', sio, file_mode)
 
2884
        transport.put_bytes('pending-merges', '', file_mode)
 
2885
 
 
2886
    def initialize(self, a_bzrdir, revision_id=None, from_branch=None,
 
2887
                   accelerator_tree=None, hardlink=False):
2698
2888
        """See WorkingTreeFormat.initialize()."""
2699
2889
        if not isinstance(a_bzrdir.transport, LocalTransport):
2700
2890
            raise errors.NotLocalUrl(a_bzrdir.transport.base)
2701
 
        branch = a_bzrdir.open_branch()
 
2891
        if from_branch is not None:
 
2892
            branch = from_branch
 
2893
        else:
 
2894
            branch = a_bzrdir.open_branch()
2702
2895
        if revision_id is None:
2703
2896
            revision_id = _mod_revision.ensure_null(branch.last_revision())
2704
2897
        branch.lock_write()
2706
2899
            branch.generate_revision_history(revision_id)
2707
2900
        finally:
2708
2901
            branch.unlock()
2709
 
        inv = Inventory()
 
2902
        inv = inventory.Inventory()
2710
2903
        wt = WorkingTree2(a_bzrdir.root_transport.local_abspath('.'),
2711
2904
                         branch,
2712
2905
                         inv,
2715
2908
                         _bzrdir=a_bzrdir)
2716
2909
        basis_tree = branch.repository.revision_tree(revision_id)
2717
2910
        if basis_tree.inventory.root is not None:
2718
 
            wt.set_root_id(basis_tree.inventory.root.file_id)
 
2911
            wt.set_root_id(basis_tree.get_root_id())
2719
2912
        # set the parent list and cache the basis tree.
2720
2913
        if _mod_revision.is_null(revision_id):
2721
2914
            parent_trees = []
2757
2950
        - is new in bzr 0.8
2758
2951
        - uses a LockDir to guard access for writes.
2759
2952
    """
2760
 
    
 
2953
 
2761
2954
    upgrade_recommended = True
2762
2955
 
2763
2956
    def get_format_string(self):
2780
2973
 
2781
2974
    def _open_control_files(self, a_bzrdir):
2782
2975
        transport = a_bzrdir.get_workingtree_transport(None)
2783
 
        return LockableFiles(transport, self._lock_file_name, 
 
2976
        return LockableFiles(transport, self._lock_file_name,
2784
2977
                             self._lock_class)
2785
2978
 
2786
 
    def initialize(self, a_bzrdir, revision_id=None):
 
2979
    def initialize(self, a_bzrdir, revision_id=None, from_branch=None,
 
2980
                   accelerator_tree=None, hardlink=False):
2787
2981
        """See WorkingTreeFormat.initialize().
2788
 
        
2789
 
        revision_id allows creating a working tree at a different
2790
 
        revision than the branch is at.
 
2982
 
 
2983
        :param revision_id: if supplied, create a working tree at a different
 
2984
            revision than the branch is at.
 
2985
        :param accelerator_tree: A tree which can be used for retrieving file
 
2986
            contents more quickly than the revision tree, i.e. a workingtree.
 
2987
            The revision tree will be used for cases where accelerator_tree's
 
2988
            content is different.
 
2989
        :param hardlink: If true, hard-link files from accelerator_tree,
 
2990
            where possible.
2791
2991
        """
2792
2992
        if not isinstance(a_bzrdir.transport, LocalTransport):
2793
2993
            raise errors.NotLocalUrl(a_bzrdir.transport.base)
2795
2995
        control_files = self._open_control_files(a_bzrdir)
2796
2996
        control_files.create_lock()
2797
2997
        control_files.lock_write()
2798
 
        control_files.put_utf8('format', self.get_format_string())
2799
 
        branch = a_bzrdir.open_branch()
 
2998
        transport.put_bytes('format', self.get_format_string(),
 
2999
            mode=a_bzrdir._get_file_mode())
 
3000
        if from_branch is not None:
 
3001
            branch = from_branch
 
3002
        else:
 
3003
            branch = a_bzrdir.open_branch()
2800
3004
        if revision_id is None:
2801
3005
            revision_id = _mod_revision.ensure_null(branch.last_revision())
2802
3006
        # WorkingTree3 can handle an inventory which has a unique root id.
2817
3021
            basis_tree = branch.repository.revision_tree(revision_id)
2818
3022
            # only set an explicit root id if there is one to set.
2819
3023
            if basis_tree.inventory.root is not None:
2820
 
                wt.set_root_id(basis_tree.inventory.root.file_id)
2821
 
            if revision_id == NULL_REVISION:
 
3024
                wt.set_root_id(basis_tree.get_root_id())
 
3025
            if revision_id == _mod_revision.NULL_REVISION:
2822
3026
                wt.set_parent_trees([])
2823
3027
            else:
2824
3028
                wt.set_parent_trees([(revision_id, basis_tree)])
2831
3035
        return wt
2832
3036
 
2833
3037
    def _initial_inventory(self):
2834
 
        return Inventory()
 
3038
        return inventory.Inventory()
2835
3039
 
2836
3040
    def __init__(self):
2837
3041
        super(WorkingTreeFormat3, self).__init__()
2852
3056
 
2853
3057
    def _open(self, a_bzrdir, control_files):
2854
3058
        """Open the tree itself.
2855
 
        
 
3059
 
2856
3060
        :param a_bzrdir: the dir for the tree.
2857
3061
        :param control_files: the control files for the tree.
2858
3062
        """
2866
3070
        return self.get_format_string()
2867
3071
 
2868
3072
 
2869
 
__default_format = WorkingTreeFormat4()
 
3073
__default_format = WorkingTreeFormat6()
2870
3074
WorkingTreeFormat.register_format(__default_format)
 
3075
WorkingTreeFormat.register_format(WorkingTreeFormat5())
 
3076
WorkingTreeFormat.register_format(WorkingTreeFormat4())
2871
3077
WorkingTreeFormat.register_format(WorkingTreeFormat3())
2872
3078
WorkingTreeFormat.set_default_format(__default_format)
2873
3079
# formats which have no format string are not discoverable