Преглед на файлове

update existing determine_wants callbacks to accept depth kwarg

Peter Rowlands преди 4 години
родител
ревизия
a56e0d8a39
променени са 3 файла, в които са добавени 5 реда и са изтрити 5 реда
  1. 1 1
      dulwich/cli.py
  2. 1 1
      dulwich/porcelain.py
  3. 3 3
      dulwich/server.py

+ 1 - 1
dulwich/cli.py

@@ -116,7 +116,7 @@ class cmd_fetch_pack(Command):
             determine_wants = r.object_store.determine_wants_all
         else:
 
-            def determine_wants(x):
+            def determine_wants(x, **kwargs):
                 return [y for y in args if y not in r.object_store]
 
         client.fetch(path, r, determine_wants)

+ 1 - 1
dulwich/porcelain.py

@@ -1154,7 +1154,7 @@ def pull(
             refspecs = [b"HEAD"]
         selected_refs = []
 
-        def determine_wants(remote_refs):
+        def determine_wants(remote_refs, **kwargs):
             selected_refs.extend(
                 parse_reftuples(remote_refs, r.refs, refspecs, force=force)
             )

+ 3 - 3
dulwich/server.py

@@ -372,8 +372,8 @@ class UploadPackHandler(PackHandler):
         )
         wants = []
 
-        def wants_wrapper(refs):
-            wants.extend(graph_walker.determine_wants(refs))
+        def wants_wrapper(refs, **kwargs):
+            wants.extend(graph_walker.determine_wants(refs, **kwargs))
             return wants
 
         objects_iter = self.repo.fetch_objects(
@@ -573,7 +573,7 @@ class _ProtocolGraphWalker(object):
         self._cache_index = 0
         self._impl = None
 
-    def determine_wants(self, heads):
+    def determine_wants(self, heads, depth=None):
         """Determine the wants for a set of heads.
 
         The given heads are advertised to the client, who then specifies which