From f97653aa06cf84bcf160be3786b6fce49ef52961 Mon Sep 17 00:00:00 2001
From: Sebastian Thiel <byronimo@gmail.com>
Date: Mon, 15 Nov 2010 18:42:44 +0100
Subject: Repo: added submodule query and iteration methods similar to the ones
 provided for Remotes, including test

---
 lib/git/objects/util.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'lib/git/objects/util.py')

diff --git a/lib/git/objects/util.py b/lib/git/objects/util.py
index 21833080..9a54e031 100644
--- a/lib/git/objects/util.py
+++ b/lib/git/objects/util.py
@@ -4,6 +4,8 @@
 # This module is part of GitPython and is released under
 # the BSD License: http://www.opensource.org/licenses/bsd-license.php
 """Module for general utility functions"""
+from git.util import IterableList
+
 import re
 from collections import deque as Deque
 import platform
@@ -273,6 +275,12 @@ class Traversable(object):
 		"""
 		raise NotImplementedError("To be implemented in subclass")
 			
+	def list_traverse(self, *args, **kwargs):
+		""":return: IterableList with the results of the traversal as produced by
+			traverse()"""
+		out = IterableList(self._id_attribute_)
+		out.extend(self.traverse(*args, **kwargs))
+		return out
 	
 	def traverse( self, predicate = lambda i,d: True,
 						   prune = lambda i,d: False, depth = -1, branch_first=True,
-- 
cgit v1.2.1