| """ |
| Monkey patching of distutils. |
| """ |
| |
| import sys |
| import distutils.filelist |
| import platform |
| import types |
| import functools |
| |
| from .py26compat import import_module |
| from setuptools.extern import six |
| |
| import setuptools |
| |
| __all__ = [] |
| """ |
| Everything is private. Contact the project team |
| if you think you need this functionality. |
| """ |
| |
| |
| def get_unpatched(item): |
| lookup = ( |
| get_unpatched_class if isinstance(item, six.class_types) else |
| get_unpatched_function if isinstance(item, types.FunctionType) else |
| lambda item: None |
| ) |
| return lookup(item) |
| |
| |
| def get_unpatched_class(cls): |
| """Protect against re-patching the distutils if reloaded |
| |
| Also ensures that no other distutils extension monkeypatched the distutils |
| first. |
| """ |
| while cls.__module__.startswith('setuptools'): |
| cls, = cls.__bases__ |
| if not cls.__module__.startswith('distutils'): |
| msg = "distutils has already been patched by %r" % cls |
| raise AssertionError(msg) |
| return cls |
| |
| |
| def patch_all(): |
| # we can't patch distutils.cmd, alas |
| distutils.core.Command = setuptools.Command |
| |
| has_issue_12885 = ( |
| sys.version_info < (3, 4, 6) |
| or |
| (3, 5) < sys.version_info <= (3, 5, 3) |
| or |
| (3, 6) < sys.version_info |
| ) |
| |
| if has_issue_12885: |
| # fix findall bug in distutils (http://bugs.python.org/issue12885) |
| distutils.filelist.findall = setuptools.findall |
| |
| needs_warehouse = ( |
| sys.version_info < (2, 7, 13) |
| or |
| (3, 0) < sys.version_info < (3, 3, 7) |
| or |
| (3, 4) < sys.version_info < (3, 4, 6) |
| or |
| (3, 5) < sys.version_info <= (3, 5, 3) |
| or |
| (3, 6) < sys.version_info |
| ) |
| |
| if needs_warehouse: |
| warehouse = 'https://upload.pypi.org/legacy/' |
| distutils.config.PyPIRCCommand.DEFAULT_REPOSITORY = warehouse |
| |
| _patch_distribution_metadata_write_pkg_file() |
| _patch_distribution_metadata_write_pkg_info() |
| |
| # Install Distribution throughout the distutils |
| for module in distutils.dist, distutils.core, distutils.cmd: |
| module.Distribution = setuptools.dist.Distribution |
| |
| # Install the patched Extension |
| distutils.core.Extension = setuptools.extension.Extension |
| distutils.extension.Extension = setuptools.extension.Extension |
| if 'distutils.command.build_ext' in sys.modules: |
| sys.modules['distutils.command.build_ext'].Extension = ( |
| setuptools.extension.Extension |
| ) |
| |
| patch_for_msvc_specialized_compiler() |
| |
| |
| def _patch_distribution_metadata_write_pkg_file(): |
| """Patch write_pkg_file to also write Requires-Python/Requires-External""" |
| distutils.dist.DistributionMetadata.write_pkg_file = ( |
| setuptools.dist.write_pkg_file |
| ) |
| |
| |
| def _patch_distribution_metadata_write_pkg_info(): |
| """ |
| Workaround issue #197 - Python 3 prior to 3.2.2 uses an environment-local |
| encoding to save the pkg_info. Monkey-patch its write_pkg_info method to |
| correct this undesirable behavior. |
| """ |
| environment_local = (3,) <= sys.version_info[:3] < (3, 2, 2) |
| if not environment_local: |
| return |
| |
| distutils.dist.DistributionMetadata.write_pkg_info = ( |
| setuptools.dist.write_pkg_info |
| ) |
| |
| |
| def patch_func(replacement, target_mod, func_name): |
| """ |
| Patch func_name in target_mod with replacement |
| |
| Important - original must be resolved by name to avoid |
| patching an already patched function. |
| """ |
| original = getattr(target_mod, func_name) |
| |
| # set the 'unpatched' attribute on the replacement to |
| # point to the original. |
| vars(replacement).setdefault('unpatched', original) |
| |
| # replace the function in the original module |
| setattr(target_mod, func_name, replacement) |
| |
| |
| def get_unpatched_function(candidate): |
| return getattr(candidate, 'unpatched') |
| |
| |
| def patch_for_msvc_specialized_compiler(): |
| """ |
| Patch functions in distutils to use standalone Microsoft Visual C++ |
| compilers. |
| """ |
| # import late to avoid circular imports on Python < 3.5 |
| msvc = import_module('setuptools.msvc') |
| |
| if platform.system() != 'Windows': |
| # Compilers only availables on Microsoft Windows |
| return |
| |
| def patch_params(mod_name, func_name): |
| """ |
| Prepare the parameters for patch_func to patch indicated function. |
| """ |
| repl_prefix = 'msvc9_' if 'msvc9' in mod_name else 'msvc14_' |
| repl_name = repl_prefix + func_name.lstrip('_') |
| repl = getattr(msvc, repl_name) |
| mod = import_module(mod_name) |
| if not hasattr(mod, func_name): |
| raise ImportError(func_name) |
| return repl, mod, func_name |
| |
| # Python 2.7 to 3.4 |
| msvc9 = functools.partial(patch_params, 'distutils.msvc9compiler') |
| |
| # Python 3.5+ |
| msvc14 = functools.partial(patch_params, 'distutils._msvccompiler') |
| |
| try: |
| # Patch distutils.msvc9compiler |
| patch_func(*msvc9('find_vcvarsall')) |
| patch_func(*msvc9('query_vcvarsall')) |
| except ImportError: |
| pass |
| |
| try: |
| # Patch distutils._msvccompiler._get_vc_env |
| patch_func(*msvc14('_get_vc_env')) |
| except ImportError: |
| pass |
| |
| try: |
| # Patch distutils._msvccompiler.gen_lib_options for Numpy |
| patch_func(*msvc14('gen_lib_options')) |
| except ImportError: |
| pass |