/brz/remove-bazaar : contents of breezy/memorytree.py at revision 7265.2.1

: (revision 7265.2.1)

To get this branch, use:

bzr branch
http://gegoxaren.bato24.eu/bzr/brz/remove-bazaar

# Copyright (C) 2006 Canonical Ltd
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA

"""MemoryTree object.

See MemoryTree for more details.
"""

from __future__ import absolute_import

import os

from . import (
    errors,
    lock,
    revision as _mod_revision,
    )
from .bzr.inventory import Inventory
from .bzr.inventorytree import MutableInventoryTree
from .osutils import sha_file
from .transport.memory import MemoryTransport


class MemoryTree(MutableInventoryTree):
    """A MemoryTree is a specialisation of MutableTree.

    It maintains nearly no state outside of read_lock and write_lock
    transactions. (it keeps a reference to the branch, and its last-revision
    only).
    """

    def __init__(self, branch, revision_id):
        """Construct a MemoryTree for branch using revision_id."""
        self.branch = branch
        self.controldir = branch.controldir
        self._branch_revision_id = revision_id
        self._locks = 0
        self._lock_mode = None

    def get_config_stack(self):
        return self.branch.get_config_stack()

    def is_control_filename(self, filename):
        # Memory tree doesn't have any control filenames
        return False

    def _add(self, files, ids, kinds):
        """See MutableTree._add."""
        with self.lock_tree_write():
            for f, file_id, kind in zip(files, ids, kinds):
                if kind is None:
                    kind = 'file'
                if file_id is None:
                    self._inventory.add_path(f, kind=kind)
                else:
                    self._inventory.add_path(f, kind=kind, file_id=file_id)

    def basis_tree(self):
        """See Tree.basis_tree()."""
        return self._basis_tree

    @staticmethod
    def create_on_branch(branch):
        """Create a MemoryTree for branch, using the last-revision of branch."""
        revision_id = _mod_revision.ensure_null(branch.last_revision())
        return MemoryTree(branch, revision_id)

    def _gather_kinds(self, files, kinds):
        """See MutableTree._gather_kinds.

        This implementation does not care about the file kind of
        missing files, so is a no-op.
        """

    def get_file(self, path):
        """See Tree.get_file."""
        return self._file_transport.get(path)

    def get_file_sha1(self, path, stat_value=None):
        """See Tree.get_file_sha1()."""
        stream = self._file_transport.get(path)
        return sha_file(stream)

    def get_root_id(self):
        return self.path2id('')

    def _comparison_data(self, entry, path):
        """See Tree._comparison_data."""
        if entry is None:
            return None, False, None
        return entry.kind, entry.executable, None

    def rename_one(self, from_rel, to_rel):
        with self.lock_tree_write():
            file_id = self.path2id(from_rel)
            to_dir, to_tail = os.path.split(to_rel)
            to_parent_id = self.path2id(to_dir)
            self._file_transport.move(from_rel, to_rel)
            self._inventory.rename(file_id, to_parent_id, to_tail)

    def path_content_summary(self, path):
        """See Tree.path_content_summary."""
        id = self.path2id(path)
        if id is None:
            return 'missing', None, None, None
        kind = self.kind(path, id)
        if kind == 'file':
            bytes = self._file_transport.get_bytes(path)
            size = len(bytes)
            executable = self._inventory[id].executable
            sha1 = None  # no stat cache
            return (kind, size, executable, sha1)
        elif kind == 'directory':
            # memory tree does not support nested trees yet.
            return kind, None, None, None
        elif kind == 'symlink':
            raise NotImplementedError('symlink support')
        else:
            raise NotImplementedError('unknown kind')

    def get_parent_ids(self):
        """See Tree.get_parent_ids.

        This implementation returns the current cached value from
            self._parent_ids.
        """
        with self.lock_read():
            return list(self._parent_ids)

    def has_filename(self, filename):
        """See Tree.has_filename()."""
        return self._file_transport.has(filename)

    def is_executable(self, path):
        return self._inventory.get_entry_by_path(path).executable

    def kind(self, path):
        file_id = self.path2id(path)
        return self._inventory[file_id].kind

    def mkdir(self, path, file_id=None):
        """See MutableTree.mkdir()."""
        self.add(path, file_id, 'directory')
        if file_id is None:
            file_id = self.path2id(path)
        self._file_transport.mkdir(path)
        return file_id

    def last_revision(self):
        """See MutableTree.last_revision."""
        with self.lock_read():
            return self._branch_revision_id

    def lock_read(self):
        """Lock the memory tree for reading.

        This triggers population of data from the branch for its revision.
        """
        self._locks += 1
        try:
            if self._locks == 1:
                self.branch.lock_read()
                self._lock_mode = "r"
                self._populate_from_branch()
            return lock.LogicalLockResult(self.unlock)
        except BaseException:
            self._locks -= 1
            raise

    def lock_tree_write(self):
        """See MutableTree.lock_tree_write()."""
        self._locks += 1
        try:
            if self._locks == 1:
                self.branch.lock_read()
                self._lock_mode = "w"
                self._populate_from_branch()
            elif self._lock_mode == "r":
                raise errors.ReadOnlyError(self)
        except BaseException:
            self._locks -= 1
            raise
        return lock.LogicalLockResult(self.unlock)

    def lock_write(self):
        """See MutableTree.lock_write()."""
        self._locks += 1
        try:
            if self._locks == 1:
                self.branch.lock_write()
                self._lock_mode = "w"
                self._populate_from_branch()
            elif self._lock_mode == "r":
                raise errors.ReadOnlyError(self)
            return lock.LogicalLockResult(self.unlock)
        except BaseException:
            self._locks -= 1
            raise

    def _populate_from_branch(self):
        """Populate the in-tree state from the branch."""
        self._set_basis()
        if self._branch_revision_id == _mod_revision.NULL_REVISION:
            self._parent_ids = []
        else:
            self._parent_ids = [self._branch_revision_id]
        self._inventory = Inventory(None, self._basis_tree.get_revision_id())
        self._file_transport = MemoryTransport()
        # TODO copy the revision trees content, or do it lazy, or something.
        inventory_entries = self._basis_tree.iter_entries_by_dir()
        for path, entry in inventory_entries:
            self._inventory.add(entry.copy())
            if path == '':
                continue
            if entry.kind == 'directory':
                self._file_transport.mkdir(path)
            elif entry.kind == 'file':
                self._file_transport.put_file(
                    path, self._basis_tree.get_file(path))
            else:
                raise NotImplementedError(self._populate_from_branch)

    def put_file_bytes_non_atomic(self, path, bytes):
        """See MutableTree.put_file_bytes_non_atomic."""
        self._file_transport.put_bytes(path, bytes)

    def unlock(self):
        """Release a lock.

        This frees all cached state when the last lock context for the tree is
        left.
        """
        if self._locks == 1:
            self._basis_tree = None
            self._parent_ids = []
            self._inventory = None
            try:
                self.branch.unlock()
            finally:
                self._locks = 0
                self._lock_mode = None
        else:
            self._locks -= 1

    def unversion(self, paths):
        """Remove the paths from the current versioned set.

        When a file_id is unversioned, all of its children are automatically
        unversioned.

        :param paths: The paths to stop versioning.
        :raises: NoSuchId if any fileid is not currently versioned.
        """
        with self.lock_tree_write():
            # XXX: This should be in mutabletree, but the inventory-save action
            # is not relevant to memory tree. Until that is done in unlock by
            # working tree, we cannot share the implementation.
            file_ids = set()
            for path in paths:
                file_id = self.path2id(path)
                if file_id is None:
                    raise errors.NoSuchFile(path)
                file_ids.add(file_id)
            for file_id in file_ids:
                if self._inventory.has_id(file_id):
                    self._inventory.remove_recursive_id(file_id)

    def set_parent_ids(self, revision_ids, allow_leftmost_as_ghost=False):
        """See MutableTree.set_parent_trees()."""
        for revision_id in revision_ids:
            _mod_revision.check_not_reserved_id(revision_id)
        if len(revision_ids) == 0:
            self._parent_ids = []
            self._branch_revision_id = _mod_revision.NULL_REVISION
        else:
            self._parent_ids = revision_ids
            self._branch_revision_id = revision_ids[0]
        self._allow_leftmost_as_ghost = allow_leftmost_as_ghost
        self._set_basis()

    def _set_basis(self):
        try:
            self._basis_tree = self.branch.repository.revision_tree(
                self._branch_revision_id)
        except errors.NoSuchRevision:
            if self._allow_leftmost_as_ghost:
                self._basis_tree = self.branch.repository.revision_tree(
                    _mod_revision.NULL_REVISION)
            else:
                raise

    def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
        """See MutableTree.set_parent_trees()."""
        if len(parents_list) == 0:
            self._parent_ids = []
            self._basis_tree = self.branch.repository.revision_tree(
                _mod_revision.NULL_REVISION)
        else:
            if parents_list[0][1] is None and not allow_leftmost_as_ghost:
                # a ghost in the left most parent
                raise errors.GhostRevisionUnusableHere(parents_list[0][0])
            self._parent_ids = [parent_id for parent_id, tree in parents_list]
            if parents_list[0][1] is None or parents_list[0][1] == b'null:':
                self._basis_tree = self.branch.repository.revision_tree(
                    _mod_revision.NULL_REVISION)
            else:
                self._basis_tree = parents_list[0][1]
            self._branch_revision_id = parents_list[0][0]

1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	1	# Copyright (C) 2006 Canonical Ltd
	2	#
	3	# This program is free software; you can redistribute it and/or modify
	4	# it under the terms of the GNU General Public License as published by
	5	# the Free Software Foundation; either version 2 of the License, or
	6	# (at your option) any later version.
	7	#
	8	# This program is distributed in the hope that it will be useful,
	9	# but WITHOUT ANY WARRANTY; without even the implied warranty of
	10	# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	11	# GNU General Public License for more details.
	12	#
	13	# You should have received a copy of the GNU General Public License
	14	# along with this program; if not, write to the Free Software
4183.7.1 by Sabin Iacob update FSF mailing address	15	# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	16
	17	"""MemoryTree object.
	18
	19	See MemoryTree for more details.
	20	"""
	21
6379.6.7 by Jelmer Vernooij Move importing from future until after doc string, otherwise the doc string will disappear.	22	from __future__ import absolute_import
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	23
3567.5.1 by John Arbash Meinel Implement rename_one on MemoryTree, and expose that in the Branch Builder	24	import os
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	25
6624 by Jelmer Vernooĳ Merge Python3 porting work ('py3 pokes')	26	from . import (
2598.5.1 by Aaron Bentley Start eliminating the use of None to indicate null revision	27	errors,
6754.8.3 by Jelmer Vernooĳ Use context manager in decorators.	28	lock,
2598.5.1 by Aaron Bentley Start eliminating the use of None to indicate null revision	29	revision as _mod_revision,
	30	)
6670.4.1 by Jelmer Vernooij Update imports.	31	from .bzr.inventory import Inventory
6670.4.12 by Jelmer Vernooĳ Move inventorytree to breezy.bzr.	32	from .bzr.inventorytree import MutableInventoryTree
6624 by Jelmer Vernooĳ Merge Python3 porting work ('py3 pokes')	33	from .osutils import sha_file
	34	from .transport.memory import MemoryTransport
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	35
	36
6672.2.2 by Jelmer Vernooij Fix imports.	37	class MemoryTree(MutableInventoryTree):
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	38	"""A MemoryTree is a specialisation of MutableTree.
3943.8.1 by Marius Kruger remove all trailing whitespace from bzr source	39
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	40	It maintains nearly no state outside of read_lock and write_lock
	41	transactions. (it keeps a reference to the branch, and its last-revision
	42	only).
	43	"""
	44
	45	def __init__(self, branch, revision_id):
	46	"""Construct a MemoryTree for branch using revision_id."""
	47	self.branch = branch
6653.6.1 by Jelmer Vernooĳ Rename a number of attributes from bzrdir to controldir.	48	self.controldir = branch.controldir
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	49	self._branch_revision_id = revision_id
	50	self._locks = 0
	51	self._lock_mode = None
	52
6449.6.7 by Jelmer Vernooij Fix tests.	53	def get_config_stack(self):
	54	return self.branch.get_config_stack()
	55
5699.2.1 by Jelmer Vernooij Move is_control_filename() from Tree to MutableTree.	56	def is_control_filename(self, filename):
	57	# Memory tree doesn't have any control filenames
	58	return False
	59
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	60	def _add(self, files, ids, kinds):
	61	"""See MutableTree._add."""
6754.8.18 by Jelmer Vernooĳ Get rid of lock_tree_write.	62	with self.lock_tree_write():
	63	for f, file_id, kind in zip(files, ids, kinds):
	64	if kind is None:
	65	kind = 'file'
	66	if file_id is None:
	67	self._inventory.add_path(f, kind=kind)
	68	else:
	69	self._inventory.add_path(f, kind=kind, file_id=file_id)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	70
	71	def basis_tree(self):
	72	"""See Tree.basis_tree()."""
	73	return self._basis_tree
	74
	75	@staticmethod
	76	def create_on_branch(branch):
	77	"""Create a MemoryTree for branch, using the last-revision of branch."""
2598.5.4 by Aaron Bentley Restore original Branch.last_revision behavior, fix bits that care	78	revision_id = _mod_revision.ensure_null(branch.last_revision())
2598.5.1 by Aaron Bentley Start eliminating the use of None to indicate null revision	79	return MemoryTree(branch, revision_id)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	80
	81	def _gather_kinds(self, files, kinds):
	82	"""See MutableTree._gather_kinds.
3943.8.1 by Marius Kruger remove all trailing whitespace from bzr source	83
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	84	This implementation does not care about the file kind of
	85	missing files, so is a no-op.
	86	"""
	87
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	88	def get_file(self, path):
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	89	"""See Tree.get_file."""
2743.3.3 by Ian Clatworthy Skip path lookup for tree.get_file() when we already know the path	90	return self._file_transport.get(path)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	91
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	92	def get_file_sha1(self, path, stat_value=None):
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	93	"""See Tree.get_file_sha1()."""
	94	stream = self._file_transport.get(path)
	95	return sha_file(stream)
	96
2946.3.5 by John Arbash Meinel MemoryTree is not tested under the tree_implementations tests.	97	def get_root_id(self):
	98	return self.path2id('')
	99
2564.2.1 by Ian Clatworthy refactor commit to support alternative population meothds	100	def _comparison_data(self, entry, path):
	101	"""See Tree._comparison_data."""
	102	if entry is None:
	103	return None, False, None
	104	return entry.kind, entry.executable, None
	105
3567.5.1 by John Arbash Meinel Implement rename_one on MemoryTree, and expose that in the Branch Builder	106	def rename_one(self, from_rel, to_rel):
6754.8.18 by Jelmer Vernooĳ Get rid of lock_tree_write.	107	with self.lock_tree_write():
	108	file_id = self.path2id(from_rel)
	109	to_dir, to_tail = os.path.split(to_rel)
	110	to_parent_id = self.path2id(to_dir)
	111	self._file_transport.move(from_rel, to_rel)
	112	self._inventory.rename(file_id, to_parent_id, to_tail)
3514.4.38 by John Arbash Meinel Use direct access to the inventory instead of path2id.	113
2776.4.2 by Robert Collins nuke _read_tree_state and snapshot from inventory, moving responsibility into the commit builder.	114	def path_content_summary(self, path):
	115	"""See Tree.path_content_summary."""
	116	id = self.path2id(path)
	117	if id is None:
	118	return 'missing', None, None, None
6809.4.7 by Jelmer Vernooĳ Swap arguments for get_symlink_target and kind/stored_kind.	119	kind = self.kind(path, id)
2776.4.2 by Robert Collins nuke _read_tree_state and snapshot from inventory, moving responsibility into the commit builder.	120	if kind == 'file':
	121	bytes = self._file_transport.get_bytes(path)
	122	size = len(bytes)
	123	executable = self._inventory[id].executable
7143.15.2 by Jelmer Vernooĳ Run autopep8.	124	sha1 = None # no stat cache
2776.4.2 by Robert Collins nuke _read_tree_state and snapshot from inventory, moving responsibility into the commit builder.	125	return (kind, size, executable, sha1)
	126	elif kind == 'directory':
	127	# memory tree does not support nested trees yet.
	128	return kind, None, None, None
	129	elif kind == 'symlink':
	130	raise NotImplementedError('symlink support')
	131	else:
	132	raise NotImplementedError('unknown kind')
	133
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	134	def get_parent_ids(self):
	135	"""See Tree.get_parent_ids.
	136
	137	This implementation returns the current cached value from
	138	self._parent_ids.
	139	"""
6754.8.16 by Jelmer Vernooĳ Get rid of all uses of needs_read_lock	140	with self.lock_read():
	141	return list(self._parent_ids)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	142
	143	def has_filename(self, filename):
	144	"""See Tree.has_filename()."""
	145	return self._file_transport.has(filename)
	146
7141.7.1 by Jelmer Vernooĳ Get rid of file_ids in most of Tree.	147	def is_executable(self, path):
	148	return self._inventory.get_entry_by_path(path).executable
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	149
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	150	def kind(self, path):
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	151	file_id = self.path2id(path)
1959.4.2 by Aaron Bentley Merge bzr.dev	152	return self._inventory[file_id].kind
1959.4.2 by Aaron Bentley Merge bzr.dev	153
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	154	def mkdir(self, path, file_id=None):
	155	"""See MutableTree.mkdir()."""
	156	self.add(path, file_id, 'directory')
	157	if file_id is None:
	158	file_id = self.path2id(path)
	159	self._file_transport.mkdir(path)
	160	return file_id
	161
1986.1.6 by Robert Collins Add MemoryTree.last_revision.	162	def last_revision(self):
1986.1.6 by Robert Collins Add MemoryTree.last_revision.	163	"""See MutableTree.last_revision."""
6754.8.16 by Jelmer Vernooĳ Get rid of all uses of needs_read_lock	164	with self.lock_read():
	165	return self._branch_revision_id
1986.1.6 by Robert Collins Add MemoryTree.last_revision.	166
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	167	def lock_read(self):
	168	"""Lock the memory tree for reading.
	169
	170	This triggers population of data from the branch for its revision.
	171	"""
	172	self._locks += 1
	173	try:
	174	if self._locks == 1:
	175	self.branch.lock_read()
	176	self._lock_mode = "r"
	177	self._populate_from_branch()
6754.8.3 by Jelmer Vernooĳ Use context manager in decorators.	178	return lock.LogicalLockResult(self.unlock)
7143.15.5 by Jelmer Vernooĳ More PEP8 fixes.	179	except BaseException:
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	180	self._locks -= 1
	181	raise
	182
1986.1.8 by Robert Collins Update to bzr.dev, which involves adding lock_tree_write to MutableTree and MemoryTree.	183	def lock_tree_write(self):
	184	"""See MutableTree.lock_tree_write()."""
	185	self._locks += 1
	186	try:
	187	if self._locks == 1:
	188	self.branch.lock_read()
	189	self._lock_mode = "w"
	190	self._populate_from_branch()
	191	elif self._lock_mode == "r":
	192	raise errors.ReadOnlyError(self)
7143.15.5 by Jelmer Vernooĳ More PEP8 fixes.	193	except BaseException:
1986.1.8 by Robert Collins Update to bzr.dev, which involves adding lock_tree_write to MutableTree and MemoryTree.	194	self._locks -= 1
	195	raise
6754.8.9 by Jelmer Vernooĳ Fix more tests.	196	return lock.LogicalLockResult(self.unlock)
1986.1.8 by Robert Collins Update to bzr.dev, which involves adding lock_tree_write to MutableTree and MemoryTree.	197
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	198	def lock_write(self):
	199	"""See MutableTree.lock_write()."""
	200	self._locks += 1
	201	try:
	202	if self._locks == 1:
	203	self.branch.lock_write()
	204	self._lock_mode = "w"
	205	self._populate_from_branch()
	206	elif self._lock_mode == "r":
	207	raise errors.ReadOnlyError(self)
6754.8.3 by Jelmer Vernooĳ Use context manager in decorators.	208	return lock.LogicalLockResult(self.unlock)
7143.15.5 by Jelmer Vernooĳ More PEP8 fixes.	209	except BaseException:
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	210	self._locks -= 1
	211	raise
	212
	213	def _populate_from_branch(self):
	214	"""Populate the in-tree state from the branch."""
4190.1.4 by Robert Collins Cache ghosts when we can get them from a RemoteRepository in get_parent_map.	215	self._set_basis()
3668.5.4 by Jelmer Vernooij Eliminate more uses of Repository.revision_tree(None).	216	if self._branch_revision_id == _mod_revision.NULL_REVISION:
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	217	self._parent_ids = []
	218	else:
	219	self._parent_ids = [self._branch_revision_id]
5802.1.1 by Jelmer Vernooij Move Inventory._get_mutable_inventory -> mutable_inventory_from_tree.	220	self._inventory = Inventory(None, self._basis_tree.get_revision_id())
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	221	self._file_transport = MemoryTransport()
	222	# TODO copy the revision trees content, or do it lazy, or something.
5802.1.1 by Jelmer Vernooij Move Inventory._get_mutable_inventory -> mutable_inventory_from_tree.	223	inventory_entries = self._basis_tree.iter_entries_by_dir()
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	224	for path, entry in inventory_entries:
5802.1.1 by Jelmer Vernooij Move Inventory._get_mutable_inventory -> mutable_inventory_from_tree.	225	self._inventory.add(entry.copy())
1731.1.50 by Aaron Bentley Merge bzr.dev	226	if path == '':
1731.1.50 by Aaron Bentley Merge bzr.dev	227	continue
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	228	if entry.kind == 'directory':
	229	self._file_transport.mkdir(path)
	230	elif entry.kind == 'file':
7143.15.5 by Jelmer Vernooĳ More PEP8 fixes.	231	self._file_transport.put_file(
7143.15.15 by Jelmer Vernooĳ Merge trunk.	232	path, self._basis_tree.get_file(path))
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	233	else:
	234	raise NotImplementedError(self._populate_from_branch)
	235
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	236	def put_file_bytes_non_atomic(self, path, bytes):
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	237	"""See MutableTree.put_file_bytes_non_atomic."""
6809.4.8 by Jelmer Vernooĳ Fix some test failures.	238	self._file_transport.put_bytes(path, bytes)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	239
	240	def unlock(self):
	241	"""Release a lock.
	242
	243	This frees all cached state when the last lock context for the tree is
	244	left.
	245	"""
	246	if self._locks == 1:
	247	self._basis_tree = None
	248	self._parent_ids = []
	249	self._inventory = None
	250	try:
	251	self.branch.unlock()
	252	finally:
	253	self._locks = 0
	254	self._lock_mode = None
	255	else:
	256	self._locks -= 1
	257
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	258	def unversion(self, paths):
6809.4.25 by Jelmer Vernooĳ Add paths argument to .unversion.	259	"""Remove the paths from the current versioned set.
1986.1.3 by Robert Collins Merge bzr.dev.	260
	261	When a file_id is unversioned, all of its children are automatically
	262	unversioned.
	263
6809.4.25 by Jelmer Vernooĳ Add paths argument to .unversion.	264	:param paths: The paths to stop versioning.
1986.1.3 by Robert Collins Merge bzr.dev.	265	:raises: NoSuchId if any fileid is not currently versioned.
1986.1.3 by Robert Collins Merge bzr.dev.	266	"""
6754.8.18 by Jelmer Vernooĳ Get rid of lock_tree_write.	267	with self.lock_tree_write():
	268	# XXX: This should be in mutabletree, but the inventory-save action
	269	# is not relevant to memory tree. Until that is done in unlock by
	270	# working tree, we cannot share the implementation.
7192.5.1 by Jelmer Vernooĳ Remove more file ids.	271	file_ids = set()
	272	for path in paths:
	273	file_id = self.path2id(path)
	274	if file_id is None:
	275	raise errors.NoSuchFile(path)
	276	file_ids.add(file_id)
6754.8.18 by Jelmer Vernooĳ Get rid of lock_tree_write.	277	for file_id in file_ids:
	278	if self._inventory.has_id(file_id):
	279	self._inventory.remove_recursive_id(file_id)
1986.1.3 by Robert Collins Merge bzr.dev.	280
2418.5.1 by John Arbash Meinel Make a Branch helper which can create a very basic MemoryTree with history.	281	def set_parent_ids(self, revision_ids, allow_leftmost_as_ghost=False):
	282	"""See MutableTree.set_parent_trees()."""
2598.5.2 by Aaron Bentley Got all tests passing with Branch returning 'null:' for null revision	283	for revision_id in revision_ids:
	284	_mod_revision.check_not_reserved_id(revision_id)
2418.5.1 by John Arbash Meinel Make a Branch helper which can create a very basic MemoryTree with history.	285	if len(revision_ids) == 0:
	286	self._parent_ids = []
4190.1.4 by Robert Collins Cache ghosts when we can get them from a RemoteRepository in get_parent_map.	287	self._branch_revision_id = _mod_revision.NULL_REVISION
2418.5.1 by John Arbash Meinel Make a Branch helper which can create a very basic MemoryTree with history.	288	else:
	289	self._parent_ids = revision_ids
	290	self._branch_revision_id = revision_ids[0]
4190.1.4 by Robert Collins Cache ghosts when we can get them from a RemoteRepository in get_parent_map.	291	self._allow_leftmost_as_ghost = allow_leftmost_as_ghost
	292	self._set_basis()
6747.2.2 by Jelmer Vernooĳ Fix tests.	293
4190.1.4 by Robert Collins Cache ghosts when we can get them from a RemoteRepository in get_parent_map.	294	def _set_basis(self):
	295	try:
	296	self._basis_tree = self.branch.repository.revision_tree(
	297	self._branch_revision_id)
	298	except errors.NoSuchRevision:
	299	if self._allow_leftmost_as_ghost:
	300	self._basis_tree = self.branch.repository.revision_tree(
	301	_mod_revision.NULL_REVISION)
	302	else:
	303	raise
2418.5.1 by John Arbash Meinel Make a Branch helper which can create a very basic MemoryTree with history.	304
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	305	def set_parent_trees(self, parents_list, allow_leftmost_as_ghost=False):
	306	"""See MutableTree.set_parent_trees()."""
	307	if len(parents_list) == 0:
	308	self._parent_ids = []
3668.5.1 by Jelmer Vernooij Use NULL_REVISION rather than None for Repository.revision_tree().	309	self._basis_tree = self.branch.repository.revision_tree(
7143.15.2 by Jelmer Vernooĳ Run autopep8.	310	_mod_revision.NULL_REVISION)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	311	else:
	312	if parents_list[0][1] is None and not allow_leftmost_as_ghost:
	313	# a ghost in the left most parent
	314	raise errors.GhostRevisionUnusableHere(parents_list[0][0])
	315	self._parent_ids = [parent_id for parent_id, tree in parents_list]
6973.14.6 by Jelmer Vernooĳ Fix some more tests.	316	if parents_list[0][1] is None or parents_list[0][1] == b'null:':
3668.5.1 by Jelmer Vernooij Use NULL_REVISION rather than None for Repository.revision_tree().	317	self._basis_tree = self.branch.repository.revision_tree(
7143.15.2 by Jelmer Vernooĳ Run autopep8.	318	_mod_revision.NULL_REVISION)
1986.1.2 by Robert Collins Various changes to allow non-workingtree specific tests to run entirely	319	else:
	320	self._basis_tree = parents_list[0][1]
1986.1.6 by Robert Collins Add MemoryTree.last_revision.	321	self._branch_revision_id = parents_list[0][0]