zmc
2023-08-08 e792e9a60d958b93aef96050644f369feb25d61b
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
# -----------------------------------------------------------------------------
# Copyright (c) 2005-2023, PyInstaller Development Team.
#
# Distributed under the terms of the GNU General Public License (version 2
# or later) with exception for distributing the bootloader.
#
# The full license is in the file COPYING.txt, distributed with this software.
#
# SPDX-License-Identifier: (GPL-2.0-or-later WITH Bootloader-exception)
# -----------------------------------------------------------------------------
"""
`distutils`-specific pre-find module path hook.
 
When run from within a virtual environment, this hook changes the `__path__` of the `distutils` package to
that of the system-wide rather than virtual-environment-specific `distutils` package. While the former is suitable for
freezing, the latter is intended for use _only_ from within virtual environments.
 
NOTE: this behavior seems to be specific to virtual environments created by (an old?) version of `virtualenv`; it is not
applicable to virtual environments created by the `venv`.
"""
 
import pathlib
 
from PyInstaller.utils.hooks import logger, get_module_file_attribute
 
 
def pre_find_module_path(api):
    # Absolute path of the system-wide "distutils" package when run from within a venv or None otherwise.
 
    # opcode is not a virtualenv module, so we can use it to find the stdlib. Technique taken from virtualenv's
    # "distutils" package detection at
    # https://github.com/pypa/virtualenv/blob/16.3.0/virtualenv_embedded/distutils-init.py#L5
    # As opcode is a module, stdlib path corresponds to the parent directory of its ``__file__`` attribute.
    stdlib_path = pathlib.Path(get_module_file_attribute('opcode')).parent.resolve()
    # As distutils is a package, we need to consider the grandparent directory of its ``__file__`` attribute.
    distutils_path = pathlib.Path(get_module_file_attribute('distutils')).parent.parent.resolve()
 
    if distutils_path.name == 'setuptools':
        logger.debug("distutils: provided by setuptools")
    elif distutils_path == stdlib_path:
        logger.debug("distutils: provided by stdlib")
    else:
        # Find this package in stdlib.
        stdlib_path = str(stdlib_path)
        logger.debug("distutils: virtualenv shim - retargeting to stdlib dir %r", stdlib_path)
        api.search_dirs = [stdlib_path]