fix support of free-threaded Python if Poetry itself is not installed with a free-threaded Python

Author: radoering

src/poetry/utils/env/base_env.py

@@ -51,6 +51,7 @@ class MarkerEnv(TypedDict):
     interpreter_name: str
     interpreter_version: str
     sysconfig_platform: str
+    free_threading: bool
 
 
 class Env(ABC):

src/poetry/utils/env/script_strings.py

@@ -86,6 +86,7 @@ def _version_nodot(version):
     ),
     "interpreter_version": interpreter_version(),
     "sysconfig_platform": sysconfig.get_platform(),
+    "free_threading": sysconfig.get_config_var("Py_GIL_DISABLED") or False,
 }
 
 print(json.dumps(env))

src/poetry/utils/env/system_env.py

@@ -74,6 +74,7 @@ def get_marker_env(self) -> MarkerEnv:
             "interpreter_name": interpreter_name(),
             "interpreter_version": interpreter_version(),
             "sysconfig_platform": sysconfig.get_platform(),
+            "free_threading": sysconfig.get_config_var("Py_GIL_DISABLED") or False,
         }
 
     def is_venv(self) -> bool:

src/poetry/utils/env/virtual_env.py

@@ -58,11 +58,15 @@ def get_supported_tags(self) -> list[Tag]:
         interpreter_name = self.marker_env["interpreter_name"]
         interpreter_version = self.marker_env["interpreter_version"]
         sysconfig_platform = self.marker_env["sysconfig_platform"]
+        free_threading = self.marker_env["free_threading"]
 
+        abis: list[str] | None = None
         if interpreter_name == "pp":
             interpreter = "pp3"
         elif interpreter_name == "cp":
             interpreter = f"{interpreter_name}{interpreter_version}"
+            if free_threading:
+                abis = [f"{interpreter}t"]
         else:
             interpreter = None
 
@@ -83,7 +87,7 @@ def get_supported_tags(self) -> list[Tag]:
 
         return [
             *(
-                cpython_tags(python, platforms=platforms)
+                cpython_tags(python, abis=abis, platforms=platforms)
                 if interpreter_name == "cp"
                 else generic_tags(platforms=platforms)
             ),

tests/utils/env/test_env.py

@@ -116,6 +116,25 @@ def test_env_get_supported_tags_matches_inside_virtualenv(
     assert run_python_script_spy.call_count == expected_call_count
 
 
+@pytest.mark.skipif(
+    sys.implementation.name != "cpython",
+    reason="free threading is only relevant for CPython",
+)
+def test_env_get_supported_tags_free_threading(
+    tmp_path: Path, manager: EnvManager
+) -> None:
+    venv_path = tmp_path / "Virtual Env"
+    manager.build_venv(venv_path)
+    venv = VirtualEnv(venv_path)
+
+    if venv.marker_env["free_threading"]:
+        assert venv.get_supported_tags() == list(packaging.tags.sys_tags())
+    else:
+        assert not any(t.abi.endswith("t") for t in venv.get_supported_tags())
+        venv.marker_env["free_threading"] = True
+        assert any(t.abi.endswith("t") for t in venv.get_supported_tags())
+
+
 @pytest.mark.skipif(os.name == "nt", reason="Symlinks are not support for Windows")
 def test_env_has_symlinks_on_nix(tmp_path: Path, tmp_venv: VirtualEnv) -> None:
     assert os.path.islink(tmp_venv.python)