Add make_archive overloads

Author: Avasam

distutils/archive_util.py

@@ -4,7 +4,7 @@
 that sort of thing)."""
 
 import os
-from typing import Literal
+from typing import Literal, overload
 
 try:
     import zipfile
@@ -128,7 +128,7 @@ def make_zipfile(
     base_dir: str | os.PathLike[str],
     verbose: bool = False,
     dry_run: bool = False,
-) -> str:  # noqa: C901
+) -> str:
     """Create a zip file from all the files under 'base_dir'.
 
     The output zip file will be named 'base_name' + ".zip".  Uses either the
@@ -210,6 +210,7 @@ def check_archive_formats(formats):
     return None
 
 
+@overload
 def make_archive(
     base_name: str,
     format: str,
@@ -219,6 +220,27 @@ def make_archive(
     dry_run: bool = False,
     owner: str | None = None,
     group: str | None = None,
+) -> str: ...
+@overload
+def make_archive(
+    base_name: str | os.PathLike[str],
+    format: str,
+    root_dir: str | os.PathLike[str] | bytes | os.PathLike[bytes],
+    base_dir: str | None = None,
+    verbose: bool = False,
+    dry_run: bool = False,
+    owner: str | None = None,
+    group: str | None = None,
+) -> str: ...
+def make_archive(
+    base_name: str | os.PathLike[str],
+    format: str,
+    root_dir: str | os.PathLike[str] | bytes | os.PathLike[bytes] | None = None,
+    base_dir: str | None = None,
+    verbose: bool = False,
+    dry_run: bool = False,
+    owner: str | None = None,
+    group: str | None = None,
 ) -> str:
     """Create an archive file (eg. zip or tar).
 

distutils/cmd.py

@@ -475,6 +475,7 @@ def spawn(
 
         spawn(cmd, search_path, dry_run=self.dry_run)
 
+    @overload
     def make_archive(
         self,
         base_name: str,
@@ -483,6 +484,25 @@ def make_archive(
         base_dir: str | None = None,
         owner: str | None = None,
         group: str | None = None,
+    ) -> str: ...
+    @overload
+    def make_archive(
+        self,
+        base_name: str | os.PathLike[str],
+        format: str,
+        root_dir: str | os.PathLike[str] | bytes | os.PathLike[bytes],
+        base_dir: str | None = None,
+        owner: str | None = None,
+        group: str | None = None,
+    ) -> str: ...
+    def make_archive(
+        self,
+        base_name: str | os.PathLike[str],
+        format: str,
+        root_dir: str | os.PathLike[str] | bytes | os.PathLike[bytes] | None = None,
+        base_dir: str | None = None,
+        owner: str | None = None,
+        group: str | None = None,
     ) -> str:
         return archive_util.make_archive(
             base_name,