Extend ObjectIterator a bit.

diff --git a/dulwich/object_store.py b/dulwich/object_store.py
index dc282bbf23d3d81ab89fe5df4fa7105ef2ff0709..71d605ddde0c86efae95fc21a6af575e0c30773b 100644 (file)
--- a/dulwich/object_store.py
+++ b/dulwich/object_store.py
@@ -197,5 +197,19 @@ class ObjectIterator(object):
     def __iter__(self):
         return ((self.store.get_object(sha), path) for sha, path in self.shas)
 
+    def iterobjects(self):
+        for o, path in self:
+            yield o
+
+    def __contains__(self, needle):
+        # FIXME: This could be more efficient
+        for sha, path in self.shas:
+            if sha == needle:
+                return True
+        return False
+
+    def __getitem__(self, key):
+        return self.store[key]
+
     def __len__(self):
         return len(self.shas)