소스 검색

Extend ObjectIterator a bit.

Jelmer Vernooij 16 년 전
부모
커밋
28980f1f1c
1개의 변경된 파일14개의 추가작업 그리고 0개의 파일을 삭제
  1. 14 0
      dulwich/object_store.py

+ 14 - 0
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)