mirror of
https://github.com/frappe/frappe_docker.git
synced 2026-06-19 06:35:10 +00:00
280 lines
8.4 KiB
Python
280 lines
8.4 KiB
Python
# tree.py
|
|
# Copyright (C) 2008, 2009 Michael Trier (mtrier@gmail.com) and contributors
|
|
#
|
|
# This module is part of GitPython and is released under
|
|
# the BSD License: http://www.opensource.org/licenses/bsd-license.php
|
|
import util
|
|
from base import IndexObject
|
|
from git.util import join_path
|
|
from blob import Blob
|
|
from submodule.base import Submodule
|
|
import git.diff as diff
|
|
|
|
from fun import (
|
|
tree_entries_from_data,
|
|
tree_to_stream
|
|
)
|
|
|
|
from gitdb.util import (
|
|
to_bin_sha,
|
|
)
|
|
|
|
__all__ = ("TreeModifier", "Tree")
|
|
|
|
class TreeModifier(object):
|
|
"""A utility class providing methods to alter the underlying cache in a list-like fashion.
|
|
|
|
Once all adjustments are complete, the _cache, which really is a refernce to
|
|
the cache of a tree, will be sorted. Assuring it will be in a serializable state"""
|
|
__slots__ = '_cache'
|
|
|
|
def __init__(self, cache):
|
|
self._cache = cache
|
|
|
|
def _index_by_name(self, name):
|
|
""":return: index of an item with name, or -1 if not found"""
|
|
for i, t in enumerate(self._cache):
|
|
if t[2] == name:
|
|
return i
|
|
# END found item
|
|
# END for each item in cache
|
|
return -1
|
|
|
|
#{ Interface
|
|
def set_done(self):
|
|
"""Call this method once you are done modifying the tree information.
|
|
It may be called several times, but be aware that each call will cause
|
|
a sort operation
|
|
:return self:"""
|
|
self._cache.sort(key=lambda t: t[2]) # sort by name
|
|
return self
|
|
#} END interface
|
|
|
|
#{ Mutators
|
|
def add(self, sha, mode, name, force=False):
|
|
"""Add the given item to the tree. If an item with the given name already
|
|
exists, nothing will be done, but a ValueError will be raised if the
|
|
sha and mode of the existing item do not match the one you add, unless
|
|
force is True
|
|
|
|
:param sha: The 20 or 40 byte sha of the item to add
|
|
:param mode: int representing the stat compatible mode of the item
|
|
:param force: If True, an item with your name and information will overwrite
|
|
any existing item with the same name, no matter which information it has
|
|
:return: self"""
|
|
if '/' in name:
|
|
raise ValueError("Name must not contain '/' characters")
|
|
if (mode >> 12) not in Tree._map_id_to_type:
|
|
raise ValueError("Invalid object type according to mode %o" % mode)
|
|
|
|
sha = to_bin_sha(sha)
|
|
index = self._index_by_name(name)
|
|
item = (sha, mode, name)
|
|
if index == -1:
|
|
self._cache.append(item)
|
|
else:
|
|
if force:
|
|
self._cache[index] = item
|
|
else:
|
|
ex_item = self._cache[index]
|
|
if ex_item[0] != sha or ex_item[1] != mode:
|
|
raise ValueError("Item %r existed with different properties" % name)
|
|
# END handle mismatch
|
|
# END handle force
|
|
# END handle name exists
|
|
return self
|
|
|
|
def add_unchecked(self, binsha, mode, name):
|
|
"""Add the given item to the tree, its correctness is assumed, which
|
|
puts the caller into responsibility to assure the input is correct.
|
|
For more information on the parameters, see ``add``
|
|
:param binsha: 20 byte binary sha"""
|
|
self._cache.append((binsha, mode, name))
|
|
|
|
def __delitem__(self, name):
|
|
"""Deletes an item with the given name if it exists"""
|
|
index = self._index_by_name(name)
|
|
if index > -1:
|
|
del(self._cache[index])
|
|
|
|
#} END mutators
|
|
|
|
|
|
class Tree(IndexObject, diff.Diffable, util.Traversable, util.Serializable):
|
|
"""Tree objects represent an ordered list of Blobs and other Trees.
|
|
|
|
``Tree as a list``::
|
|
|
|
Access a specific blob using the
|
|
tree['filename'] notation.
|
|
|
|
You may as well access by index
|
|
blob = tree[0]
|
|
"""
|
|
|
|
type = "tree"
|
|
__slots__ = "_cache"
|
|
|
|
# actual integer ids for comparison
|
|
commit_id = 016 # equals stat.S_IFDIR | stat.S_IFLNK - a directory link
|
|
blob_id = 010
|
|
symlink_id = 012
|
|
tree_id = 004
|
|
|
|
_map_id_to_type = {
|
|
commit_id : Submodule,
|
|
blob_id : Blob,
|
|
symlink_id : Blob
|
|
# tree id added once Tree is defined
|
|
}
|
|
|
|
|
|
def __init__(self, repo, binsha, mode=tree_id<<12, path=None):
|
|
super(Tree, self).__init__(repo, binsha, mode, path)
|
|
|
|
@classmethod
|
|
def _get_intermediate_items(cls, index_object):
|
|
if index_object.type == "tree":
|
|
return tuple(index_object._iter_convert_to_object(index_object._cache))
|
|
return tuple()
|
|
|
|
def _set_cache_(self, attr):
|
|
if attr == "_cache":
|
|
# Set the data when we need it
|
|
ostream = self.repo.odb.stream(self.binsha)
|
|
self._cache = tree_entries_from_data(ostream.read())
|
|
else:
|
|
super(Tree, self)._set_cache_(attr)
|
|
# END handle attribute
|
|
|
|
def _iter_convert_to_object(self, iterable):
|
|
"""Iterable yields tuples of (binsha, mode, name), which will be converted
|
|
to the respective object representation"""
|
|
for binsha, mode, name in iterable:
|
|
path = join_path(self.path, name)
|
|
try:
|
|
yield self._map_id_to_type[mode >> 12](self.repo, binsha, mode, path)
|
|
except KeyError:
|
|
raise TypeError("Unknown mode %o found in tree data for path '%s'" % (mode, path))
|
|
# END for each item
|
|
|
|
def __div__(self, file):
|
|
"""Find the named object in this tree's contents
|
|
:return: ``git.Blob`` or ``git.Tree`` or ``git.Submodule``
|
|
|
|
:raise KeyError: if given file or tree does not exist in tree"""
|
|
msg = "Blob or Tree named %r not found"
|
|
if '/' in file:
|
|
tree = self
|
|
item = self
|
|
tokens = file.split('/')
|
|
for i,token in enumerate(tokens):
|
|
item = tree[token]
|
|
if item.type == 'tree':
|
|
tree = item
|
|
else:
|
|
# safety assertion - blobs are at the end of the path
|
|
if i != len(tokens)-1:
|
|
raise KeyError(msg % file)
|
|
return item
|
|
# END handle item type
|
|
# END for each token of split path
|
|
if item == self:
|
|
raise KeyError(msg % file)
|
|
return item
|
|
else:
|
|
for info in self._cache:
|
|
if info[2] == file: # [2] == name
|
|
return self._map_id_to_type[info[1] >> 12](self.repo, info[0], info[1], join_path(self.path, info[2]))
|
|
# END for each obj
|
|
raise KeyError( msg % file )
|
|
# END handle long paths
|
|
|
|
|
|
@property
|
|
def trees(self):
|
|
""":return: list(Tree, ...) list of trees directly below this tree"""
|
|
return [ i for i in self if i.type == "tree" ]
|
|
|
|
@property
|
|
def blobs(self):
|
|
""":return: list(Blob, ...) list of blobs directly below this tree"""
|
|
return [ i for i in self if i.type == "blob" ]
|
|
|
|
@property
|
|
def cache(self):
|
|
"""
|
|
:return: An object allowing to modify the internal cache. This can be used
|
|
to change the tree's contents. When done, make sure you call ``set_done``
|
|
on the tree modifier, or serialization behaviour will be incorrect.
|
|
See the ``TreeModifier`` for more information on how to alter the cache"""
|
|
return TreeModifier(self._cache)
|
|
|
|
def traverse( self, predicate = lambda i,d: True,
|
|
prune = lambda i,d: False, depth = -1, branch_first=True,
|
|
visit_once = False, ignore_self=1 ):
|
|
"""For documentation, see util.Traversable.traverse
|
|
Trees are set to visit_once = False to gain more performance in the traversal"""
|
|
return super(Tree, self).traverse(predicate, prune, depth, branch_first, visit_once, ignore_self)
|
|
|
|
# List protocol
|
|
def __getslice__(self, i, j):
|
|
return list(self._iter_convert_to_object(self._cache[i:j]))
|
|
|
|
def __iter__(self):
|
|
return self._iter_convert_to_object(self._cache)
|
|
|
|
def __len__(self):
|
|
return len(self._cache)
|
|
|
|
def __getitem__(self, item):
|
|
if isinstance(item, int):
|
|
info = self._cache[item]
|
|
return self._map_id_to_type[info[1] >> 12](self.repo, info[0], info[1], join_path(self.path, info[2]))
|
|
|
|
if isinstance(item, basestring):
|
|
# compatability
|
|
return self.__div__(item)
|
|
# END index is basestring
|
|
|
|
raise TypeError( "Invalid index type: %r" % item )
|
|
|
|
|
|
def __contains__(self, item):
|
|
if isinstance(item, IndexObject):
|
|
for info in self._cache:
|
|
if item.binsha == info[0]:
|
|
return True
|
|
# END compare sha
|
|
# END for each entry
|
|
# END handle item is index object
|
|
# compatability
|
|
|
|
# treat item as repo-relative path
|
|
path = self.path
|
|
for info in self._cache:
|
|
if item == join_path(path, info[2]):
|
|
return True
|
|
# END for each item
|
|
return False
|
|
|
|
def __reversed__(self):
|
|
return reversed(self._iter_convert_to_object(self._cache))
|
|
|
|
def _serialize(self, stream):
|
|
"""Serialize this tree into the stream. Please note that we will assume
|
|
our tree data to be in a sorted state. If this is not the case, serialization
|
|
will not generate a correct tree representation as these are assumed to be sorted
|
|
by algorithms"""
|
|
tree_to_stream(self._cache, stream.write)
|
|
return self
|
|
|
|
def _deserialize(self, stream):
|
|
self._cache = tree_entries_from_data(stream.read())
|
|
return self
|
|
|
|
|
|
# END tree
|
|
|
|
# finalize map definition
|
|
Tree._map_id_to_type[Tree.tree_id] = Tree
|