Merge branch 'master' into typecheck-part-1

Author: DinhHuy2010

pygit2/__init__.py

@@ -23,14 +23,15 @@
 # the Free Software Foundation, 51 Franklin Street, Fifth Floor,
 # Boston, MA 02110-1301, USA.
 
+# ruff: noqa: F401 F403 F405
+
 # Standard Library
-import functools  # noqa: I001
-from os import PathLike
+import functools
+import os
 import typing
 
 # Low level API
-from ._pygit2 import *  # noqa: F403
-from ._pygit2 import _cache_enums
+from ._pygit2 import *
 
 # High level API
 from . import enums
@@ -65,8 +66,9 @@
 _cache_enums()
 
 
+
 def init_repository(  # noqa: PLR0913
-    path: typing.Union[str, bytes, PathLike, None],
+    path: typing.Union[str, bytes, os.PathLike, None],
     bare: bool = False,
     flags: enums.RepositoryInitFlag = enums.RepositoryInitFlag.MKPATH,
     mode: typing.Union[

pygit2/ffi.py

@@ -24,4 +24,5 @@
 # Boston, MA 02110-1301, USA.
 
 # Import from pygit2
+
 from ._libgit2 import ffi, lib as C  # noqa: I001, F401

pygit2/remotes.py

@@ -237,7 +237,7 @@ def push_refspecs(self):
         check_error(err)
         return strarray_to_strings(specs)
 
-    def push(self, specs, callbacks=None, proxy=None, push_options=None):
+    def push(self, specs, callbacks=None, proxy=None, push_options=None, threads=1):
         """
         Push the given refspec to the remote. Raises ``GitError`` on protocol
         error or unpack failure.
@@ -263,9 +263,19 @@ def push(self, specs, callbacks=None, proxy=None, push_options=None):
         push_options : [str]
             Push options to send to the server, which passes them to the
             pre-receive as well as the post-receive hook.
+
+        threads : int
+            If the transport being used to push to the remote requires the
+            creation of a pack file, this controls the number of worker threads
+            used by the packbuilder when creating that pack file to be sent to
+            the remote.
+
+            If set to 0, the packbuilder will auto-detect the number of threads
+            to create. The default value is 1.
         """
         with git_push_options(callbacks) as payload:
             opts = payload.push_options
+            opts.pb_parallelism = threads
             self.__set_proxy(opts.proxy_opts, proxy)
             with StrArray(specs) as refspecs, StrArray(push_options) as pushopts:
                 pushopts.assign_to(opts.remote_push_options)

test/test_remote.py

@@ -466,3 +466,19 @@ def test_push_options(origin, clone, remote):  # noqa: ARG001
     remote_push_options = callbacks.push_options.remote_push_options
     assert remote_push_options.count == 2  # noqa: PLR2004
     # strings pointed to by remote_push_options.strings[] are already freed
+
+
+def test_push_threads(origin, clone, remote):
+    from pygit2 import RemoteCallbacks
+
+    callbacks = RemoteCallbacks()
+    remote.push(['refs/heads/master'], callbacks)
+    assert callbacks.push_options.pb_parallelism == 1
+
+    callbacks = RemoteCallbacks()
+    remote.push(['refs/heads/master'], callbacks, threads=0)
+    assert callbacks.push_options.pb_parallelism == 0
+
+    callbacks = RemoteCallbacks()
+    remote.push(['refs/heads/master'], callbacks, threads=1)
+    assert callbacks.push_options.pb_parallelism == 1