mirror of
https://github.com/python/cpython.git
synced 2025-07-08 03:45:36 +00:00
99fcf15052
The getpath.py file is frozen at build time and executed as code over a namespace. It is never imported, nor is it meant to be importable or reusable. However, it should be easier to read, modify, and patch than the previous code. This commit attempts to preserve every previously tested quirk, but these may be changed in the future to better align platforms.
879 lines
28 KiB
Python
879 lines
28 KiB
Python
import copy
|
|
import ntpath
|
|
import pathlib
|
|
import posixpath
|
|
import sys
|
|
import unittest
|
|
|
|
from test.support import verbose
|
|
|
|
try:
|
|
# If we are in a source tree, use the original source file for tests
|
|
SOURCE = (pathlib.Path(__file__).absolute().parent.parent.parent / "Modules/getpath.py").read_bytes()
|
|
except FileNotFoundError:
|
|
# Try from _testcapimodule instead
|
|
from _testinternalcapi import get_getpath_codeobject
|
|
SOURCE = get_getpath_codeobject()
|
|
|
|
|
|
class MockGetPathTests(unittest.TestCase):
|
|
def __init__(self, *a, **kw):
|
|
super().__init__(*a, **kw)
|
|
self.maxDiff = None
|
|
|
|
def test_normal_win32(self):
|
|
"Test a 'standard' install layout on Windows."
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\Python\python.exe",
|
|
real_executable=r"C:\Python\python.exe",
|
|
)
|
|
ns.add_known_xfile(r"C:\Python\python.exe")
|
|
ns.add_known_file(r"C:\Python\Lib\os.py")
|
|
ns.add_known_dir(r"C:\Python\DLLs")
|
|
expected = dict(
|
|
executable=r"C:\Python\python.exe",
|
|
base_executable=r"C:\Python\python.exe",
|
|
prefix=r"C:\Python",
|
|
exec_prefix=r"C:\Python",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\Python\python98.zip",
|
|
r"C:\Python\Lib",
|
|
r"C:\Python\DLLs",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_buildtree_win32(self):
|
|
"Test an in-build-tree layout on Windows."
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\CPython\PCbuild\amd64\python.exe",
|
|
real_executable=r"C:\CPython\PCbuild\amd64\python.exe",
|
|
)
|
|
ns.add_known_xfile(r"C:\CPython\PCbuild\amd64\python.exe")
|
|
ns.add_known_file(r"C:\CPython\Lib\os.py")
|
|
ns.add_known_file(r"C:\CPython\PCbuild\amd64\pybuilddir.txt", [""])
|
|
expected = dict(
|
|
executable=r"C:\CPython\PCbuild\amd64\python.exe",
|
|
base_executable=r"C:\CPython\PCbuild\amd64\python.exe",
|
|
prefix=r"C:\CPython",
|
|
exec_prefix=r"C:\CPython",
|
|
build_prefix=r"C:\CPython",
|
|
_is_python_build=1,
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\CPython\PCbuild\amd64\python98.zip",
|
|
r"C:\CPython\Lib",
|
|
r"C:\CPython\PCbuild\amd64",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_venv_win32(self):
|
|
"""Test a venv layout on Windows.
|
|
|
|
This layout is discovered by the presence of %__PYVENV_LAUNCHER__%,
|
|
specifying the original launcher executable. site.py is responsible
|
|
for updating prefix and exec_prefix.
|
|
"""
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\Python\python.exe",
|
|
ENV___PYVENV_LAUNCHER__=r"C:\venv\Scripts\python.exe",
|
|
real_executable=r"C:\Python\python.exe",
|
|
)
|
|
ns.add_known_xfile(r"C:\Python\python.exe")
|
|
ns.add_known_xfile(r"C:\venv\Scripts\python.exe")
|
|
ns.add_known_file(r"C:\Python\Lib\os.py")
|
|
ns.add_known_dir(r"C:\Python\DLLs")
|
|
ns.add_known_file(r"C:\venv\pyvenv.cfg", [
|
|
r"home = C:\Python"
|
|
])
|
|
expected = dict(
|
|
executable=r"C:\venv\Scripts\python.exe",
|
|
prefix=r"C:\Python",
|
|
exec_prefix=r"C:\Python",
|
|
base_executable=r"C:\Python\python.exe",
|
|
base_prefix=r"C:\Python",
|
|
base_exec_prefix=r"C:\Python",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\Python\python98.zip",
|
|
r"C:\Python\Lib",
|
|
r"C:\Python",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_registry_win32(self):
|
|
"""Test registry lookup on Windows.
|
|
|
|
On Windows there are registry entries that are intended for other
|
|
applications to register search paths.
|
|
"""
|
|
hkey = rf"HKLM\Software\Python\PythonCore\9.8-XY\PythonPath"
|
|
winreg = MockWinreg({
|
|
hkey: None,
|
|
f"{hkey}\\Path1": "path1-dir",
|
|
f"{hkey}\\Path1\\Subdir": "not-subdirs",
|
|
})
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\Python\python.exe",
|
|
real_executable=r"C:\Python\python.exe",
|
|
winreg=winreg,
|
|
)
|
|
ns.add_known_xfile(r"C:\Python\python.exe")
|
|
ns.add_known_file(r"C:\Python\Lib\os.py")
|
|
ns.add_known_dir(r"C:\Python\DLLs")
|
|
expected = dict(
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\Python\python98.zip",
|
|
"path1-dir",
|
|
# should not contain not-subdirs
|
|
r"C:\Python\Lib",
|
|
r"C:\Python\DLLs",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
ns["config"]["use_environment"] = 0
|
|
ns["config"]["module_search_paths_set"] = 0
|
|
ns["config"]["module_search_paths"] = None
|
|
expected = dict(
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\Python\python98.zip",
|
|
r"C:\Python\Lib",
|
|
r"C:\Python\DLLs",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_normal_win32(self):
|
|
"Test a 'standard' install layout via symlink on Windows."
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\LinkedFrom\python.exe",
|
|
real_executable=r"C:\Python\python.exe",
|
|
)
|
|
ns.add_known_xfile(r"C:\LinkedFrom\python.exe")
|
|
ns.add_known_xfile(r"C:\Python\python.exe")
|
|
ns.add_known_link(r"C:\LinkedFrom\python.exe", r"C:\Python\python.exe")
|
|
ns.add_known_file(r"C:\Python\Lib\os.py")
|
|
ns.add_known_dir(r"C:\Python\DLLs")
|
|
expected = dict(
|
|
executable=r"C:\LinkedFrom\python.exe",
|
|
base_executable=r"C:\LinkedFrom\python.exe",
|
|
prefix=r"C:\Python",
|
|
exec_prefix=r"C:\Python",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\Python\python98.zip",
|
|
r"C:\Python\Lib",
|
|
r"C:\Python\DLLs",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_buildtree_win32(self):
|
|
"Test an in-build-tree layout via symlink on Windows."
|
|
ns = MockNTNamespace(
|
|
argv0=r"C:\LinkedFrom\python.exe",
|
|
real_executable=r"C:\CPython\PCbuild\amd64\python.exe",
|
|
)
|
|
ns.add_known_xfile(r"C:\LinkedFrom\python.exe")
|
|
ns.add_known_xfile(r"C:\CPython\PCbuild\amd64\python.exe")
|
|
ns.add_known_link(r"C:\LinkedFrom\python.exe", r"C:\CPython\PCbuild\amd64\python.exe")
|
|
ns.add_known_file(r"C:\CPython\Lib\os.py")
|
|
ns.add_known_file(r"C:\CPython\PCbuild\amd64\pybuilddir.txt", [""])
|
|
expected = dict(
|
|
executable=r"C:\LinkedFrom\python.exe",
|
|
base_executable=r"C:\LinkedFrom\python.exe",
|
|
prefix=r"C:\CPython",
|
|
exec_prefix=r"C:\CPython",
|
|
build_prefix=r"C:\CPython",
|
|
_is_python_build=1,
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
r"C:\CPython\PCbuild\amd64\python98.zip",
|
|
r"C:\CPython\Lib",
|
|
r"C:\CPython\PCbuild\amd64",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_normal_posix(self):
|
|
"Test a 'standard' install layout on *nix"
|
|
ns = MockPosixNamespace(
|
|
PREFIX="/usr",
|
|
argv0="python",
|
|
ENV_PATH="/usr/bin",
|
|
)
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_file("/usr/lib/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib/python9.8/lib-dynload")
|
|
expected = dict(
|
|
executable="/usr/bin/python",
|
|
base_executable="/usr/bin/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib/python98.zip",
|
|
"/usr/lib/python9.8",
|
|
"/usr/lib/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_buildpath_posix(self):
|
|
"""Test an in-build-tree layout on POSIX.
|
|
|
|
This layout is discovered from the presence of pybuilddir.txt, which
|
|
contains the relative path from the executable's directory to the
|
|
platstdlib path.
|
|
"""
|
|
ns = MockPosixNamespace(
|
|
argv0=r"/home/cpython/python",
|
|
PREFIX="/usr/local",
|
|
)
|
|
ns.add_known_xfile("/home/cpython/python")
|
|
ns.add_known_xfile("/usr/local/bin/python")
|
|
ns.add_known_file("/home/cpython/pybuilddir.txt", ["build/lib.linux-x86_64-9.8"])
|
|
ns.add_known_file("/home/cpython/Lib/os.py")
|
|
ns.add_known_dir("/home/cpython/lib-dynload")
|
|
expected = dict(
|
|
executable="/home/cpython/python",
|
|
prefix="/usr/local",
|
|
exec_prefix="/usr/local",
|
|
base_executable="/home/cpython/python",
|
|
build_prefix="/home/cpython",
|
|
_is_python_build=1,
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/local/lib/python98.zip",
|
|
"/home/cpython/Lib",
|
|
"/home/cpython/build/lib.linux-x86_64-9.8",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_venv_posix(self):
|
|
"Test a venv layout on *nix."
|
|
ns = MockPosixNamespace(
|
|
argv0="python",
|
|
PREFIX="/usr",
|
|
ENV_PATH="/venv/bin:/usr/bin",
|
|
)
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_xfile("/venv/bin/python")
|
|
ns.add_known_file("/usr/lib/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib/python9.8/lib-dynload")
|
|
ns.add_known_file("/venv/pyvenv.cfg", [
|
|
r"home = /usr/bin"
|
|
])
|
|
expected = dict(
|
|
executable="/venv/bin/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
base_executable="/usr/bin/python",
|
|
base_prefix="/usr",
|
|
base_exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib/python98.zip",
|
|
"/usr/lib/python9.8",
|
|
"/usr/lib/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_normal_posix(self):
|
|
"Test a 'standard' install layout via symlink on *nix"
|
|
ns = MockPosixNamespace(
|
|
PREFIX="/usr",
|
|
argv0="/linkfrom/python",
|
|
)
|
|
ns.add_known_xfile("/linkfrom/python")
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_link("/linkfrom/python", "/usr/bin/python")
|
|
ns.add_known_file("/usr/lib/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib/python9.8/lib-dynload")
|
|
expected = dict(
|
|
executable="/linkfrom/python",
|
|
base_executable="/linkfrom/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib/python98.zip",
|
|
"/usr/lib/python9.8",
|
|
"/usr/lib/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_buildpath_posix(self):
|
|
"""Test an in-build-tree layout on POSIX.
|
|
|
|
This layout is discovered from the presence of pybuilddir.txt, which
|
|
contains the relative path from the executable's directory to the
|
|
platstdlib path.
|
|
"""
|
|
ns = MockPosixNamespace(
|
|
argv0=r"/linkfrom/python",
|
|
PREFIX="/usr/local",
|
|
)
|
|
ns.add_known_xfile("/linkfrom/python")
|
|
ns.add_known_xfile("/home/cpython/python")
|
|
ns.add_known_link("/linkfrom/python", "/home/cpython/python")
|
|
ns.add_known_xfile("/usr/local/bin/python")
|
|
ns.add_known_file("/home/cpython/pybuilddir.txt", ["build/lib.linux-x86_64-9.8"])
|
|
ns.add_known_file("/home/cpython/Lib/os.py")
|
|
ns.add_known_dir("/home/cpython/lib-dynload")
|
|
expected = dict(
|
|
executable="/linkfrom/python",
|
|
prefix="/usr/local",
|
|
exec_prefix="/usr/local",
|
|
base_executable="/linkfrom/python",
|
|
build_prefix="/home/cpython",
|
|
_is_python_build=1,
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/local/lib/python98.zip",
|
|
"/home/cpython/Lib",
|
|
"/home/cpython/build/lib.linux-x86_64-9.8",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_custom_platlibdir_posix(self):
|
|
"Test an install with custom platlibdir on *nix"
|
|
ns = MockPosixNamespace(
|
|
PREFIX="/usr",
|
|
argv0="/linkfrom/python",
|
|
PLATLIBDIR="lib64",
|
|
)
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_file("/usr/lib64/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib64/python9.8/lib-dynload")
|
|
expected = dict(
|
|
executable="/linkfrom/python",
|
|
base_executable="/linkfrom/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib64/python98.zip",
|
|
"/usr/lib64/python9.8",
|
|
"/usr/lib64/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_venv_macos(self):
|
|
"""Test a venv layout on macOS.
|
|
|
|
This layout is discovered when 'executable' and 'real_executable' match,
|
|
but $__PYVENV_LAUNCHER__ has been set to the original process.
|
|
"""
|
|
ns = MockPosixNamespace(
|
|
os_name="darwin",
|
|
argv0="/usr/bin/python",
|
|
PREFIX="/usr",
|
|
ENV___PYVENV_LAUNCHER__="/framework/Python9.8/python",
|
|
real_executable="/usr/bin/python",
|
|
)
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_xfile("/framework/Python9.8/python")
|
|
ns.add_known_file("/usr/lib/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib/python9.8/lib-dynload")
|
|
ns.add_known_file("/framework/Python9.8/pyvenv.cfg", [
|
|
"home = /usr/bin"
|
|
])
|
|
expected = dict(
|
|
executable="/framework/Python9.8/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
base_executable="/usr/bin/python",
|
|
base_prefix="/usr",
|
|
base_exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib/python98.zip",
|
|
"/usr/lib/python9.8",
|
|
"/usr/lib/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_normal_macos(self):
|
|
"Test a 'standard' install layout via symlink on macOS"
|
|
ns = MockPosixNamespace(
|
|
os_name="darwin",
|
|
PREFIX="/usr",
|
|
argv0="python",
|
|
ENV_PATH="/linkfrom:/usr/bin",
|
|
# real_executable on macOS matches the invocation path
|
|
real_executable="/linkfrom/python",
|
|
)
|
|
ns.add_known_xfile("/linkfrom/python")
|
|
ns.add_known_xfile("/usr/bin/python")
|
|
ns.add_known_link("/linkfrom/python", "/usr/bin/python")
|
|
ns.add_known_file("/usr/lib/python9.8/os.py")
|
|
ns.add_known_dir("/usr/lib/python9.8/lib-dynload")
|
|
expected = dict(
|
|
executable="/linkfrom/python",
|
|
base_executable="/linkfrom/python",
|
|
prefix="/usr",
|
|
exec_prefix="/usr",
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/lib/python98.zip",
|
|
"/usr/lib/python9.8",
|
|
"/usr/lib/python9.8/lib-dynload",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_symlink_buildpath_macos(self):
|
|
"""Test an in-build-tree layout via symlink on macOS.
|
|
|
|
This layout is discovered from the presence of pybuilddir.txt, which
|
|
contains the relative path from the executable's directory to the
|
|
platstdlib path.
|
|
"""
|
|
ns = MockPosixNamespace(
|
|
os_name="darwin",
|
|
argv0=r"python",
|
|
ENV_PATH="/linkfrom:/usr/bin",
|
|
PREFIX="/usr/local",
|
|
# real_executable on macOS matches the invocation path
|
|
real_executable="/linkfrom/python",
|
|
)
|
|
ns.add_known_xfile("/linkfrom/python")
|
|
ns.add_known_xfile("/home/cpython/python")
|
|
ns.add_known_link("/linkfrom/python", "/home/cpython/python")
|
|
ns.add_known_xfile("/usr/local/bin/python")
|
|
ns.add_known_file("/home/cpython/pybuilddir.txt", ["build/lib.macos-9.8"])
|
|
ns.add_known_file("/home/cpython/Lib/os.py")
|
|
ns.add_known_dir("/home/cpython/lib-dynload")
|
|
expected = dict(
|
|
executable="/linkfrom/python",
|
|
prefix="/usr/local",
|
|
exec_prefix="/usr/local",
|
|
base_executable="/linkfrom/python",
|
|
build_prefix="/home/cpython",
|
|
_is_python_build=1,
|
|
module_search_paths_set=1,
|
|
module_search_paths=[
|
|
"/usr/local/lib/python98.zip",
|
|
"/home/cpython/Lib",
|
|
"/home/cpython/build/lib.macos-9.8",
|
|
],
|
|
)
|
|
actual = getpath(ns, expected)
|
|
self.assertEqual(expected, actual)
|
|
|
|
|
|
# ******************************************************************************
|
|
|
|
DEFAULT_NAMESPACE = dict(
|
|
PREFIX="",
|
|
EXEC_PREFIX="",
|
|
PYTHONPATH="",
|
|
VPATH="",
|
|
PLATLIBDIR="",
|
|
PYDEBUGEXT="",
|
|
VERSION_MAJOR=9, # fixed version number for ease
|
|
VERSION_MINOR=8, # of testing
|
|
PYWINVER=None,
|
|
EXE_SUFFIX=None,
|
|
|
|
ENV_PATH="",
|
|
ENV_PYTHONHOME="",
|
|
ENV_PYTHONEXECUTABLE="",
|
|
ENV___PYVENV_LAUNCHER__="",
|
|
argv0="",
|
|
py_setpath="",
|
|
real_executable="",
|
|
executable_dir="",
|
|
library="",
|
|
winreg=None,
|
|
build_prefix=None,
|
|
venv_prefix=None,
|
|
)
|
|
|
|
DEFAULT_CONFIG = dict(
|
|
home=None,
|
|
platlibdir=None,
|
|
pythonpath=None,
|
|
program_name=None,
|
|
prefix=None,
|
|
exec_prefix=None,
|
|
base_prefix=None,
|
|
base_exec_prefix=None,
|
|
executable=None,
|
|
base_executable="",
|
|
stdlib_dir=None,
|
|
platstdlib_dir=None,
|
|
module_search_paths=None,
|
|
module_search_paths_set=0,
|
|
pythonpath_env=None,
|
|
argv=None,
|
|
orig_argv=None,
|
|
|
|
isolated=0,
|
|
use_environment=1,
|
|
use_site=1,
|
|
)
|
|
|
|
class MockNTNamespace(dict):
|
|
def __init__(self, *a, argv0=None, config=None, **kw):
|
|
self.update(DEFAULT_NAMESPACE)
|
|
self["config"] = DEFAULT_CONFIG.copy()
|
|
self["os_name"] = "nt"
|
|
self["PLATLIBDIR"] = "DLLs"
|
|
self["PYWINVER"] = "9.8-XY"
|
|
self["VPATH"] = r"..\.."
|
|
super().__init__(*a, **kw)
|
|
if argv0:
|
|
self["config"]["orig_argv"] = [argv0]
|
|
if config:
|
|
self["config"].update(config)
|
|
self._files = {}
|
|
self._links = {}
|
|
self._dirs = set()
|
|
self._warnings = []
|
|
|
|
def add_known_file(self, path, lines=None):
|
|
self._files[path.casefold()] = list(lines or ())
|
|
self.add_known_dir(path.rpartition("\\")[0])
|
|
|
|
def add_known_xfile(self, path):
|
|
self.add_known_file(path)
|
|
|
|
def add_known_link(self, path, target):
|
|
self._links[path.casefold()] = target
|
|
|
|
def add_known_dir(self, path):
|
|
p = path.rstrip("\\").casefold()
|
|
while p:
|
|
self._dirs.add(p)
|
|
p = p.rpartition("\\")[0]
|
|
|
|
def __missing__(self, key):
|
|
try:
|
|
return getattr(self, key)
|
|
except AttributeError:
|
|
raise KeyError(key) from None
|
|
|
|
def abspath(self, path):
|
|
if self.isabs(path):
|
|
return path
|
|
return self.joinpath("C:\\Absolute", path)
|
|
|
|
def basename(self, path):
|
|
return path.rpartition("\\")[2]
|
|
|
|
def dirname(self, path):
|
|
name = path.rstrip("\\").rpartition("\\")[0]
|
|
if name[1:] == ":":
|
|
return name + "\\"
|
|
return name
|
|
|
|
def hassuffix(self, path, suffix):
|
|
return path.casefold().endswith(suffix.casefold())
|
|
|
|
def isabs(self, path):
|
|
return path[1:3] == ":\\"
|
|
|
|
def isdir(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a dir")
|
|
return path.casefold() in self._dirs
|
|
|
|
def isfile(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a file")
|
|
return path.casefold() in self._files
|
|
|
|
def ismodule(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a module")
|
|
path = path.casefold()
|
|
return path in self._files and path.rpartition(".")[2] == "py".casefold()
|
|
|
|
def isxfile(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a executable")
|
|
path = path.casefold()
|
|
return path in self._files and path.rpartition(".")[2] == "exe".casefold()
|
|
|
|
def joinpath(self, *path):
|
|
return ntpath.normpath(ntpath.join(*path))
|
|
|
|
def readlines(self, path):
|
|
try:
|
|
return self._files[path.casefold()]
|
|
except KeyError:
|
|
raise FileNotFoundError(path) from None
|
|
|
|
def realpath(self, path, _trail=None):
|
|
if verbose:
|
|
print("Read link from", path)
|
|
try:
|
|
link = self._links[path.casefold()]
|
|
except KeyError:
|
|
return path
|
|
if _trail is None:
|
|
_trail = set()
|
|
elif link.casefold() in _trail:
|
|
raise OSError("circular link")
|
|
_trail.add(link.casefold())
|
|
return self.realpath(link, _trail)
|
|
|
|
def warn(self, message):
|
|
self._warnings.append(message)
|
|
if verbose:
|
|
print(message)
|
|
|
|
|
|
class MockWinreg:
|
|
HKEY_LOCAL_MACHINE = "HKLM"
|
|
HKEY_CURRENT_USER = "HKCU"
|
|
|
|
def __init__(self, keys):
|
|
self.keys = {k.casefold(): v for k, v in keys.items()}
|
|
self.open = {}
|
|
|
|
def __repr__(self):
|
|
return "<MockWinreg>"
|
|
|
|
def __eq__(self, other):
|
|
return isinstance(other, type(self))
|
|
|
|
def open_keys(self):
|
|
return list(self.open)
|
|
|
|
def OpenKeyEx(self, hkey, subkey):
|
|
if verbose:
|
|
print(f"OpenKeyEx({hkey}, {subkey})")
|
|
key = f"{hkey}\\{subkey}".casefold()
|
|
if key in self.keys:
|
|
self.open[key] = self.open.get(key, 0) + 1
|
|
return key
|
|
raise FileNotFoundError()
|
|
|
|
def CloseKey(self, hkey):
|
|
if verbose:
|
|
print(f"CloseKey({hkey})")
|
|
hkey = hkey.casefold()
|
|
if hkey not in self.open:
|
|
raise RuntimeError("key is not open")
|
|
self.open[hkey] -= 1
|
|
if not self.open[hkey]:
|
|
del self.open[hkey]
|
|
|
|
def EnumKey(self, hkey, i):
|
|
if verbose:
|
|
print(f"EnumKey({hkey}, {i})")
|
|
hkey = hkey.casefold()
|
|
if hkey not in self.open:
|
|
raise RuntimeError("key is not open")
|
|
prefix = f'{hkey}\\'
|
|
subkeys = [k[len(prefix):] for k in sorted(self.keys) if k.startswith(prefix)]
|
|
subkeys[:] = [k for k in subkeys if '\\' not in k]
|
|
for j, n in enumerate(subkeys):
|
|
if j == i:
|
|
return n.removeprefix(prefix)
|
|
raise OSError("end of enumeration")
|
|
|
|
def QueryValue(self, hkey):
|
|
if verbose:
|
|
print(f"QueryValue({hkey})")
|
|
hkey = hkey.casefold()
|
|
if hkey not in self.open:
|
|
raise RuntimeError("key is not open")
|
|
try:
|
|
return self.keys[hkey]
|
|
except KeyError:
|
|
raise OSError()
|
|
|
|
|
|
class MockPosixNamespace(dict):
|
|
def __init__(self, *a, argv0=None, config=None, **kw):
|
|
self.update(DEFAULT_NAMESPACE)
|
|
self["config"] = DEFAULT_CONFIG.copy()
|
|
self["os_name"] = "posix"
|
|
self["PLATLIBDIR"] = "lib"
|
|
super().__init__(*a, **kw)
|
|
if argv0:
|
|
self["config"]["orig_argv"] = [argv0]
|
|
if config:
|
|
self["config"].update(config)
|
|
self._files = {}
|
|
self._xfiles = set()
|
|
self._links = {}
|
|
self._dirs = set()
|
|
self._warnings = []
|
|
|
|
def add_known_file(self, path, lines=None):
|
|
self._files[path] = list(lines or ())
|
|
self.add_known_dir(path.rpartition("/")[0])
|
|
|
|
def add_known_xfile(self, path):
|
|
self.add_known_file(path)
|
|
self._xfiles.add(path)
|
|
|
|
def add_known_link(self, path, target):
|
|
self._links[path] = target
|
|
|
|
def add_known_dir(self, path):
|
|
p = path.rstrip("/")
|
|
while p:
|
|
self._dirs.add(p)
|
|
p = p.rpartition("/")[0]
|
|
|
|
def __missing__(self, key):
|
|
try:
|
|
return getattr(self, key)
|
|
except AttributeError:
|
|
raise KeyError(key) from None
|
|
|
|
def abspath(self, path):
|
|
if self.isabs(path):
|
|
return path
|
|
return self.joinpath("/Absolute", path)
|
|
|
|
def basename(self, path):
|
|
return path.rpartition("/")[2]
|
|
|
|
def dirname(self, path):
|
|
return path.rstrip("/").rpartition("/")[0]
|
|
|
|
def hassuffix(self, path, suffix):
|
|
return path.endswith(suffix)
|
|
|
|
def isabs(self, path):
|
|
return path[0:1] == "/"
|
|
|
|
def isdir(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a dir")
|
|
return path in self._dirs
|
|
|
|
def isfile(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a file")
|
|
return path in self._files
|
|
|
|
def ismodule(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is a module")
|
|
return path in self._files and path.rpartition(".")[2] == "py"
|
|
|
|
def isxfile(self, path):
|
|
if verbose:
|
|
print("Check if", path, "is an xfile")
|
|
return path in self._xfiles
|
|
|
|
def joinpath(self, *path):
|
|
return posixpath.normpath(posixpath.join(*path))
|
|
|
|
def readlines(self, path):
|
|
try:
|
|
return self._files[path]
|
|
except KeyError:
|
|
raise FileNotFoundError(path) from None
|
|
|
|
def realpath(self, path, _trail=None):
|
|
if verbose:
|
|
print("Read link from", path)
|
|
try:
|
|
link = self._links[path]
|
|
except KeyError:
|
|
return path
|
|
if _trail is None:
|
|
_trail = set()
|
|
elif link in _trail:
|
|
raise OSError("circular link")
|
|
_trail.add(link)
|
|
return self.realpath(link, _trail)
|
|
|
|
def warn(self, message):
|
|
self._warnings.append(message)
|
|
if verbose:
|
|
print(message)
|
|
|
|
|
|
def diff_dict(before, after, prefix="global"):
|
|
diff = []
|
|
for k in sorted(before):
|
|
if k[:2] == "__":
|
|
continue
|
|
if k == "config":
|
|
diff_dict(before[k], after[k], prefix="config")
|
|
continue
|
|
if k in after and after[k] != before[k]:
|
|
diff.append((k, before[k], after[k]))
|
|
if not diff:
|
|
return
|
|
max_k = max(len(k) for k, _, _ in diff)
|
|
indent = " " * (len(prefix) + 1 + max_k)
|
|
if verbose:
|
|
for k, b, a in diff:
|
|
if b:
|
|
print("{}.{} -{!r}\n{} +{!r}".format(prefix, k.ljust(max_k), b, indent, a))
|
|
else:
|
|
print("{}.{} +{!r}".format(prefix, k.ljust(max_k), a))
|
|
|
|
|
|
def dump_dict(before, after, prefix="global"):
|
|
if not verbose or not after:
|
|
return
|
|
max_k = max(len(k) for k in after)
|
|
for k, v in sorted(after.items(), key=lambda i: i[0]):
|
|
if k[:2] == "__":
|
|
continue
|
|
if k == "config":
|
|
dump_dict(before[k], after[k], prefix="config")
|
|
continue
|
|
try:
|
|
if v != before[k]:
|
|
print("{}.{} {!r} (was {!r})".format(prefix, k.ljust(max_k), v, before[k]))
|
|
continue
|
|
except KeyError:
|
|
pass
|
|
print("{}.{} {!r}".format(prefix, k.ljust(max_k), v))
|
|
|
|
|
|
def getpath(ns, keys):
|
|
before = copy.deepcopy(ns)
|
|
failed = True
|
|
try:
|
|
exec(SOURCE, ns)
|
|
failed = False
|
|
finally:
|
|
if failed:
|
|
dump_dict(before, ns)
|
|
else:
|
|
diff_dict(before, ns)
|
|
return {
|
|
k: ns['config'].get(k, ns.get(k, ...))
|
|
for k in keys
|
|
}
|