Jelajahi Sumber

Support progress argument to generate_pack_contents.

Jelmer Vernooij 15 tahun lalu
induk
melakukan
d253d5a499
1 mengubah file dengan 3 tambahan dan 2 penghapusan
  1. 3 2
      dulwich/object_store.py

+ 3 - 2
dulwich/object_store.py

@@ -225,13 +225,14 @@ class BaseObjectStore(object):
         """
         return ObjectStoreGraphWalker(heads, lambda sha: self[sha].parents)
 
-    def generate_pack_contents(self, have, want):
+    def generate_pack_contents(self, have, want, progress=None):
         """Iterate over the contents of a pack file.
 
         :param have: List of SHA1s of objects that should not be sent
         :param want: List of SHA1s of objects that should be sent
+        :param progress: Optional progress reporting method
         """
-        return self.iter_shas(self.find_missing_objects(have, want))
+        return self.iter_shas(self.find_missing_objects(have, want, progress))
 
 
 class PackBasedObjectStore(BaseObjectStore):