1 files changed, 89 insertions, 0 deletions
diff --git a/lib/git/tree.py b/lib/git/tree.py
new file mode 100644
index 00000000..9c4dab1d
--- /dev/null
+++ b/lib/git/tree.py
@@ -0,0 +1,89 @@
+import os
+from lazy import LazyMixin
+import blob
+
+class Tree(LazyMixin):
+    def __init__(self, repo, **kwargs):
+        LazyMixin.__init__(self)
+        self.repo = repo
+        self.id = None
+        self.mode = None
+        self.name = None
+        self.contents = None
+
+        for k, v in kwargs.items():
+            setattr(self, k, v)
+
+    def __bake__(self):
+        temp = Tree.construct(self.repo, self.id)
+        self.contents = temp.contents
+
+    @classmethod
+    def construct(cls, repo, treeish, paths = []):
+        output = repo.git.ls_tree(treeish, *paths)
+        return Tree(repo, **{'id': treeish}).construct_initialize(repo, treeish, output)
+
+    def construct_initialize(self, repo, id, text):
+        self.repo = repo
+        self.id = id
+        self.contents = []
+        self.__baked__ = False
+
+        for line in text.splitlines():
+            self.contents.append(self.content_from_string(self.repo, line))
+
+        self.contents = [c for c in self.contents if c is not None]
+
+        self.__bake_it__()
+        return self
+
+    def content_from_string(self, repo, text):
+        """
+        Parse a content item and create the appropriate object
+
+        ``repo``
+            is the Repo
+
+         ``text``
+            is the single line containing the items data in `git ls-tree` format
+
+        Returns
+            ``GitPython.Blob`` or ``GitPython.Tree``
+        """
+        try:
+            mode, typ, id, name = text.expandtabs(1).split(" ", 4)
+        except:
+            return None
+
+        if typ == "tree":
+            return Tree(repo, **{'id': id, 'mode': mode, 'name': name})
+        elif typ == "blob":
+            return blob.Blob(repo, **{'id': id, 'mode': mode, 'name': name})
+        elif typ == "commit":
+            return None
+        else:
+          raise(TypeError, "Invalid type: %s" % typ)
+
+    def __div__(self, file):
+        """
+        Find the named object in this tree's contents
+
+        Examples::
+
+            >>> Repo('/path/to/python-git').tree/'lib'
+            <GitPython.Tree "6cc23ee138be09ff8c28b07162720018b244e95e">
+            >>> Repo('/path/to/python-git').tree/'README.txt'
+            <GitPython.Blob "8b1e02c0fb554eed2ce2ef737a68bb369d7527df">
+
+        Returns
+            ``GitPython.Blob`` or ``GitPython.Tree`` or ``None`` if not found
+        """
+        contents = [c for c in self.contents if c.name == file]
+        return contents and contents[0] or None
+
+    @property
+    def basename(self):
+        os.path.basename(self.name)
+
+    def __repr__(self):
+        return '<GitPython.Tree "%s">' % self.id