pbs-installer 2024.10.2__py3-none-any.whl → 2024.10.10__py3-none-any.whl

pbs_installer/_install.py

@@ -10,11 +10,11 @@ from urllib.parse import unquote
 from ._utils import PythonVersion, get_arch_platform
 
 if TYPE_CHECKING:
+    from typing import Literal
+
     import httpx
     from _typeshed import StrPath
 
-    from typing import Literal
-
     PythonImplementation = Literal["cpython", "pypy"]
 
 logger = logging.getLogger(__name__)
@@ -38,6 +38,7 @@ def get_download_link(
     platform: str = THIS_PLATFORM,
     implementation: PythonImplementation = "cpython",
     build_dir: bool = False,
+    free_threaded: bool = False,
 ) -> tuple[PythonVersion, PythonFile]:
     """Get the download URL matching the given requested version.
 
@@ -47,6 +48,7 @@ def get_download_link(
         platform: The platform to install, e.g. linux, macos
         implementation: The implementation of Python to install, allowed values are 'cpython' and 'pypy'
         build_dir: Whether to include the `build/` directory from indygreg builds
+        free_threaded: Whether to install the freethreaded version of Python
 
     Returns:
         A tuple of the PythonVersion and the download URL
@@ -62,10 +64,16 @@ def get_download_link(
         if not py_ver.matches(request, implementation):
             continue
 
-        matched = urls.get((platform, arch, not build_dir))
+        if request.endswith("t"):
+            free_threaded = True
+
+        matched = urls.get((platform, arch, not build_dir, free_threaded))
         if matched is not None:
             return py_ver, matched
-        if build_dir and (matched := urls.get((platform, arch, False))) is not None:
+        if (
+            not build_dir
+            and (matched := urls.get((platform, arch, False, free_threaded))) is not None
+        ):
             return py_ver, matched
     raise ValueError(
         f"Could not find a version matching version={request!r}, implementation={implementation}"
@@ -162,6 +170,7 @@ def install(
     platform: str | None = None,
     implementation: PythonImplementation = "cpython",
     build_dir: bool = False,
+    free_threaded: bool = False,
 ) -> None:
     """Download and install the requested python version.
 
@@ -177,7 +186,7 @@ def install(
         platform: The platform to install, e.g. linux, macos
         implementation: The implementation of Python to install, allowed values are 'cpython' and 'pypy'
         build_dir: Whether to include the `build/` directory from indygreg builds
-
+        free_threaded: Whether to install the freethreaded version of Python
     Examples:
         >>> install("3.10", "./python")
         Installing cpython@3.10.4 to ./python
@@ -190,7 +199,12 @@ def install(
         arch = THIS_ARCH
 
     ver, python_file = get_download_link(
-        request, arch=arch, platform=platform, implementation=implementation, build_dir=build_dir
+        request,
+        arch=arch,
+        platform=platform,
+        implementation=implementation,
+        build_dir=build_dir,
+        free_threaded=free_threaded,
     )
     if version_dir:
         destination = os.path.join(destination, str(ver))

pbs_installer/_utils.py

@@ -27,7 +27,7 @@ class PythonVersion(NamedTuple):
         if implementation != self.implementation:
             return False
         try:
-            parts = tuple(int(v) for v in request.split("."))
+            parts = tuple(int(v) for v in request.rstrip("t").split("."))
         except ValueError:
             raise ValueError(
                 f"Invalid version: {request!r}, each part must be an integer"