gh-117786: Fix venv created from Windows Store install by restoring __PYVENV_LAUNCHER__ smuggling (GH-117814)

This commit is contained in:
Steve Dower 2024-04-24 23:00:55 +01:00 committed by GitHub
parent 8942bf41da
commit 4b10e209c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 21 additions and 12 deletions

View File

@ -747,6 +747,9 @@ class InitConfigTests(EmbeddingTestsMixin, unittest.TestCase):
if value is self.IGNORE_CONFIG:
config.pop(key, None)
del expected[key]
# Resolve bool/int mismatches to reduce noise in diffs
if isinstance(value, (bool, int)) and isinstance(config.get(key), (bool, int)):
expected[key] = type(config[key])(expected[key])
self.assertEqual(config, expected)
def check_global_config(self, configs):

View File

@ -0,0 +1,2 @@
Fixes virtual environments not correctly launching when created from a Store
install.

View File

@ -310,7 +310,10 @@ if ENV_PYTHONEXECUTABLE or ENV___PYVENV_LAUNCHER__:
# and should not affect base_executable.
base_executable = f"{dirname(library)}/bin/python{VERSION_MAJOR}.{VERSION_MINOR}"
else:
base_executable = executable
# Use the real executable as our base, or argv[0] otherwise
# (on Windows, argv[0] is likely to be ENV___PYVENV_LAUNCHER__; on
# other platforms, real_executable is likely to be empty)
base_executable = real_executable or executable
if not real_executable:
real_executable = base_executable
@ -408,13 +411,14 @@ if not base_executable:
if not real_executable:
real_executable = base_executable
try:
real_executable = realpath(real_executable)
except OSError as ex:
# Only warn if the file actually exists and was unresolvable
# Otherwise users who specify a fake executable may get spurious warnings.
if isfile(real_executable):
warn(f'Failed to find real location of {base_executable}')
if real_executable:
try:
real_executable = realpath(real_executable)
except OSError as ex:
# Only warn if the file actually exists and was unresolvable
# Otherwise users who specify a fake executable may get spurious warnings.
if isfile(real_executable):
warn(f'Failed to find real location of {base_executable}')
if not executable_dir and os_name == 'darwin' and library:
# QUIRK: macOS checks adjacent to its library early
@ -427,12 +431,12 @@ if not executable_dir and os_name == 'darwin' and library:
# If we do not have the executable's directory, we can calculate it.
# This is the directory used to find prefix/exec_prefix if necessary.
if not executable_dir:
if not executable_dir and real_executable:
executable_dir = real_executable_dir = dirname(real_executable)
# If we do not have the real executable's directory, we calculate it.
# This is the directory used to detect build layouts.
if not real_executable_dir:
if not real_executable_dir and real_executable:
real_executable_dir = dirname(real_executable)
# ******************************************************************************

View File

@ -484,8 +484,8 @@ process(int argc, wchar_t ** argv)
// We do not update argv[0] to point at the target runtime, and so we do not
// pass through our original argv[0] in an environment variable.
//exitCode = smuggle_path();
//if (exitCode) return exitCode;
exitCode = smuggle_path();
if (exitCode) return exitCode;
exitCode = launch(home_path, GetCommandLineW());
return exitCode;