Merge typeshed's setuptools._distutils annotations

Author: Avasam

distutils/_modified.py

@@ -2,20 +2,32 @@
 
 import functools
 import os.path
+from collections.abc import Callable, Iterable
+from typing import Literal, TypeVar
 
 from jaraco.functools import splat
 
 from .compat.py39 import zip_strict
 from .errors import DistutilsFileError
 
+_SourcesT = TypeVar(
+    "_SourcesT", bound=str | bytes | os.PathLike[str] | os.PathLike[bytes]
+)
+_TargetsT = TypeVar(
+    "_TargetsT", bound=str | bytes | os.PathLike[str] | os.PathLike[bytes]
+)
+
 
 def _newer(source, target):
     return not os.path.exists(target) or (
         os.path.getmtime(source) > os.path.getmtime(target)
     )
 
 
-def newer(source, target):
+def newer(
+    source: str | bytes | os.PathLike[str] | os.PathLike[bytes],
+    target: str | bytes | os.PathLike[str] | os.PathLike[bytes],
+) -> bool:
     """
     Is source modified more recently than target.
 
@@ -25,12 +37,16 @@ def newer(source, target):
     Raises DistutilsFileError if 'source' does not exist.
     """
     if not os.path.exists(source):
-        raise DistutilsFileError(f"file '{os.path.abspath(source)}' does not exist")
+        raise DistutilsFileError(f"file {os.path.abspath(source)!r} does not exist")
 
     return _newer(source, target)
 
 
-def newer_pairwise(sources, targets, newer=newer):
+def newer_pairwise(
+    sources: Iterable[_SourcesT],
+    targets: Iterable[_TargetsT],
+    newer: Callable[[_SourcesT, _TargetsT], bool] = newer,
+) -> tuple[list[_SourcesT], list[_TargetsT]]:
     """
     Filter filenames where sources are newer than targets.
 
@@ -43,7 +59,11 @@ def newer_pairwise(sources, targets, newer=newer):
     return tuple(map(list, zip(*newer_pairs))) or ([], [])
 
 
-def newer_group(sources, target, missing='error'):
+def newer_group(
+    sources: Iterable[str | bytes | os.PathLike[str] | os.PathLike[bytes]],
+    target: str | bytes | os.PathLike[str] | os.PathLike[bytes],
+    missing: Literal["error", "ignore", "newer"] = "error",
+) -> bool:
     """
     Is target out-of-date with respect to any file in sources.
 

distutils/archive_util.py

@@ -4,6 +4,7 @@
 that sort of thing)."""
 
 import os
+from typing import Literal
 
 try:
     import zipfile
@@ -54,14 +55,14 @@ def _get_uid(name):
 
 
 def make_tarball(
-    base_name,
-    base_dir,
-    compress="gzip",
-    verbose=False,
-    dry_run=False,
-    owner=None,
-    group=None,
-):
+    base_name: str,
+    base_dir: str | os.PathLike[str],
+    compress: Literal["gzip", "bzip2", "xz"] | None = "gzip",
+    verbose: bool = False,
+    dry_run: bool = False,
+    owner: str | None = None,
+    group: str | None = None,
+) -> str:
     """Create a (possibly compressed) tar file from all the files under
     'base_dir'.
 
@@ -122,7 +123,12 @@ def _set_uid_gid(tarinfo):
     return archive_name
 
 
-def make_zipfile(base_name, base_dir, verbose=False, dry_run=False):  # noqa: C901
+def make_zipfile(
+    base_name: str,
+    base_dir: str | os.PathLike[str],
+    verbose: bool = False,
+    dry_run: bool = False,
+) -> str:  # noqa: C901
     """Create a zip file from all the files under 'base_dir'.
 
     The output zip file will be named 'base_name' + ".zip".  Uses either the
@@ -205,15 +211,15 @@ def check_archive_formats(formats):
 
 
 def make_archive(
-    base_name,
-    format,
-    root_dir=None,
-    base_dir=None,
-    verbose=False,
-    dry_run=False,
-    owner=None,
-    group=None,
-):
+    base_name: 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).
 
     'base_name' is the name of the file to create, minus any format-specific